U Xi|0@spddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z dZ dZdZdZdZd Zeed d d Zeed ddZeedddZeeedddZeeedddZdddeeeeeeeeeeeeed ddZdddeeeeeeeeeeeeed dd Zeeedd!d"Zed#Zddd$ddd%eeeeeeeeeeeeeeeeed&d'd(Z dS))N)OptionalLiteral)TIKTOK_CLIENT_KEYTIKTOK_CLIENT_SECRETTIKTOK_REDIRECT_URI TIKTOK_SCOPESTIKTOK_PRODUCTIONz)https://www.tiktok.com/v2/auth/authorize/z+https://open.tiktokapis.com/v2/oauth/token/z=https://open.tiktokapis.com/v2/post/publish/inbox/video/init/z7https://open.tiktokapis.com/v2/post/publish/video/init/z?https://open.tiktokapis.com/v2/post/publish/creator_info/query/z9https://open.tiktokapis.com/v2/post/publish/status/fetch/)statereturncCs$tdtt|d}tdtj|S)Ncode) client_keyZ response_typescope redirect_urir ?)rrrTIKTOK_OAUTH_AUTHORIZE_URLurllibparse urlencode)r paramsr+/var/www/html/luxverbi-app/tiktok_client.pybuild_authorize_url"sr)r r cCs0ttd|td}tjt|dd}||S)NZauthorization_code)r Z client_secretZ grant_typer r)datatimeout)rrrrequestspostTIKTOK_OAUTH_TOKEN_URLraise_for_statusjson)r payloadresprrrexchange_code_for_token-sr") access_tokenr cCsfd|dd}tjt|idd}||}|di}|ddkrZtd ||d iS) a Mandatory for audit: retrieve latest creator info right before rendering Post-to-TikTok page. Returns data dict with fields such as: - creator_nickname - privacy_level_options (list) - comment_disabled / duet_disabled / stitch_disabled (bool) - max_video_post_duration_sec (int) - may also include "can_post" flags depending on account state Bearer application/json; charset=UTF-8 Authorization Content-Typerheadersrrerrorr okzcreator_info error: r)rrCREATOR_INFO_URLrrget RuntimeError)r#r*r!r errrrrquery_creator_info>s  r1)r# publish_idr cCsjd|dd}tjt|d|idd}||}|di}|dd kr^td ||d iS) z= Poll post publish status so users can see progress. r$r%r&r2rr)r+r r,zstatus_fetch error: r)rrSTATUS_FETCH_URLrrr.r/)r#r2r*r!r r0rrrfetch_post_statusXs r4) file_pathr#r c Cstj|}d|dd}dd||ddi}tjt||dd }||}|d i}|d d krxt d ||di}|d} | st d||d} dd| d|d} t |d} tj | | | dd} W5QRX| |S)z Upload as draft (inbox) using video.upload: 1) INIT -> /post/publish/inbox/video/init/ 2) PUT -> upload_url with full file (Content-Range) r$r%r& source_info FILE_UPLOADsourceZ video_size chunk_sizeZtotal_chunk_count<r)r+r r,zTikTok inbox init error: r upload_urlz(No upload_url in TikTok inbox response: video/mp4bytes 0-/r(z Content-Rangerb,r*rr) ospathgetsizerrINBOX_INIT_URLrrr.r/openput)r5r# file_sizer*body init_resp init_datar+rr=end_byte put_headersfput_resprrr_upload_video_as_draftqs:      rST)is_aigcvideo_cover_timestamp_ms) r# video_urlcaption privacy_leveldisable_comment disable_duetdisable_stitchbrand_content_togglebrand_organic_togglerTrUr c  Cs|rt|tstdd|dd} ||p.d|||||| d} | dk rTt| | d<d |d | d } tjt| | d d }|jdkrtd|jd|j | | }| di}| ddkrtd||S)a Direct Post using PULL_FROM_URL (TikTok pulls the video from your URL). This is the correct mode when the MP4 already exists on your server/storage. Important: - video_url must be reachable by TikTok servers. - Your domain/prefix must be allowed in TT4D "Manage URL properties". - privacy_level must be explicitly selected by the user from privacy_level_options. z'video_url is required for PULL_FROM_URLr$r%r&rXtitlerYrZr[r\r]rTNrU PULL_FROM_URL)r:rVr6 post_infor<r)Direct init failed : r+r r,z*TikTok direct init (PULL_FROM_URL) error: ) isinstancestr ValueErrorintrrDIRECT_POST_INIT_URL status_coder/textrrr.)r#rVrWrXrYrZr[r\r]rTrUr*rcrLrMrNr+rrr_publish_video_from_urls<    rn) r5r#rWrXrYrZr[r\r]rTrUr c  CsDtj|} d|dd} ||p$d|||||| d} | dk rJt| | d<d| | d d | d }tjt| |d d }|jdkrtd|jd|j | | }| di}| ddkrtd|| di}| d}|std|| d }dd|d| d}t |d}tj|||dd}W5QRX| |S)af Direct Post (FILE_UPLOAD): 1) INIT -> /v2/post/publish/video/init/ with source_info + post_info 2) PUT -> upload_url with full file (Content-Range) 3) TikTok publishes based on post_info NOTE: Not recommended when the video is already on your server-side storage. Use PULL_FROM_URL for audit compliance in that scenario. r$r%r&r^r_NrUr7r8r9rbr<r)rdrerfr+r r,z(TikTok direct init (FILE_UPLOAD) error: rr=z)No upload_url in TikTok direct response: r>r?r@rArBrCrD)rErFrGrjrrrkrlr/rmrrr.rIrJ)r5r#rWrXrYrZr[r\r]rTrUrKr*rcrLrMrNr+rr=rOrPrQrRrrr%_upload_and_publish_video_file_uploadsT         rocCs t||dS)Nr5r#)rSrprrrupload_video_draftAsrq)rar7ra)rTrUmoderVr5)r#rWrXrYrZr[r\r]rTrUrrrVr5r c Csv| dkr2| stdt|| ||||||||| d S| dkrd| sFtdt| |||||||||| d Std| dS) z Unified entrypoint: - mode="PULL_FROM_URL": requires video_url (recommended for audit if video is on your server) - mode="FILE_UPLOAD": requires file_path (fallback/tests) raz/video_url is required when mode='PULL_FROM_URL') r#rVrWrXrYrZr[r\r]rTrUr7z-file_path is required when mode='FILE_UPLOAD') r5r#rWrXrYrZr[r\r]rTrUzUnsupported mode: N)rirnro) r#rWrXrYrZr[r\r]rTrUrrrVr5rrrupload_video_direct_postGsBrs)!rEr urllib.parsertypingrrconfigrrrrrrrrHrkr-r3rhrdictr"r1r4rSboolrjrnrorqZ PublishModersrrrrs ? Q V