skdbj/llm-quant
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix company mapping initialization with flag to prevent redundant calls

Browse Source
This commit is contained in:
sam 2025-10-07 22:33:19 +08:00
parent 37aaeb92f6
commit d2a214cd28
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/ingest/entity_recognition.py
View File

@ -9,6 +9,8 @@ from app.utils.logging import get_logger
LOGGER = get_logger(__name__) LOGGER = get_logger(__name__)
_COMPANY_MAPPING_INITIALIZED = False
# 股票代码正则表达式 # 股票代码正则表达式
A_SH_CODE_PATTERN = re.compile(r"\b(\d{6})(\.(?:SH|SZ))?\b", re.IGNORECASE) A_SH_CODE_PATTERN = re.compile(r"\b(\d{6})(\.(?:SH|SZ))?\b", re.IGNORECASE)
HK_CODE_PATTERN = re.compile(r"\b(\d{4})\.HK\b", re.IGNORECASE) HK_CODE_PATTERN = re.compile(r"\b(\d{4})\.HK\b", re.IGNORECASE)
@ -136,6 +138,11 @@ def initialize_company_mapping(db_connection) -> None:
Args: Args:
db_connection: SQLite数据库连接 db_connection: SQLite数据库连接
""" """
global _COMPANY_MAPPING_INITIALIZED
if _COMPANY_MAPPING_INITIALIZED:
return
_COMPANY_MAPPING_INITIALIZED = True
cursor = db_connection.cursor() cursor = db_connection.cursor()
try: try:
cursor.execute( cursor.execute(