v3.4

- Removed youtube-dl
- Replaced m3u8 with flv

Author: Michele0303

enums.py

@@ -46,7 +46,7 @@ class Info(Enum):
     def __str__(self):
         return str(self.value)
 
-    VERSION = 3.3
+    VERSION = 3.4
 
     BANNER = f"""
 

prova.py

@@ -0,0 +1,26 @@
+import requests
+from urllib.parse import urljoin
+from moviepy.editor import VideoFileClip, concatenate_videoclips
+
+playlist_url = 'https://pull-hls-f16-gcp01.tiktokcdn.com/stage/stream-3283529153973059756_or4/index.m3u8'
+response = requests.get(playlist_url)
+playlist = response.text.strip().split('\n')
+
+# Find the media segment lines in the playlist
+media_segments = [line for line in playlist if not line.startswith('#')]
+
+# Combine the media segment URLs with the playlist URL to create valid URLs
+media_urls = [urljoin(playlist_url, segment_url) for segment_url in media_segments]
+
+# Download each media segment and save it to a file
+clips = []
+for i, segment_url in enumerate(media_urls):
+    response = requests.get(segment_url)
+    with open(f'segment_{i}.ts', 'wb') as f:
+        f.write(response.content)
+    clip = VideoFileClip(f'segment_{i}.ts')
+    clips.append(clip)
+
+# Concatenate the clips and save the result as an .mp4 file
+final_clip = concatenate_videoclips(clips)
+final_clip.write_videofile('merged_video.mp4')

requirements.txt

@@ -1,3 +1,2 @@
 requests
-argparse
-streamlink
+argparse

tiktokbot.py

@@ -1,12 +1,9 @@
-import signal
-import sys
 import time
 import requests as req
 import re
-import subprocess
 import os
 from enums import Mode, Error, StatusCode, TimeOut
-
+import shutil
 
 class TikTok:
 
@@ -65,14 +62,23 @@ def start_recording(self):
 
         print("\n[*] STARTED RECORDING... [PRESS ONLY ONCE CTRL + C TO STOP]")
 
-        cmd = f"streamlink {live_url} best -o {output}"
+        try:
+            response = req.get(live_url, stream=True)
+            with open(output, "wb") as out_file:
+                shutil.copyfileobj(response.raw, out_file)
+        except KeyboardInterrupt:
+            pass
+
+        print("FINISH", output)
+        
+        #cmd = f"streamlink {live_url} best -o {output}"
         #cmd = f"youtube-dl --hls-prefer-ffmpeg --no-continue --no-part -o {output} {live_url}"
+        '''
         with open("error.log", "w") as error_log, open("info.log", "w") as info:
             p = subprocess.Popen(cmd, stderr=error_log, stdout=info, shell=True)
             signal.signal(signal.SIGINT, lambda x, y: sys.exit(0))
             p.communicate()
-
-        print(f"[*] FINISH {output}")
+        '''
 
     def get_live_url(self) -> str:
         try:
@@ -81,9 +87,10 @@ def get_live_url(self) -> str:
             json = req.get(url).json()
 
             live_url_m3u8 = json['data']['stream_url']['hls_pull_url']
-            print("[*] URL M3U8", live_url_m3u8)
+            live_url_flv = json['data']['stream_url']['rtmp_pull_url']
+            print("[*] URL FLV", live_url_flv)
 
-            return live_url_m3u8
+            return live_url_flv
         except Exception as ex:
             print(ex)