db2_hj3415 0.1.9__py2.py3-none-any.whl → 0.1.10__py2.py3-none-any.whl

db2_hj3415/nfs/_ops.py

@@ -1,3 +1,5 @@
+from typing import Literal
+
 from motor.motor_asyncio import AsyncIOMotorClient
 from db2_hj3415.nfs import DB_NAME, connection
 from db2_hj3415.nfs.models import CodeName
@@ -48,20 +50,20 @@ def get_all_codes_sync() -> list[str]:
         connection.close_mongo_client_sync()
 
 
-async def get_all_codes_names(client: AsyncIOMotorClient) -> list[CodeName]:
+async def get_all_codes_names(client: AsyncIOMotorClient, sort_by:Literal['종목명', '코드']='종목명') -> list[CodeName]:
     collection = client[DB_NAME]['c101']
-    cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0})
+    cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0}).sort(sort_by, 1)
     result = []
     async for doc in cursor:
         result.append(CodeName(**doc))
     return result
 
 
-def get_all_codes_names_sync() -> list[CodeName] | None:
+def get_all_codes_names_sync(sort_by:Literal['종목명', '코드']='종목명') -> list[CodeName] | None:
     client = connection.get_mongo_client_sync()
     try:
         collection = client[DB_NAME]['c101']
-        cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0})
+        cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0}).sort(sort_by, 1)
         return [CodeName(**doc) for doc in cursor]
     finally:
         connection.close_mongo_client_sync()

db2_hj3415/nfs/c108.py

