mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-11 06:39:59 +08:00
62 lines
8.9 KiB
Plaintext
62 lines
8.9 KiB
Plaintext
|
U
|
|||
|
|
d’y`B ã @ sš d dl Z d dlmZmZ d dlmZ d dlmZ d dlZd dlZd dlm Z d dl
|
|||
|
|
Z
|
|||
|
|
d dl Z d dl m
|
|||
|
|
Z
|
|||
|
|
edƒjZi ZG dd „ d ƒZG d
|
|||
|
|
d „ d ƒZG d d
|
|||
|
|
„ d
|
|||
|
|
ƒZedƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZedœdd„Zd7eeeedœdd„Z dd „ Z eed œ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/d0„ Z'd1d2„ Z(d3d4„ Z)d5d6„ Z*dS )8é N)ÚdatetimeÚ timedelta)Ú defaultdict)Úrequire)ÚSELF_ID)ÚTXT_PATHZnonebot_plugin_apschedulerc @ s* e Zd Zdd„ Zdd„ Zedœdd„ZdS ) Ú CountLimiterc C s t tƒ| _|| _d S ©N)r ÚintÚcountÚmax)Úselfr © r ú$/home/hibiki/hibikibot/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 0 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 9 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_timeC s zFreqLimiter.left_timeN)r ) r r r r r r r* r& r+ r r r r r% 8 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 L s
|
|||
|
|
|
|||
|
|
zBanCheckLimiter.__init__c C s2 | j | dkr t ¡ | j|< | j | d7 < d S r )r2 r r3 r r r r r R 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 W s
|
|||
|
|
(
|
|||
|
|
zBanCheckLimiter.checkN)r. r/ ) r r r r& r
|
|||
|
|
r r r r r r r r r- K 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 i 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 n s
|
|||
|
|
|
|||
|
|
zDailyNumberLimiter.checkc C s
|
|||
|
|
| j | S r r r r r r Úget_numv s zDailyNumberLimiter.get_numr c C s | j | |7 < d S r r )r
|
|||
|
|
r Únumr r r Úincreasey s zDailyNumberLimiter.increasec C s d| j |< d S r) r r r r r Úreset| s zDailyNumberLimiter.resetN)r ) r r r ÚpytzÚtimezoner9 r r r r< r> r? r r r r r4 f s
|
|||
|
|
|
|||
|
|
r4 c @ sH e Zd Zdd„ Zdd„ Zdd„ Zedœdd „Zedœd
|
|||
|
|
d „Z d d
|
|||
|
|
„ Z
|
|||
|
|
dS )ÚFuduc C s t tƒ| _d S r )r ÚlistÚmlistr! r r r r � s z
|
|||
|
|
Fudu.__init__c C s | j | |¡ d S r )rD Úappend©r
|
|||
|
|
r Úcontentr r r rE „ s z Fudu.appendc C s g | j |< d S r ©rD r r r r r; ‡ s z
|
|||
|
|
Fudu.clearr c C s t | j| ƒS r )ÚlenrD r r r r ÚsizeŠ s z Fudu.sizec C s | j | d |kS r) rH rF r r r r � s z
|
|||
|
|
Fudu.checkc C s | j | d S r) rH r r r r Úget� s zFudu.getN) r r r r rE r; r
|
|||
|
|
rJ r r rK r r r r rB € s rB r c C sZ zt | ƒ W dS tk
|
|||
|
|
r" Y nX zdd l}| | ¡ W dS ttfk
|
|||
|
|
rT Y nX dS )NTr F)r& Ú
|
|||
|
|
ValueErrorÚ unicodedataÚnumericÚ TypeError)ÚsrM r r r Ú is_number” s
|
|||
|
|
rQ )ÚpathÚstartÚendr c C st g }t | dddd��V}| ¡ }|D ]B}|dkr"|dkr"|dkrR| ||d … ¡ q"| |||… ¡ q"W 5 Q R X |S )NÚrÚignorezUTF-8)ÚerrorsÚencodingÚ
|