Fj*ndZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZdd lmZmZdd lmZmZmZmZej4d k\rdd lmZndd lmZe rddlmZddl Z ddl m!Z!GddZ"Gdde"Z#Gdde"Z$Gdde"Z%Gdde"Z&Gdde"Z'y)z.Provide classes for uploading media to Reddit.) annotationsN)cached_property)Path)IO TYPE_CHECKINGAnyClassVarcast) ElementTree ServerError)API_PATH JPEG_HEADER)ClientExceptionRedditAPIExceptionRedditErrorItemTooLargeMediaException) )guess_file_type) guess_type)Response)modelsceZdZUdZded<dZded<eddZddZddZ ddd Z dd Z dd Z dd Z eddZ ddZddZddZy )MediazBase class representing media that can be uploaded to Reddit. Use one of the subclasses, e.g., :class:`.EmojiMedia` or :class:`.PostMedia`, depending on what the media is being uploaded for. z ClassVar[str]LEASE_API_PATH s3UploadLeaseLEASE_RESPONSE_KEYct|Nr responses O/root/workspace/ytshorts/venv/lib/python3.12/site-packages/praw/models/media.py_raise_upload_errorzMedia._raise_upload_error,s (##ct|t|uxr4|j|jk(xr|j|jk(S)z5Return whether the other instance equals the current.)typename_data)selfothers r#__eq__z Media.__eq__0s<E{d4j(bTYY%**-DbW\WbWbIbbr%cnt|jj|j|jfS)z(Return the hash of the current instance.)hash __class____name__r(r)r*s r#__hash__zMedia.__hash__4s&T^^,,diiDEEr%Nct|tr|s d}t||}n+t|}|xs |j}|j }||_||_y)aInitialize a :class:`.Media` instance. :param fp: The path to a media file as a string, or the content of a media file as a :class:`bytes` object. :param name: The name of the media file, e.g., ``"picture.png"``. The name is used to infer the media's MIME type. When ``fp`` is a path the name is derived from it, otherwise this parameter is required. z/'name' is required when 'fp' is a bytes object.N) isinstancebytes ValueErrorrr( read_bytesr))r*fpr(msgdatapaths r#__init__zMedia.__init__8sV b% G o%D8D$499D??$D  r%cPd|jjd|jdS)z/Return a string representation of the instance.)r/r0r(r1s r#__repr__zMedia.__repr__Ns&4>>**+6$))a@@r%c 8|j|jd|S)N)filepathmimetype)r( _mime_type)r*additional_datas r#_build_lease_datazMedia._build_lease_dataRs II4??VoVVr%c |jdi|}|j|||\}}}|j|||}|js|j ||||fS)N)rF _obtain_lease _post_to_s3okr$) r* lease_urlredditadditional_lease_data lease_datalease_response upload_data upload_urlr"s r#_lease_and_postzMedia._lease_and_postUsm,T++D.CD 262D2DZQZ\b2c/ Z##FKD{{  $ $X .{J66r%cpt|j\}}|d|jd}t||S)Nz%Unable to determine the MIME type of .)rr(r)r* mime_type_r9s r#rDzMedia._mime_type_s=&tyy1 1  9$))aHC!#& &r%c|j||}||j}|dDcic] }|d|d }}d|d}|||fScc}w)N)r:fieldsr(valuezhttps:action)postr) r*rOrLrMrP upload_leaseitemrQrRs r#rIzMedia._obtain_leasegsp YZ@%d&=&=> ?KH?UVttF|T']2V Vl8456 {J66WsA c|jJ|jjjd||d|j|jfiS)NPOSTfiler:files)_core requestorrequestr(r))r*rMrQrRs r#rJzMedia._post_to_s3psR||'''||%%--  DIItzz23 .  r%c |jj|}|j||jfi|\}}}|d|dS)aUpload the media to Reddit. :param subreddit: The subreddit the media is associated with. :param additional_lease_data: Additional data to include in the upload lease request. :returns: The URL of the uploaded media.  subreddit/keyrformatrS_reddit)r*rirNrLrWrQrRs r#_uploadz Media._uploadys[''...C %9T%9%9)YEVEV%pZo%p"; Q{51233r%r"rreturnNone)r+objectrqbool)rqintr )r( str | Noner8z str | bytesrqrrrqstr)rErxrqdict[str, str])rLrxrM praw.RedditrNrxrq*tuple[dict[str, Any], dict[str, str], str])rOryrLrxrMrzrqr{)rMrzrQryrRrxrqrrimodels.SubredditrNrxrqrx)r0 __module__ __qualname____doc____annotations__r staticmethodr$r,r2r<r@rFrSrrDrIrJrorHr%r#rr!s"!(7 7$$cF,AW77&17OR7 377(7587BM7 37  4r%rc"eZdZdZedZddZy) EmojiMediazVMedia to be uploaded as a subreddit emoji. See :meth:`.SubredditEmoji.add`. emoji_leasec |jj|}|j||jfi|\}}}|dS)a Upload the media to Reddit. :param subreddit: The subreddit the emoji is associated with. :param additional_lease_data: Additional data to include in the upload lease request. :returns: The S3 key of the uploaded media. rhrkrl)r*rirNrLrWrQs r#rozEmojiMedia._uploadsO''...C 0D00Iz0PostMedia._parse_xml_response..s'M 'Msr)actual maximum_size)r fromstringrtagrru)r"rootrtagsrWrrs r#_parse_xml_responsezPostMedia._parse_xml_responses%%hmm4+/0 00 8L L'MD!H'M $Q (FKa0@sS_SdcdOef f M1sA;cXtj|tj|yr )rrrr$r!s r#r$zPostMedia._raise_upload_errors%%h/ !!(+r%Nlink)expected_mime_prefix upload_typec |L|jjdd|k7r+d|d|jd|jd}t||j |j |\}}}|dk(r |d|dS|d d S) aUpload the media to Reddit (undocumented endpoint). Unlike the other :class:`.Media` subclasses, post media is not associated with a subreddit when uploaded, so this method takes a :class:`.Reddit` instance. :param reddit: The :class:`.Reddit` instance to upload with. :param expected_mime_prefix: If provided, enforce that the media has a MIME type that starts with the provided prefix. :param upload_type: One of ``"link"``, ``"gallery"``, or ``"selfpost"`` (default: ``"link"``). :returns: The URL of the uploaded media when ``upload_type`` is ``"link"``, otherwise the media's asset ID. rjrz"Expected a mimetype starting with z but got mimetype z (from file name z).rrkassetasset_id)rD partitionr(rrSr)r*rMrrr9rPrQrRs r#rozPostMedia._uploads. +0I0I#0Nq0QUi0i67K6NN`aeapap`stEFJFOFOERRTUC!#& &262F2FtGZGZ\b2c/ Z &  \;u#5"67 7g&z22r%rp)rMrzrrvrrxrqrx) r0r~rrrrrrrr$rorHr%r#rrst m,Ngg,,,0! 33 ) 3  3 3r%rc:eZdZdZedZ dfd ZxZS)StylesheetAssetz{Media to be uploaded as a subreddit style asset, e.g., a banner. See :meth:`.SubredditStylesheet.upload_banner`. style_asset_leasec&t|||S)zUpload the media to Reddit. :param subreddit: The subreddit the style asset is associated with. :param image_type: The type of style asset, e.g., ``"bannerBackgroundImage"``. :returns: The URL of the uploaded media. ) imagetype)superro)r*ri image_typer/s r#rozStylesheetAsset._uploadswyJ??r%)rir}rrxrqrx)r0r~rrrrro __classcell__)r/s@r#rrs< 12N @) @=@ @  @ @r%rc@eZdZdZedZeddZ ddZy)StylesheetImageziMedia to be uploaded as a subreddit stylesheet image. See :meth:`.SubredditStylesheet.upload`. upload_imagecH|jjtrdSdS)Njpgpng)r) startswithrr1s r# _image_typezStylesheetImage._image_types --k:uEEr%c vd|ji|}|jj|}tdd|j|j fi}|j j|||}|drE|dd}|jdd gd}|d vsJd tt|d |xsd g|S) a8Upload the media to Reddit. :param subreddit: The subreddit the stylesheet image is associated with. :param additional_data: Additional data to include in the upload request. :returns: A dictionary containing a link to the uploaded image under the key ``img_src``. img_typerhzdict[str, IO[Any]]rarberrorsr errors_values> IMAGE_ERROR BAD_CSS_NAMEzPlease file a bug with PRAW.) error_typefieldmessage) rUPLOAD_API_PATHrmr r(r)rnr\getrr) r*rirEr:urlrcr"r error_values r#rozStylesheetImage._uploadsD,,@@""))I)>)FTYY 4K+LM$$))#D)F H !(+A.J",,=a@K" .. .%oSU_j_pnp&q%rs sr%Nrw)rir}rErxrqzdict[str, Any]) r0r~rrrrrrrorHr%r#rrsD ~.OFF)AD r%rceZdZdZedZy) WidgetMediaztMedia to be uploaded for use in a subreddit widget. See :meth:`.SubredditWidgetsModeration.upload_image`. widget_leaseN)r0r~rrrrrHr%r#rrs n-Nr%r)(r __future__rsys functoolsrpathlibrtypingrrrr r defusedxmlr prawcore.exceptionsr praw.constrrpraw.exceptionsrrrr version_info mimetypesrrrequestsrprawrrrrrrrrHr%r#rs4" %99"+,w)7!d4d4N""06363r@e@.'e'T.%.r%