fix(media): 修复类型错误

This commit is contained in:
Attente
2025-11-23 13:28:01 +08:00
parent 56c684ec06
commit 5162b2748e
2 changed files with 17 additions and 16 deletions
+16 -15
View File
@@ -85,25 +85,26 @@ async def search(title: str,
return obj.get("source") return obj.get("source")
return obj.source return obj.source
result = []
media_chain = MediaChain() media_chain = MediaChain()
if type == "media": if type == "media":
_, medias = await media_chain.async_search(title=title) _, medias = await media_chain.async_search(title=title)
if medias: result = [media.to_dict() for media in medias] if medias else []
result = [media.to_dict() for media in medias]
elif type == "collection": elif type == "collection":
result = await media_chain.async_search_collections(name=title) collections = await media_chain.async_search_collections(name=title)
else: result = [collection.to_dict() for collection in collections] if collections else []
result = await media_chain.async_search_persons(name=title) else: # person
if result: persons = await media_chain.async_search_persons(name=title)
# 按设置的顺序对结果进行排序 result = [person.model_dump() for person in persons] if persons else []
setting_order = settings.SEARCH_SOURCE.split(',') or []
sort_order = {} if not result:
for index, source in enumerate(setting_order): return []
sort_order[source] = index
result = sorted(result, key=lambda x: sort_order.get(__get_source(x), 4)) # 排序和分页
return result[(page - 1) * count:page * count] setting_order = settings.SEARCH_SOURCE.split(',') if settings.SEARCH_SOURCE else []
return [] sort_order = {source: index for index, source in enumerate(setting_order)}
sorted_result = sorted(result, key=lambda x: sort_order.get(__get_source(x), 4))
return sorted_result[(page - 1) * count:page * count]
@router.post("/scrape/{storage}", summary="刮削媒体信息", response_model=schemas.Response) @router.post("/scrape/{storage}", summary="刮削媒体信息", response_model=schemas.Response)
+1 -1
View File
@@ -14,7 +14,7 @@ class CommingMessage(BaseModel):
# 用户ID # 用户ID
userid: Optional[Union[str, int]] = None userid: Optional[Union[str, int]] = None
# 用户名称 # 用户名称
username: Optional[str] = None username: Optional[Union[str, int]] = None
# 消息渠道 # 消息渠道
channel: Optional[MessageChannel] = None channel: Optional[MessageChannel] = None
# 来源(渠道名称) # 来源(渠道名称)