db2_hj3415 0.1.5__tar.gz → 0.1.7__tar.gz

{db2_hj3415-0.1.5 → db2_hj3415-0.1.7}/PKG-INFO

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

db2_hj3415-0.1.7/db2_hj3415/nfs/dart.py

@@ -0,0 +1,27 @@
+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)}

{db2_hj3415-0.1.5 → db2_hj3415-0.1.7}/db2_hj3415/nfs/models.py

@@ -2,7 +2,7 @@ from pydantic import BaseModel, Field, field_serializer, ConfigDict, field_valid
 from datetime import datetime
 
 class C101(BaseModel):
-    id: str | None = Field(alias="_id")
+    id: str | None = Field(default=None, alias="_id")
     날짜: datetime
     코드: str
     bps: int | None
@@ -58,7 +58,7 @@ class 항목값q(BaseModel):
 
 
 class C103(BaseModel):
-    id: str | None = Field(alias="_id")
+    id: str | None = Field(default=None, alias="_id")
     코드: str
     날짜: datetime
     손익계산서q: list[항목값q]
@@ -76,7 +76,7 @@ class C103(BaseModel):
 
 
 class C104(BaseModel):
-    id: str | None = Field(alias="_id")
+    id: str | None = Field(default=None, alias="_id")
     코드: str
     날짜: datetime
 
@@ -109,7 +109,7 @@ class 기업데이터(BaseModel):
     )
 
 class C106(BaseModel):
-    id: str | None = Field(alias="_id")
+    id: str | None = Field(default=None, alias="_id")
     코드: str
     날짜: datetime
 
@@ -124,7 +124,7 @@ class C106(BaseModel):
 
 
 class C108(BaseModel):
-    id: str | None = Field(alias="_id")
+    id: str | None = Field(default=None, alias="_id")
     코드: str
     날짜: datetime
 
@@ -147,9 +147,9 @@ class C108(BaseModel):
     )
 
 class Dart(BaseModel):
-    id: str | None = Field(alias="_id")
-    코드: str = Field(default=None, alias="stock_code") # 종목 코드 (6자리)
-    날짜: datetime = Field(default=None, alias="rcept_dt") # 접수 일자 (YYYYMMDD)
+    id: str | None = Field(default=None, alias="_id")
+    코드: str | None= Field(default=None, alias="stock_code") # 종목 코드 (6자리)
+    날짜: datetime | None = Field(default=None, alias="rcept_dt") # 접수 일자 (YYYYMMDD)
 
     corp_cls: str  # 기업 구분 (예: 'K', 'Y', 'N')
     corp_code: str  # 고유 회사 코드 (8자리)

{db2_hj3415-0.1.5 → db2_hj3415-0.1.7}/pyproject.toml

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