a NbP@sLddlZddlmZddlZddlmZddlZddlZGdddeZ dS)N) text_type)parse_timestampc@s"eZdZdFddZddZddZdd Zd d Zed d Z eddZ ddZ dGddZ dHddZ ddZdIddZdJddZdKd d!Zd"d#Zd$d%Zd&d'ZdLd(d)ZdMd*d+Zd,d-ZdNd/d0ZdOd1d2ZdPd3d4Zd5d6ZdQd7d8Zd9d:Zd;d<ZdRd=d>ZdSdBdCZ dTdDdEZ!dS)UPageNcCst|t|ur"|j|jdS||_||_i|_|s|rpddt|}g}t |D]}| |q^nd}d}t|t ur|jj dg|R||dd}n|jj dg|R||dd}t t |dd }||_d |vrtj| d | d d |_| dd|_|jr.||j|_n|j|_|jdd |_|jdd |_t| d|_| dd |_d|v|_| d|_dd| ddD|_d|v|_| dd|_| dd|_ | dd|_!| dd|_"d|_#d|_$dS)Nzinfo||infoquery protection)propZpageidsinprop)r titlesr ZpagesinvalidZ invalidreasonnsrtitle/touchedZ lastrevidmissinglengthcSs&i|]}|r|d|d|dfqS)typelevelZexpiryr).0irrJ/var/www/html/diaryflask/venv/lib/python3.9/site-packages/mwclient/page.py 7sz!Page.__init__..redirectpageid contentmodel pagelanguagerestrictiontypes)%r__dict__updatesitename _textcachejoinsixiterkeys itervaluesextendintgetnext_infomwclienterrorsZInvalidPageTitle namespacestrip_namespaceZ page_titlesplitZ base_title base_namerrrevisionexistsrr rrrrr last_rev_time edit_time)selfr"r#rZextra_propertiesr Z extra_propsZ extra_proprrr__init__ sd        z Page.__init__cCsb|jjdd|jdd}d|dvrZ|ddD]&}|d|jkr.t|j|dSq.dSdSdS) zE Get the redirect target page, or None if the page is not a redirect.rZ pagepropsr)r r redirectsr:fromtoN)r"r+r#r)r8rpagerrr redirects_toEs zPage.redirects_tocCs|}|dur|S|SdS)zI Get the redirect target page, or the current page if its not a redirect.N)r>)r8Z target_pagerrrresolve_redirectPszPage.resolve_redirectcCsd|jd|jfS)Nzzutf-8)r#encoder"r8rrr__repr__Xsz Page.__repr__cCs|jS)N)r#rArrr __unicode__[szPage.__unicode__cCs.|ddkr|dd}||dddS)Nr:)findrrrrr1^s  zPage.strip_namespacecCsH|}|ddkr |dd}|d|dd}|dd}|S)NrrDrE _)stripupperreplacerGrrrnormalize_titleds    zPage.normalize_titlecCs,|j||fd}|dkr d}||jjvS)zCheck if the current user has the right to carry out some action with the current page. Example: >>> page.can('edit') True rZsysopZ editprotected)r r+r"Zrights)r8actionrrrrcanns zPage.canFcCs|jj|||jdS)NrG)r" get_tokenr#)r8rforcerrrrP}szPage.get_tokenTmainc Cs|dstj||js dS|dur0t|}t||f}|rT||jvrT|j|S|jdd||d}z8t |}d|vr|d|d}n|d}|d |_ Wnt yd}d|_ Yn0|st |_n |j|}|r||j|<|S) atGet the current wikitext of the page, or of a specific section. If the page does not exist, an empty string is returned. By default, results will be cached and if you call text() again with the same section and expandtemplates the result will come from the cache. The cache is stored on the instance, so it lives as long as the instance does. Args: section (int): Section number, to only get text from a single section. expandtemplates (bool): Expand templates (default: `False`) cache (bool): Use in-memory caching (default: `True`) readrNzcontent|timestamprE)r limitsectionslotsrV* timestamp)rOr.r/InsufficientPermissionr5rhashr$ revisionsr,r6 StopIterationtimegmtimer7r"expandtemplates) r8rUr_cacheZslotkeyrevsrevtextrrrrds6          z Page.textcOs|j|i|S)z8Alias for edit, for maintaining backwards compatibility.)edit)r8argskwargsrrrsavesz Page.savercKs|j||||fd|i|S)zXUpdate the text of a section or the whole page by performing an edit operation. rd_editr8rdsummaryminorbotrUrgrrrresz Page.editcKs|j||||fd|i|S)zTAppend text to a section or the whole page by performing an edit operation. Z appendtextrirkrrrappendsz Page.appendcKs|j||||fd|i|S)zUPrepend text to a section or the whole page by performing an edit operation. Z prependtextrirkrrrprependsz Page.prependc sjjsjjrtjjjr2tjjjdsHtj jj s\tj i|rldd<|sxdd<j rt dj d<jrt djd<|rdd<|dur|d <|jjrd d <fd d }z |}Wntjjy}zv|jdkrljdddz |}Wn6tjjyh}z|WYd}~n d}~00n |WYd}~n d}~00d|dvrt|dd_ jjjD],} d| jvrjjj| j| j| jqi_|dS)Nre1rmZnotminorz %Y%m%d%H%M%SZ basetimestampZstarttimestamprnrUuserassertcsNjjdjdd}|dddkrJtj|d|S)Nre)rrltokenresultZfailure)re) r"postr#rPr+lowerr.r/ EditError)rudatar8rlrrdo_editszPage._edit..do_editZbadtokenT)rQZ newtimestampZPostEditRevision)r"Z logged_inZ force_loginr.r/AssertUserFailedErrorZblockedZ UserBlockedrOProtectedPageErrorwriteapi NoWriteApir6r]strftimer7r!ZAPIErrorcoderPhandle_edit_errorkeysrr+ connectioncookiesr#cleardomainpathr$) r8rlrmrnrUrgr{ruecookierryrrjsX        $" z Page._editcCs\|jdkrtj|||jn:|jdvr>tj||j|jn|jdkrTtjn|dS)NZ editconflict> zcantcreate-anonZcascadeprotectedzprotectednamespace-interfaceZnoimageredirectz noedit-anonZprotectednamespaceZ protectedpageznoimageredirect-anonZ cantcreateZcustomcssjsprotectedZprotectedtitleZnoeditZassertuserfailed)rr.r/rxrr}r|)r8rrlrrrrs    zPage.handle_edit_errorcCs|js dS|ddS)aLPerform a "null edit" on the page to update the wiki's cached data of it. This is useful in contrast to purge when needing to update stored data on a wiki, for example Semantic MediaWiki properties or Cargo table values, since purge only forces update of a page's displayed values and not its store. Nr)r5rorArrrtouchsz Page.touchcCsx|dstj||jjs*tj|i}|r:d|d<|rFd|d<|jjdd|jff|| d|d|}|dS)aMove (rename) page to new_title. If user account is an administrator, specify no_redirect as True to not leave a redirect. If user does not have permission to move page, an InsufficientPermission exception is raised. moverqZmovetalkZ noredirectr;)r<rtreason rOr.r/rYr"r~rrvr#rP)r8Z new_titlerZ move_talkZ no_redirectrzrurrrrs    z Page.movecCs||dstj||jjs*tj|i}|r:d|d<|rFd|d<|rR||d<|jjd|j| d|d|}|dS)zDelete page. If user does not have permission to delete page, an InsufficientPermission exception is raised. deleterqwatchunwatcholdimage)rrtr)rr)r8rrrrrzrurrrr8s$    z Page.deletecCs|jjd|jddS)zkPurge server-side cache of page. This will re-render templates and other dynamic content. purge)r N)r"rvr#rArrrrQsz Page.purgeallcCsltjjd|}ttjjj|||d}|r8d|d|<|j||d<tjj||jddf|dd|S)zList pages that link to the current page, similar to Special:Whatlinkshere. API doc: https://www.mediawiki.org/wiki/API:Backlinks blr0 filterredirrqz %sredirectr backlinksrT return_values r.listingList get_prefixdictgenerate_kwargsr#Zget_listr")r8r0rrrT generatorprefixrgrrrr[s   zPage.backlinkscCsbtjjd|}ttjjj||d}|rBtjj|ddfi|Stjj|ddfddi|SdS)aList categories used on the current page. API doc: https://www.mediawiki.org/wiki/API:Categories Args: generator (bool): Return generator (Default: True) show (str): Set to 'hidden' to only return hidden categories or '!hidden' to only return non-hidden ones. Returns: mwclient.listings.PagePropertyGenerator Zcl)show categoriesrrNr.rrrrrPagePropertyGenerator PageProperty)r8rrrrgrrrros"  zPage.categoriescCs\tjjd|}ttjjj|||d}|j||d<tjj||jddf|dd|S)a List pages that transclude the current page. API doc: https://www.mediawiki.org/wiki/API:Embeddedin Args: namespace (int): Restricts search to a given namespace (Default: None) filterredir (str): How to filter redirects, either 'all' (default), 'redirects' or 'nonredirects'. limit (int): Maximum amount of pages to return per request generator (bool): Return generator (Default: True) Returns: mwclient.listings.List: Page iterator eirr embeddedinrr)r8r0rrTrrrgrrrrs zPage.embeddedincCstjj|ddddS)zrList external links from the current page. API doc: https://www.mediawiki.org/wiki/API:Extlinks extlinkselrWrr.rrrArrrrsz Page.extlinkscCs,|rtj|ddStjj|ddddSdS)zuList files/images embedded in the current page. API doc: https://www.mediawiki.org/wiki/API:Images imagesrrrN)r.rrr)r8rrrrrs  z Page.imagescCstjj|ddddS)zrList interwiki links from the current page. API doc: https://www.mediawiki.org/wiki/API:Iwlinks iwlinksiw)rrWrrrArrrrs z Page.iwlinkscKstjj|ddfddi|S)zxList interlanguage links from the current page. API doc: https://www.mediawiki.org/wiki/API:Langlinks langlinksZllr)langrWr)r8rgrrrrs zPage.langlinkscCsntjjd|}ttjjj||d}|r2d|d<|rNtjj|ddfi|Stjj|ddfddi|SdS) zuList links to other pages from the current page. API doc: https://www.mediawiki.org/wiki/API:Links plr0rqr:linksrrNr)r8r0rr:rrgrrrrsz Page.linksolder2 ids|timestamp|flags|comment|userc Csttjjjd||||||| | d }|jjdddkrFd|vrF|d=||d<| |d<||d <| rjd |d <| durz| |d <tjj|d dfd|i|S)aList revisions of the current page. API doc: https://www.mediawiki.org/wiki/API:Revisions Args: startid (int): Revision ID to start listing from. endid (int): Revision ID to stop listing at. start (str): Timestamp to start listing from. end (str): Timestamp to end listing at. dir (str): Direction to list in: 'older' (default) or 'newer'. user (str): Only list revisions made by this user. excludeuser (str): Exclude revisions made by this user. limit (int): The maximum number of revisions to return per request. prop (str): Which properties to get for each revision, default: 'ids|timestamp|flags|comment|user' expandtemplates (bool): Expand templates in rvprop=content output section (int): Section number. If rvprop=content is set, only the contents of this section will be retrieved. diffto (str): Revision ID to diff each revision to. Use "prev", "next" and "cur" for the previous, next and current revision respectively. slots (str): The content slot (Mediawiki >= 1.32) to retrieve content from. uselang (str): Language to use for parsed edit comments and other localized messages. Returns: mwclient.listings.List: Revision iterator rv)startidendidstartendrr excludeuserdifftorVN)rE ZrvslotsZrvdirZrvpropuselangrqZrvexpandtemplatesZ rvsectionr[rT)rr.rrrr"versionZRevisionsIterator)r8rrrrdirrrrrTr r_rUrrVrrgrrrr[s  zPage.revisionscCsbtjjd|}ttjjj||d}|rBtjj|d|fi|Stjj|d|fddi|SdS)zqList templates used on the current page. API doc: https://www.mediawiki.org/wiki/API:Templates tlr templatesrrNr)r8r0rrrgrrrrszPage.templates)NN)F)NFTrR)rFTN)rFTN)rFTN)rTF)rFFF)NrFNT)TN)NrNT)T)NTF)NNNNrNNrrFNNNN)NT)"__name__ __module__ __qualname__r9r>r?rBrC staticmethodr1rMrOrPrdrhrerorprjrrrrrrrrrrrrrr[rrrrrr sN :     1   A          4r) r&rr]Z mwclient.utilrZmwclient.listingr.Zmwclient.errorsobjectrrrrrs