U

JD�g��@sxdZd
dl
Z
d
dlZd
dlZd
dlmZ
ej�d
ejj�Z	dZ
dZe�dej
�Zdd�Zdd
d�
ZGdd
�d
ej�ZdS)zHelpers for :mod:`datetime`.�N)
�
timestamp_pb2z%Y-%m-%dT%H:%M:%S.%fZz%Y-%m-%dT%H:%M:%Sah

    (?P<no_fraction>
        \d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}  # YYYY-MM-DDTHH:MM:SS
    )
    (                                        # Optional decimal part
     \.                                      # decimal point
     (?P<nanos>\d{1,9})                      # nanoseconds, maybe truncated
    )?
    Z                                        # Zulu
c

Cstt
j|d
�
S)a
Convert timestamp in microseconds since the unix epoch to datetime.

    Args:
        value (float): The timestamp to convert, in microseconds.

    Returns:
        datetime.datetime: The datetime object equivalent to the timestamp in
            UTC.
    )
�microseconds)�
_UTC_EPOCH�datetime�	timedelta)
�value�r�C/opt/cppython/lib/python3.8/site-packages/proto/datetime_helpers.py�_from_microseconds,s
r
TcCs,|
s"|jd
k	r"|j
d
d�
|��}|�t�
S)a�
Convert a datetime to an RFC3339 timestamp string.

    Args:
        value (datetime.datetime):
            The datetime object to be converted to a string.
        ignore_zone (bool): If True, then the timezone (if any) of the
            datetime object is ignored and the datetime is treated as UTC.

    Returns:
        str: The RFC3339 formatted string representing the datetime.
    N�
�tzinfo)r�replace�	utcoffset�strftime�_RFC3339_MICROS)rZignore_zonerrr	�_to_rfc33399srcs`eZ
dZd
ZdZdd�Z�f
dd�Zedd��
Zd	d
�Z	e
dd��
Zd
d�Ze
dd��
Z
�ZS)�DatetimeWithNanosecondszuTrack nanosecond in addition to normal datetime attrs.

    Nanosecond can be passed only as a keyword argument.
    )
�_nanosecondc
OsR|�d
d�}|dkr0d|kr$t
d�
�
|d|d<tjj|f
|
�|�
}|pJd|_|S)N�
nanosecondr
�microsecond�1Specify only one of 'microsecond' or 'nanosecond'��)�pop�	TypeErrorr�__new__r)�cls�args�kw�nanos�instrrr	rUs








zDatetimeWithNanoseconds.__new__c
sxd
|k}d|k}|�dd�}|j
}|r2|r2td�
�
|rB|d|d
<t�j|
|�
}|rb|jd|_n|rn||_n||_|S)a@
Return a date with the same value, except for those parameters given
        new values by whichever keyword arguments are specified. For example,
        if d == date(2002, 12, 31), then
        d.replace(day=26) == date(2002, 12, 26).
        NOTE: nanosecond and microsecond are mutually exclusive arguments.
        rrr
rr)rrr�superr
rr)�selfrrZms_providedZns_providedZprovided_nsZ
prev_nanosr�
�	__class__rr	r
`s




zDatetimeWithNanoseconds.replacec

Cs|jp|j
d
S)z Read-only: nanosecond precision.r)rr)
r!rrr	r�sz"DatetimeWithNanoseconds.nanosecondc
Cs<|jd
krt
|�
St|j�
�dd��d�
}
d�|�t�
|
�S)z�Return an RFC3339-compliant timestamp.

        Returns:
            (str): Timestamp string according to RFC3339 spec.
        r
�	�
0z{}.{}Z)rr�str�rjust�rstrip�formatr�_RFC3339_NO_FRACTION)r!rrrr	�rfc3339�s



zDatetimeWithNanoseconds.rfc3339c
Cs�t�
|
�
}|d
kr$td�|
tj��
�
tj�|�d�
t�}|�d�
}|d
krPd}ndt	|�
}t
|�
d|}||j|j|j
|j|j|j|tjjd�S)	ak
Parse RFC3339-compliant timestamp, preserving nanoseconds.

        Args:
            stamp (str): RFC3339 stamp, with up to nanosecond precision

        Returns:
            :class:`DatetimeWithNanoseconds`:
                an instance matching the timestamp string

        Raises:
            ValueError: if `stamp` does not match the expected format
        Nz)Timestamp: {}, does not match pattern: {}Zno_fractionrr
r$�
�rr)�_RFC3339_NANOS�match�
ValueErrorr)�patternr�strptime�groupr*�len�int�year�month�day�hour�minute�second�timezone�utc)r�stampZ
with_nanos�bareZfractionr�scalerrr	�from_rfc3339�s6




��
�












�z$DatetimeWithNanoseconds.from_rfc3339c
CsP|jd
k	r|n|j
tjjd�
}
|
t}t|���
}|jp@|j	d}t
j||d�S)z�Return a timestamp message.

        Returns:
            (:class:`~google.protobuf.timestamp_pb2.Timestamp`): Timestamp message
        Nrr)�secondsr)rr
rr<r=rr5�
total_secondsrrr�	Timestamp)r!r�deltarBrrrr	�timestamp_pb�s��


z$DatetimeWithNanoseconds.timestamp_pbc
Cs@t|
j
d
�
}t|�
}||j|j|j|j|j|j|
j	t
jjd�S)a&
Parse RFC3339-compliant timestamp, preserving nanoseconds.

        Args:
            stamp (:class:`~google.protobuf.timestamp_pb2.Timestamp`): timestamp message

        Returns:
            :class:`DatetimeWithNanoseconds`:
                an instance matching the timestamp message
        g��.Ar-)
r5rBr
r6r7r8r9r:r;rrr<r=)rr>rr?rrr	�from_timestamp_pb�s









�z)DatetimeWithNanoseconds.from_timestamp_pb)�__name__�
__module__�__qualname__�__doc__�	__slots__rr
�propertyrr+�classmethodrArFrG�
__classcell__rrr"r	rLs
"



(
r)
T)rK�calendarr�reZgoogle.protobufr�
fromtimestampr<r=rrr*�compile�VERBOSEr.r
rrrrrr	�<module>s




