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

{neurostats_API-0.0.4 → neurostats_api-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: neurostats_API
-Version: 0.0.4
+Version: 0.0.5
 Summary: The service of NeuroStats website
 Home-page: https://github.com/NeurowattStats/NeuroStats_API.git
 Author: JasonWang@Neurowatt

neurostats_api-0.0.5/neurostats_API/__init__.py

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

{neurostats_API-0.0.4 → neurostats_api-0.0.5}/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.5}/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.5}/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.5}/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.5}/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.5/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.5}/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.5}/neurostats_API.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
-Name: neurostats-API
-Version: 0.0.4
+Name: neurostats_API
+Version: 0.0.5
 Summary: The service of NeuroStats website
 Home-page: https://github.com/NeurowattStats/NeuroStats_API.git
 Author: JasonWang@Neurowatt

{neurostats_API-0.0.4 → neurostats_api-0.0.5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='neurostats_API',
-    version='0.0.4',
+    version='0.0.5',
     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.5}/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)