neurostats-API 0.0.4__tar.gz → 0.0.6__tar.gz

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: neurostats_API
-Version: 0.0.4
+Version: 0.0.6
 Summary: The service of NeuroStats website
 Home-page: https://github.com/NeurowattStats/NeuroStats_API.git
 Author: JasonWang@Neurowatt
@@ -118,9 +118,10 @@ data = stats_fetcher.query_data()
 
 ### 回傳月營收表
 ``` Python
-from neurostats_API.utils import StatsFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = 2330 # 換成tw50內任意ticker
-fetcher = StatsFetcher()
+fetcher = StatsFetcher(db_client)
 data = stats_fetcher.get_month_revenue_sheet(ticker)
 ```
 
@@ -170,9 +171,10 @@ YoY_10          None         None  ...           None           None
 ### 財務分析: 重要指標
 對應https://ifa.ai/tw-stock/2330/finance-overview
 ```Python
-from neurostats_API.fetchers import FinanceOverviewFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = "2330"
-fetcher = FinanceOverviewFetcher(ticker = "2330")
+fetcher = FinanceOverviewFetcher(ticker = "2330", db_client = db_client)
 data = fetcher.query_data()
 ```
 
@@ -253,8 +255,9 @@ markdown
 
 ### 損益表
 ```Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 data = fetcher.get_profit_lose(ticker)
 ```
@@ -301,8 +304,9 @@ data = fetcher.get_profit_lose(ticker)
 
 ### 資產負債表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_balance_sheet(ticker)
 ```
@@ -360,8 +364,9 @@ equity  1.094370e+09               53.3  ...  3.390706e+09              56.68
 
 ### 現金流量表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_cash_flow(ticker)
 ```

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/README.md

@@ -108,9 +108,10 @@ data = stats_fetcher.query_data()
 
 ### 回傳月營收表
 ``` Python
-from neurostats_API.utils import StatsFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = 2330 # 換成tw50內任意ticker
-fetcher = StatsFetcher()
+fetcher = StatsFetcher(db_client)
 data = stats_fetcher.get_month_revenue_sheet(ticker)
 ```
 
@@ -160,9 +161,10 @@ YoY_10          None         None  ...           None           None
 ### 財務分析: 重要指標
 對應https://ifa.ai/tw-stock/2330/finance-overview
 ```Python
-from neurostats_API.fetchers import FinanceOverviewFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = "2330"
-fetcher = FinanceOverviewFetcher(ticker = "2330")
+fetcher = FinanceOverviewFetcher(ticker = "2330", db_client = db_client)
 data = fetcher.query_data()
 ```
 
@@ -243,8 +245,9 @@ markdown
 
 ### 損益表
 ```Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 data = fetcher.get_profit_lose(ticker)
 ```
@@ -291,8 +294,9 @@ data = fetcher.get_profit_lose(ticker)
 
 ### 資產負債表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_balance_sheet(ticker)
 ```
@@ -350,8 +354,9 @@ equity  1.094370e+09               53.3  ...  3.390706e+09              56.68
 
 ### 現金流量表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_cash_flow(ticker)
 ```

neurostats_API-0.0.6/neurostats_API/__init__.py

@@ -0,0 +1 @@
+__version__='0.0.6'

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/fetchers/base.py

@@ -5,14 +5,11 @@ import pytz
 from datetime import datetime, timedelta, date
 from ..utils import StatsDateTime, StatsProcessor
 import yaml
-from ..utils import shared_client 
-
 
 class StatsFetcher:
-
-    def __init__(self, ticker):
+    def __init__(self, ticker, db_client):
         self.ticker = ticker