@@ -75,14 +75,14 @@ async def save_many(many_data: dict[str, pd.DataFrame], client: AsyncIOMotorClie
     return total_result
 
 
-async def get_latest(code: str, client: AsyncIOMotorClient, days:int = 30) -> list[C108]:
+async def get_latest(code: str, client: AsyncIOMotorClient, days:int = 60) -> list[C108]:
     """
     최근 N일 이내의 C108 리포트 도큐먼트를 조회합니다.
 
     Args:
         code (str): 종목 코드 (예: "005930").
         client (AsyncIOMotorClient): 비동기 MongoDB 클라이언트.
-        days (int, optional): 현재 시점에서 며칠 전까지의 데이터를 조회할지 설정합니다. 기본값은 30일.
+        days (int, optional): 현재 시점에서 며칠 전까지의 데이터를 조회할지 설정합니다. 기본값은 60일.
 
     Returns:
         list[C108]: 조건에 해당하는 C108 도큐먼트 리스트. 일치하는 문서가 없을 경우 빈 리스트를 반환합니다.

db2_hj3415/nfs/dart.py

@@ -1,6 +1,7 @@
 from motor.motor_asyncio import AsyncIOMotorClient
-from pymongo import InsertOne
+from pymongo import InsertOne, DESCENDING
 from db2_hj3415.nfs import Dart, DB_NAME, get_collection
+from datetime import datetime, timezone, timedelta
 
 COL_NAME = "dart"
 
@@ -25,3 +26,76 @@ async def save_many(many_data: list[Dart], client: AsyncIOMotorClient) -> dict:
         return {"inserted_count": result.inserted_count, "skipped": skipped}
     except Exception as e:
         return {"error": str(e)}
+
+
+async def get_data_last_n_days(code: str, client: AsyncIOMotorClient, days:int = 30) -> list[Dart]:
+    """
+    지정한 종목 코드(code)에 대해 최근 N일(days) 동안의 DART 데이터를 비동기로 조회합니다.
+
+    MongoDB 컬렉션에서 날짜 기준으로 N일 이전(cutoff) 이후의 문서들을 조회하며,
+    결과는 최신 날짜순(DESCENDING)으로 정렬됩니다.
+
+    Parameters:
+        code (str): 조회할 종목 코드 (예: '005930').
+        client (AsyncIOMotorClient): 비동기 MongoDB 클라이언트.
+        days (int, optional): 며칠 전까지의 데이터를 가져올지 설정 (기본값: 30일).
+
+    Returns:
+        list[Dart]: Dart Pydantic 모델의 리스트. 데이터가 없으면 빈 리스트 반환.
+    """
+    collection = get_collection(client, DB_NAME, COL_NAME)
+    now = datetime.now(timezone.utc)
+    cutoff = now - timedelta(days=days)
+
+    cursor = collection.find(
+        {
+            "코드": code,
+            "날짜": {"$gte": cutoff}
+        }
+    ).sort("날짜", DESCENDING)
+
+    docs = await cursor.to_list(length=None)
+    if not docs:
+        return []
+    else:
+        darts = []
+        for doc in docs:
+            doc["_id"] = str(doc["_id"])
+            darts.append(Dart(**doc))
+        return darts
+
+
+async def get_data_today(client: AsyncIOMotorClient) -> list[Dart]:
+    """
+    오늘 날짜의 DART 데이터를 비동기로 조회합니다.
+
+    MongoDB 컬렉션에서 오늘 날짜에 해당하는 모든 종목의 문서를 조회하며,
+    결과는 '날짜' 필드를 기준으로 내림차순 정렬됩니다.
+
+    Parameters:
+        client (AsyncIOMotorClient): 비동기 MongoDB 클라이언트.
+
+    Returns:
+        List[Dart]: Dart Pydantic 모델의 리스트. 없으면 빈 리스트.
+    """
+    collection = get_collection(client, DB_NAME, COL_NAME)
+
+    # 오늘 00:00 ~ 내일 00:00 사이의 UTC 기준
+    now = datetime.now(timezone.utc)
+    start_of_day = datetime(now.year, now.month, now.day, tzinfo=timezone.utc)
+    start_of_next_day = start_of_day + timedelta(days=1)
+
+    cursor = collection.find(
+        {
+            "날짜": {
+                "$gte": start_of_day,
+                "$lt": start_of_next_day
+            }
+        }
+    ).sort("날짜", DESCENDING)
+
+    docs = await cursor.to_list(length=None)
+    if not docs:
+        return []
+
+    return [Dart(**{**doc, "_id": str(doc["_id"])}) for doc in docs]

{db2_hj3415-0.1.9.dist-info → db2_hj3415-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: db2_hj3415
-Version: 0.1.9
+Version: 0.1.10
 Summary: Gathering the stock data by playwright
 Author-email: Hyungjin Kim <hj3415@gmail.com>
 Description-Content-Type: text/markdown

{db2_hj3415-0.1.9.dist-info → db2_hj3415-0.1.10.dist-info}/RECORD

@@ -20,13 +20,13 @@ db2_hj3415/mi/usdkrw.py,sha256=QjcHqFSpSgyzI5lzRfYycNUzolzM-ikOvPDFy4FPadM,876
 db2_hj3415/mi/wti.py,sha256=wMoS5yHGOlUS2b2P7Eok5U_59eehYXQGfqWa56J_dkk,768
 db2_hj3415/nfs/__init__.py,sha256=JA3ZZHIKFTzDkRp2DC_WCBocotjU5ZlNTFT3JrastXI,230
 db2_hj3415/nfs/_c10346.py,sha256=9cb6yZp5ZfPY5JZO7_4BuHiV_md4yDrM3XFrUf7oH5g,9392
-db2_hj3415/nfs/_ops.py,sha256=a6p0pxW805aaKETM-w9I6p5g7PP22QT1U4DQfgJkF0c,2519
+db2_hj3415/nfs/_ops.py,sha256=oJW86wMxLrfbxv-1qlrtJDk-vKoJ8L-k4YSR-N7sLRw,2683
 db2_hj3415/nfs/c101.py,sha256=5gndh6OAQPHc6cvBBnBqdfHuMt34oEoFaXFOdCRv62I,5077
 db2_hj3415/nfs/c103.py,sha256=HfbknyRcC7-mKF5vnZzlHnESl9fBRwoIEZ_WCy6-tsM,1864
 db2_hj3415/nfs/c104.py,sha256=yYV2dxble1pXx4f_X7q6bKY54WDHgJ3btTbvX0C3Mik,1864
 db2_hj3415/nfs/c106.py,sha256=lzbMROOQ-w1AU8oeSYsYFhYuk0EuOux97CKuJrWhx1A,1864
-db2_hj3415/nfs/c108.py,sha256=F_kP1HMhpe9E6VKm_m-aDl6PZ-FqVs5-O2_-rU8f34Y,3761
-db2_hj3415/nfs/dart.py,sha256=yaqs7Qt8AYzbKWdPSNsaryOBOpVmNudTooBEaj2N-aY,876
+db2_hj3415/nfs/c108.py,sha256=WvT7Fs-BOd0nm_aZGWI4g8Kyo-fFu1dUP7QYyTL87tk,3761
+db2_hj3415/nfs/dart.py,sha256=vVDP2ee5akaoR_m17eN4PxVxxfg5K77TLxVXzQ9ZLyU,3392
 db2_hj3415/nfs/models.py,sha256=Y5FgSiB_BPzX5hY8RLWAFSXCl22LTECTi6RJAtXbaMU,5005
 db2_hj3415/valuation/__init__.py,sha256=SxzWnyeo412A6q7yqYHtQz0Qv-haoAN6qr5Q4h2NRqw,274
 db2_hj3415/valuation/_ops.py,sha256=NuTtlLo30lGaiTPHKJP81yP8nfifFvEneCUUjChhLlQ,2480
@@ -35,7 +35,7 @@ db2_hj3415/valuation/growth.py,sha256=bpixczYpGx6cfb2ChsZXM8YkNDrHMaNWdkuMyPZuld
 db2_hj3415/valuation/mil.py,sha256=kOc9MYrzHYTL7oBSOI95gTD2Edu3H2yOd3R-0qdaH8c,480
 db2_hj3415/valuation/models.py,sha256=fBmWhtdUE56E7AaFs-04aw4WC8ptm-Thl_IX7Yex0js,5592
 db2_hj3415/valuation/red.py,sha256=YV3uavn0A6-T5fk5YiD5MdpC2XhjZMN1mP-e2n_VJrs,481
-db2_hj3415-0.1.9.dist-info/entry_points.txt,sha256=JAGFsxKtvbNgaKCSKo-QRYwCyToabWmdRW0MZWtg9kg,45
-db2_hj3415-0.1.9.dist-info/WHEEL,sha256=Dyt6SBfaasWElUrURkknVFAZDHSTwxg3PaTza7RSbkY,100
-db2_hj3415-0.1.9.dist-info/METADATA,sha256=fLFeHOr2kGOTyrbnIHVIqTbYhVFm_7QYUIaNaxqVg-Q,688
-db2_hj3415-0.1.9.dist-info/RECORD,,
+db2_hj3415-0.1.10.dist-info/entry_points.txt,sha256=JAGFsxKtvbNgaKCSKo-QRYwCyToabWmdRW0MZWtg9kg,45
+db2_hj3415-0.1.10.dist-info/WHEEL,sha256=Dyt6SBfaasWElUrURkknVFAZDHSTwxg3PaTza7RSbkY,100
+db2_hj3415-0.1.10.dist-info/METADATA,sha256=nTnOhMzON77D0gN7YMDOSOE4SVAqa1BgV8X3AuRUTiE,689
+db2_hj3415-0.1.10.dist-info/RECORD,,