148 lines
4.1 KiB
Python
148 lines
4.1 KiB
Python
from backend.schemas.stock_schemas import TickerInfo
|
|
import yahooquery as yq
|
|
from fastapi import HTTPException
|
|
|
|
COMPANY_MAP = {
|
|
"NVDA": {
|
|
"company_name": "NVIDIA",
|
|
"aliases": ["엔비디아", "nvidia", "nvda"],
|
|
},
|
|
"AAPL": {
|
|
"company_name": "Apple",
|
|
"aliases": ["애플", "apple", "aapl"],
|
|
},
|
|
"MSFT": {
|
|
"company_name": "Microsoft",
|
|
"aliases": ["마이크로소프트", "microsoft", "msft"],
|
|
},
|
|
"TSLA": {
|
|
"company_name": "Tesla",
|
|
"aliases": ["테슬라", "tesla", "tsla"],
|
|
},
|
|
"GOOGL": {
|
|
"company_name": "Alphabet (Google)",
|
|
"aliases": ["구글", "google", "알파벳", "alphabet", "googl"],
|
|
},
|
|
"AMZN": {
|
|
"company_name": "Amazon",
|
|
"aliases": ["아마존", "amazon", "amzn"],
|
|
},
|
|
"META": {
|
|
"company_name": "Meta Platforms",
|
|
"aliases": ["메타", "meta", "페이스북", "facebook", "fb"],
|
|
},
|
|
"AMD": {
|
|
"company_name": "Advanced Micro Devices",
|
|
"aliases": ["amd"],
|
|
},
|
|
"INTC": {
|
|
"company_name": "Intel",
|
|
"aliases": ["인텔", "intel", "intc"],
|
|
},
|
|
"TSM": {
|
|
"company_name": "TSMC",
|
|
"aliases": ["tsmc", "tsm", "대만반도체"],
|
|
},
|
|
"QCOM": {
|
|
"company_name": "Qualcomm",
|
|
"aliases": ["퀄컴", "qualcomm", "qcom"],
|
|
},
|
|
"AVGO": {
|
|
"company_name": "Broadcom",
|
|
"aliases": ["브로드컴", "broadcom", "avgo"],
|
|
},
|
|
"MU": {
|
|
"company_name": "Micron Technology",
|
|
"aliases": ["마이크론", "micron", "mu"],
|
|
},
|
|
"V": {
|
|
"company_name": "Visa",
|
|
"aliases": ["비자", "visa", "v"],
|
|
},
|
|
"MA": {
|
|
"company_name": "Mastercard",
|
|
"aliases": ["마스터카드", "mastercard", "ma"],
|
|
},
|
|
"JPM": {
|
|
"company_name": "JPMorgan Chase",
|
|
"aliases": ["jp모건", "jpmorgan", "jpm"],
|
|
},
|
|
"JNJ": {
|
|
"company_name": "Johnson & Johnson",
|
|
"aliases": ["존슨앤존슨", "johnson", "jnj"],
|
|
},
|
|
"MSFT": {
|
|
"company_name": "Microsoft",
|
|
"aliases": ["마이크로소프트", "microsoft", "msft"],
|
|
},
|
|
}
|
|
|
|
class StockService:
|
|
"""주식 분석"""
|
|
async def _extract(self, query: str):
|
|
"""
|
|
사용자 쿼리에서 티커와 회사명 추출
|
|
1차 : Map 이용
|
|
2차 : yahooquery 이용
|
|
|
|
query : 엔비디아 분석해줘
|
|
"""
|
|
keyword = self._extract_company_keyword(query)
|
|
|
|
# 1 차 Map
|
|
result = self._extract_ticker_from_query(keyword)
|
|
|
|
if result:
|
|
return result
|
|
|
|
# 2 차 yahooquery
|
|
result = await self._search_yahoo_symbol(keyword)
|
|
|
|
if result:
|
|
return result
|
|
|
|
raise HTTPException(status_code=30000, detail="종목을 찾을 수 없습니다.")
|
|
|
|
def _extract_ticker_from_query(self, query:str):
|
|
"""
|
|
COMPANY_MAP 안에서 일치하는 회사 찾기
|
|
"""
|
|
|
|
query = query.lower()
|
|
|
|
for ticker, info in COMPANY_MAP.items():
|
|
for alias in info["aliases"]:
|
|
if alias.lower() in query:
|
|
return TickerInfo(ticker = ticker, company_name = info["company_name"])
|
|
|
|
def _extract_company_keyword(self, query:str):
|
|
"""
|
|
query : 엔비디아 분석해줘
|
|
=> 필요없는 문장 제거한 후 키워드만 추출
|
|
"""
|
|
|
|
stop_words = [
|
|
"분석해줘", "분석해", "분석", "해줘", "해", "주가", "전망", "재무", "알려줘", "어때", "어떻게", ""
|
|
]
|
|
|
|
keyword = query
|
|
|
|
for word in stop_words:
|
|
keyword = keyword.replace(word, "")
|
|
|
|
return keyword.strip()
|
|
|
|
async def _search_yahoo_symbol(self, keyword: str):
|
|
"""
|
|
yahooquery 이용
|
|
"""
|
|
try:
|
|
result = yq.search(keyword, first_quote=True)
|
|
|
|
if not result or (isinstance(result, dict) and "explanation" in result):
|
|
return None
|
|
|
|
return TickerInfo(ticker=result["symbol"], company_name=result["longname"])
|
|
|
|
except Exception:
|
|
return None |