U
    ;»{j^  ã                   @   sD   d dl Z d dlZd dlmZ G dd„ deddƒƒZG dd„ dƒZdS )é    N)Ú
namedtuplec                       s    e Zd ZdZ‡ fdd„Z‡  ZS )ÚExtTypez'ExtType represents ext type in msgpack.c                    sR   t |tƒstdƒ‚t |tƒs$tdƒ‚d|  kr8dksBn tdƒ‚tƒ  | ||¡S )Nzcode must be intzdata must be bytesr   é   zcode must be 0~127)Ú
isinstanceÚintÚ	TypeErrorÚbytesÚ
ValueErrorÚsuperÚ__new__)ÚclsÚcodeÚdata©Ú	__class__© ú8/tmp/pip-install-zsn1f9xv/pip/pip/_vendor/msgpack/ext.pyr   	   s    

zExtType.__new__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Ú__classcell__r   r   r   r   r      s   r   z	code datac                   @   s’   e Zd ZdZddgZd dd„Zdd„ Zd	d
„ Zdd„ Zdd„ Z	e
dd„ ƒZdd„ Ze
dd„ ƒZdd„ Ze
dd„ ƒZdd„ Zdd„ Ze
dd„ ƒZdS )!Ú	TimestampaV  Timestamp represents the Timestamp extension type in msgpack.

    When built with Cython, msgpack uses C methods to pack and unpack `Timestamp`.
    When using pure-Python msgpack, :func:`to_bytes` and :func:`from_bytes` are used to pack and
    unpack `Timestamp`.

    This class is immutable: Do not override seconds and nanoseconds.
    ÚsecondsÚnanosecondsr   c                 C   sR   t |tƒstdƒ‚t |tƒs$tdƒ‚d|  kr8dk sBn tdƒ‚|| _|| _dS )aÆ  Initialize a Timestamp object.

        :param int seconds:
            Number of seconds since the UNIX epoch (00:00:00 UTC Jan 1 1970, minus leap seconds).
            May be negative.

        :param int nanoseconds:
            Number of nanoseconds to add to `seconds` to get fractional time.
            Maximum is 999_999_999.  Default is 0.

        Note: Negative times (before the UNIX epoch) are represented as neg. seconds + pos. ns.
        zseconds must be an integerznanoseconds must be an integerr   é Êš;z?nanoseconds must be a non-negative integer less than 999999999.N)r   r   r   r	   r   r   )Úselfr   r   r   r   r   Ú__init__   s    

zTimestamp.__init__c                 C   s   d| j › d| j› d�S )z#String representation of Timestamp.zTimestamp(seconds=z, nanoseconds=ú)©r   r   ©r   r   r   r   Ú__repr__5   s    zTimestamp.__repr__c                 C   s*   t |ƒ| jkr&| j|jko$| j|jkS dS )z0Check for equality with another Timestamp objectF)Útyper   r   r   ©r   Úotherr   r   r   Ú__eq__9   s    zTimestamp.__eq__c                 C   s   |   |¡ S )z(not-equals method (see :func:`__eq__()`))r%   r#   r   r   r   Ú__ne__?   s    zTimestamp.__ne__c                 C   s   t | j| jfƒS )N)Úhashr   r   r    r   r   r   Ú__hash__C   s    zTimestamp.__hash__c                 C   s€   t | ƒdkr"t d| ¡d }d}nTt | ƒdkrPt d| ¡d }|d@ }|d? }n&t | ƒdkrnt d	| ¡\}}ntd
ƒ‚t||ƒS )a  Unpack bytes into a `Timestamp` object.

        Used for pure-Python msgpack unpacking.

        :param b: Payload from msgpack ext message with code -1
        :type b: bytes

        :returns: Timestamp object unpacked from msgpack ext payload
        :rtype: Timestamp
        é   ú!Lr   é   ú!Ql   ÿÿ é"   é   ú!IqzFTimestamp type can only be created from 32, 64, or 96-bit byte objects)ÚlenÚstructÚunpackr	   r   )Úbr   r   Údata64r   r   r   Ú
from_bytesF   s    
ÿzTimestamp.from_bytesc                 C   s\   | j d? dkrF| jd> | j B }|d@ dkr8t d|¡}qXt d|¡}nt d| j| j ¡}|S )zÂPack this Timestamp object into bytes.

        Used for pure-Python msgpack packing.

        :returns data: Payload for EXT message with code -1 (timestamp type)
        :rtype: bytes
        r-   r   l       üÿ r*   r,   r/   )r   r   r1   Úpack)r   r4   r   r   r   r   Úto_bytesa   s    zTimestamp.to_bytesc                 C   s&   t | d ƒ}t | d d ƒ}t||ƒS )z›Create a Timestamp from posix timestamp in seconds.

        :param unix_float: Posix timestamp in seconds.
        :type unix_float: int or float
        é   r   )r   r   )Zunix_secr   r   r   r   r   Ú	from_unixv   s    zTimestamp.from_unixc                 C   s   | j | jd  S )znGet the timestamp as a floating-point value.

        :returns: posix timestamp
        :rtype: float
        g    eÍÍAr   r    r   r   r   Úto_unix�   s    zTimestamp.to_unixc                 C   s   t t| dƒŽ S )z—Create a Timestamp from posix timestamp in nanoseconds.

        :param int unix_ns: Posix timestamp in nanoseconds.
        :rtype: Timestamp
        r   )r   Údivmod)Zunix_nsr   r   r   Úfrom_unix_nano‰   s    zTimestamp.from_unix_nanoc                 C   s   | j d | j S )z~Get the timestamp as a unixtime in nanoseconds.

        :returns: posix timestamp in nanoseconds
        :rtype: int
        r   r   r    r   r   r   Úto_unix_nano’   s    zTimestamp.to_unix_nanoc                 C   s,   t jj}t j  d|¡t j| j| jd d� S )zRGet the timestamp as a UTC datetime.

        :rtype: `datetime.datetime`
        r   éè  )r   Úmicroseconds)ÚdatetimeÚtimezoneÚutcÚfromtimestampÚ	timedeltar   r   )r   rB   r   r   r   Úto_datetimeš   s
     ÿzTimestamp.to_datetimec                 C   s   t t|  ¡ ƒ| jd d�S )zQCreate a Timestamp from datetime with tzinfo.

        :rtype: Timestamp
        r>   r   )r   r   Ú	timestampÚmicrosecond)Údtr   r   r   Úfrom_datetime¤   s    zTimestamp.from_datetimeN)r   )r   r   r   r   Ú	__slots__r   r!   r%   r&   r(   Ústaticmethodr5   r7   r9   r:   r<   r=   rE   rI   r   r   r   r   r      s&   	





r   )r@   r1   Úcollectionsr   r   r   r   r   r   r   Ú<module>   s   