U
    \Sha                     @  s   U 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	 erRddl
mZmZ dZded	< d
Zded< dddddZddddddddZdS )z
    babel.messages.mofile
    ~~~~~~~~~~~~~~~~~~~~~

    Writing of files in the ``gettext`` MO (machine object) format.

    :copyright: (c) 2013-2025 by the Babel Team.
    :license: BSD, see LICENSE for more details.
    )annotationsN)TYPE_CHECKING)CatalogMessage)SupportsReadSupportsWritel   * intLE_MAGICl   $< BE_MAGICzSupportsRead[bytes]r   )fileobjreturnc                   s\  t   i }t| dd}|  }t|}tj}|d|dd d }|tkrj|d|dd \}}}	}
d	}n4|tkr|d
|dd \}}}	}
d}ntdd|t	|D ]}||||	|	d  \}}|| }||||
|
d  \}}|| }||k r||k r||| }||| }ntdd||dkrd }}|
 D ]j}| }|sZqDd|kr|dd\}}|   }}| ||< n|rD||  d| 7  < qDd|kr|d\}}nd}d|kr|d}|d} fdd|D } fdd|D }n| j}| j}t|||d |< |	d7 }	|
d7 }
q|  _ S )aa  Read a binary MO file from the given file-like object and return a
    corresponding `Catalog` object.

    :param fileobj: the file-like object to read the MO file from

    :note: The implementation of this function is heavily based on the
           ``GNUTranslations._parse`` method of the ``gettext`` module in the
           standard library.
    name z<IN   r   z<4I   z<IIz>4Iz>IIzBad magic number   zFile is corrupt   :      
       c                   s   g | ]}|  jqS  decodecharset.0xcatalogr   H/root/rtd-docs/venv/lib/python3.8/site-packages/babel/messages/mofile.py
<listcomp>\   s     zread_mo.<locals>.<listcomp>c                   s   g | ]}|  jqS r   r   r   r   r   r    r!   ]   s     )context)r   getattrreadlenstructunpackr	   r
   OSErrorrange
splitlinesstripsplitlowerr   r   r   itemsZmime_headers)r   headersfilenamebufbuflenr'   magicversionmsgcountZorigidxtransidxiiZ_imlenmoffmendtlentofftendmsgtmsgZlastkeykeyitemvalueZctxtr   r   r    read_mo   sd    








rC   FzSupportsWrite[bytes]boolNone)r   r   	use_fuzzyr   c                   s  t  }fdd|dd D |dd< |  d }}g }|D ]}|jrd fdd|jD }g }	t|jD ]2\}
}|s|	|jtt	|
d  qt|	| qtd fdd|	D }n|j
 j}|j
 j}|jrd	|j
 j|g}|t|t|t|t|f ||d 7 }||d 7 }qBd
dt|  }|t| }g }g }|D ].\}}}}|||| g7 }|||| g7 }q`|| }| tdtdt|d
d
t|d  ddtjtd| | |  dS )a  Write a catalog to the specified file-like object using the GNU MO file
    format.

    >>> import sys
    >>> from babel.messages import Catalog
    >>> from gettext import GNUTranslations
    >>> from io import BytesIO

    >>> catalog = Catalog(locale='en_US')
    >>> catalog.add('foo', 'Voh')
    <Message ...>
    >>> catalog.add((u'bar', u'baz'), (u'Bahr', u'Batz'))
    <Message ...>
    >>> catalog.add('fuz', 'Futz', flags=['fuzzy'])
    <Message ...>
    >>> catalog.add('Fizz', '')
    <Message ...>
    >>> catalog.add(('Fuzz', 'Fuzzes'), ('', ''))
    <Message ...>
    >>> buf = BytesIO()

    >>> write_mo(buf, catalog)
    >>> x = buf.seek(0)
    >>> translations = GNUTranslations(fp=buf)
    >>> if sys.version_info[0] >= 3:
    ...     translations.ugettext = translations.gettext
    ...     translations.ungettext = translations.ngettext
    >>> translations.ugettext('foo')
    u'Voh'
    >>> translations.ungettext('bar', 'baz', 1)
    u'Bahr'
    >>> translations.ungettext('bar', 'baz', 2)
    u'Batz'
    >>> translations.ugettext('fuz')
    u'fuz'
    >>> translations.ugettext('Fizz')
    u'Fizz'
    >>> translations.ugettext('Fuzz')
    u'Fuzz'
    >>> translations.ugettext('Fuzzes')
    u'Fuzzes'

    :param fileobj: the file-like object to write to
    :param catalog: the `Catalog` instance
    :param use_fuzzy: whether translations marked as "fuzzy" should be included
                      in the output
    c                   s    g | ]}|j r s|js|qS r   )stringZfuzzy)r   m)rF   r   r    r!      s
      zwrite_mo.<locals>.<listcomp>r   N    r   c                   s   g | ]}|  jqS r   encoder   )r   msgidr   r   r    r!      s    c                   s   g | ]}|  jqS r   rJ   )r   msgstrr   r   r    r!      s    r         ZIiiiiiir   r   i)listsortZpluralizablejoinid	enumeraterG   appendminr   rK   r   r"   r%   writer&   packr	   arraytobytes)r   r   rF   messagesZidsstrsoffsetsmessagerL   ZmsgstrsidxrG   rM   ZkeystartZ
valuestartZkoffsetsZvoffsetsZo1l1o2l2r   )r   rF   r    write_mok   sd    0"
"
 rd   )F)__doc__
__future__r   rZ   r&   typingr   Zbabel.messages.catalogr   r   Z	_typeshedr   r   r	   __annotations__r
   rC   rd   r   r   r   r    <module>   s   	R