sticker-convert 2.8.12__py3-none-any.whl → 2.17.0.0__py3-none-any.whl

sticker_convert/uploaders/upload_viber.py

@@ -0,0 +1,178 @@
+#!/usr/bin/env python3
+import copy
+import json
+import zipfile
+from pathlib import Path
+from typing import Any, Dict, List, Tuple
+
+import requests
+
+from sticker_convert.converter import StickerConvert
+from sticker_convert.job_option import CompOption, CredOption, OutputOption
+from sticker_convert.uploaders.upload_base import UploadBase
+from sticker_convert.utils.callback import CallbackProtocol, CallbackReturn
+from sticker_convert.utils.files.metadata_handler import MetadataHandler
+from sticker_convert.utils.files.sanitize_filename import sanitize_filename
+from sticker_convert.utils.media.format_verify import FormatVerify
+from sticker_convert.utils.translate import I
+
+
+class UploadViber(UploadBase):
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+        self.base_spec.set_size_max(0)
+
+        self.png_spec = copy.deepcopy(self.base_spec)
+        self.png_spec.set_res_max(490)
+        self.png_spec.set_format((".png",))
+        self.opt_comp_merged = copy.deepcopy(self.opt_comp)
+        self.opt_comp_merged.merge(self.png_spec)
+
+        self.png_cover_spec = copy.deepcopy(self.base_spec)
+        self.png_cover_spec.set_res_max(120)
+        self.png_cover_spec.set_format((".png",))
+        self.opt_comp_cover_merged = copy.deepcopy(self.opt_comp)
+        self.opt_comp_cover_merged.merge(self.png_cover_spec)
+
+    def upload_stickers_viber(self) -> Tuple[int, int, List[str]]:
+        urls: List[str] = []
+
+        if not self.opt_cred.viber_auth:
+            self.cb.put(I("Viber auth required for uploading to viber"))
+            return 0, 0, urls
+
+        upload_data_base: Dict[str, str] = {}
+        for i in self.opt_cred.viber_auth.split(";"):
+            j = i.split(":")
+            upload_data_base[j[0]] = j[1]
+
+        if upload_data_base.get("member_id") is None:
+            self.cb.put(I("Invalid Viber auth: Missing member_id"))
+            return 0, 0, urls
+        if upload_data_base.get("m_token") is None:
+            self.cb.put(I("Invalid Viber auth: Missing m_token"))
+            return 0, 0, urls
+        if upload_data_base.get("m_ts") is None:
+            self.cb.put(I("Invalid Viber auth: Missing m_ts"))
+            return 0, 0, urls
+
+        title, author, _ = MetadataHandler.get_metadata(
+            self.opt_output.dir,
+            title=self.opt_output.title,
+            author=self.opt_output.author,
+        )
+        if title is None:
+            raise TypeError(I("title cannot be {}").format(title))
+        if author is None:
+            author = ""
+
+        packs = MetadataHandler.split_sticker_packs(
+            self.opt_output.dir,
+            title=title,
+            file_per_pack=24,
+            separate_image_anim=False,
+        )
+
+        cover_path = MetadataHandler.get_cover(self.opt_output.dir)
+        if cover_path is None:
+            cover_path = MetadataHandler.get_stickers_present(self.opt_output.dir)[0]
+
+        if FormatVerify.check_file(cover_path, spec=self.png_cover_spec):
+            with open(cover_path, "rb") as f:
+                cover_bytes = f.read()
+        else:
+            _, _, cover_bytes, _ = StickerConvert.convert(  # type: ignore
+                cover_path,
+                Path("bytes.png"),
+                self.opt_comp_cover_merged,
+                self.cb,
+                self.cb_return,
+            )
+            assert isinstance(cover_bytes, bytes)
+
+        stickers_total = 0
+        stickers_ok = 0
+        for pack_title, stickers in packs.items():
+            stickers_total += len(stickers)
+            out_f = Path(
+                self.opt_output.dir, sanitize_filename(pack_title + ".zip")
+            ).as_posix()
+            with zipfile.ZipFile(out_f, "w", zipfile.ZIP_DEFLATED) as zipf:
+                for num, src in enumerate(stickers):
+                    self.cb.put(I("Verifying {} for uploading to Viber").format(src))
+
+                    if not FormatVerify.check_file(src, spec=self.png_spec):
+                        success, _, image_data, _ = StickerConvert.convert(
+                            src,
+                            Path("bytes.png"),
+                            self.opt_comp_merged,
+                            self.cb,
+                            self.cb_return,
+                        )
+                        assert isinstance(image_data, bytes)
+                        if not success:
+                            self.cb.put(
+                                I(
+                                    "Warning: Cannot compress file {}, skip this file..."
+                                ).format(src.name)
+                            )
+                            continue
+                    else:
+                        with open(src, "rb") as f:
+                            image_data = f.read()
+
+                    zipf.writestr(f"{str(num).zfill(2)}.png", image_data)
+
+            upload_data = copy.deepcopy(upload_data_base)
+            upload_data["title"] = pack_title
+            upload_data["description"] = author
+            upload_data["shareable"] = "1"
+
+            with open(out_f, "rb") as f:
+                r = requests.post(
+                    "https://market.api.viber.com/2/users/custom-sticker-packs/create",
+                    files={
+                        "file": ("upload.zip", f.read()),
+                        "file_icon": ("color_icon.png", cover_bytes),
+                    },
+                    data=upload_data,
+                )
+
+            if r.ok:
+                rjson = json.loads(r.text)
+                if rjson["status"] == 1:
+                    pack_id = rjson["custom_sticker_pack"]["id"]
+                    url = f"https://stickers.viber.com/pages/custom-sticker-packs/{pack_id}"
+                    urls.append(url)
+                    self.cb.put(I("Uploaded {}").format(pack_title))
+                    stickers_ok += len(stickers)
+                else:
+                    self.cb.put(
+                        I("Failed to upload {}: {} {}").format(
+                            pack_title, r.status_code, r.text
+                        )
+                    )
+                if rjson["status"] == 103:
+                    self.cb.put(
+                        "Viber auth data may have expired. Try to regenerate it?"
+                    )
+            else:
+                self.cb.put(
+                    I("Failed to upload {}: {} {}").format(
+                        pack_title, r.status_code, r.text
+                    )
+                )
+
+        return stickers_ok, stickers_total, urls
+
+    @staticmethod
+    def start(
+        opt_output: OutputOption,
+        opt_comp: CompOption,
+        opt_cred: CredOption,
+        cb: CallbackProtocol,
+        cb_return: CallbackReturn,
+    ) -> Tuple[int, int, List[str]]:
+        exporter = UploadViber(opt_output, opt_comp, opt_cred, cb, cb_return)
+        return exporter.upload_stickers_viber()