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 setu.py

Browse Source
This commit is contained in:
HibiKier 2021-09-09 14:30:42 +08:00 committed by GitHub
parent 1f9fa2adb9
commit c7b341393d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
models/setu.py
View File

@ -57,6 +57,7 @@ class Setu(db.Model):
local_id: Optional[int] = None,
tags: Optional[List[str]] = None,
r18: int = 0,
limit: int = 50,
):
"""
说明：
@ -65,6 +66,7 @@ class Setu(db.Model):
:param local_id: 本地色图 id
:param tags: tags
:param r18: 是否 r18，0：非r18 1：r18 2：混合
:param limit: 获取数量
"""
if local_id:
flag = True if r18 == 1 else False
@ -80,7 +82,7 @@ class Setu(db.Model):
if tags:
for tag in tags:
query = query.where(cls.tags.contains(tag) | cls.title.contains(tag) | cls.author.contains(tag))
query = query.order_by(db.func.random()).limit(50)
query = query.order_by(db.func.random()).limit(limit)
return await query.gino.all()
@classmethod
@ -89,8 +91,7 @@ class Setu(db.Model):
说明：
查询图片数量
"""
return len(await cls.query_image(r18=r18))
# return await db.func.count(cls.local_id).gino.scalar()
return len(await cls.query_image(r18=r18, limit=999999))
@classmethod
async def get_image_in_hash(cls, img_hash: str) -> "Setu":