U
    \Shce                     @  s  d Z ddlmZ ddlZddlZddlmZ ddlmZm	Z	 ddl
mZ ddlmZmZ ddlmZmZ erdd	lmZmZ dd
lmZ dddddZdddddZdddddZG dd deZG dd dZG dd dZdAddddddd d!d"d#Zed$Zdddd%d&Z dBddd)dd*d+d,Z!ddd-d.d/Z"dCd1d d)dddddddd2d3d4d5Z#dd0ddddd(d6d dddddd7d)d8d9	d:d;Z$d<d7d=d>d?d@Z%dS )Dz
    babel.messages.pofile
    ~~~~~~~~~~~~~~~~~~~~~

    Reading and writing of files in the ``gettext`` PO (portable object)
    format.

    :copyright: (c) 2013-2025 by the Babel Team.
    :license: BSD, see LICENSE for more details.
    )annotationsN)Iterable)TYPE_CHECKINGLiteral)Locale)CatalogMessage)TextWrapper_cmp)IOAnyStr)SupportsWritestr)stringreturnc                 C  s"   dd }t d|| dd S )zReverse `escape` the given string.

    >>> print(unescape('"Say:\\n  \\"hello, world!\\"\\n"'))
    Say:
      "hello, world!"
    <BLANKLINE>

    :param string: the string to unescape
    c                 S  s2   |  d}|dkrdS |dkr"dS |dkr.dS |S )N   n
t	r)group)matchm r   H/root/rtd-docs/venv/lib/python3.8/site-packages/babel/messages/pofile.pyreplace_escapes&   s    
z!unescape.<locals>.replace_escapesz\\([\\trn"])r   )recompilesub)r   r   r   r   r   unescape   s    

r"   c                 C  sF   d| kr:|   }| dr&|dd }tt|}d|S t| S dS )a  Reverse the normalization done by the `normalize` function.

    >>> print(denormalize(r'''""
    ... "Say:\n"
    ... "  \"hello, world!\"\n"'''))
    Say:
      "hello, world!"
    <BLANKLINE>

    >>> print(denormalize(r'''""
    ... "Say:\n"
    ... "  \"Lorem ipsum dolor sit "
    ... "amet, consectetur adipisicing"
    ... " elit, \"\n"'''))
    Say:
      "Lorem ipsum dolor sit amet, consectetur adipisicing elit, "
    <BLANKLINE>

    :param string: the string to denormalize
    r   ""r   N )
splitlines
startswithmapr"   join)r   Zescaped_lineslinesr   r   r   denormalize3   s    


r*   z	list[str])liner   c                 C  s   d| krd| kr|    S g }d}d}| D ]n}|dkrL|rDtdd}q,q,|dkrh|s`tdd}q,q,|dkr|r~||7 }q|r|| d}q,||7 }q,|r|rtd|| |S )	zExtract locations from location comments.

    Locations are extracted while properly handling First Strong
    Isolate (U+2068) and Pop Directional Isolate (U+2069), used by
    gettext to enclose filenames with spaces and tabs in their names.
       ⁨   ⁩r$   Fzglocation comment contains more First Strong Isolate characters, than Pop Directional Isolate charactersTzglocation comment contains more Pop Directional Isolate characters, than First Strong Isolate characters )lstripsplit
ValueErrorappend)r+   	locationslocationZin_filenamecr   r   r   _extract_locationsR   s8    



r6   c                      s.   e Zd ZdZdddddd fddZ  ZS )	PoFileErrorzDException thrown by PoParser when an invalid po file is encountered.r   r   intNone)messagecatalogr+   linenor   c                   s,   t  | d|  || _|| _|| _d S )Nz on )super__init__r;   r+   r<   )selfr:   r;   r+   r<   	__class__r   r   r>      s    zPoFileError.__init__)__name__
__module____qualname____doc__r>   __classcell__r   r   r@   r   r7   ~   s   r7   c                   @  s   e Zd ZdddddZdddddZdd	d
dZdd	ddZdd	ddZdddddZdddddZ	dddddZ
dddddZdddddZdddddZdddd d!Zd"S )#_NormalizedStringr   r9   )argsr   c                 G  s   g | _ |D ]}| | q
d S N)_strsr2   )r?   rH   argr   r   r   r>      s    z_NormalizedString.__init__)sr   c                 C  s   | j |  d S rI   )rJ   r2   strip)r?   rL   r   r   r   r2      s    z_NormalizedString.appendr   c                 C  s   d tt| jS )Nr$   )r(   r'   r"   rJ   r?   r   r   r   r*      s    z_NormalizedString.denormalizeboolc                 C  s
   t | jS rI   )rP   rJ   rO   r   r   r   __bool__   s    z_NormalizedString.__bool__c                 C  s   t j| jS rI   )oslinesepr(   rJ   rO   r   r   r   __repr__   s    z_NormalizedString.__repr__objectr8   )otherr   c                 C  s   |sdS t t| t|S )Nr   )r
   r   r?   rV   r   r   r   __cmp__   s    z_NormalizedString.__cmp__c                 C  s   |  |dkS Nr   rX   rW   r   r   r   __gt__   s    z_NormalizedString.__gt__c                 C  s   |  |dk S rY   rZ   rW   r   r   r   __lt__   s    z_NormalizedString.__lt__c                 C  s   |  |dkS rY   rZ   rW   r   r   r   __ge__   s    z_NormalizedString.__ge__c                 C  s   |  |dkS rY   rZ   rW   r   r   r   __le__   s    z_NormalizedString.__le__c                 C  s   |  |dkS rY   rZ   rW   r   r   r   __eq__   s    z_NormalizedString.__eq__c                 C  s   |  |dkS rY   rZ   rW   r   r   r   __ne__   s    z_NormalizedString.__ne__N)rB   rC   rD   r>   r2   r*   rQ   rT   rX   r[   r\   r]   r^   r_   r`   r   r   r   r   rG      s   rG   c                   @  s   e Zd ZdZddddgZd#dddd	d
