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

Update omega_pixiv_illusts.py

Browse Source
This commit is contained in:
HibiKier 2021-09-09 16:01:50 +08:00 committed by GitHub
parent b61cb49c1a
commit dc7681357e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
models/omega_pixiv_illusts.py
View File

@ -118,14 +118,15 @@ class OmegaPixivIllusts(db.Model):
参数：
:param tags: 关键词/Tag
"""
query = cls.query
setattr(OmegaPixivIllusts, 'count', db.func.count(cls.pid).label('count'))
query = cls.select('count')
if tags:
for tag in tags:
query = query.where(cls.tags.contains(tag))
count = len(await query.where(cls.nsfw_tag == 0).gino.all())
setu_count = len(await query.where(cls.nsfw_tag == 1).gino.all())
r18_count = len(await query.where(cls.nsfw_tag == 2).gino.all())
return count, setu_count, r18_count
count = await query.where(cls.nsfw_tag == 0).gino.first()
setu_count = await query.where(cls.nsfw_tag == 1).gino.first()
r18_count = await query.where(cls.nsfw_tag == 2).gino.first()
return count[0], setu_count[0], r18_count[0]
@classmethod
async def get_all_pid(cls) -> List[int]:
@ -133,6 +134,6 @@ class OmegaPixivIllusts(db.Model):
说明：
获取所有图片PID
"""
data = await cls.query.gino.all()
return [x.pid for x in data]
data = await cls.select('pid').gino.all()
return [x[0] for x in data]