Pasing cookie params in test

Author: avilash

TikTokAPI/tiktokapi.py

@@ -9,16 +9,18 @@
 class VideoException(Exception):
     pass
 
+
 class TikTokAPI(object):
 
-    def __init__(self, cookie={}, language='en', browser_lang="en-US", timezone="Asia/Kolkata", region='IN'):
+    def __init__(self, cookie=None, language='en', browser_lang="en-US", timezone="Asia/Kolkata", region='IN'):
         self.base_url = "https://t.tiktok.com/api"
         self.user_agent = "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:79.0) Gecko/20100101 Firefox/79.0"
 
+        if cookie is None:
+            cookie = {}
         self.verifyFp = cookie.get("s_v_web_id", "verify_kjf974fd_y7bupmR0_3uRm_43kF_Awde_8K95qt0GcpBk")
         self.tt_webid = cookie.get("tt_webid", "6913027209393473025")
 
-
         self.headers = {
             'Host': 't.tiktok.com',
             'User-Agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:79.0) Gecko/20100101 Firefox/79.0',
@@ -134,7 +136,7 @@ def getVideosByUserName(self, user_name, count=30):
         for key, val in self.default_params.items():
             params[key] = val
         return self.send_get_request(url, params)
-    
+
     def getLikesByUserName(self, user_name, count=30):
         user_data = self.getUserByName(user_name)
         user_obj = user_data["userInfo"]["user"]
@@ -249,7 +251,7 @@ def downloadVideoByIdNoWatermark(self, video_id, save_path):
             raise VideoException("Video without watermark not available in new videos")
         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])
+            .format(video_data[pos + 4:pos + 36])
 
         video_data_no_wm = get_req_content(video_url_no_wm, params=None, headers=self.headers)
         with open(save_path, 'wb') as f:

tests/cookie.json

@@ -0,0 +1,4 @@
+{
+  "s_v_web_id": "verify_kjf974fd_y7bupmR0_3uRm_43kF_Awde_8K95qt0GcpBk",
+  "tt_webid": "6913027209393473025"
+}

tests/test_hashtag.py

@@ -1,14 +1,15 @@
 import argparse
 from TikTokAPI import TikTokAPI
+from utils import read_json_from_file
 
 
 def getHashTag(hashTag):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getHashTag(hashTag)
 
 
 def getVideosByHashTag(hashTag):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getVideosByHashTag(hashTag, count=10)
 
 

tests/test_music.py

@@ -1,14 +1,15 @@
 import argparse
 from TikTokAPI import TikTokAPI
+from utils import read_json_from_file
 
 
 def getMusic(music_id):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getMusic(music_id)
 
 
 def getVideosByMusic(music_id):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getVideosByMusic(music_id, count=10)
 
 

tests/test_trending.py

@@ -1,6 +1,7 @@
 from TikTokAPI import TikTokAPI
+from utils import read_json_from_file
 
-api = TikTokAPI()
+api = TikTokAPI(read_json_from_file("cookie.json"))
 retval = api.getTrending(count=5)
 print("Trending Videos")
 print(retval)

tests/test_user.py

@@ -1,19 +1,20 @@
 import argparse
 from TikTokAPI import TikTokAPI
+from utils import read_json_from_file
 
 
 def getUser(user_name):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getUserByName(user_name)
 
 
 def getVideosByUserName(user_name):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getVideosByUserName(user_name, count=1)
 
 
 def getLikesByUserName(user_name):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getLikesByUserName(user_name, count=1)
 
 

tests/test_video.py

@@ -1,19 +1,20 @@
 import argparse
 from TikTokAPI import TikTokAPI
+from utils import read_json_from_file
 
 
 def getVideoById(video_id):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     return api.getVideoById(video_id)
 
 
 def downloadVideoById(video_id):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     api.downloadVideoById(video_id, video_id+".mp4")
 
 
 def downloadVideoByIdNoWatermark(video_id):
-    api = TikTokAPI()
+    api = TikTokAPI(read_json_from_file("cookie.json"))
     api.downloadVideoByIdNoWatermark(video_id, video_id+"_no_wm.mp4")
 
 

tests/utils.py

@@ -0,0 +1,9 @@
+import json
+
+
+def read_json_from_file(filepath):
+    with open(filepath, 'r', encoding='utf-8') as json_file:
+        data = json.load(json_file)
+    if type(data) is str:
+        return json.loads(data)
+    return data