KJi\VdZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZmZejd ej dej!Z"e"j#Z$dZ%e%dZ&ej dZ'ej dZ(dZ)dddddddfdZ*d6dZ+de zdzZ,de zdzZ-dezdzZ.e+e,e-e.Z/e+e,e-e.dZ0d Z1d7d!Z2ej d"Z3d8d#Z4d$Z5d%Z6d&Z7eeeed'Z8e9d(e:ej;DZd*Z?d+Z@d,ZAd-ZBd9d.ZCd/ZDej d0ZEej d1ZFd2ZGd:d3ZHd7d4ZId;d5ZJdS).cgiPrint@sU EGO P P 1A    a   8#677A)rr s` rmakeCgiPrintEncodingr#>s# Or!utf8z %\((.+?)\)sz!(.+?)cfd}|S)aCreate a function useful to handle text[::optional_note] values. replaceTxtNotes is a format string, which can include the following values: %(text)s and %(notes)s. Portions of the text can be conditionally excluded, if one of the values is absent. E.g.: [%(notes)s] will be replaced with '[notes]' if notes exists, or by an empty string otherwise. The returned function is suitable be passed as applyToValues argument of the makeObject2Txt function.c}t|ts|S|dd}|d}i|rdd<|d|}t |dkr"dd<|d |d}n|d d }fd }t |r5t ||}t |5|S) N::rrTtextz%(text)snotesz %(notes)sc`|dvr|dSdSNrr)r+)group)matchobjkeysDicts r_excludeFalseConditionalszCmakeTextNotes.._replacer.._excludeFalseConditionalsms1~~a  H,,~~a(((2r!)rrsplitreplacelenre_conditionalsearchsub)routSssplitr(r1r0replaceTxtNotess @r _replacerz makeTextNotes.._replacer\s!S!! Hq!!ay  $#HV ||J-- v;;!   $HW << VAY77DD<< R00D      ##D)) G!%%&?FFD##D)) G r!r")r:r;s` r makeTextNotesr<Ss$4 r!z / c|SNr")xs rr@{s1r!Tc Pdddddfd S)a"Return a function useful to pretty-print Movie, Person, Character and Company instances. *movieTxt* -- how to format a Movie object. *personTxt* -- how to format a Person object. *characterTxt* -- how to format a Character object. *companyTxt* -- how to format a Company object. *joiner* -- string used to join a list of objects. *applyToValues* -- function to apply to values. *_recurse* -- if True (default) manage only the given object. Nz%(long imdb title)sz%(long imdb name)scdndkrdSdz tttfr"  fdDSttrA  fdtDSi}tt r j|d< }nhttr j |d <}nFttr j |d < }n$ttr j |d < }nSfd }t|r5t||}t|5t"|D]}|pt)|d}t|t*s#s|rt+|}|r  | }n|rt+|}|sd}n$t|t*st+|}|d|zdz|}|S)zPretty-print objects.Nrr+rc*g|]}|S)_limitRecursionr").0orF object2txts r z6makeObject2Txt..object2txt..s8 . . .$%!+ 1o N N N . . .r!cPg|]"\}}|d|#S)rEr'r")rGkvrFrIs rrJz6makeObject2Txt..object2txt..s]000Q(Z?KKKKK'Z?KKKKM000r!movieIDpersonID characterID companyIDc|d}|pt|d}|r|dSdSr-)r.getgetattr)r/ conditionproceedobjs rr1zEmakeObject2Txt..object2txt.._excludeFalseConditionalssU q))Iggi((IGCD,I,IG ~~a(((rr!rEz%(z)s)rlisttuplejoindictitemsr rNrrOr rPr rQr5r6r7re_substfindallrSrTrr3)rWrFobjDataoutsr1keyvalue_recurse applyToValues characterTxt companyTxtjoinermovieTxtrI personTxts`` rrIz"makeObject2Txt..object2txts  "OO q 21 cD%= ) ) ;; . . . . .), . . .// / T " " ;;00000!#))++..000   c5 ! ! !$GI DD V $ $ "%,GJ DD Y ' ' %(_GM "DD W % % #&=GK DDJ     ##D)) G!%%&?FFD##D)) G##D)) : :CGGCLL;GCd$;$;EeS)) 2++ #E O&JuoNNNE 2% c%jj11 #s++ #E <<s T 1599DD r!r>r")rhrirerfrgrdrcrIs```````@rmakeObject2Txtrjys(( + ) ;;;;;;;;;;;;;x r!cltdfd }|_|_|_|S)adMake a function used to pretty-print movies and persons refereces; movieTxt and personTxt are the strings used for the substitutions. movieTxt must contains %(movieID)s and %(title)s, while personTxt must contains %(personID)s and %(name)s and characterTxt %(characterID)s and %(name)s; characterTxt is optional, for backward compatibility.Nc8i fd}  fd} fd}|dddd}td |}tj||}t j||}t j||}|S) z)Substitute movies and persons references.Nc|d}|}|r'|j}|t|ddz}|S)Nrr)rNtitle)r.rSrNr)match to_replaceitemrN _cgiPrintrrh titlesRefss r _replaceMoviez;makeModCGILinks..modCGILinks.._replaceMovieskQJ>>*--D ,%& :!6!6BUVV))  r!c|d}|}|r'|j}|t|ddz}|S)Nrr)rOname)r.rSrOr)rorprqrOrrr namesRefsris r_replacePersonz.modCGILinks.._replacePersonskQJ==,,D =& ( * 5 5xATUU**  r!c|d}|S|}|r+|j}||S|t|ddz}|S)Nrr)rPrv)r.rSrPr)rorprqrPrr characterRefsrers r_replaceCharacterz?makeModCGILinks..modCGILinks.._replaceCharactersQJ#!! $$Z00D ". &%%)#. * 5 5xATUU--  r!z>z\1)r3 _re_hrefsubrr7rr) rrsrwrzrtrxr{rrrerrhris ``` r modCGILinksz$makeModCGILinks..modCGILinkss  M                         IIc6 " " * *3 7 7 . 2 2 OM1 - - N>1 - -   11 5 5r!r>)r#rhrire)rhrirerrrrs```` @rmakeModCGILinksrsi %X..I//////////`$K%K+K r!z %(title)sznm%(personID)s">%(name)szch%(characterID)s">%(name)s)rhrireascii)rhrirerct|di}||S)z4Return a sorted list of seasons of the given series.episodes)rXrSkeyssort)mseasonss r sortedSeasonsrs;155R((--//00G LLNNN Nr!cg}|}|t|}nt|ttfs|g}|D]}t|di|i}||D])}||d||*|S)z{Return a sorted list of episodes of the given series, considering only the specified season(s) (every season, if None).Nr)rrrYrXrSrrappend)rseasonrrreps_indxes rsortedEpisodesr!sHG ~""&5$-00 hG 11j"--11!R88==??@@  1 1A OOAjM!,Q/ 0 0 0 0 1 Or!z\b(nm|tt|ch|co)([0-9]{7,8})\bc|g}|i}t|i|}t|}|sdS|d}|d}|dkr|||S|dkr|||S|dkr|||S|dkr|||SdS) aJReturn a Movie, Person, Character or Company object for the given URL; info is the info set to retrieve, args and kwds are respectively a list and a dictionary or arguments to initialize the data access system. Returns None if unable to correctly parse the url; can raise exceptions if unable to retrieve the data.Nrr)tt)infonmchco)r _re_imdbIDurlr6r. get_movie get_person get_character get_company)urlrargskwdsiaroimdbtypeimdbIDs r get_byURLr7s  | | t t  B   % %E t{{1~~H [[^^F4||F|... T  }}V$}/// T  T222 T  ~~f4~000 4r!c*|S)aGGiven an URL string or a Movie, Person or Character instance, returns an URL to the full-size version of the cover/headshot, or None otherwise. This function is obsolete: the same information are available as keys: 'full-size cover url' and 'full-size headshot', respectively for movies and persons/characters.)get_fullsizeURL)rWs rfullSizeCoverURLrUs     r!c.t|ddS)zReturn a key (the ones used to access information in Movie and other classes instances) converted to the style of the XML output.r+r)rras rkeyToXMLr^s C  Q r!c:tt|S)zTranslate a given key.)r rrs r translateKeyrds Xc]]  r!)personmovie charactercompanyc"g|] }|ddf S)rNr"rGr?s rrJrJrs JJJqqtTlJJJr!c|d}|r*|ddkrt|}|S|jS)zOReturn the name of the tag, taking it from the 'key' attribute, if present.rakeytypeint)rSrtag)rkeyAttrs rtagToKeyrvsIggennG 779   & &'llG 7Nr!c|d}|dkrt|}n|dkrt|}|S)z0Return tagValue, handling some type conversions.typerfloat)rSrr)rtagValuetagTypes r_valueWithTypersDggfooG%x== G  ?? Or!) imdbindexkindyear)r)rcountrycl d|i}|i}t||pdgd}|jpd}|sJdkrD|}|r.|djr!|djpd}|d}|r.||<||g|jtvrt|d} | r|s| }n|} | _ g} |d} dkrU| _ t} | d } | / | j|| nʼnd kr;| _t } | d } | s| d } nSd kr$| _t } | d } n)dkr#| _t&} | d } | /| j|| | D]M}|jvr | |}|*t-||jjt|<N| d}|!|j_||| d}|1t1|d|||jd<||| d}|0t1|d|||}|_|||sSfd}n|rO| d}|&|jpd}|r|d|zz }nt-||}|S|(t|}|t4vrgfd}nifd}nifd}|D]X}t|}|jtvr|vr6t1|d|||}|r |||Y|r=tvr4|_|_t=|_ S)z!Recursively parse a tree of tags.Nrr+rqinfosetz access-systemidrrnrzlong imdb canonical namervrrr*z episode-ofF) _topLevel_as _infoset2keys _key2infosetz episode ofz current-rolec<j||iSr>)dataupdaterarbrqs rr@zparseTags..sDI$4$4c5\$B$Br!z::%sc.|Sr>)rrs rr@zparseTags..s E(:(:r!c2||iSr>rrs rr@zparseTags..s S%L(A(Ar!c2|iSr>r)rarbrqrvs rr@zparseTags..sDKKu $>$>r!)!r getchildrenr(striprS setdefaultrr _MAP_TOP_OBJ accessSystemrN _titleTagsfind set_titleremoverO _nameTagsrPrQ _companyTagsset_namerrr* parseTags currentRole _TAGS_TO_LIST_additional_keys infoset2keys key2infosetrXr current_info)rrrrr firstChildtagStr tagContentritemAs tagsToGettheIDttitletheNametdataTagnotesTag episodeOfcRolecr_addertagNotesr*firstChildNamesubTag subTagKeysubItemrqrvs @@rrrsK D  C==D//##-vq1Jhn" # # % %F 8dfnn__&&  8*Q-, 8 m(.B5577Fggi  G;$ T  "--44T::: w,D!##))   F"   7?? DL"IXXg&&F!v{+++ 6"""x % % ((#=>>/!hhv..G$$#( % ((6**""!&( ((6**" gl+++ 7### U UADI~~hhqkkG"/=gw|/T/T (7++,88G$$  !DJ JJx HH\**  &/ U47}=I'K'K'KDIl # JJy ! ! !((  5Es)6\SSSB!D  JJu    KBBBB ?88G$$  ](b//11E )&5.(#C00F  !*-- ] * *D::::FFDAAAAFF>>>>>//##''V$$ 7l " "yD4I4I4K4K'K'K Fe*7lTTT  ' F9g & & &7T\)))' !3!3!5!566 Kr!cXddl}t|j|S)ziParse a XML string, returning an appropriate object (usually an instance of a subclass of _Container.rN) lxml.etreeretree fromstring)xmllxmls rparseXMLrs. TZ**3// 0 00r!z (?:[(])([a-zA-Z]+?)(?: title[)])z\(.*?\)cg}t|dpg|dpgz}|D]}tjd|dd}t |dkrt |d}|r|d}ne|ddd}t d| }tj|}nd }| ||df|S) zhGiven a movie, return a list of tuples in (lang, AKA) format; lang can be None, if unable to detect.akaszakas from release infoz^(.*) \((.*?)\)rr)r,r+N)setrSrer6r.r4 _re_akas_langgroupsr2_re_akas_countryr7rr r)r lang_and_akarakalanguagers r akasLanguagesrs.L  &!!'REII6N,O,O,USUV W WD00i*C0066q!<< s88q==$++CF33H 5#??,,Q/a&,,s++A.*..r7;;AACC'+G44HXs1v.//// r!cN|}|d}|}g}tjd||}|||ddft |D]f\}} | }tjd||}|r ||kr|dz }||| |fg|d|r d|DS|S)a@Return a list of movie AKAs, sorted by their similarity to the given title. If _titlesOnly is not True, similarity information are returned. If _preferredLang is specified, AKAs in the given language will get a higher score. The return is a list of title, or a list of tuples if _titlesOnly is False.rnNrTreversecg|] }|d Srr"rs rrJz(sortAKAsBySimilarity..O%%%!%%%r!) guessLanguagelowerdifflibSequenceMatcherratiorrr) rrn _titlesOnly_preferredLangrm_titlel_titlescoresscorers rsortAKAsBySimilarityr6s-""$$HGn""$$GkkmmG F  #D'7 ; ; A A C CE MM5%.$/000&u--.. #))++'gw??EEGG  n88 QJE uc8,---- KKK&%%f%%%% Mr!c \g}t|D] \}}||kr||!|rwg}|D]P}|tjd|||Q|dd|D}|S)zReturn a list of AKAs of a movie, in the specified language. If _searchedTitle is given, the AKAs are sorted by their similarity to it.NTr cg|] }|d Sr r"rs rrJz%getAKAsInLanguage..ar r!)rrrrrr)rlang_searchedTitlerrrrs rgetAKAsInLanguagerSs D&u-- # 8   KK   & P PC MM'1$ 2@2F2F2H2HJJKN P P P P D !!!%%f%%% Kr!c,|r|nd} tj||d}n #t$rtd|zwxYwd|vr|dz }d|vr|dz }n|dz }|r|d|zz }|r|d |zz }|rd |}|d |zz }|d z }|S) z%Return resized and cropped image url.z(https://m.media-amazon.com/images/\w/\w+rz)Image url not matched. Original url: "%s"z@@@z._V1_z@._V1_zSX%s_zSY%s_rzCR%s_z.jpg)rr^ IndexErrorrrZ)imagewidthheightcrop custom_regex regexString resultImagecropValss r resizeImager*es#/_,,4_KOje44Q7 OOOCuMNNNO u}}s  %w x  'w&  (w''  *88D>>w)) 6K s $A)Nr$r>)NNN)TNNN)TN)NNNN)K__doc__rgettextrhtmlrimdbrrrrimdb._exceptionsrimdb.Characterr imdb.Companyr imdb.linguisticsr imdb.localer imdb.Movier imdb.Personr imdb.utilsrrrrr textdomaincompileI_re_hrefr7r~r#r r]r5r<rjr _movieTxt _personTxt _characterTxt modHtmlLinksmodHtmlLinksASCIIrrrrrrrrr[rXvaluesrrrrrrrrrrrrrrr*r"r!rrAs0$  '&&&&&$$$$$$ ))))))8 2:-rt 4 4l      ' ' 2:n % %@AA###L!Dt"5!,tTTTTn;;;;~ , ,/M M . .1O O 4412  Z,9;;; #OY*1>-4666 & ;<< <!!!      JJDD1F1F1H1H,I,IJJJKK \"""+  ' zzzzz111 =>> 2:j)) 6:$r!