fix: cli render snaptik

krypton-byte · Nov 28, 2023 · 7830652 · 7830652 · vercel · Nov 28, 2023
1 parent d723744
commit 7830652
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tiktok_downloader/__main__.py b/tiktok_downloader/__main__.py
@@ -99,8 +99,9 @@ def info(url: str, js: Optional[bool] = False):
                 json.dumps(
                     [
                         {
-                            'type': i.type,
+                            'type': i.type.value,
                             'url': i.json,
+                            "render": bool(i.render),
                             'watermark': i.watermark
                         } for i in ok
                     ],
@@ -110,7 +111,10 @@ def info(url: str, js: Optional[bool] = False):
         elif parse.save:
             if parse.info:
                 info(parse.url, js=parse.json)
-            ok[0].download(parse.save, bar=True)
+            if ok[0].render:
+                ok[0].get_render().download(parse.save, bar=True)
+            else:
+                ok[0].download(parse.save, bar=True)
         else:
             os.system("python3 -m tiktok_downloader --help")
     except requests.exceptions.ConnectionError: