feat(config): add TOKENIZED_SEARCH

This commit is contained in:
InfinityPacer
2024-12-26 13:56:08 +08:00
parent 0a1dc1724c
commit d02bf33345
2 changed files with 7 additions and 0 deletions
+5
View File
@@ -1,10 +1,12 @@
from typing import List, Any from typing import List, Any
import jieba
from fastapi import APIRouter, Depends from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from app import schemas from app import schemas
from app.chain.storage import StorageChain from app.chain.storage import StorageChain
from app.core.config import settings
from app.core.event import eventmanager from app.core.event import eventmanager
from app.core.security import verify_token from app.core.security import verify_token
from app.db import get_db from app.db import get_db
@@ -57,6 +59,9 @@ def transfer_history(title: str = None,
status = True status = True
if title: if title:
if settings.TOKENIZED_SEARCH:
words = jieba.cut(title, HMM=False)
title = "%".join(words)
total = TransferHistory.count_by_title(db, title=title, status=status) total = TransferHistory.count_by_title(db, title=title, status=status)
result = TransferHistory.list_by_title(db, title=title, page=page, result = TransferHistory.list_by_title(db, title=title, page=page,
count=count, status=status) count=count, status=status)
+2
View File
@@ -240,6 +240,8 @@ class ConfigModel(BaseModel):
RENAME_FORMAT_S0_NAMES: List[str] = Field( RENAME_FORMAT_S0_NAMES: List[str] = Field(
default_factory=lambda: ["Specials", "SPs"] default_factory=lambda: ["Specials", "SPs"]
) )
# 启用分词搜索
TOKENIZED_SEARCH: bool = False
class Settings(BaseSettings, ConfigModel): class Settings(BaseSettings, ConfigModel):