云端粗出

2025-07-12 17:07:14 +08:00
parent bb14fba3fa
commit 5f1fc7d9c8
9 changed files with 659 additions and 165 deletions
--- a/python_core/kv.py
+++ b/python_core/kv.py
@@ -0,0 +1,84 @@
+
+from python_core.config import settings
+import httpx
+from python_core.utils.logger import setup_logger
+from typing import Dict
+
+logger = setup_logger(__name__)
+
+class Kv:
+    kv: Dict[str, str]
+    def __init__(self):
+        self.cf_account_id = settings.cloudflare_account_id
+        self.cf_kv_api_token = settings.cloudflare_api_key
+        self.cf_kv_id = settings.cloudflare_kv_id
+        
+    def gets(self, keys: list[str]):
+        try:
+            with httpx.Client() as client:
+                response = client.post(
+                    f"https://api.cloudflare.com/client/v4/accounts/{self.cf_account_id}/storage/kv/namespaces/{self.cf_kv_id}/bulk/get",
+                    headers={"Authorization": f"Bearer {self.cf_kv_api_token}"},
+                    json={
+                        "keys": keys,
+                        "type": "json"
+                    }
+                )
+                response.raise_for_status()
+        except httpx.RequestError as e:
+            logger.error(f"An error occurred while put kv to cloudflare")
+            raise e
+        except httpx.HTTPStatusError as e:
+            logger.error(f"HTTP error occurred while get kv from cloudflare {str(e)}")
+            raise e
+        except Exception as e:
+            logger.error(f"An unexpected error occurred: {str(e)}")
+            raise e
+        
+    def sets(self, caches: Dict[str, str]):
+        try:
+            with httpx.Client() as client:
+                response = client.put(
+                    f"https://api.cloudflare.com/client/v4/accounts/{self.cf_account_id}/storage/kv/namespaces/{self.cf_kv_id}/bulk",
+                    headers={"Authorization": f"Bearer {self.cf_kv_api_token}"},
+                    json=[
+                        {
+                            "based64": False,
+                            "key": key,
+                            "value": value,
+                        }
+                        for (key, value) in caches.items()
+                    ]
+                )
+                response.raise_for_status()
+        except httpx.RequestError as e:
+            logger.error(f"An error occurred while put kv to cloudflare")
+            raise e
+        except httpx.HTTPStatusError as e:
+            logger.error(f"HTTP error occurred while get kv from cloudflare {str(e)}")
+            raise e
+        except Exception as e:
+            logger.error(f"An unexpected error occurred: {str(e)}")
+            raise e
+       
+    def remove(self, keys: list[str]):
+        with httpx.Client() as client:
+            try:
+                response = client.post(
+                    f"https://api.cloudflare.com/client/v4/accounts/{self.cf_account_id}/storage/kv/namespaces/{self.cf_kv_id}/bulk/delete",
+                    headers={"Authorization": f"Bearer {self.cf_kv_api_token}"},
+                    json=keys
+                )
+                response.raise_for_status()
+            except httpx.RequestError as e:
+                logger.error(f"An error occurred while put kv to cloudflare")
+                raise e
+            except httpx.HTTPStatusError as e:
+                logger.error(f"HTTP error occurred while get kv from cloudflare {str(e)}")
+                raise e
+            except Exception as e:
+                logger.error(f"An unexpected error occurred: {str(e)}")
+                raise e
+            
+            
+kv = Kv()