CasperSecurity
o
;s*bX � @ s6 d Z dZdZddlZddlZddlZG dd� d�ZdS )z
Cyril Jaquierz Copyright (c) 2004 Cyril Jaquier�GPL� Nc
@ sD e Zd ZdZdZdZdZedd� �Zedd� �Z edd� �Z
ed d
� �Z ed d � �Z ed&d
d��Z
ed'dd��Ze�d�Ze�d�ZdZe�ed �ded� fe�ed �ded� fe�ed �ded� fe�ed �ded � fe�ed �ded � fe�ed �ded!� fe�ed" �ded#� ffZed$d%� �ZdS )(�MyTimea` A wrapper around time module primarily for testing purposes
This class is a wrapper around time.time() and time.gmtime(). When
performing unit test, it is very useful to get a fixed value from
these functions. Thus, time.time() and time.gmtime() should never
be called directly. This wrapper should be called instead. The API
are equivalent.
Nc C s* | t _| durtj�| �t _dS dt _dS �zrSet current time.
Use None in order to always get the real current time.
@param t the time to set or None
N)r �alternateNowTime�datetime�
fromtimestamp� alternateNow��t� r �8/usr/lib/python3/dist-packages/fail2ban/server/mytime.py�setAlternateNow/ s
��zMyTime.setAlternateNowc C s
| t _dS r )r �myTimer r r r �setTime< s
zMyTime.setTimec C s t jdu r t�� S t jS )zqDecorate time.time() for the purpose of testing mocking
@return time.time() if setTime was called with None
N)r r �timer r r r r G s
z MyTime.timec C s t jdu r t�� S t�t j�S )zuDecorate time.gmtime() for the purpose of testing mocking
@return time.gmtime() if setTime was called with None
N)r r r �gmtimer r r r r S s
z
MyTime.gmtimec C s4 t jdu r
tj�� S t jt jkrt jS tj�t j�S )zwDecorate datetime.now() for the purpose of testing mocking
@return datetime.now() if setTime was called with None
N)r r r �nowr r r r r r r r ^ s
z
MyTime.nowc C s( t jdu s | durt�| �S t�t j�S )z{Decorate time.localtime() for the purpose of testing mocking
@return time.localtime() if setTime was called with None
N)r r r � localtime)�xr r r r j s
zMyTime.localtime�%Y-%m-%d %H:%M:%Sc C s: | dk rt j �| �jdd�n t � dddddd�}|�|�S ) z�Convert time to a string representing as date and time using given format.
Default format is ISO 8601, YYYY-MM-DD HH:MM:SS without microseconds.
@return ISO-capable string representation of given unixTime
l p�� r )� microsecondi' � � � �; )r r �replace�strftime)�unixTime�format�dtr r r �time2stru s
����
zMyTime.time2strz(?i)(?<=[a-z])(\d)z (\d)\s+(\d)z(?i)(?<=[\d\s])(%s)\bz days?|da|dd?�*i�Q zweeks?|wee?|ww?i�: z months?|mon?g TDAzyears?|yea?|yy?g ~~Azseconds?|sec?|ss?� zminutes?|min?|mm?�< zhours?|hou?|hh?i c C sT t | tttf�r
| S tj�d| �} tjD ]
\}}|�|| �} qtj�d| �} t | �S )a) Wraps string expression like "1h 2m 3s" into number contains seconds (3723).
The string expression will be evaluated as mathematical expression, spaces between each groups
will be wrapped to "+" operand (only if any operand does not specified between).
Because of case insensitivity and overwriting with minutes ("m" or "mm"), the short replacement for month
are "mo" or "mon".
Ex: 1hour+30min = 5400
0d 1h 30m = 5400
1year-6mo = 15778800
6 months = 15778800
warn: month is not 30 days, it is a year in seconds / 12, the leap years will be respected also:
>>>> float(str2seconds("1month")) / 60 / 60 / 24
30.4375
>>>> float(str2seconds("1year")) / 60 / 60 / 24
365.25
@returns number (calculated seconds from expression "val")
z \1z\1+\2)
�
isinstance�int�float�complexr �
_str2sec_prep�sub�_str2sec_parts�
_str2sec_fini�eval)�val�rexp�rplr r r � str2seconds� s zMyTime.str2seconds)N)r )�__name__�
__module__� __qualname__�__doc__r r r � staticmethodr
r r r r r r �re�compiler( r+ �_str2sec_subpart�strr* r0 r r r r r ! s@
�
r )�
__author__�
__copyright__� __license__r r6 r r r r r r �<module> s