Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-06 12:20:00 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

refactor code

Browse Source
This commit is contained in:
hibiki
2021-07-30 21:21:51 +08:00
parent 2ad891aa1e
commit cc24822dca
165 changed files with 7815 additions and 8174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/russian_user.py
+60 -32
View File
@@ -3,7 +3,7 @@ from typing import List
class RussianUser(db.Model):
__tablename__ = 'russian_users'
__tablename__ = "russian_users"
id = db.Column(db.Integer(), primary_key=True)
user_qq = db.Column(db.BigInteger(), nullable=False)
@@ -13,34 +13,49 @@ class RussianUser(db.Model):
make_money = db.Column(db.Integer(), default=0)
lose_money = db.Column(db.Integer(), default=0)
_idx1 = db.Index('russian_group_users_idx1', 'user_qq', 'group_id', unique=True)
_idx1 = db.Index("russian_group_users_idx1", "user_qq", "group_id", unique=True)
@classmethod
async def ensure(cls, user_qq: int, group_id: int) -> 'RussianUser':
user = await cls.query.where(
(cls.user_qq == user_qq) & (cls.group_id == group_id)
).with_for_update().gino.first()
return user or await cls.create(
user_qq=user_qq,
group_id=group_id
async def ensure(cls, user_qq: int, group_id: int) -> "RussianUser":
"""
说明：
获取用户对象
参数：
:param user_qq: qq号
:param group_id: 群号
"""
user = (
await cls.query.where((cls.user_qq == user_qq) & (cls.group_id == group_id))
.with_for_update()
.gino.first()
)
return user or await cls.create(user_qq=user_qq, group_id=group_id)
@classmethod
async def add_count(cls, user_qq: int, group_id: int, itype: str) -> bool:
"""
说明：
添加用户输赢次数
说明：
:param user_qq: qq号
:param group_id: 群号
:param itype: 输或赢 'win' or 'lose'
"""
try:
user = await cls.query.where(
(cls.user_qq == user_qq) & (cls.group_id == group_id)
).with_for_update().gino.first()
if not user:
user = await cls.create(
user_qq=user_qq,
group_id=group_id
user = (
await cls.query.where(
(cls.user_qq == user_qq) & (cls.group_id == group_id)
)
if itype == 'win':
.with_for_update()
.gino.first()
)
if not user:
user = await cls.create(user_qq=user_qq, group_id=group_id)
if itype == "win":
await user.update(
win_count=user.win_count + 1,
).apply()
elif itype == 'lose':
elif itype == "lose":
await user.update(
fail_count=user.fail_count + 1,
).apply()
@@ -50,20 +65,30 @@ class RussianUser(db.Model):
@classmethod
async def money(cls, user_qq: int, group_id: int, itype: str, count: int) -> bool:
"""
说明：
添加用户输赢金钱
参数：
:param user_qq: qq号
:param group_id: 群号
:param itype: 输或赢 'win' or 'lose'
:param count: 金钱数量
"""
try:
user = await cls.query.where(
(cls.user_qq == user_qq) & (cls.group_id == group_id)
).with_for_update().gino.first()
if not user:
user = await cls.create(
user_qq=user_qq,
group_id=group_id
user = (
await cls.query.where(
(cls.user_qq == user_qq) & (cls.group_id == group_id)
)
if itype == 'win':
.with_for_update()
.gino.first()
)
if not user:
user = await cls.create(user_qq=user_qq, group_id=group_id)
if itype == "win":
await user.update(
make_money=user.make_money + count,
).apply()
elif itype == 'lose':
elif itype == "lose":
await user.update(
lose_money=user.lose_money + count,
).apply()
@@ -72,9 +97,12 @@ class RussianUser(db.Model):
return False
@classmethod
async def all_user(cls, group_id: int) -> List['RussianUser']:
users = await cls.query.where(
(cls.group_id == group_id)
).gino.all()
async def get_all_user(cls, group_id: int) -> List["RussianUser"]:
"""
说明：
获取该群所有用户对象
参数：
:param group_id: 群号
"""
users = await cls.query.where((cls.group_id == group_id)).gino.all()
return users