-        self.db = shared_client[
+        self.db = db_client[
             "company"]  # Replace with your database name
         self.collection = self.db["twse_stats"]
 

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/fetchers/finance_overview.py

@@ -11,8 +11,8 @@ class FinanceOverviewFetcher(StatsFetcher):
     對應iFa.ai -> 財務分析 -> 重要指標(finance_overview)
     """
 
-    def __init__(self, ticker):
-        super().__init__(ticker)
+    def __init__(self, ticker, db_client):
+        super().__init__(ticker, db_client)
 
         self.target_fields = StatsProcessor.load_yaml("finance_overview_dict.yaml")
         self.inverse_dict = StatsProcessor.load_txt("seasonal_data_field_dict.txt", json_load=True)

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/fetchers/tech.py

@@ -3,9 +3,8 @@ import pandas as pd
 
 class TechFetcher(StatsFetcher):
     
-    def __init__(self, ticker:str):
-        super().__init__()
-        self.ticker = ticker
+    def __init__(self, ticker:str, db_client):
+        super().__init__(ticker, db_client)
         self.full_ohlcv = self._get_ohlcv()
         self.basic_indexes = ['SMA5', 'SMA20', 'SMA60', 'EMA5', 'EMA20', 
                               'EMA40', 'EMA12', 'EMA26', 'RSI7', 'RSI14',

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/fetchers/value_invest.py

@@ -6,8 +6,8 @@ from ..utils import StatsDateTime, StatsProcessor
 
 class ValueFetcher(StatsFetcher):
 
-    def __init__(self, ticker: str):
-        super().__init__(ticker)
+    def __init__(self, ticker: str, db_client):
+        super().__init__(ticker, db_client)
 
     def prepare_query(self, start_date, end_date):
         pipeline = super().prepare_query()

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/utils/__init__.py

@@ -1,4 +1,4 @@
 from .datetime import StatsDateTime
-from .db_client import shared_client
+from .db_client import DBClient
 from .data_process import StatsProcessor
 from .fetcher import StatsFetcher

neurostats_API-0.0.6/neurostats_API/utils/db_client.py

@@ -0,0 +1,16 @@
+import os
+from pymongo import MongoClient
+from dotenv import load_dotenv
+
+class DBClient:
+    def __init__(self, mongo_uri):
+        """初始化時接收 MongoDB 連接 URI"""
+        self.client = MongoClient(mongo_uri)
+        self.db = self.client["twse_company"]
+    
+    def get_client(self):
+        return self.client
+
+    def get_collection(self, collection_name):
+        """返回指定名稱的集合"""
+        return self.db[collection_name]

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API/utils/fetcher.py

@@ -1,4 +1,3 @@
-from .db_client import shared_client
 import pandas as pd
 import json
 import pytz
@@ -9,8 +8,8 @@ import yaml
 
 class StatsFetcher:
 
-    def __init__(self):
-        self.db = shared_client["company"]  # Replace with your database name
+    def __init__(self, db_client):
+        self.db = db_client["company"]  # Replace with your database name
         self.collection = self.db["twse_stats"]
 
         self.timezone = pytz.timezone("Asia/Taipei")

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/neurostats_API.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: neurostats-API
-Version: 0.0.4
+Version: 0.0.6
 Summary: The service of NeuroStats website
 Home-page: https://github.com/NeurowattStats/NeuroStats_API.git
 Author: JasonWang@Neurowatt
@@ -118,9 +118,10 @@ data = stats_fetcher.query_data()
 
 ### 回傳月營收表
 ``` Python
-from neurostats_API.utils import StatsFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = 2330 # 換成tw50內任意ticker
-fetcher = StatsFetcher()
+fetcher = StatsFetcher(db_client)
 data = stats_fetcher.get_month_revenue_sheet(ticker)
 ```
 
@@ -170,9 +171,10 @@ YoY_10          None         None  ...           None           None
 ### 財務分析: 重要指標
 對應https://ifa.ai/tw-stock/2330/finance-overview
 ```Python
-from neurostats_API.fetchers import FinanceOverviewFetcher
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
 ticker = "2330"
-fetcher = FinanceOverviewFetcher(ticker = "2330")
+fetcher = FinanceOverviewFetcher(ticker = "2330", db_client = db_client)
 data = fetcher.query_data()
 ```
 
@@ -253,8 +255,9 @@ markdown
 
 ### 損益表
 ```Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 data = fetcher.get_profit_lose(ticker)
 ```
@@ -301,8 +304,9 @@ data = fetcher.get_profit_lose(ticker)
 
 ### 資產負債表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_balance_sheet(ticker)
 ```
@@ -360,8 +364,9 @@ equity  1.094370e+09               53.3  ...  3.390706e+09              56.68
 
 ### 現金流量表
 ``` Python
-from neurostats_API.utils import StatsFetcher
-fetcher = StatsFetcher()
+from neurostats_API.utils import StatsFetcher, DBClient
+db_client = DBClient("<連接的DB位置>").get_client()
+fetcher = StatsFetcher(db_client)
 ticker = 2330 # 換成tw50內任意ticker
 stats_fetcher.get_cash_flow(ticker)
 ```

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='neurostats_API',
-    version='0.0.4',
+    version='0.0.6',
     long_description=open('README.md', 'r', encoding='utf-8').read(),
     long_description_content_type='text/markdown',
     install_requires=[

{neurostats_API-0.0.4 → neurostats_API-0.0.6}/test/test_fetchers.py

@@ -1,12 +1,13 @@
 import pytest
 import pandas as pd
 import yaml
-from neurostats_API.utils import StatsProcessor
+from neurostats_API.utils import StatsProcessor, DBClient
 
+db_client = DBClient("mongodb://neurowatt:neurodb123@db.neurowatt.ai:27017/neurowatt").get_client()
 
 def test_value():
     from neurostats_API.fetchers import ValueFetcher
-    fetcher = ValueFetcher(ticker='2330')
+    fetcher = ValueFetcher(ticker='2330', db_client=db_client)
 
     fetched_data = fetcher.query_data()
 
@@ -19,7 +20,7 @@ def test_value():
 def test_profit_lose():
     from neurostats_API.utils import StatsFetcher
 
-    fetcher = StatsFetcher()
+    fetcher = StatsFetcher(db_client)
 
     data = fetcher.get_profit_lose("2330")
 
@@ -31,7 +32,7 @@ def test_profit_lose():
 def test_cash_flow():
     from neurostats_API.utils import StatsFetcher
 
-    fetcher = StatsFetcher()
+    fetcher = StatsFetcher(db_client)
     data = fetcher.get_cash_flow("2330")
 
     assert("cash_flow" in data.keys())
@@ -39,7 +40,7 @@ def test_cash_flow():
 def test_month_revenue():
     from neurostats_API.utils import StatsFetcher
 
-    fetcher = StatsFetcher()
+    fetcher = StatsFetcher(db_client)
     data = fetcher.get_month_revenue_sheet("2330")
 
     assert("month_revenue" in data.keys())
@@ -47,7 +48,7 @@ def test_month_revenue():
 def test_balance_sheet():
     from neurostats_API.utils import StatsFetcher
 
-    fetcher = StatsFetcher()
+    fetcher = StatsFetcher(db_client)
 
     data = fetcher.get_balance_sheet("2330")
 
@@ -57,7 +58,7 @@ def test_balance_sheet():
 
 def test_finance_overview():
     from neurostats_API.fetchers import FinanceOverviewFetcher
-    fetcher = FinanceOverviewFetcher(ticker='2330')
+    fetcher = FinanceOverviewFetcher(ticker='2330', db_client=db_client)
     fetched_data = fetcher.query_data()
 
     expected_keys = [

neurostats_API-0.0.4/neurostats_API/__init__.py

@@ -1 +0,0 @@
-__version__='0.0.4'

neurostats_API-0.0.4/neurostats_API/utils/db_client.py

@@ -1,10 +0,0 @@
-import os
-from pymongo import MongoClient
-from dotenv import load_dotenv
-
-# 載入 .env 檔案
-load_dotenv()
-
-# 創建並共享 MongoClient 連線
-mongo_uri = os.getenv('MONGO_URI', 'mongodb://localhost:27017')
-shared_client = MongoClient(mongo_uri)