Adding User Likes, get Videos by Hashtags, Music

Author: avilash

TikTokAPI/tiktokapi.py

@@ -26,10 +26,18 @@ def __init__(self, language='en', region='IN', cookie=None):
         self.signature_key = "_signature"
         self.tiktok_browser = TikTokBrowser(self.user_agent)
 
-    def send_get_request(self, url, params):
+    def send_get_request(self, url, params, extra_headers=None):
         url = build_get_url(url, params)
         signature = self.tiktok_browser.fetch_auth_params(url, language=self.language)
         url = build_get_url(url, {self.signature_key: signature}, append=True)
+        if extra_headers is None:
+            headers = self.headers
+        else:
+            headers = {}
+            for key, val in extra_headers.items():
+                headers[key] = val
+            for key, val in self.headers.items():
+                headers[key] = val
         data = get_req_json(url, params=None, headers=self.headers)
         return data
 
@@ -63,7 +71,12 @@ def getUserByName(self, user_name):
             params[key] = val
         return self.send_get_request(url, params)
 
-    def getUserVideos(self, user_id, secUid, count=30):
+    def getVideosByUserName(self, user_name, count=30):
+        user_data = self.getUserByName(user_name)
+        user_obj = user_data["userInfo"]["user"]
+        user_id = user_obj["id"]
+        secUid = user_obj["secUid"]
+
         url = self.base_url + "/item_list/"
         req_default_params = {
             "type": "1",
@@ -83,3 +96,101 @@ def getUserVideos(self, user_id, secUid, 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"]
+        user_id = user_obj["id"]
+        secUid = user_obj["secUid"]
+
+        url = self.base_url + "/item_list/"
+        req_default_params = {
+            "type": "2",
+            "maxCursor": "0",
+            "minCursor": "0",
+            "sourceType": "9",
+            "appId": "1180",
+            "region": self.region
+        }
+        params = {
+            "id": user_id,
+            "secUid": secUid,
+            "count": str(count)
+        }
+        for key, val in req_default_params.items():
+            params[key] = val
+        for key, val in self.default_params.items():
+            params[key] = val
+        return self.send_get_request(url, params)
+
+    def getHashTag(self, hashTag):
+        url = self.base_url + "/challenge/detail/"
+        params = {
+            "challengeName": hashTag.replace("#", "")
+        }
+        for key, val in self.default_params.items():
+            params[key] = val
+        return self.send_get_request(url, params)
+
+    def getVideosByHashTag(self, hashTag, count=30):
+        hashTag = hashTag.replace("#", "")
+        hashTag_obj = self.getHashTag(hashTag)
+        hashTag_id = hashTag_obj["challengeInfo"]["challenge"]["id"]
+        url = "https://m.tiktok.com/share/item/list"
+        req_default_params = {
+            "secUid": "",
+            "type": "3",
+            "minCursor": "0",
+            "maxCursor": "0",
+            "shareUid": "",
+            "recType": ""
+        }
+        params = {
+            "id": str(hashTag_id),
+            "count": str(count),
+        }
+        for key, val in req_default_params.items():
+            params[key] = val
+        for key, val in self.default_params.items():
+            params[key] = val
+        extra_headers = {"Referer": "https://www.tiktok.com/tag/" + str(hashTag)}
+        return self.send_get_request(url, params, extra_headers=extra_headers)
+
+    def getMusic(self, music_id):
+        url = self.base_url + "/music/detail/"
+        params = {
+            "musicId": music_id
+        }
+        for key, val in self.default_params.items():
+            params[key] = val
+        return self.send_get_request(url, params)
+
+    def getVideosByMusic(self, music_id, count=30):
+        url = "https://m.tiktok.com/share/item/list"
+        req_default_params = {
+            "secUid": "",
+            "type": "4",
+            "minCursor": "0",
+            "maxCursor": "0",
+            "shareUid": "",
+            "recType": ""
+        }
+        params = {
+            "id": str(music_id),
+            "count": str(count),
+        }
+        for key, val in req_default_params.items():
+            params[key] = val
+        for key, val in self.default_params.items():
+            params[key] = val
+        extra_headers = {"Referer": "https://www.tiktok.com/music/original-sound-" + str(music_id)}
+        return self.send_get_request(url, params, extra_headers=extra_headers)
+
+    def getVideoById(self, video_id):
+        url = self.base_url + "/item/detail/"
+        params = {
+            "itemId": str(video_id)
+        }
+        for key, val in self.default_params.items():
+            params[key] = val
+        return self.send_get_request(url, params)

setup.py

@@ -4,7 +4,7 @@
     long_description = fh.read()
 
 setuptools.setup(
-    name="TikTokAPI",
+    name="TikTokAPI-Python",
     version="0.0.1",
     license='MIT',
     author="Avilash Kumar",

tests/test_hashtag.py

@@ -0,0 +1,29 @@
+import argparse
+from TikTokAPI import TikTokAPI
+
+
+def getHashTag(hashTag):
+    api = TikTokAPI()
+    return api.getHashTag(hashTag)
+
+
+def getVideosByHashTag(hashTag):
+    api = TikTokAPI()
+    return api.getVideosByHashTag(hashTag, count=10)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='')
+    parser.add_argument('--data', dest='data', type=str, help='hashtag/videos')
+    args = parser.parse_args()
+
+    var = "#fcbarcelona"
+    if args.data == 'hashtag':
+        retval = getHashTag(var)
+        print(retval)
+    elif args.data == 'videos':
+        retval = getVideosByHashTag(var)
+        print(retval)
+    else:
+        print("Invalid Argument")
+

tests/test_music.py

@@ -0,0 +1,29 @@
+import argparse
+from TikTokAPI import TikTokAPI
+
+
+def getMusic(music_id):
+    api = TikTokAPI()
+    return api.getMusic(music_id)
+
+
+def getVideosByMusic(music_id):
+    api = TikTokAPI()
+    return api.getVideosByMusic(music_id, count=10)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='')
+    parser.add_argument('--data', dest='data', type=str, help='music/videos')
+    args = parser.parse_args()
+
+    var = "6704854531001289474"
+    if args.data == 'music':
+        retval = getMusic(var)
+        print(retval)
+    elif args.data == 'videos':
+        retval = getVideosByMusic(var)
+        print(retval)
+    else:
+        print("Invalid Argument")
+

tests/test_user.py

@@ -1,12 +1,36 @@
+import argparse
 from TikTokAPI import TikTokAPI
 
-user = "fcbarcelona"
-print("Testing For User - {}".format(user))
-api = TikTokAPI()
-retval = api.getUserByName(user)
-print("User Details")
-print(retval)
-user_obj = retval["userInfo"]["user"]
-retval = api.getUserVideos(user_obj["id"], user_obj["secUid"], count=1)
-print("User Videos")
-print(retval)
+
+def getUser(user_name):
+    api = TikTokAPI()
+    return api.getUserByName(user_name)
+
+
+def getVideosByUserName(user_name):
+    api = TikTokAPI()
+    return api.getVideosByUserName(user_name, count=1)
+
+
+def getLikesByUserName(user_name):
+    api = TikTokAPI()
+    return api.getLikesByUserName(user_name, count=1)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='')
+    parser.add_argument('--data', dest='data', type=str, help='user/videos/likes')
+    args = parser.parse_args()
+
+    var = "fcbarcelona"
+    if args.data == 'user':
+        retval = getUser(var)
+        print(retval)
+    elif args.data == 'videos':
+        retval = getVideosByUserName(var)
+        print(retval)
+    elif args.data == 'likes':
+        retval = getLikesByUserName(var)
+        print(retval)
+    else:
+        print("Invalid Argument")

tests/test_video.py

@@ -0,0 +1,7 @@
+from TikTokAPI import TikTokAPI
+
+
+video_id = "6843481669886954757"
+api = TikTokAPI()
+video_obj = api.getVideoById(video_id)
+print(video_obj)