RiT dZddlZddlZddlZddlmZddlmZmZm Z eje Z gdZ GddeZddZ dd ed ed ed ed ef dZdZdZdZdZdZdZddZdZdZdZdZdZy)z A set of functions useful for customizing bibtex fields. You can find inspiration from these functions to design yours. Each of them takes a record and return the modified record. N)str)latex_to_unicodestring_to_latexprotect_uppercase) splitnamegetnamesauthoreditorjournalkeywordlinkpage_double_hyphendoitypeconvert_to_unicodehomogenize_latex_encodingadd_plaintext_fieldsceZdZdZy) InvalidNamez[Exception raised by :py:func:`customization.splitname` when an invalid name is input. N)__name__ __module__ __qualname____doc__R/home/agent/.friday_env/lib/python3.12/site-packages/bibtexparser/customization.pyrrs  rrcj td}gg}gg}g}d}d}d}d} d} t|} | D]9} | dk(rt| } | |vr|j| | } nd|rd}| j } d} n*|dk(r%| j r| j rd}nd}|j| |j| | d k(r|dz }|j| d}d} d} d}| d k(rM|r|dz}n.|rt d j||jdd d} d} |j| |rT| r| j s/d} n,| r*|dk(r%| j r| j rd}nd}|j| \| d k(s| |vr|r?|djd j|g}|dj|d}d} d} | d k(rMt|dkr#|jg|jgn|rt dj||j| |dk(s| j s$| j rd}8d}<|r7|rt dj||r|jd |dz}|r|r7|djd j||dj||dsLt|dkDr|rt dj||jd|jd|rtd|DsiSggggd}t|dk(r|d}t|dk(r||d<|St|dk(r|dd|d<|dd|d<|S|d}d|vrb|jdt|z }|dddjd dz }|dk(r|dz}|d||d<|||dz|d<||dzd|d<|S|dd|d<|dd|d<|S|d}|r |dr||d<t|dk(r|d}|r |dr||d<|d}t|dk(r||d<|S|d}d|vrKt||dddjdz }|t|k(rd}|dd||d<|d|d|d<|S|d|d<|S)a Break a name into its constituent parts: First, von, Last, and Jr. :param string name: a string containing a single name :param Boolean strict_mode: whether to use strict mode :returns: dictionary of constituent parts :raises `customization.InvalidName`: If an invalid name is given and ``strict_mode = True``. In BibTeX, a name can be represented in any of three forms: * First von Last * von Last, First * von Last, Jr, First This function attempts to split a given name into its four parts. The returned dictionary has keys of ``first``, ``last``, ``von`` and ``jr``. Each value is a list of the words making up that part; this may be an empty list. If the input has no non-whitespace characters, a blank dictionary is returned. It is capable of detecting some errors with the input name. If the ``strict_mode`` parameter is ``True``, which is the default, this results in a :class:`customization.InvalidName` exception being raised. If it is ``False``, the function continues, working around the error as best it can. The errors that can be detected are listed below along with the handling for non-strict mode: * Name finishes with a trailing comma: delete the comma * Too many parts (e.g., von Last, Jr, First, Error): merge extra parts into First * Unterminated opening brace: add closing brace to end of input * Unmatched closing brace: add opening brace at start of word z ~ rFTN\{}z(Unmatched closing brace in name {{{0}}}.,z$Too many commas in the name {{{0}}}.z/Unterminated opening brace in the name {{{0}}}.z&Trailing comma at end of name {{{0}}}.c32K|]}t|ywN)bool).0sections r zsplitname..sEtG}Es)firstlastvonjrr-r,r.r/)setiternextappendisalphaisupperrformatinsertjoinlenpopanyindex)name strict_mode whitespacesectionscaseswordcaselevel bracestart controlseq specialcharnameitercharescapedpartsp0firstllastlr,r/r-lcasessplits rrrsPZJ tH DE D D EJJKDzHi 4<8nG*$ D!!&J!(!2J"&Kbjgoo&7(   D! G$ 3; QJE KK JJK   3; %&P&W&WX\&]^^ As#JK KK   ||~!&JBJDLLN||~   KK   3;$*, ##BGGDM2b   &" # s{x=1$OOB'LL$ %&L&S&STX&YZZ  D BJDLLN||~SiX OVVW[\] ] KK  QJE   BGGDM* b  B< MA ;FMMdST T R "  3EHEE "Rr :E 8} a[ r7a<E&MP LKW\VE'NqrFE&MF L!HEEzQ#e*4tt**1--1B;QJE"$GVg!&3e "519:f Z LS"$CRg "23f P LG  U1X"E'N x=A "Bbe d { t9> E&M* L#1XF F{F fTrTl&8&8&;;CK'E'{6E2e ( EF 3f  L!) f  Lrtextopeningclosingignore_escapedreturncg}g}|r d|z}d|z}tj||D]$}|j|jd&tj||D]'}|jd|jdz)t |dk(rt |dk(riSt |t |k7rt d|d|dt ||zd}i}g} |D]Q}|dk\r| j|t | dk(rt d |d |d | j} d|z|| <St | dkDr t d |d | jd |S)af Find matching 'brackets'. :param text: The string to consider. :param opening: The opening bracket (e.g. "(", "[", "{"). :param closing: The closing bracket (e.g. ")", "]", "}"). :param ignore_escaped: Ignore escaped bracket (e.g. "\(", "\[", "\{", "\)", "\]", "\}"). :return: Dictionary with ``{index_opening: index_closing}`` z(?zfind_matching..Qs 3q6r)keyz No closing z at: dz No opening z at )refinditerr5spanr; IndexErrorsortedr<) rSrTrUrVabr[bracketsretstackjs r find_matchingrj.s} A Aw&w& [[$ ' ![[$ '# affhqk!"# 1v{s1v{  1vQ;wis7)6BCCa!e!12H C E  6 LLO5zQ ;wiuQqE!BCC A!VCF 5zA~;witEIIK?CDD Jrcg}|D]}|j}t|dkr#d|vrT|jdd}|dj}|djDcgc]}|j}}nd|vrd|vr{ t|dd}g}d}d} ||vr||}n|dz }|t|k\rnX||dk(r|j ||||dz}n%|t|dz k(r|j ||df|j}|j }|Dcgc]"}|jdd j$}}|d vr|j }|D]} | d vs|j dz|z}|j |d zdj|z|Scc}w#t$r|j |YwxYwcc}w) a^Convert people names as surname, firstnames or surname, initials. :param names: a list of names :type names: list :returns: list -- Correctly formated names .. Note:: This function is known to be too simple to handle properly the complex rules. We would like to enhance this in forthcoming releases. r r#rr!r" N.z. )jnrr/junior)benvanderdelalez, ) stripr;rRrjrbr5r<replacer:) names tidynames namestring namesplitr-r[firstsrfstartitems rrresI'9 %%' z?Q   * "((a0IQ<%%'D)21););)=>Aaggi>F>j SJ%6,ZcBH H}$QKQC O+!!}+!((E!)<= !Ac*o11!((EF);<',,. ==?D>zz|c)D0 1 sxx'778O'9P C? "$$Z0*Gs&F1 F6''G6GGc d|vrk|drcttjd|djddtjDcgc]}|j c}|d<|S|d=|Scc}w)z Split author field into a list of "Name, Surname". :param record: the record. :type record: dict :returns: dict -- the modified record. r z\ and\  rl)flags)rr_rRrw IGNORECASErvrecordr[s rr r s6 ( 'BHHZQWX`QaQiQijnpsQtKM===Z)[q)[ \F8  Mx M )[s A2c`d|vr|drt|djddjdDcgc]}|jc}|d<|dDcgc]7}||jddjddjddd9c}|d<|S|d=|Scc}wcc}w) z Turn the editor field into a dict composed of the original editor name and a editor id (without coma or blank). :param record: the record. :type record: dict :returns: dict -- the modified record. r rrlz and r#r$rmr?ID)rrwrRrvrs rr r s6 ( 'F8|d|djddjddjddd|d<|S)z Turn the journal field into a dict composed of the original journal name and a journal id (without coma or blank). :param record: the record. :type record: dict :returns: dict -- the modified record. r r#r$rlrmrrwrs rr r scF ) )/ ):'-i'8'@'@b'I'Q'QRUWY'Z'b'bcfhj'k!mF9  Mrcd|vrHtj||djddDcgc]}|jc}|d<|Scc}w)z Split keyword field into a list. :param record: the record. :type record: dict :param sep: pattern used for the splitting regexp. :type record: string, optional :returns: dict -- the modified record. r rr$)r_rRrwrv)rsepr[s rr r sQF02fY>O>W>WX\^`>a0bc1QWWYcy MdsAcd|vr|djdDcgc]"}|jjdd$}}g|d<|D]l}|jd}d|di}t|dkDr|d|d<t|d kDr|d |d <t|ddkDsY|dj |n|Scc}w) h :param record: the record. :type record: dict :returns: dict -- the modified record. r rz rlurlrr anchorr0r8)rRrvrwr;r5)rr[linksr rMlinkobjs rr r s7=f~7K7KD7QR!""4-RRv /DJJsOEeAh'G5zA~$)!H!5zA~$)!H!75>"Q&v%%g. / MSs'B<cd|vrPd|vrg|d<d}|dD] }d|vsd} |r2|d}|jdrd|z}|dj|dd|S)rrr TF10zhttps://doi.org/)rr) startswithr5)rnodoir~r s rrr"s   F6N6N D}  %=Dt$)D0 6N ! !$%"@ A Mrc @|D]}t||tr ||Dcgc] }t|c}||<6t||tr4||j Dcic]\}}|t|c}}||<}t||||<|Scc}wcc}}w)z Convert accent from latex to unicode style. :param record: the record. :type record: dict :returns: dict -- the modified record. ) isinstancelistrdictitems)rvalxkvs rrr9s 8 fSk4 (-3C[() #F3Ks T *39#;3D3D3F+/1a#A&&F3K+6#;7F3K 8 Ms B"Bct|}|D]}|dvstjd|t||tr ||Dcgc] }t |c}||<nBt||t rt ||||<ntjd|zt|dk(stjdtjd||t||||<tjd|||Scc}w)z Homogenize the latex enconding style for bibtex This function is experimental. :param record: the record. :type record: dict :returns: dict -- the modified record. )rzApply string_to_latex to: %szDUnable to homogenize latex encoding for %s: Expected string or list,titlezProtect uppercase in titlez Before: %sz After: %s) rloggerdebugrrrrwarningswarnRuntimeWarningr)rrrs rrrOs  'F7 g  LL7 =&+t,06s +,OA&s F3K--fSk:s  dgjj,.g~ 9: \6#;7/s <s  [&+6!7" MsDc d}t|jD]}dj|}||||<t||tr|||||<>t||t r1||j Dcic]\}}|||c}}||<t||ts||Dcgc] }|| c}||<|Scc}}wcc}w)a% For each field in the record, add a `plain_` field containing the plaintext, stripped from braces and similar. See https://github.com/sciunto-org/python-bibtexparser/issues/116. :param record: the record. :type record: dict :returns: dict -- the modified record. c8dD]}|j|d}|S)N)r!r"r$r)stringstrippeds r _strip_stringz+add_plaintext_fields.._strip_string{s&" 2H^^Hb1F 2 rzplain_{})rkeysr8rrrr)rrr] plain_keysubkeyvalues rrrps FKKM"%%c* "3Ky fY' - -fY.? @F9  y)4 0&,I%6%<%<%>!!FE e,,!F9 y)4 0$I.!e$!F9 " M! !s C:C)T)z,|;) rloggingr_rbuiltinsrbibtexparser.latexencrrr getLoggerrr__all__ ValueErrorrrr(rrjrr r rrr r r rrrrrrrrs  VV   8 $ @  * Mh 4 4 44 4  4n6r$*& &"0.,B!r