linkmerce 0.3.2__tar.gz → 0.3.3__tar.gz

{linkmerce-0.3.2 → linkmerce-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: linkmerce
-Version: 0.3.2
+Version: 0.3.3
 Summary: E-commerce API integration management
 Requires-Dist: aiohttp>=3.12.15
 Requires-Dist: bs4>=0.0.2

{linkmerce-0.3.2 → linkmerce-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "linkmerce"
-version = "0.3.2"
+version = "0.3.3"
 description = "E-commerce API integration management"
 readme = "README.md"
 requires-python = ">=3.10"

{linkmerce-0.3.2 → linkmerce-0.3.3}/src/linkmerce/common/load.py

@@ -179,8 +179,8 @@ class Connection(metaclass=ABCMeta):
 ###################################################################
 
 class DuckDBConnection(Connection):
-    def __init__(self, **kwargs):
-        self.set_connection(**kwargs)
+    def __init__(self, tzinfo: str | None = None, **kwargs):
+        self.set_connection(tzinfo, **kwargs)
 
     @property
     def conn(self) -> DuckDBPyConnection:
@@ -189,9 +189,11 @@ class DuckDBConnection(Connection):
     def get_connection(self) -> DuckDBPyConnection:
         return self.__conn
 
-    def set_connection(self, **kwargs):
+    def set_connection(self, tzinfo: str | None = None, **kwargs):
         import duckdb
         self.__conn = duckdb.connect(**kwargs)
+        if tzinfo is not None:
+            self.conn.execute(f"SET TimeZone = '{tzinfo}';")
 
     def close(self):
         try:
@@ -495,6 +497,9 @@ class DuckDBConnection(Connection):
             return bool(self.conn.execute(query).fetchone())
         return False
 
+    def count_table(self, table: str) -> int:
+        return self.conn.execute(f"SELECT COUNT(*) FROM {table}").fetchall()[0][0]
+
     def get_columns(self, obj: str | DuckDBPyConnection) -> list[str]:
         if isinstance(obj, str):
             obj = self.conn.execute(f"DESCRIBE {obj}")

{linkmerce-0.3.2 → linkmerce-0.3.3}/src/linkmerce/core/naver/openapi/search/models.sql

@@ -240,6 +240,7 @@ CREATE OR REPLACE TABLE {{ table }} (
     nvMid BIGINT PRIMARY KEY
   , mallPid BIGINT
   , productName VARCHAR
+  , wholeCategoryName VARCHAR
   , mallName VARCHAR
   , brandName VARCHAR
   , updatedAt TIMESTAMP NOT NULL
@@ -250,6 +251,7 @@ SELECT
     TRY_CAST(productId AS BIGINT) AS nvMid
   , TRY_CAST(REGEXP_EXTRACT(link, '/products/(\d+)$', 1) AS BIGINT) AS mallPid
   , REGEXP_REPLACE(title, '<[^>]+>', '', 'g') AS productName
+  , CONCAT_WS('>', category1, category2, category3, category4) AS wholeCategoryName
   , NULLIF(mallName, '네이버') AS mallName
   , NULLIF(brand, '') AS brandName
   , CAST(DATE_TRUNC('second', CURRENT_TIMESTAMP) AS TIMESTAMP) AS updatedAt
@@ -261,6 +263,7 @@ INSERT INTO {{ table }} {{ values }}
 ON CONFLICT DO UPDATE SET
     mallPid = COALESCE(excluded.mallPid, mallPid)
   , productName = COALESCE(excluded.productName, productName)
+  , wholeCategoryName = COALESCE(excluded.wholeCategoryName, wholeCategoryName)
   , mallName = COALESCE(excluded.mallName, mallName)
   , brandName = COALESCE(excluded.brandName, brandName)
   , updatedAt = excluded.updatedAt;

{linkmerce-0.3.2 → linkmerce-0.3.3}/src/linkmerce/core/searchad/manage/exposure/models.sql

@@ -29,7 +29,7 @@ SELECT
   , NULLIF(fmpBrand, '') AS mallName
   , NULLIF(fmpMaker, '') AS makerName
   , imageUrl
-  , TRY_CAST(COALESCE(lowPrice, mobileLowPrice, NULL) AS INTEGER) AS salesPrice
+  , TRY_CAST(COALESCE(lowPrice, mobileLowPrice) AS INTEGER) AS salesPrice
 FROM {{ array }}
 WHERE ($is_own IS NULL) OR (isOwn = $is_own);
 
@@ -71,6 +71,7 @@ INSERT INTO {{ table }} {{ values }} ON CONFLICT DO NOTHING;
 CREATE OR REPLACE TABLE {{ table }} (
     productId BIGINT PRIMARY KEY
   , isNvMid BOOLEAN
+  , mallPid BIGINT
   , productName VARCHAR
   , wholeCategoryName VARCHAR
   , mallName VARCHAR
@@ -89,10 +90,11 @@ FROM (
           TRY_CAST(TRY_CAST(FROM_BASE64(REGEXP_EXTRACT(imageUrl, '^https://[^/]+/[^/]+/([^.]+)', 1)) AS VARCHAR) AS BIGINT)
         ELSE NULL END) AS productId
     , PREFIX(imageUrl, 'https://shopping-') AS isNvMid
+    , NULL AS mallPid
     , productTitle AS productName
     , categoryNames AS wholeCategoryName
     , NULLIF(fmpBrand, '') AS mallName
-    , TRY_CAST(COALESCE(lowPrice, mobileLowPrice, NULL) AS INTEGER) AS salesPrice
+    , TRY_CAST(COALESCE(lowPrice, mobileLowPrice) AS INTEGER) AS salesPrice
     , CAST(DATE_TRUNC('second', CURRENT_TIMESTAMP) AS TIMESTAMP) AS updatedAt
   FROM {{ array }}
   WHERE ($is_own IS NULL) OR (isOwn = $is_own)

{linkmerce-0.3.2 → linkmerce-0.3.3}/src/linkmerce/core/searchad/manage/exposure/transform.py

@@ -52,8 +52,12 @@ class ExposureRank(ExposureDiagnosis):
             params: dict = dict(),
             **kwargs
         ) -> tuple[DuckDBPyRelation,DuckDBPyRelation]:
+        def reparse_object(obj: list[dict]) -> list[dict]:
+            obj[0] = dict(obj[0], lowPrice=obj[0].get("lowPrice", None), mobileLowPrice=obj[0].get("mobileLowPrice", None))
+            return obj
         def split_params(keyword: str, is_own: bool | None = None, **kwargs) -> tuple[dict,dict]:
             return dict(keyword=keyword, is_own=is_own), dict(is_own=is_own)
+        obj = reparse_object(obj)
         rank_params, product_params = split_params(**params)
         rank = super().insert_into_table(obj, key="insert_rank", table=rank_table, values=":select_rank:", params=rank_params)
         product = super().insert_into_table(obj, key="upsert_product", table=product_table, values=":select_product:", params=product_params)

{linkmerce-0.3.2 → linkmerce-0.3.3}/src/linkmerce/utils/tqdm.py

@@ -10,7 +10,7 @@ if TYPE_CHECKING:
 
 
 ###################################################################
-############################## Select #############################
+############################## Gather #############################
 ###################################################################
 
 def gather(
@@ -81,7 +81,7 @@ def _get_seconds(value: Real | Sequence[Real,Real]) -> Real:
 
 
 ###################################################################
-############################## Select #############################
+############################## Expand #############################
 ###################################################################
 
 def expand(