mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-12 15:30:04 +08:00
51 lines
8.2 KiB
Plaintext
51 lines
8.2 KiB
Plaintext
|
a
|
|||
|
|
§¢` ã @ s– d dl Z d dlmZmZ d dlmZ d dlmZ d dlZd dlZd dlZd dl m
|
|||
|
|
Z
|
|||
|
|
d dl m Z d dl
|
|||
|
|
Z
|
|||
|
|
edƒjZG dd „ d ƒZG d
|
|||
|
|
d „ d ƒZG d d
|
|||
|
|
„ d
|
|||
|
|
ƒZedƒZG dd„ dƒZG dd„ dƒZedœdd„Zd7eeeedœdd„Zdd„ Zeedœd d „Z eedœd d „Z eedœd d!„Z eedœd"d#„Z eedœd$d%„Z ee!dœd&d'„Z"d(d)„ Z#eed*œd+d,„Z$d-d.„ Z%d/d0„ Z&d1d2„ Z'd3d4„ Z(edœd5d6„Z)dS )8é N)ÚdatetimeÚ timedelta)Ú defaultdict)Úrequire)ÚTXT_PATH©Ú system_proxyZnonebot_plugin_apschedulerc @ s* e Zd Zdd„ Zdd„ Zedœdd„ZdS ) Ú CountLimiterc C s t tƒ| _|| _d S ©N)r ÚintÚcountÚmax)Úselfr
|
|||
|
|
© r ú4D:\Python\Python_work\hibikibot\hibiki\util\utils.pyÚ__init__ s
|
|||
|
|
zCountLimiter.__init__c C s | j | d7 < d S ©Né ©r ©r Úkeyr r r Úadd s zCountLimiter.add©Úreturnc C s" | j | | jkr d| j |< dS dS )Nr TF)r r
|
|||
|
|
r r r r Úcheck s
|
|||
|
|
zCountLimiter.checkN)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__r r Úboolr r r r r r s r c @ s, e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd S )
|
|||
|
|
ÚUserExistLimiterc C s t tƒ| _t ¡ | _d S r
|
|||
|
|
)r r ÚmboolÚtime)r r r r r " s
|
|||
|
|
zUserExistLimiter.__init__c C s t ¡ | _ d| j|< d S )NT)r! r r r r r Úset_True& s
|
|||
|
|
zUserExistLimiter.set_Truec C s d| j |< d S )NF)r r r r r Ú set_False* s zUserExistLimiter.set_Falsec C s* t ¡ | j dkr | |¡ dS | j| S )Né F)r! r# r r r r r r - s
|
|||
|
|
zUserExistLimiter.checkN)r r r r r" r# r r r r r r ! s r c @ s: e Zd Zdd„ Zedœdd„Zd dd„Zedœd d
|
|||
|
|
„Zd S )
|
|||
|
|
Ú FreqLimiterc C s t tƒ| _|| _d S r
|
|||
|
|
)r ÚfloatÚ next_timeÚ
|
|||
|
|
default_cd)r Zdefault_cd_secondsr r r r 6 s
|
|||
|
|
zFreqLimiter.__init__r c C s t ¡ | j| kS r
|
|||
|
|
)r! r' r r r r r : s zFreqLimiter.checkr c C s$ t ¡ |dkr|n| j | j|< d S ©Nr )r! r( r' )r r Zcd_timer r r Ústart_cd= s zFreqLimiter.start_cdc C s | j | t ¡ S r
|
|||
|
|
)r' r! r r r r Ú left_time@ s zFreqLimiter.left_timeN)r ) r r r r r r r* r& r+ r r r r r% 5 s
|
|||
|
|
r% é c @ s4 e Zd Zd eedœdd„Zdd„ Zedœd d
|
|||
|
|
„Zd S )
|
|||
|
|
ÚBanCheckLimiteré é )Údefault_check_timeÚ
|
|||
|
|
default_countc C s$ t tƒ| _t tƒ| _|| _|| _d S r
|
|||
|
|
)r r Úmintr& Úmtimer0 r1 )r r0 r1 r r r r I s
|
|||
|
|
|
|||
|
|
zBanCheckLimiter.__init__c C s2 | j | dkr t ¡ | j|< | j | d7 < d S r )r2 r! r3 r r r r r O s zBanCheckLimiter.addr c C s| t ¡ | j| | jkr4t ¡ | j|< d| j|< dS | j| | jkrxt ¡ | j| | jk rxt ¡ | j|< d| j|< dS dS )Nr FT)r! r3 r0 r2 r1 r r r r r T s
|
|||
|
|
(
|
|||
|
|
zBanCheckLimiter.checkN)r. r/ ) r r r r& r r r r r r r r r r- H s r- c @ sF e Zd Ze d¡Zdd„ Zedœdd„Zdd„ Z dd
|
|||
|
|
d „Z
|
|||
|
|
d d
|
|||
|
|
„ Z dS )ÚDailyNumberLimiterz
|
|||
|
|
Asia/Shanghaic C s d| _ ttƒ| _|| _d S )Néÿÿÿÿ)Útodayr r r r
|
|||
|
|
)r Zmax_numr r r r f s
|
|||
|
|
zDailyNumberLimiter.__init__r c C sJ t | j¡}|tdd� j}|| jkr6|| _| j ¡ t| j| | j k ƒS )Nr. )Úhours)
|
|||
|
|
r ÚnowÚtzr Údayr6 r Úclearr r
|
|||
|
|
)r r r8 r: r r r r k s
|
|||
|
|
|
|||
|
|
zDailyNumberLimiter.checkc C s
|
|||
|
|
| j | S r
|
|||
|
|
r r r r r Úget_nums s zDailyNumberLimiter.get_numr c C s | j | |7 < d S r
|
|||
|
|
r )r r Únumr r r Úincreasev s zDailyNumberLimiter.increasec C s d| j |< d S r) r r r r r Úresety s zDailyNumberLimiter.resetN)r ) r r r ÚpytzÚtimezoner9 r r r r< r> r? r r r r r4 c s
|
|||
|
|
|
|||
|
|
r4 r c C sV zt | ƒ W dS ty Y n0 zdd l}| | ¡ W dS ttfyP Y n0 dS )NTr F)r& Ú
|
|||
|
|
ValueErrorÚ unicodedataÚnumericÚ TypeError)ÚsrC r r r Ú is_number} s
|