
    OJni)	                     J    d dl Z ddlmZmZ ddlmZ ddlmZ  G d de      Zy)	    N   )
StreamDatawrap_stream)	async_req   )BaseLiveStreamc                        e Zd ZdZddedz  dedz  f fdZdefdZdded	edefd
Z	e
ddedeez  dz  defd       Z xZS )LangLiveStreamzK
    A class for fetching and processing Lang live stream information.
    N
proxy_addrcookiesc                 P    t         |   ||       | j                         | _        y N)super__init___get_pc_headers
pc_headers)selfr   r   	__class__s      Y/home/uftp/myenv/lib/python3.12/site-packages/streamget/platforms/langlive/live_stream.pyr   zLangLiveStream.__init__   s"    W-..0    returnc                 ,    ddd| j                   xs ddS )Nzhttps://www.lang.livezhttps://www.lang.live/z;ios/7.830 (ios 17.0; ; iPhone 15 (A2846/A3089/A3090/A3092)) )originrefererz
user-agentcookie)r   )r   s    r   r   zLangLiveStream._get_pc_headers   s!    -/Wll(b	
 	
r   urlprocess_datac                 j  K   |j                  d      d   j                  dd      d   }d| }t        || j                  | j                         d	{   }t        j                  |      }|s|S |d
   d   }|d   }|d   }	|dd}
|	dk(  r |d
   d   d   }|d
   d   d   }|
d|||dz  }
|
S 7 [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.
        ?r   /   )maxsplitz7https://api.lang.live/langweb/v1/room/liveinfo?room_id=)r   headersNdata	live_infonicknamelive_statusF)anchor_nameis_liveliveurlliveurl_hlsT)r+   m3u8_urlflv_url
record_url)splitrsplitr   r   r   jsonloads)r   r   r   room_idapi_1json_str	json_datar'   r*   r)   resultr/   r.   s                r   fetch_web_stream_dataz$LangLiveStream.fetch_web_stream_data   s      ))C.#**3*;B?I'S"5T__doo^^JJx(	f%k2	
+.!,?!'4Y?G (5mDH$H`hiiF _s   AB3B1AB3r8   video_qualityc                 .   K   | ddiz  } t        |       S w)z_
        Fetches the stream URL for a live room and wraps it into a StreamData object.
        platformu   浪Live)r   )r8   r;   s     r   fetch_stream_urlzLangLiveStream.fetch_stream_url3   s"     
 	j),,	9%%s   )NN)Tr   )__name__
__module____qualname____doc__strr   dictr   boolr:   staticmethodintr   r>   __classcell__)r   s   @r   r
   r
      s}    13: 1sTz 1
 
s $ RV 6 &$ &sSy4?O &[e & &r   r
   )	r3   r&   r   r   requests.async_httpr   baser   r
    r   r   <module>rL      s     + , !1&^ 1&r   