git reimport

src/zvk/plugins/vk/upload.py

@@ -0,0 +1,65 @@
+import mimetypes
+
+from sqlalchemy import Integer, String
+
+from zvk.plugins.vk.api import VKApi
+from zvk.util.db import DBBase, NNColumn, Database
+from zvk.util.network import Network
+from zvk.util.zlogging import logger
+
+
+class CachedUpload(DBBase):
+  id = NNColumn(Integer, primary_key=True)
+  local_path = NNColumn(String)
+  vk_object = NNColumn(String)
+
+
+def is_local_file_an_image(local_path):
+  mimetype, _ = mimetypes.guess_type(local_path)
+  return mimetype in ['image/jpeg', 'image/png', 'image/gif']
+
+
+async def upload_image(db: Database,
+                       net: Network,
+                       api: VKApi,
+                       local_path: str,
+                       use_cache: bool = True) -> str:
+  logger.info(f'Uploading local image {local_path}')
+
+  if use_cache:
+    with db as session:
+      cached_upload = session \
+        .query(CachedUpload) \
+        .filter_by(local_path=local_path) \
+        .first()
+      if cached_upload:
+        logger.info(f'Found {local_path} in upload cache: {cached_upload.vk_object}')
+        return cached_upload.vk_object
+
+  if not is_local_file_an_image(local_path):
+    raise ValueError(f'{local_path} does not look like an image!')
+
+  response = await api.photos.getMessagesUploadServer()
+  upload_url = response['upload_url']
+
+  with open(local_path, 'rb') as f:
+    files = {'photo': f}
+
+    _, response = await net.post_json(upload_url, data=files)
+
+    if 'server' not in response or 'photo' not in response or 'hash' not in response:
+      raise RuntimeError(f'Could not upload {local_path}')
+
+    logger.info(f'Uploaded {local_path} as {response}')
+
+  response = await api.photos.saveMessagesPhoto(**response)
+  if len(response) < 1:
+    raise RuntimeError(f'Could not save image {local_path}')
+
+  vk_object = f'photo{response[0]["owner_id"]}_{response[0]["id"]}'
+
+  if use_cache:
+    with db as session:
+      session.add(CachedUpload(local_path=local_path, vk_object=vk_object))
+
+  return vk_object