Add sm.ms to photostore

fishroom/photostore.py

@@ -88,6 +88,39 @@ def upload_image(self, filename=None, filedata=None, **kwargs) -> str:
         return r.text.strip()
 
 
+class SMMS(BasePhotoStore):
+
+    url = "https://sm.ms/api/upload"
+
+    def __init__(self, **kwargs):
+        pass
+
+    def upload_image(self, filename=None, filedata=None, **kwargs) -> str:
+        if filedata is None:
+            with open(filename, 'rb') as f:
+                filedata = f.read()
+        files = {"smfile": filedata}
+
+        try:
+            r = requests.post(self.url, files=files, timeout=5)
+        except requests.exceptions.Timeout:
+            logger.error("Timeout uploading to SMMS")
+            return None
+        except:
+            logger.exception("Unknown errror uploading to SMMS")
+            return None
+
+        try:
+            ret = json.loads(r.text)
+        except:
+            return None
+        if ret.get('code', None) != 'success':
+            logger.error("Error: SMMS returned error")
+            return None
+
+        return ret.get('data', {}).get('url', None)
+
+
 if __name__ == "__main__":
     import sys
     imgur = Imgur(sys.argv[1])

fishroom/telegram.py

@@ -703,7 +703,7 @@ def Fishroom2TelegramThread(tg: Telegram, bus: MessageBus):
 def init():
     from .db import get_redis
     from .filestore import get_qiniu
-    from .photostore import Imgur, VimCN
+    from .photostore import Imgur, VimCN, SMMS
     redis_client = get_redis()
 
     def photo_store_init():
@@ -713,6 +713,8 @@ def photo_store_init():
             return Imgur(**options)
         elif provider == "vim-cn":
             return VimCN()
+        elif provider == "sm.ms":
+            return SMMS()
         elif provider == "qiniu":
             return get_qiniu(redis_client, config)