db2_hj3415 0.1.8__tar.gz → 0.1.9__tar.gz

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

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

@@ -1,5 +1,6 @@
 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 +48,24 @@ def get_all_codes_sync() -> list[str]:
         connection.close_mongo_client_sync()
 
 
+async def get_all_codes_names(client: AsyncIOMotorClient) -> list[CodeName]:
+    collection = client[DB_NAME]['c101']
+    cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0})
+    result = []
+    async for doc in cursor:
+        result.append(CodeName(**doc))
+    return result
+
+
+def get_all_codes_names_sync() -> list[CodeName] | None:
+    client = connection.get_mongo_client_sync()
+    try:
+        collection = client[DB_NAME]['c101']
+        cursor = collection.find({}, {"코드": 1, "종목명": 1, "_id": 0})
+        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.9}/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.9}/pyproject.toml

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