OJnix b d dl Z d dlZd dlmZ ddlmZmZ ddlmZm Z ddl
m Z G d d e Z y)
N)
itemgetter )
StreamData wrap_stream) async_reqget_response_status )BaseLiveStreamc ~ e Zd ZdZddedz dedz f fd
ZdefdZdded edefd
Z dd ed ee
z dz de fd
Z xZ
S )TikTokLiveStreamzM
A class for fetching and processing TikTok live stream information.
N
proxy_addrcookiesc P t | || | j | _ y N)super__init___get_pc_headers
pc_headers)selfr
r __class__s W/home/uftp/myenv/lib/python3.12/site-packages/streamget/platforms/tiktok/live_stream.pyr zTikTokLiveStream.__init__ s"
W- ..0 returnc , dd| j xs dddS )NzPMozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/115.0z;zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2ztt-target-idc=useast5zhttps://www.tiktok.com/live)z
user-agentzaccept-languagecookiereferer)r )r s r r z TikTokLiveStream._get_pc_headers s" l \ll=&=4
r url process_datac K t || j | j d d{ }d|v r=t j d| }t d|r|j
d d d |vrLt j d
|t j }|s t d t j |d }|S y7 w)
a)
Fetches web stream data for a live room.
Args:
url (str): The room URL.
process_data (bool): Whether to process the data. Defaults to True.
Returns:
dict: A dictionary containing anchor name, live status, room URL, and title.
Fr r
headershttp2NzBWe regret to inform you that we have discontinued operating TikTokzE
\s+(We regret to inform you that we have discontinu.*?)\.
\s+
zzYour proxy node's regional network is blocked from accessing TikTok; please switch to a node in another region to access. UNEXPECTED_EOF_WHILE_READINGz>zCPlease check if your network can access the TikTok website normallyr ) r r
r researchConnectionErrorgroupfindallDOTALLjsonloads)r r r html_strmsgjson_str json_datas r fetch_web_stream_dataz&TikTokLiveStream.fetch_web_stream_data s #stPTP_P_glmm OS[ [)) hjrsC!-=@SYYq\,IK
FH,IK
* 9zzPRZ\^\e\egH%&kll
8A;/I
: ns 'CC BCr1
video_qualityc K |st dddd S dt fd}|d d }|d }|d
d |d }|j d
d }d|dd}|dk( rd|d vr t d |d d d d } t j
| j di } || d }
|| d } t
|
dk r#|
j |
d t
|
dk r#t
| dk r#| j | d t
| dk r#| j | \ }} |
| }
| | }|j d xs |
j d }t || j | j d d{ }|s| dk r| d z n| d z
}|
| }
| | }|
d j d d }|d j d d }|d |d d ||||xs |d!z }t | S 7 jw)"z_
Fetches the stream URL for a live room and wraps it into a StreamData object.
TikTokNF)platform anchor_nameis_liver c , g }| D ] }| | d }|d }t j | }t |d }|j dd }d}|j | r?|| j d s|| j d r || dz |z }n || d z |z }|d
} |d k7 s| st t | j
d \ }
} |j |||
| fd
|j t d d |j d |S )Nmain
sdk_paramsvbitrateVCodecr$ z.flvz.m3u8z?codec=z&codec=
resolutionr x)r r< r> T)keyreversec 0 | d | d d | d d fS )Nr< r> r r# )r? s r zRTikTokLiveStream.fetch_stream_url..get_video_quality_url..O s/ 1Z=.1\?1;M:MPQR^P_`aPbOb)c r )r@ )
r, r- intgetendswithmapsplitappendsortr ) streamq_key play_listr@ url_infor; r< v_codecplay_urlr> widthheights r get_video_quality_urlz@TikTokLiveStream.fetch_stream_url..get_video_quality_url; s. I
m!#;v. %l 3
!ZZ
3
z* 56$..26 <<& //78E?;S;ST[;\#+E?Y#>#H#+E?Y#>#H ' 5
q=Z$'Z-=-=c-B$CME6 $$X8\aci[j%kl
m"
NNz* 5tN D NN cN d r LiveRoomliveRoomUserInfousernickname-uniqueIdstatus r
streamDataliveRoomzRThis live stream may be uncomfortable for some viewers. Log in to confirm your age pull_data stream_datadataflvhls r r r# zhttps://zhttp://Ttitle)r8 rf qualitym3u8_urlflv_url
record_url)
r listrF Exceptionr, r- lenrJ get_quality_indexr r
r replace)r r1 r3 rT live_roomrW r7 r[ resultra flv_url_list
m3u8_url_list
quality_indexflv_dict m3u8_dict check_urlokindexri rh s r fetch_stream_urlz!TikTokLiveStream.fetch_stream_url4 sZ HTV[ \] ] D . j)*<= j)*!D,<+=> (A& !&
Q;9Z#88 tuu Z( 6{CMRD::d#''3D0u=L 1$ >Ml#a' ##L$45 l#a'm$q( $$]2%67 m$q(+/+A+A-+P (M= )- 8H +M :I!
e,C U0CI* $//4??Z_a aB -:Q->
)MTUDU!-e!4"/"6 uo--j)DG '//
IFH ":.w7( $" & 1'
F 6""%as CG$ 0G$A(G$7G"8A+G$)NN)Tr )__name__
__module__ __qualname____doc__strr dictr boolr2 rE r rz
__classcell__)r s @r r r
ss 13: 1sTz 1
s $ RV 4L# L#S3YQUEU L#ak L#r r )
r, r& operatorr ra r r requests.async_httpr r baser
r rC r r r s% + A !v#~ v#r