ddZd	dddZd	dddZd	dddZd$d	dddZ	d%d	dddZ
d	dddZd	dddZdd	dddZd	dd d!Zd"S )&PoFileParserzSupport class to  read messages from a ``gettext`` PO (portable object) file
    and add them to a `Catalog`

    See `read_po` for simple cases.
    msgidmsgstrmsgctxtmsgid_pluralFr   rP   r9   )r;   ignore_obsoleteabort_invalidr   c                 C  s*   || _ || _d| _d| _|| _|   d S rY   )r;   rf   counteroffsetrg   _reset_message_state)r?   r;   rf   rg   r   r   r   r>      s    zPoFileParser.__init__rN   c                 C  sF   g | _ g | _g | _g | _g | _g | _d | _d| _d| _d| _	d| _
d S )NF)messagestranslationsr3   flagsuser_commentsauto_commentscontextobsoletein_msgid	in_msgstr
in_msgctxtrO   r   r   r   rj      s    z!PoFileParser._reset_message_statec              
   C  sD  | j   t| jdkr.tdd | jD }n| jd  }t|ttfrdd t| j	j
D }| j D ]2\}}|| j	j
kr| d| jd qf| ||< qft|}n| j d d  }| jr| j nd	}t||t| jt| j| j| j| jd |d
}| jr | js*|| j	j| j	||< n
|| j	|< |  jd7  _|   d	S )z
        Add a message to the catalog based on the current parser state and
        clear the state ready to process the next message.
        r   c                 s  s   | ]}|  V  qd S rI   )r*   ).0r   r   r   r   	<genexpr>   s     z,PoFileParser._add_message.<locals>.<genexpr>r   c                 S  s   g | ]}d qS )r$   r   )ru   _r   r   r   
<listcomp>   s     z-PoFileParser._add_message.<locals>.<listcomp>r$   z5msg has more translations than num_plurals of catalogN)r<   rp   )rl   sortlenrk   tupler*   
isinstancelistranger;   num_plurals_invalid_pofileri   rp   r   r3   setrm   ro   rn   rq   rf   Z_key_forrh   rj   )r?   rb   r   idxtranslationrd   r:   r   r   r   _add_message   s4    

  
zPoFileParser._add_messagec                 C  sN   | j rJ| jsB| d| jd| j d   d | jdtdg |   d S )Nr$   zmissing msgstr for msgid 'r   ')rk   rl   r   ri   r*   r2   rG   r   rO   r   r   r   _finish_current_message   s
    "z$PoFileParser._finish_current_messagec                 C  s*   | dr| || n| ||| d S )N")r&   !_process_string_continuation_line_process_keyword_line)r?   r<   r+   rq   r   r   r   _process_message_line   s    
