mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from services.db_context import db
|
|
|
|
|
|
class UserCount(db.Model):
|
|
__tablename__ = "count_users"
|
|
|
|
user_qq = db.Column(db.BigInteger(), nullable=False, primary_key=True)
|
|
reimu_count = db.Column(db.Integer(), nullable=False, default=0)
|
|
setu_r18_count = db.Column(db.Integer(), nullable=False, default=0)
|
|
|
|
_idx1 = db.Index("sign_reimu_users_idx1", "user_qq", unique=True)
|
|
|
|
@classmethod
|
|
async def add_count(cls, user_qq: int, name: str, count: int = 1):
|
|
"""
|
|
说明：
|
|
用户添加次数
|
|
参数：
|
|
:param user_qq: qq号
|
|
:param name: 目标名称
|
|
:param count: 增加次数
|
|
"""
|
|
query = cls.query.where((cls.user_qq == user_qq))
|
|
query = query.with_for_update()
|
|
user = await query.gino.first()
|
|
user = user if user else await cls.create(user_qq=user_qq)
|
|
if name == "reimu":
|
|
await user.update(reimu_count=cls.reimu_count + count).apply()
|
|
if name == "setu_r18":
|
|
await user.update(setu_r18_count=cls.setu_r18_count + count).apply()
|
|
|
|
@classmethod
|
|
async def check_count(cls, user_qq: int, name: str, max_count: int) -> bool:
|
|
"""
|
|
说明：
|
|
检测次数是否到达最大值
|
|
参数：
|
|
:param user_qq: qq号
|
|
:param name: 目标名称
|
|
:param max_count: 最大值
|
|
"""
|
|
query = cls.query.where((cls.user_qq == user_qq))
|
|
user = await query.gino.first()
|
|
user = user if user else await cls.create(user_qq=user_qq)
|
|
if name == "reimu":
|
|
if user.reimu_count == max_count:
|
|
return True
|
|
else:
|
|
return False
|
|
if name == "setu_r18":
|
|
if user.setu_r18_count == max_count:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@classmethod
|
|
async def reset_count(cls):
|
|
"""
|
|
说明：
|
|
重置每日次数
|
|
"""
|
|
for user in await cls.query.gino.all():
|
|
await user.update(reimu_count=0, setu_r18_count=0).apply()
|