add gitee oss

app/middleware/oss/__init__.py

@@ -1,6 +1,8 @@
 from app.core.configuration import SystemConfiguration
 from app.middleware.oss.aliyun import AliyunOss
 from app.middleware.oss.files import OssFile
+from app.middleware.oss.gitee import GiteeOss
+from config import Config
 
 
 class OssClient(object):
@@ -21,7 +23,9 @@ def get_oss_client(cls) -> OssFile:
             endpoint = oss_config.get("endpoint")
             if oss_config is None:
                 raise Exception("服务器未配置oss信息, 请在configuration.json中添加")
-            if oss_config.get("type").lower() == "aliyun":
+            if oss_config.get("type").lower() == Config.ALIYUN:
                 return AliyunOss(access_key_id, access_key_secret, endpoint, bucket)
+            if oss_config.get("type").lower() == Config.GITEE:
+                return GiteeOss(access_key_secret, bucket, access_key_id)
             raise Exception("不支持的oss类型")
         return OssClient._client

app/middleware/oss/files.py

@@ -7,11 +7,11 @@
 class OssFile(ABC):
 
     @abstractmethod
-    def create_file(self, filepath: str, content: ByteString):
+    def create_file(self, filepath: str, content: bytes):
         pass
 
     @abstractmethod
-    def update_file(self, filepath: str, content: ByteString):
+    def update_file(self, filepath: str, content: bytes):
         pass
 
     @abstractmethod

app/middleware/oss/gitee.py

@@ -0,0 +1,37 @@
+import base64
+
+import requests
+
+from app.middleware.oss import OssFile
+
+
+class GiteeOss(OssFile):
+    _create_url = "https://gitee.com/api/v5/repos/{}/{}/contents/{}"
+
+    def __init__(self, user, repo, token):
+        self.user = user
+        self.repo = repo
+        self.token = token
+
+    def create_file(self, filepath: str, content: bytes):
+        gitee_url = self._create_url.format(self.user, self.repo, filepath)
+        data = base64.b64encode(content)
+        json_data = {"access_token": self.token, "message": "pity create file", "content": data}
+        r = requests.post(url=gitee_url, json=json_data)
+        if r.status_code != 200:
+            raise Exception("上传文件到gitee失败")
+
+    def update_file(self, filepath: str, content: bytes):
+        pass
+
+    def delete_file(self, filepath: str):
+        pass
+
+    def list_file(self):
+        pass
+
+    def download_file(self, filepath, filename):
+        pass
+
+    def get_file_object(self, filepath):
+        pass

config.py

@@ -57,6 +57,9 @@ class Config(object):
 
     SERVER_REPORT = "http://test.pity.fun/#/record/report/"
 
+    ALIYUN = "aliyun"
+    GITEE = "gitee"
+
     # 请求类型
     class BodyType:
         none = 0

configuration.json

@@ -9,10 +9,10 @@
     "token": ""
   },
   "oss": {
-    "type": "aliyun",
-    "access_key_id": "",
-    "access_key_secret": "",
-    "bucket": "",
+    "type": "gitee",
+    "access_key_id": "56a5797a2fb4a3cdfca439082da7dee1",
+    "access_key_secret": "woodywrx",
+    "bucket": "oss",
     "endpoint": ""
   }
 }