z"PoFileParser._process_message_linec              	   C  sH  | j D ]\}z4||r<|t| dkr<|t|d  }W  qvW q tk
r`   | ||d Y qX q| ||d d S |dkr|   || _|dkr|| _|dkrd| _d| _	| j
t| n|d	kr*d| _	d| _|d
r|dd  dd\}}| jt|t|g n| jdt|g n|dkrDd| _t|| _d S )N)r.   [z$Keyword must be followed by a stringz0Start of line didn't match any expected keyword.)rb   rd   rb   )rb   re   FTrc   r   r   ]r   rd   )	_keywordsr&   rz   
IndexErrorr   r   rq   ri   rt   rr   rk   r2   rG   rs   r0   rl   r8   rp   )r?   r<   r+   rq   keywordrK   r   msgr   r   r   r     s8    



z"PoFileParser._process_keyword_linec                 C  sV   | j r| jd }n6| jr(| jd d }n | jr6| j}n| ||d d S || d S )Nr   r   z<Got line starting with " but not in msgid, msgstr or msgctxt)rr   rk   rs   rl   rt   rp   r   r2   )r?   r+   r<   rL   r   r   r   r   -  s    z.PoFileParser._process_string_continuation_linec              	   C  s4  |    |dd  drt|dd  D ]r}|d}|dkrzt||d d  }W n tk
rp   Y q*Y nX | j|d | |f q*| j|d f q*n|dd  dr|dd   	dD ]}| j
|  qnP|dd  dr|dd   }|r0| j| n| j|dd    d S )Nr   :   r   ,.)r   r&   r6   rfindr8   r1   r3   r2   r/   r0   rm   rM   ro   rn   )r?   r+   r4   posr<   flagcommentr   r   r   _process_comment9  s&    

zPoFileParser._process_commentIO[AnyStr] | Iterable[AnyStr])fileobjr   c                 C  s  t |D ]\}}| }t|ts0|| jj}|s6q|dr|dd drp| j||dd 	 dd qz| 
| W q tk
r } z| ||t| W 5 d}~X Y qX q| || q|   | js| js| js| jr| jtd | jd	tdg |   dS )
z
        Reads from the file-like object `fileobj` and adds any po file
        units found in it to the `Catalog` supplied to the constructor.
        #r   N~r   T)rq   r#   r   )	enumeraterM   r|   r   decoder;   charsetr&   r   r/   r   r1   r   r   rh   rm   rn   ro   rk   r2   rG   rl   r   )r?   r   r<   r+   excr   r   r   parseT  s&    

&zPoFileParser.parsec                 C  sJ   t |tst| jr$t|| j||td| td|d  d| d S )NzWARNING:zWARNING: Problem on line r   z: )r|   r   AssertionErrorrg   r7   r;   print)r?   r+   r<   r   r   r   r   r   t  s
    
zPoFileParser._invalid_pofileN)FF)F)F)rB   rC   rD   rE   r   r>   rj   r   r   r   r   r   r   r   r   r   r   r   r   ra      s     ) ra   Fr   zLocale | str | Nonez
str | NonerP   r   )r   localedomainrf   r   rg   r   c                 C  s*   t |||d}t|||d}||  |S )a  Read messages from a ``gettext`` PO (portable object) file from the given
    file-like object (or an iterable of lines) and return a `Catalog`.

    >>> from datetime import datetime
    >>> from io import StringIO
    >>> buf = StringIO('''
    ... #: main.py:1
    ... #, fuzzy, python-format
    ... msgid "foo %(name)s"
    ... msgstr "quux %(name)s"
    ...
    ... # A user comment
    ... #. An auto comment
    ... #: main.py:3
    ... msgid "bar"
    ... msgid_plural "baz"
    ... msgstr[0] "bar"
    ... msgstr[1] "baaz"
    ... ''')
    >>> catalog = read_po(buf)
    >>> catalog.revision_date = datetime(2007, 4, 1)

    >>> for message in catalog:
    ...     if message.id:
    ...         print((message.id, message.string))
    ...         print(' ', (message.locations, sorted(list(message.flags))))
    ...         print(' ', (message.user_comments, message.auto_comments))
    (u'foo %(name)s', u'quux %(name)s')
      ([(u'main.py', 1)], [u'fuzzy', u'python-format'])
      ([], [])
    ((u'bar', u'baz'), (u'bar', u'baaz'))
      ([(u'main.py', 3)], [])
      ([u'A user comment'], [u'An auto comment'])

    .. versionadded:: 1.0
       Added support for explicit charset argument.

    :param fileobj: the file-like object (or iterable of lines) to read the PO file from
    :param locale: the locale identifier or `Locale` object, or `None`
                   if the catalog is not bound to a locale (which basically
                   means it's a template)
    :param domain: the message domain
    :param ignore_obsolete: whether to ignore obsolete messages in the input
    :param charset: the character set of the catalog.
    :param abort_invalid: abort read if po file is invalid
    )r   r   r   )rg   )r   ra   r   )r   r   r   rf   r   rg   r;   parserr   r   r   read_po|  s    6
r   zL(\s+|[^\s\w]*\w+[a-zA-Z]-(?=\w+[a-zA-Z])|(?<=[\w\!\"\'\&\.\,\?])-{2,}(?=\w))c                 C  s0   d|  dd dd dd dd	 d
d S )zEscape the given string so that it can be included in double-quoted
    strings in ``PO`` files.

    >>> escape('''Say:
    ...   "hello, world!"
    ... ''')
    '"Say:\\n  \\"hello, world!\\"\\n"'

    :param string: the string to escape
    z"%s"\z\\r   z\tr   z\rr   z\nr   z\")replace)r   r   r   r   escape  s        r   r$   L   r8   )r   prefixwidthr   c           
        s0  |r|dkrt  }g }| dD ]}t t|| |krt|}|  |rg }d}|rt t|d d | }	||	 |k r||  ||	7 }qX|s||  qqX|d| qLq"|| q"n
| d}t |dkrt| S |r|d s|d= |d  d7  < dd fd	d
|D  S )a  Convert a string into a format that is appropriate for .po files.

    >>> print(normalize('''Say:
    ...   "hello, world!"
    ... ''', width=None))
    ""
    "Say:\n"
    "  \"hello, world!\"\n"

    >>> print(normalize('''Say:
    ...   "Lorem ipsum dolor sit amet, consectetur adipisicing elit, "
    ... ''', width=32))
    ""
    "Say:\n"
    "  \"Lorem ipsum dolor sit "
    "amet, consectetur adipisicing"
    " elit, \"\n"

    :param string: the string to normalize
    :param prefix: a string that should be prepended to every line
    :param width: the maximum line width; use `None`, 0, or a negative number
                  to completely disable line wrapping
    r   Tr   r   r$   r   r   z""
c                   s   g | ]} t | qS r   )r   )ru   r+   r   r   r   rx     s     znormalize.<locals>.<listcomp>)	rz   r%   r   WORD_SEPr0   reverser2   popr(   )
r   r   r   	prefixlenr)   r+   chunksbufsizelengthr   r   r   	normalize  s6    


r   )filenamer   c                 C  s<   d| krd| kr| S |  ds&d|  } | ds8| d7 } | S )zEnclose filenames which include white spaces or tabs.

    Do the same as gettext and enclose filenames which contain white
    spaces or tabs with First Strong Isolate (U+2068) and Pop
    Directional Isolate (U+2069).
    r.   r   r,   r-   )r&   endswith)r   r   r   r   _enclose_filename_if_necessary  s    

r   TzSupportsWrite[bytes]r9   )r   r;   r   no_locationomit_headersort_outputsort_by_filerf   include_previousinclude_linenor   c
              
   C  sZ   d}
|rd}
n|rd}
t |||	||||
|dD ]&}t|trJ||jd}| | q.dS )a  Write a ``gettext`` PO (portable object) template file for a given
    message catalog to the provided file-like object.

    >>> catalog = Catalog()
    >>> catalog.add(u'foo %(name)s', locations=[('main.py', 1)],
    ...             flags=('fuzzy',))
    <Message...>
    >>> catalog.add((u'bar', u'baz'), locations=[('main.py', 3)])
    <Message...>
    >>> from io import BytesIO
    >>> buf = BytesIO()
    >>> write_po(buf, catalog, omit_header=True)
    >>> print(buf.getvalue().decode("utf8"))
    #: main.py:1
    #, fuzzy, python-format
    msgid "foo %(name)s"
    msgstr ""
    <BLANKLINE>
    #: main.py:3
    msgid "bar"
    msgid_plural "baz"
    msgstr[0] ""
    msgstr[1] ""
    <BLANKLINE>
    <BLANKLINE>

    :param fileobj: the file-like object to write to
    :param catalog: the `Catalog` instance
    :param width: the maximum line width for the generated output; use `None`,
                  0, or a negative number to completely disable line wrapping
    :param no_location: do not emit a location comment for every message
    :param omit_header: do not include the ``msgid ""`` entry at the top of the
                        output
    :param sort_output: whether to sort the messages in the output by msgid
    :param sort_by_file: whether to sort the messages in the output by their
                         locations
    :param ignore_obsolete: whether to ignore obsolete messages and not include
                            them in the output; by default they are included as
                            comments
    :param include_previous: include the old msgid as a comment when
                             updating the catalog
    :param include_lineno: include line number in the location comment
    Nr:   r4   rf   r   r   r   r   sort_byr   backslashreplace)generate_por|   r   encoder   write)r   r;   r   r   r   r   r   rf   r   r   r   r+   r   r   r   write_po  s$    8


r   r   z'Literal[('message', 'location')] | NonezIterable[str])	r;   rf   r   r   r   r   r   r   r   c             	   #  s  rdkrnd}t |ddt ddd}	d fdd		}
d! fd
d	}t |dD ]}|js|rjqX j}rdkrg }| D ]}||	|7 }qd|}| dV  |jD ]}|
|E dH  q|jD ]}|
|ddE dH  q|sg }zt	|j
dd d}W n tk
r&   |j
}Y nX |D ]P\}}|tjd}t|}|rf|rf| d|d}||kr,|| q,|
d|ddE dH  |jrddd"t	|j dV  |jr*|r*|
dt|jd d ddE dH  t|jdkr*t|jd d}|
d| ddE dH  ||E dH  dV  qX|st j |dD ]:}|jD ]}|
|E dH  qb||ddE dH  dV  qXdS )#zYield text strings representing a ``gettext`` PO (portable object) file.

    See `write_po()` for a more detailed description.
    r   r   F)r   break_long_wordsz# )r   subsequent_indentr   r$   c                 3  s,     | D ]}d| d|  dV  q
d S )Nr   r.   r   )wraprM   )r   r   r+   )comment_wrapperr   r   _format_comment  s    z$generate_po.<locals>._format_commentc              
   3  s@  t | jttfr| jr4| dt| j|d dV  | dt| jd |d dV  | dt| jd |d dV  t jD ]P}z| j| }W n t	k
r   d}Y nX | d	|d
dt||d dV  qnf| jr| dt| j|d dV  | dt| j|d dV  | dt| jp*d|d dV  d S )Nzmsgctxt )r   r   r   msgid r   msgid_plural r   r$   zmsgstr[dz] zmsgstr )
r|   idr}   r{   rp   r   r~   r   r   r   )r:   r   r   r   )r;   r   r   r   _format_message  s    ""
(z$generate_po.<locals>._format_message)r   r   Nr   r   c                 S  s"   | d t | d tr| d pdfS )Nr   r   r   )r|   r8   )xr   r   r   <lambda>      zgenerate_po.<locals>.<lambda>key/r   r   r.   r   z, r   )r   |r   r   z#~ )r$   )r$   )r$   )r	   _sort_messagesr   Zheader_commentr%   r   r(   rn   ro   sortedr3   	TypeErrorr   rR   sepr   r2   rm   Zprevious_idr   rz   rq   values)r;   rf   r   r   r   r   r   r   Zcomment_widthZheader_wrapperr   r   r:   Zcomment_headerr)   r+   r   Zlocsr3   r   r<   r4   Znorm_previous_idr   )r;   r   r   r   r   l  sr    






r   zIterable[Message]zlist[Message])rk   r   r   c                 C  s6   t | } |dkr|   n|dkr2| jdd d | S )z
    Sort the given message iterable by the given criteria.

    Always returns a list.

    :param messages: An iterable of Messages.
    :param sort_by: Sort by which criteria? Options are `message` and `location`.
    :return: list[Message]
    r:   r4   c                 S  s   | j S rI   )r3   )r   r   r   r   r     r   z _sort_messages.<locals>.<lambda>r   )r}   ry   )rk   r   r   r   r   r     s    

r   )NNFNF)r$   r   )r   FFFFFFT)&rE   
__future__r   rR   r   collections.abcr   typingr   r   Z
babel.corer   Zbabel.messages.catalogr   r   Z
babel.utilr	   r
   r   r   Z	_typeshedr   r"   r*   r6   	Exceptionr7   rG   ra   r   r    r   r   r   r   r   r   r   r   r   r   r   <module>   s\   
,
, K     <
=        $P"m