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

feat✨: PIX图库

Browse Source
This commit is contained in:
HibiKier
2024-05-20 22:03:11 +08:00
parent c2fd8661b5
commit c6afb8c1e9
15 changed files with 1738 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zhenxun/plugins/pix_gallery/_model/__init__.py
+1
View File
@@ -0,0 +1 @@
zhenxun/plugins/pix_gallery/_model/omega_pixiv_illusts.py
+89
View File
@@ -0,0 +1,89 @@
from tortoise import fields
from tortoise.contrib.postgres.functions import Random
from zhenxun.services.db_context import Model
class OmegaPixivIllusts(Model):
id = fields.IntField(pk=True, generated=True, auto_increment=True)
"""自增id"""
pid = fields.BigIntField()
"""pid"""
uid = fields.BigIntField()
"""uid"""
title = fields.CharField(255)
"""标题"""
uname = fields.CharField(255)
"""画师名称"""
classified = fields.IntField()
"""标记标签, 0=未标记, 1=已人工标记或从可信已标记来源获取"""
nsfw_tag = fields.IntField()
"""nsfw标签,-1=未标记, 0=safe, 1=setu. 2=r18"""
width = fields.IntField()
"""宽度"""
height = fields.IntField()
"""高度"""
tags = fields.TextField()
"""tags"""
url = fields.CharField(255)
"""pixiv url链接"""
class Meta:
table = "omega_pixiv_illusts"
table_description = "omega图库数据表"
unique_together = ("pid", "url")
@classmethod
async def query_images(
cls,
keywords: list[str] | None = None,
uid: int | None = None,
pid: int | None = None,
nsfw_tag: int | None = 0,
num: int = 100,
) -> list["OmegaPixivIllusts"]:
"""查找符合条件的图片
参数:
keywords: 关键词
uid: 画师uid
pid: 图片pid
nsfw_tag: nsfw标签, 0=safe, 1=setu. 2=r18
num: 获取图片数量
"""
if not num:
return []
query = cls
if nsfw_tag is not None:
query = cls.filter(nsfw_tag=nsfw_tag)
if keywords:
for keyword in keywords:
query = query.filter(tags__contains=keyword)
elif uid:
query = query.filter(uid=uid)
elif pid:
query = query.filter(pid=pid)
query = query.annotate(rand=Random()).limit(num)
return await query.all() # type: ignore
@classmethod
async def get_keyword_num(
cls, tags: list[str] | None = None
) -> tuple[int, int, int]:
"""获取相关关键词(keyword, tag)在图库中的数量
参数:
tags: 关键词/Tag
"""
query = cls
if tags:
for tag in tags:
query = query.filter(tags__contains=tag)
else:
query = query.all()
count = await query.filter(nsfw_tag=0).count()
setu_count = await query.filter(nsfw_tag=1).count()
r18_count = await query.filter(nsfw_tag=2).count()
return count, setu_count, r18_count
zhenxun/plugins/pix_gallery/_model/pixiv.py
+91
View File
@@ -0,0 +1,91 @@
from tortoise import fields
from tortoise.contrib.postgres.functions import Random
from zhenxun.services.db_context import Model
class Pixiv(Model):
id = fields.IntField(pk=True, generated=True, auto_increment=True)
"""自增id"""
pid = fields.BigIntField()
"""pid"""
uid = fields.BigIntField()
"""uid"""
author = fields.CharField(255)
"""作者"""
title = fields.CharField(255)
"""标题"""
width = fields.IntField()
"""宽度"""
height = fields.IntField()
"""高度"""
view = fields.IntField()
"""pixiv查看数"""
bookmarks = fields.IntField()
"""收藏数"""
tags = fields.TextField()
"""tags"""
img_url = fields.CharField(255)
"""pixiv url链接"""
img_p = fields.CharField(255)
"""图片pN"""
is_r18 = fields.BooleanField()
class Meta:
table = "pixiv"
table_description = "pix图库数据表"
unique_together = ("pid", "img_url", "img_p")
# 0：非r18 1：r18 2：混合
@classmethod
async def query_images(
cls,
keywords: list[str] | None = None,
uid: int | None = None,
pid: int | None = None,
r18: int | None = 0,
num: int = 100,
) -> list["Pixiv"]:
"""查找符合条件的图片
参数:
keywords: 关键词
uid: 画师uid
pid: 图片pid
r18: 是否r18，0：非r18 1：r18 2：混合
num: 查找图片的数量
"""
if not num:
return []
query = cls
if r18 == 0:
query = query.filter(is_r18=False)
elif r18 == 1:
query = query.filter(is_r18=True)
if keywords:
for keyword in keywords:
query = query.filter(tags__contains=keyword)
elif uid:
query = query.filter(uid=uid)
elif pid:
query = query.filter(pid=pid)
query = query.annotate(rand=Random()).limit(num)
return await query.all() # type: ignore
@classmethod
async def get_keyword_num(cls, tags: list[str] | None = None) -> tuple[int, int]:
"""获取相关关键词(keyword, tag)在图库中的数量
参数:
tags: 关键词/Tag
"""
query = cls
if tags:
for tag in tags:
query = query.filter(tags__contains=tag)
else:
query = query.all()
count = await query.filter(is_r18=False).count()
r18_count = await query.filter(is_r18=True).count()
return count, r18_count
zhenxun/plugins/pix_gallery/_model/pixiv_keyword_user.py
+52
View File
@@ -0,0 +1,52 @@
from tortoise import fields
from zhenxun.services.db_context import Model
class PixivKeywordUser(Model):
id = fields.IntField(pk=True, generated=True, auto_increment=True)
"""自增id"""
user_id = fields.CharField(255)
"""用户id"""
group_id = fields.CharField(255)
"""群聊id"""
keyword = fields.CharField(255, unique=True)
"""关键词"""
is_pass = fields.BooleanField()
"""是否通过"""
class Meta:
table = "pixiv_keyword_users"
table_description = "pixiv关键词数据表"
@classmethod
async def get_current_keyword(cls) -> tuple[list[str], list[str]]:
"""获取当前通过与未通过的关键词"""
pass_keyword = []
not_pass_keyword = []
for data in await cls.all().values_list("keyword", "is_pass"):
if data[1]:
pass_keyword.append(data[0])
else:
not_pass_keyword.append(data[0])
return pass_keyword, not_pass_keyword
@classmethod
async def get_black_pid(cls) -> list[str]:
"""获取黑名单PID"""
black_pid = []
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);",
]