mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-06 04:10:02 +08:00
数据库中所有user_qq改名以及user_id和group_id改为字符串
This commit is contained in:
@@ -6,14 +6,12 @@ from services.db_context import Model
|
||||
|
||||
|
||||
class PixivKeywordUser(Model):
|
||||
__tablename__ = "pixiv_keyword_users"
|
||||
__table_args__ = {"extend_existing": True}
|
||||
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
user_qq = fields.BigIntField()
|
||||
user_id = fields.CharField(255)
|
||||
"""用户id"""
|
||||
group_id = fields.BigIntField()
|
||||
group_id = fields.CharField(255)
|
||||
"""群聊id"""
|
||||
keyword = fields.CharField(255, unique=True)
|
||||
"""关键词"""
|
||||
@@ -46,9 +44,17 @@ class PixivKeywordUser(Model):
|
||||
获取黑名单PID
|
||||
"""
|
||||
black_pid = []
|
||||
keyword_list = await cls.filter(user_qq=114514).values_list(
|
||||
keyword_list = await cls.filter(user_id="114514").values_list(
|
||||
"keyword", flat=True
|
||||
)
|
||||
for image in keyword_list:
|
||||
black_pid.append(image[6:])
|
||||
return black_pid
|
||||
|
||||
@classmethod
|
||||
async def _run_script(cls):
|
||||
return [
|
||||
"ALTER TABLE pixiv_keyword_users RENAME COLUMN user_qq TO user_id;", # 将user_qq改为user_id
|
||||
"ALTER TABLE pixiv_keyword_users ALTER COLUMN user_id TYPE character varying(255);",
|
||||
"ALTER TABLE pixiv_keyword_users ALTER COLUMN group_id TYPE character varying(255);",
|
||||
]
|
||||
|
||||
@@ -54,8 +54,8 @@ async def _(bot: Bot, event: MessageEvent, arg: Message = CommandArg()):
|
||||
# ):
|
||||
if not await PixivKeywordUser.exists(keyword=msg):
|
||||
await PixivKeywordUser.create(
|
||||
user_qq=event.user_id,
|
||||
group_id=group_id,
|
||||
user_id=str(event.user_id),
|
||||
group_id=str(group_id),
|
||||
keyword=msg,
|
||||
is_pass=str(event.user_id) in bot.config.superusers,
|
||||
)
|
||||
@@ -104,8 +104,8 @@ async def _(
|
||||
# ):
|
||||
if not await PixivKeywordUser.exists(keyword=msg):
|
||||
await PixivKeywordUser.create(
|
||||
user_qq=event.user_id,
|
||||
group_id=group_id,
|
||||
user_id=str(event.user_id),
|
||||
group_id=str(group_id),
|
||||
keyword=msg,
|
||||
is_pass=str(event.user_id) in bot.config.superusers,
|
||||
)
|
||||
@@ -138,7 +138,7 @@ async def _(bot: Bot, event: MessageEvent, arg: Message = CommandArg()):
|
||||
keyword=f"black:{pid}{f'_p{img_p}' if img_p else ''}"
|
||||
):
|
||||
await PixivKeywordUser.create(
|
||||
user_qq=114514,
|
||||
user_id=114514,
|
||||
group_id=114514,
|
||||
keyword=f"black:{pid}{f'_p{img_p}' if img_p else ''}",
|
||||
is_pass=str(event.user_id) in bot.config.superusers,
|
||||
|
||||
@@ -110,8 +110,8 @@ async def _(bot: Bot, event: MessageEvent, arg: Message = CommandArg()):
|
||||
keyword=f"black:{pid}{f'_p{img_p}' if img_p else ''}"
|
||||
):
|
||||
await PixivKeywordUser.create(
|
||||
user_qq=114514,
|
||||
group_id=114514,
|
||||
user_id="114514",
|
||||
group_id="114514",
|
||||
keyword=f"black:{pid}{f'_p{img_p}' if img_p else ''}",
|
||||
is_pass=False,
|
||||
)
|
||||
@@ -168,7 +168,7 @@ async def _(
|
||||
if data:
|
||||
data.is_pass = flag
|
||||
await data.save(update_fields=["is_pass"])
|
||||
user_id, group_id = data.user_qq, data.group_id
|
||||
user_id, group_id = data.user_id, data.group_id
|
||||
if not user_id:
|
||||
await pass_keyword.send(f"未找到关键词/UID:{x},请检查关键词/UID是否存在...")
|
||||
continue
|
||||
|
||||
@@ -37,7 +37,7 @@ show_pix = on_command("查看pix图库", priority=1, block=True)
|
||||
|
||||
@my_keyword.handle()
|
||||
async def _(event: MessageEvent):
|
||||
data = await PixivKeywordUser.filter(user_qq=event.user_id).values_list(
|
||||
data = await PixivKeywordUser.filter(user_id=str(event.user_id)).values_list(
|
||||
"keyword", flat=True
|
||||
)
|
||||
if not data:
|
||||
|
||||
Reference in New Issue
Block a user