mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 13:42:56 +08:00
100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
from datetime import datetime
|
|
|
|
from services.db_context import db
|
|
|
|
|
|
class GroupInfoUser(db.Model):
|
|
__tablename__ = 'group_info_users'
|
|
|
|
id = db.Column(db.Integer(), primary_key=True)
|
|
user_qq = db.Column(db.BigInteger(), nullable=False)
|
|
user_name = db.Column(db.Unicode(), nullable=False)
|
|
belonging_group = db.Column(db.BigInteger(), nullable=False)
|
|
user_join_time = db.Column(db.DateTime(), nullable=False)
|
|
nickname = db.Column(db.Unicode())
|
|
|
|
_idx1 = db.Index('info_group_users_idx1', 'user_qq', 'belonging_group', unique=True)
|
|
|
|
@classmethod
|
|
async def insert(cls, user_qq: int, belonging_group: int, user_name: str, user_join_time: datetime) -> 'bool':
|
|
query = cls.query.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
)
|
|
query = query.with_for_update()
|
|
try:
|
|
if await query.gino.first() is None:
|
|
await cls.create(
|
|
user_qq=user_qq,
|
|
user_name=user_name,
|
|
belonging_group=belonging_group,
|
|
user_join_time=user_join_time,
|
|
)
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
@classmethod
|
|
async def select_member_info(cls, user_qq: int, belonging_group: int) -> 'GroupInfoUser':
|
|
query = cls.query.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
)
|
|
return await query.gino.first()
|
|
|
|
@classmethod
|
|
async def delete_member_info(cls, user_qq: int, belonging_group: int) -> 'bool':
|
|
query = cls.query.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
)
|
|
query = query.with_for_update()
|
|
user = await query.gino.first()
|
|
try:
|
|
if user is None:
|
|
return True
|
|
else:
|
|
await cls.delete.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
).gino.status()
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
@classmethod
|
|
async def query_group_member_list(cls, belonging_group: int) -> 'list':
|
|
member_list = []
|
|
query = cls.query.where(
|
|
(cls.belonging_group == belonging_group)
|
|
)
|
|
for user in await query.gino.all():
|
|
member_list.append(user.user_qq)
|
|
return member_list
|
|
|
|
@classmethod
|
|
async def set_group_member_nickname(cls, user_qq: int, belonging_group: int, nickname: str) -> 'bool':
|
|
query = cls.query.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
)
|
|
user = await query.gino.first()
|
|
if user:
|
|
await user.update(
|
|
nickname=nickname
|
|
).apply()
|
|
return True
|
|
return False
|
|
|
|
@classmethod
|
|
async def get_group_member_nickname(cls, user_qq: int, belonging_group: int) -> 'str':
|
|
query = cls.query.where(
|
|
(cls.user_qq == user_qq) & (cls.belonging_group == belonging_group)
|
|
)
|
|
user = await query.gino.first()
|
|
if user:
|
|
if user.nickname:
|
|
return user.nickname
|
|
return ''
|
|
|
|
|
|
|
|
|
|
|
|
|