neurostats-API 1.0.0rc6__tar.gz → 1.0.1rc1__tar.gz

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: neurostats_API
-Version: 1.0.0rc6
+Version: 1.0.1rc1
 Summary: The service of NeuroStats website
 Home-page: https://github.com/NeurowattStats/NeuroStats_API.git
 Author: JasonWang@Neurowatt
@@ -71,7 +71,7 @@ pip install neurostats-API
 ```Python 
 >>> import neurostats_API
 >>> print(neurostats_API.__version__)
-1.0.0rc6
+1.0.0
 ```
 
 ### 下載舊版

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/README.md

@@ -54,7 +54,7 @@ pip install neurostats-API
 ```Python 
 >>> import neurostats_API
 >>> print(neurostats_API.__version__)
-1.0.0rc6
+1.0.0
 ```
 
 ### 下載舊版

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/__init__.py

@@ -1,4 +1,4 @@
-__version__='1.0.0rc5'
+__version__='1.0.0'
 
 from .fetchers import (
     AgentFinanceOverviewFetcher,

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/db_extractors/__init__.py

@@ -4,6 +4,7 @@ from .daily import (
     AsyncDailyValueDBExtractor,
     AsyncTEJDailyValueDBExtractor,
     AsyncTWSEChipDBExtractor,
+    AsyncUS_F13DBExtractor
 
 )
 

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/db_extractors/daily/__init__.py

@@ -4,4 +4,5 @@ from .value import (
     AsyncDailyValueDBExtractor, 
     AsyncTEJDailyValueDBExtractor
 )
+from .us_chip import AsyncUS_F13DBExtractor
 from .yf import AsyncYFDailyTechDBExtractor

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/db_extractors/daily/base.py

@@ -1,5 +1,5 @@
 from ..base import BaseDBExtractor
-from datetime import datetime
+from datetime import datetime, timezone
 from neurostats_API.utils import NotSupportedError
 from pymongo import ASCENDING, DESCENDING
 
@@ -73,16 +73,23 @@ class BaseDailyTechDBExtractor(BaseDBExtractor):
 
         return query
 
-    def _transform_date(self, date):
+    def _transform_date(self, date, use_UTC = False):
         if (isinstance(date, str)):
             try:
-                return datetime.strptime(date, "%Y-%m-%d")
+                date = datetime.strptime(date, "%Y-%m-%d")
+                if (use_UTC):
+                    date = date.astimezone(tz = timezone.utc)
+                return date 
             except Exception as e:
+                print(e)
                 raise ValueError(
                     "date string format imcompatible, should be \"YYYY-MM-DD\""
                 )
         elif (isinstance(date, datetime)):
-            return date
+            if (use_UTC):
+                return date.astimezone(tz = timezone.utc)
+            else:
+                return date
         else:
             raise ValueError(
                 "date type imcompatible, should be string(\"YYYY-MM-DD\") or datetime.datetime"

neurostats_api-1.0.1rc1/neurostats_API/async_mode/db_extractors/daily/us_chip.py

@@ -0,0 +1,66 @@
+from datetime import datetime, timedelta
+from pymongo import ASCENDING, DESCENDING
+from .base import BaseDailyTechDBExtractor
+from neurostats_API.utils import (
+    NotSupportedError, StatsProcessor
+)
+
+
+class AsyncUS_F13DBExtractor(BaseDailyTechDBExtractor):
+
+    def __init__(self, ticker, client, managerTicker=None):
+        """
+        ticker: Issuer Ticker, 被持有的股票代碼
+        managerTicker: 持有者的股票代碼
+        """
+        super().__init__(ticker, client)
+        self.issuerTicker = ticker
+        self.managerTicker = managerTicker
+
+        if (self.get_zone() not in ['us']):
+            raise NotSupportedError("Supports US Company Only")
+
+    def _get_collection_name(self):
+        self.collection_name_map = {"us": "us_F13"}
+
+        return self.collection_name_map.get(self.zone, None)
+
+    def _prepare_query(self, start_date=None, end_date=None, get_latest=False):
+        query = {"issuerTicker": self.issuerTicker}
+
+        if (self.managerTicker):
+            query.update(
+                {
+                    "managerTicker": self.managerTicker
+                }
+            )
+
+        query = self._update_query_with_date(query, start_date, end_date)
+
+        projection = {
+            "_id": 0,
+        }
+
+        if (get_latest):
+            sort = [("reportDate", DESCENDING)]
+        else:
+            sort = [("reportDate", ASCENDING)]
+
+        return query, projection, sort
+
+    def _update_query_with_date(self, query, start_date, end_date):
+        start_date = self._transform_date(start_date)
+        end_date = self._transform_date(end_date)
+
+        date_range = {"$gte": start_date, "$lte": end_date}
+
+        query.update({"reportDate": date_range})
+
+        return query
+
+    async def query_data(
+        self, start_date=None, end_date=None, get_latest=False
+    ):
+        result = await super().query_data(start_date, end_date, get_latest)
+
+        return result

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/factory/extractor_factory.py

@@ -5,7 +5,7 @@ from neurostats_API.async_mode.db_extractors import (
     AsyncTWSEChipDBExtractor, AsyncTWSEMonthlyRevenueExtractor,
     AsyncBalanceSheetExtractor, AsyncProfitLoseExtractor,
     AsyncCashFlowExtractor, AsyncTEJFinanceStatementDBExtractor,
-    AsyncTEJSelfSettlementDBExtractor
+    AsyncTEJSelfSettlementDBExtractor, AsyncUS_F13DBExtractor
 )
 from neurostats_API.utils import NotSupportedError
 
@@ -139,7 +139,19 @@ EXTRACTOR_MAP: dict[str, ExtractorConfig] = {
         DB_collection: twse_daily_share_price
         TWSE的評價面
         """
-    }
+    },
+
+
+    "US_Chip_F13": {
+        "class_":
+        AsyncUS_F13DBExtractor,
+        "default_kwargs": {},
+        "description":
+        """
+        DB_collection: "us_F13_flat"
+        US籌碼面(F13)
+        """
+    },
 }
 
 

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/fetchers/__init__.py

@@ -7,4 +7,5 @@ from .twse_margin import AsyncTWSEMarginFetcher
 from .month_revenue import AsyncMonthlyRevenueFetcher
 from .profit_lose import AsyncProfitLoseFetcher
 from .value import AsyncTWSEStatsValueFetcher
-from .tej import AsyncTEJDailyTechFetcher, AsyncTEJSeasonalFetcher
+from .tej import AsyncTEJDailyTechFetcher, AsyncTEJSeasonalFetcher
+from .us_chip import AsyncUSChipF13Fetcher

{neurostats_api-1.0.0rc6 → neurostats_api-1.0.1rc1}/neurostats_API/async_mode/fetchers/finance_overview.py

@@ -14,11 +14,6 @@ import pytz
 import holidays
 import warnings
 
-warnings.filterwarnings(
-    "ignore",
-    category=holidays.deprecations.v1_incompatibility.
-    FutureIncompatibilityWarning
-)
 import yaml
 
 

neurostats_api-1.0.1rc1/neurostats_API/async_mode/fetchers/us_chip.py

@@ -0,0 +1,46 @@
+from .base import AsyncBaseFetcher
+from ..db_extractors import (
+    AsyncDailyValueDBExtractor, AsyncTEJDailyValueDBExtractor
+)
+from neurostats_API.async_mode.factory import get_extractor
+from neurostats_API.transformers import US_F13_Transformer
+from neurostats_API.utils import NotSupportedError
+
+
+class AsyncUSChipF13Fetcher(AsyncBaseFetcher):
+    def __init__(self, 
+        ticker, 
+        client,
+        managerTicker = None,
+    ):
+        self.issuerTicker = ticker
+        self.managerTicker = managerTicker
+        self.transformer_map = {"us": US_F13_Transformer}
+        try:
+            self.extractor = get_extractor(
+                "US_Chip_F13", ticker, client, managerTicker=self.managerTicker
+            )
+
+            company_name = self.extractor.get_company_name()
+            zone = self.extractor.get_zone()
+
+            transformer = self.get_transformer(zone)
+            self.transformer = transformer(ticker, company_name, zone)
+
+        except NotSupportedError as e:
+            raise NotSupportedError(
+                f"{self.__class__.__name__} only support {list(self.transformer_map.keys())} companies now, {ticker} is not available"
+            ) from e
+
+    async def query_data(
+        self, start_date=None, end_date=None, get_latest=False
+    ):
+        fetched_data = await self.extractor.query_data(
+            start_date, end_date, get_latest
+        )
+
+        transformed_data = self.transformer.process_transform(
+            fetched_data
+        )
+
+        return transformed_data