Adding Apis for video download and video download without watermark

Author: avilash

TikTokAPI/tiktokapi.py

@@ -1,5 +1,5 @@
 import os
-from .utils import random_key, build_get_url, get_req_json
+from .utils import random_key, build_get_url, get_req_json, get_req_content, get_req_text
 from .tiktok_browser import TikTokBrowser
 
 
@@ -194,3 +194,25 @@ def getVideoById(self, video_id):
         for key, val in self.default_params.items():
             params[key] = val
         return self.send_get_request(url, params)
+
+    def downloadVideoById(self, video_id, save_path):
+        video_info = self.getVideoById(video_id)
+        video_url = video_info["itemInfo"]["itemStruct"]["video"]["downloadAddr"]
+        headers = {"User-Agent": "okhttp"}
+        video_data = get_req_content(video_url, params=None, headers=headers)
+        with open(save_path, 'wb') as f:
+            f.write(video_data)
+
+    def downloadVideoByIdNoWatermark(self, video_id, save_path):
+        video_info = self.getVideoById(video_id)
+        video_url = video_info["itemInfo"]["itemStruct"]["video"]["downloadAddr"]
+        headers = {"User-Agent": "okhttp"}
+        video_data = get_req_text(video_url, params=None, headers=headers)
+        pos = video_data.find("vid:")
+        video_url_no_wm = "https://api2-16-h2.musical.ly/aweme/v1/play/?video_id={" \
+                          "}&vr_type=0&is_play_url=1&source=PackSourceEnum_PUBLISH&media_type=4" \
+            .format(video_data[pos+4:pos+36])
+        headers = {"User-Agent": "okhttp"}
+        video_data_no_wm = get_req_content(video_url_no_wm, params=None, headers=headers)
+        with open(save_path, 'wb') as f:
+            f.write(video_data_no_wm)

TikTokAPI/utils.py

@@ -28,6 +28,16 @@ def get_req_json(url, params=None, headers=None):
     return json.loads(r.text)
 
 
+def get_req_content(url, params=None, headers=None):
+    r = requests.get(url, params=params, headers=headers)
+    return r.content
+
+
+def get_req_text(url, params=None, headers=None):
+    r = requests.get(url, params=params, headers=headers)
+    return r.text
+
+
 def python_list2_web_list(data):
     web_list = "[\""
     web_list += '", "'.join(data)

tests/test_video.py

@@ -1,7 +1,34 @@
+import argparse
 from TikTokAPI import TikTokAPI
 
 
-video_id = "6843481669886954757"
-api = TikTokAPI()
-video_obj = api.getVideoById(video_id)
-print(video_obj)
+def getVideoById(video_id):
+    api = TikTokAPI()
+    return api.getVideoById(video_id)
+
+
+def downloadVideoById(video_id):
+    api = TikTokAPI()
+    api.downloadVideoById(video_id, video_id+".mp4")
+
+
+def downloadVideoByIdNoWatermark(video_id):
+    api = TikTokAPI()
+    api.downloadVideoByIdNoWatermark(video_id, video_id+"_no_wm.mp4")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='')
+    parser.add_argument('--data', dest='data', type=str, help='data/video/video_no_wm')
+    args = parser.parse_args()
+
+    var = "6843481669886954757"
+    if args.data == 'data':
+        retval = getVideoById(var)
+        print(retval)
+    elif args.data == 'video':
+        downloadVideoById(var)
+    elif args.data == 'video_no_wm':
+        downloadVideoByIdNoWatermark(var)
+    else:
+        print("Invalid Argument")