Merge branch 'feature/weibo' into 'test'

微博数据入库

See merge request alpha/saturn!100

api/utils/upload.py

 import requests
 from gm_upload import upload, upload_file
 from gm_upload import IMG_TYPE
+import io
+from PIL import Image
+# import numpy as np
 
 
 def upload_image(url, img_type=IMG_TYPE.TOPIC):
@@ -10,3 +13,24 @@ def upload_image(url, img_type=IMG_TYPE.TOPIC):
         return upload(response.content, img_type=img_type)
     except:
         return None
+
+
+def upload_weibo_image(url, img_type=IMG_TYPE.TOPIC):
+    '''非站内图片处理'''
+    try:
+        response = requests.get(url)
+        img = Image.open(io.BytesIO(response.content))
+
+        w, h = img.size
+        img = img.crop((0, 0, w, h-10))
+
+        # img =  img.convert('RGB')
+        # content =  np.array(img)[..., ::-1]
+        temp = io.BytesIO()
+        # content = Image.fromarray(content)
+        img.save(temp, format="png")
+        content = temp.getvalue()
+
+        return upload(content, img_type=img_type)
+    except:
+        return None

api/views/pictorial.py

@@ -132,8 +132,8 @@ class CreatePictorial(BaseView):
             if platform == GRAP_PLATFORM.XIAOHONGSHU:
                 for topic_comment in topic_comments:
                     error, _ = self.create_comment(comment_list=topic_comment, from_id=from_id, platform=platform, topic_id=topic_obj.get('id'))
-                if error:
-                    return error, _
+                    if error:
+                        return error, _
 
         return None, None