db2_hj3415 0.1.8__tar.gz → 0.1.10__tar.gz

{db2_hj3415-0.1.8 → db2_hj3415-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: db2_hj3415
-Version: 0.1.8
+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.8 → db2_hj3415-0.1.10}/db2_hj3415/nfs/_ops.py

@@ -1,5 +1,8 @@
+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
 
 
 async def get_all_codes(client: AsyncIOMotorClient) -> list[str]:
@@ -47,6 +50,24 @@ def get_all_codes_sync() -> list[str]:
         connection.close_mongo_client_sync()
 
 
+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}).sort(sort_by, 1)
+    result = []
+    async for doc in cursor:
+        result.append(CodeName(**doc))
+    return result
+
+
+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}).sort(sort_by, 1)
+        return [CodeName(**doc) for doc in cursor]
+    finally:
+        connection.close_mongo_client_sync()
+
 async def delete_code_from_all_collections(code: str, client: AsyncIOMotorClient) -> dict[str, int]:
     db = client[DB_NAME]
 

{db2_hj3415-0.1.8 → db2_hj3415-0.1.10}/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-0.1.10/db2_hj3415/nfs/dart.py

@@ -0,0 +1,101 @@
+from motor.motor_asyncio import AsyncIOMotorClient
+from pymongo import InsertOne, DESCENDING
+from db2_hj3415.nfs import Dart, DB_NAME, get_collection
+from datetime import datetime, timezone, timedelta
+
+COL_NAME = "dart"
+
+async def save_many(many_data: list[Dart], client: AsyncIOMotorClient) -> dict:
+    if not many_data:
+        return {"inserted_count": 0, "skipped": 0}
+
+    collection = get_collection(client, DB_NAME, COL_NAME)
+
+    # unique index 보장용 (이미 설정되었는지 확인)
+    await collection.create_index("rcept_no", unique=True)
+
+    ops = []
+    skipped = 0
+
+    for item in many_data:
+        doc = item.model_dump(mode="json", exclude={"id"})  # _id는 제외
+        ops.append(InsertOne(doc))
+
+    try:
+        result = await collection.bulk_write(ops, ordered=False)
+        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.8 → db2_hj3415-0.1.10}/db2_hj3415/nfs/models.py

@@ -1,6 +1,10 @@
 from pydantic import BaseModel, Field, field_serializer, ConfigDict, field_validator
 from datetime import datetime
 
+class CodeName(BaseModel):
+    코드: str
+    종목명: str | None
+
 class C101(BaseModel):
     id: str | None = Field(default=None, alias="_id")
     코드: str

{db2_hj3415-0.1.8 → db2_hj3415-0.1.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "db2_hj3415"
-version = "0.1.8"
+version = "0.1.10"
 authors = [{name = "Hyungjin Kim", email = "hj3415@gmail.com"}]
 description = "Gathering the stock data by playwright"
 readme = "README.md"

db2_hj3415-0.1.8/db2_hj3415/nfs/dart.py

@@ -1,27 +0,0 @@
-from motor.motor_asyncio import AsyncIOMotorClient
-from pymongo import InsertOne
-from db2_hj3415.nfs import Dart, DB_NAME, get_collection
-
-COL_NAME = "dart"
-
-async def save_many(many_data: list[Dart], client: AsyncIOMotorClient) -> dict:
-    if not many_data:
-        return {"inserted_count": 0, "skipped": 0}
-
-    collection = get_collection(client, DB_NAME, COL_NAME)
-
-    # unique index 보장용 (이미 설정되었는지 확인)
-    await collection.create_index("rcept_no", unique=True)
-
-    ops = []
-    skipped = 0
-
-    for item in many_data:
-        doc = item.model_dump(mode="json", exclude={"id"})  # _id는 제외
-        ops.append(InsertOne(doc))
-
-    try:
-        result = await collection.bulk_write(ops, ordered=False)
-        return {"inserted_count": result.inserted_count, "skipped": skipped}
-    except Exception as e:
-        return {"error": str(e)}