akshare-one 0.2.3__py3-none-any.whl → 0.3.0__py3-none-any.whl

akshare_one/modules/eastmoney/client.py

@@ -1,88 +1,88 @@
-import requests
-from typing import Dict, Any
-
-
-class EastMoneyClient:
-    """
-    A client for interacting directly with EastMoney's data APIs.
-    This class handles session management, request signing, and API calls.
-    """
-
-    def __init__(self):
-        self.session = requests.Session()
-        self.session.headers.update(
-            {
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/132.0.0.0 Safari/537.36",
-                "Referer": "https://quote.eastmoney.com/",
-                "Accept": "application/json, text/plain, */*",
-                "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
-            }
-        )
-
-    def _get_security_id(self, symbol: str) -> str:
-        """
-        Converts a stock symbol to EastMoney's internal secid format.
-        e.g., '600519' -> '1.600519', '000001' -> '0.000001'
-        """
-        symbol = symbol.upper()
-        if symbol.startswith("SZ"):
-            market = "0"
-            code = symbol[2:]
-        elif symbol.startswith("SH"):
-            market = "1"
-            code = symbol[2:]
-        elif symbol.startswith("HK"):
-            market = "116"
-            code = symbol[2:]
-        elif len(symbol) == 6:
-            if symbol.startswith(("000", "001", "002", "003", "300")):
-                market = "0"
-            elif symbol.startswith(("600", "601", "603", "605", "688")):
-                market = "1"
-            else:
-                market = "0"  # Default to SZ for ambiguity
-            code = symbol
-        elif len(symbol) == 5:  # HK Market
-            market = "116"
-            code = symbol
-        else:
-            market = "0"
-            code = symbol
-        return f"{market}.{code}"
-
-    def fetch_historical_klines(
-        self, symbol: str, klt: str, fqt: str, start_date: str, end_date: str
-    ) -> Dict[str, Any]:
-        """
-        Fetches historical K-line (candlestick) data.
-        """
-        url = "https://push2his.eastmoney.com/api/qt/stock/kline/get"
-        secid = self._get_security_id(symbol)
-        params = {
-            "fields1": "f1,f2,f3,f4,f5,f6",
-            "fields2": "f51,f52,f53,f54,f55,f56,f57,f58,f59,f60,f61",
-            "klt": klt,
-            "fqt": fqt,
-            "secid": secid,
-            "beg": start_date,
-            "end": end_date,
-        }
-        response = self.session.get(url, params=params)
-        response.raise_for_status()
-        return response.json()
-
-    def fetch_realtime_quote(self, symbol: str) -> Dict[str, Any]:
-        """
-        Fetches real-time quote data for a single stock.
-        """
-        url = "https://push2.eastmoney.com/api/qt/stock/get"
-        secid = self._get_security_id(symbol)
-        params = {
-            "invt": "2",
-            "fltt": "2",
-            "fields": "f43,f57,f58,f169,f170,f46,f60,f44,f51,f168,f47,f164,f163,f116,f60,f45,f52,f50,f48,f167,f117,f71,f161,f49,f530",
-            "secid": secid,
-        }
-        response = self.session.get(url, params=params)
-        response.raise_for_status()
-        return response.json()
+import requests
+from typing import Dict, Any
+
+
+class EastMoneyClient:
+    """
+    A client for interacting directly with EastMoney's data APIs.
+    This class handles session management, request signing, and API calls.
+    """
+
+    def __init__(self):
+        self.session = requests.Session()
+        self.session.headers.update(
+            {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/132.0.0.0 Safari/537.36",
+                "Referer": "https://quote.eastmoney.com/",
+                "Accept": "application/json, text/plain, */*",
+                "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
+            }
+        )
+
+    def _get_security_id(self, symbol: str) -> str:
+        """
+        Converts a stock symbol to EastMoney's internal secid format.
+        e.g., '600519' -> '1.600519', '000001' -> '0.000001'
+        """
+        symbol = symbol.upper()
+        if symbol.startswith("SZ"):
+            market = "0"
+            code = symbol[2:]
+        elif symbol.startswith("SH"):
+            market = "1"
+            code = symbol[2:]
+        elif symbol.startswith("HK"):
+            market = "116"
+            code = symbol[2:]
+        elif len(symbol) == 6:
+            if symbol.startswith(("000", "001", "002", "003", "300")):
+                market = "0"
+            elif symbol.startswith(("600", "601", "603", "605", "688")):
+                market = "1"
+            else:
+                market = "0"  # Default to SZ for ambiguity
+            code = symbol
+        elif len(symbol) == 5:  # HK Market
+            market = "116"
+            code = symbol
+        else:
+            market = "0"
+            code = symbol
+        return f"{market}.{code}"
+
+    def fetch_historical_klines(
+        self, symbol: str, klt: str, fqt: str, start_date: str, end_date: str
+    ) -> Dict[str, Any]:
+        """
+        Fetches historical K-line (candlestick) data.
+        """
+        url = "https://push2his.eastmoney.com/api/qt/stock/kline/get"
+        secid = self._get_security_id(symbol)
+        params = {
+            "fields1": "f1,f2,f3,f4,f5,f6",
+            "fields2": "f51,f52,f53,f54,f55,f56,f57,f58,f59,f60,f61",
+            "klt": klt,
+            "fqt": fqt,
+            "secid": secid,
+            "beg": start_date,
+            "end": end_date,
+        }
+        response = self.session.get(url, params=params)
+        response.raise_for_status()
+        return response.json()
+
+    def fetch_realtime_quote(self, symbol: str) -> Dict[str, Any]:
+        """
+        Fetches real-time quote data for a single stock.
+        """
+        url = "https://push2.eastmoney.com/api/qt/stock/get"
+        secid = self._get_security_id(symbol)
+        params = {
+            "invt": "2",
+            "fltt": "2",
+            "fields": "f43,f57,f58,f169,f170,f46,f60,f44,f51,f168,f47,f164,f163,f116,f60,f45,f52,f50,f48,f167,f117,f71,f161,f49,f530",
+            "secid": secid,
+        }
+        response = self.session.get(url, params=params)
+        response.raise_for_status()
+        return response.json()

akshare_one/modules/eastmoney/utils.py

@@ -1,104 +1,104 @@
-import pandas as pd
-from typing import Dict, Any
-
-
-def parse_kline_data(data: Dict[str, Any]) -> pd.DataFrame:
-    """
-    Parses K-line data from the API response into a pandas DataFrame.
-    """
-    klines = data.get("data", {}).get("klines", [])
-    if not klines:
-        return pd.DataFrame(
-            columns=["timestamp", "open", "high", "low", "close", "volume"]
-        )
-
-    records = []
-    for kline in klines:
-        parts = kline.split(",")
-        if len(parts) >= 6:
-            records.append(
-                {
-                    "timestamp": parts[0],
-                    "open": float(parts[1]),
-                    "close": float(parts[2]),
-                    "high": float(parts[3]),
-                    "low": float(parts[4]),
-                    "volume": int(parts[5]),
-                }
-            )
-
-    df = pd.DataFrame(records)
-    if not df.empty:
-        df["timestamp"] = pd.to_datetime(df["timestamp"])
-        df["timestamp"] = (
-            df["timestamp"].dt.tz_localize("Asia/Shanghai").dt.tz_convert("UTC")
-        )
-        df = df[["timestamp", "open", "high", "low", "close", "volume"]]
-    return df
-
-
-def parse_realtime_data(data: Dict[str, Any]) -> pd.DataFrame:
-    """
-    Parses real-time quote data from the API response into a pandas DataFrame.
-    """
-    stock_data = data.get("data")
-    if not stock_data:
-        return pd.DataFrame()
-
-    df = pd.DataFrame(
-        [
-            {
-                "symbol": stock_data.get("f57"),
-                "price": stock_data.get("f43"),
-                "change": stock_data.get("f169"),
-                "pct_change": stock_data.get("f170"),
-                "volume": stock_data.get("f47"),
-                "amount": stock_data.get("f48"),
-                "open": stock_data.get("f46"),
-                "high": stock_data.get("f44"),
-                "low": stock_data.get("f45"),
-                "prev_close": stock_data.get("f60"),
-            }
-        ]
-    )
-    df["timestamp"] = pd.Timestamp.now(tz="Asia/Shanghai").tz_convert("UTC")
-    return df
-
-
-def resample_historical_data(
-    df: pd.DataFrame, interval: str, multiplier: int
-) -> pd.DataFrame:
-    """
-    Resamples historical data to a specified frequency.
-    """
-    if df.empty or multiplier <= 1:
-        return df
-
-    df = df.set_index("timestamp")
-
-    freq_map = {
-        "day": f"{multiplier}D",
-        "week": f"{multiplier}W-MON",
-        "month": f"{multiplier}MS",
-        "year": f"{multiplier * 12}MS",
-    }
-    freq = freq_map.get(interval)
-
-    if not freq:
-        return df.reset_index()
-
-    resampled = (
-        df.resample(freq)
-        .agg(
-            {
-                "open": "first",
-                "high": "max",
-                "low": "min",
-                "close": "last",
-                "volume": "sum",
-            }
-        )
-        .dropna()
-    )
-
-    return resampled.reset_index()
+import pandas as pd
+from typing import Dict, Any
+
+
+def parse_kline_data(data: Dict[str, Any]) -> pd.DataFrame:
+    """
+    Parses K-line data from the API response into a pandas DataFrame.
+    """
+    klines = data.get("data", {}).get("klines", [])
+    if not klines:
+        return pd.DataFrame(
+            columns=["timestamp", "open", "high", "low", "close", "volume"]
+        )
+
+    records = []
+    for kline in klines:
+        parts = kline.split(",")
+        if len(parts) >= 6:
+            records.append(
+                {
+                    "timestamp": parts[0],
+                    "open": float(parts[1]),
+                    "close": float(parts[2]),
+                    "high": float(parts[3]),
+                    "low": float(parts[4]),
+                    "volume": int(parts[5]),
+                }
+            )
+
+    df = pd.DataFrame(records)
+    if not df.empty:
+        df["timestamp"] = pd.to_datetime(df["timestamp"])
+        df["timestamp"] = (
+            df["timestamp"].dt.tz_localize("Asia/Shanghai").dt.tz_convert("UTC")
+        )
+        df = df[["timestamp", "open", "high", "low", "close", "volume"]]
+    return df
+
+
+def parse_realtime_data(data: Dict[str, Any]) -> pd.DataFrame:
+    """
+    Parses real-time quote data from the API response into a pandas DataFrame.
+    """
+    stock_data = data.get("data")
+    if not stock_data:
+        return pd.DataFrame()
+
+    df = pd.DataFrame(
+        [
+            {
+                "symbol": stock_data.get("f57"),
+                "price": stock_data.get("f43"),
+                "change": stock_data.get("f169"),
+                "pct_change": stock_data.get("f170"),
+                "volume": stock_data.get("f47"),
+                "amount": stock_data.get("f48"),
+                "open": stock_data.get("f46"),
+                "high": stock_data.get("f44"),
+                "low": stock_data.get("f45"),
+                "prev_close": stock_data.get("f60"),
+            }
+        ]
+    )
+    df["timestamp"] = pd.Timestamp.now(tz="Asia/Shanghai").tz_convert("UTC")
+    return df
+
+
+def resample_historical_data(
+    df: pd.DataFrame, interval: str, multiplier: int
+) -> pd.DataFrame:
+    """
+    Resamples historical data to a specified frequency.
+    """
+    if df.empty or multiplier <= 1:
+        return df
+
+    df = df.set_index("timestamp")
+
+    freq_map = {
+        "day": f"{multiplier}D",
+        "week": f"{multiplier}W-MON",
+        "month": f"{multiplier}MS",
+        "year": f"{multiplier * 12}MS",
+    }
+    freq = freq_map.get(interval)
+
+    if not freq:
+        return df.reset_index()
+
+    resampled = (
+        df.resample(freq)
+        .agg(
+            {
+                "open": "first",
+                "high": "max",
+                "low": "min",
+                "close": "last",
+                "volume": "sum",
+            }
+        )
+        .dropna()
+    )
+
+    return resampled.reset_index()

akshare_one/modules/financial/base.py

@@ -1,22 +1,22 @@
-from abc import ABC, abstractmethod
-import pandas as pd
-
-
-class FinancialDataProvider(ABC):
-    def __init__(self, symbol: str) -> None:
-        self.symbol = symbol
-
-    @abstractmethod
-    def get_balance_sheet(self) -> pd.DataFrame:
-        """Fetches balance sheet data"""
-        pass
-
-    @abstractmethod
-    def get_income_statement(self) -> pd.DataFrame:
-        """Fetches income statement data"""
-        pass
-
-    @abstractmethod
-    def get_cash_flow(self) -> pd.DataFrame:
-        """Fetches cash flow data"""
-        pass
+from abc import ABC, abstractmethod
+import pandas as pd
+
+
+class FinancialDataProvider(ABC):
+    def __init__(self, symbol: str) -> None:
+        self.symbol = symbol
+
+    @abstractmethod
+    def get_balance_sheet(self) -> pd.DataFrame:
+        """Fetches balance sheet data"""
+        pass
+
+    @abstractmethod
+    def get_income_statement(self) -> pd.DataFrame:
+        """Fetches income statement data"""
+        pass
+
+    @abstractmethod
+    def get_cash_flow(self) -> pd.DataFrame:
+        """Fetches cash flow data"""
+        pass

akshare_one/modules/financial/factory.py

@@ -1,44 +1,44 @@
-from .sina import SinaFinancialReport
-from .base import FinancialDataProvider
-
-
-class FinancialDataFactory:
-    """
-    Factory class for creating financial data providers
-    """
-
-    _providers = {
-        "sina": SinaFinancialReport,
-    }
-
-    @classmethod
-    def get_provider(cls, provider_name: str, **kwargs) -> FinancialDataProvider:
-        """
-        Get a financial data provider by name
-
-        Args:
-            provider_name: Name of the provider (e.g., 'sina')
-            **kwargs: Additional arguments to pass to the provider's constructor
-
-        Returns:
-            FinancialDataProvider: An instance of the requested provider
-
-        Raises:
-            ValueError: If the requested provider is not found
-        """
-        provider_class = cls._providers.get(provider_name.lower())
-        if not provider_class:
-            raise ValueError(f"Unknown financial data provider: {provider_name}")
-
-        return provider_class(**kwargs)
-
-    @classmethod
-    def register_provider(cls, name: str, provider_class: type):
-        """
-        Register a new financial data provider
-
-        Args:
-            name: Name to associate with this provider
-            provider_class: The provider class to register
-        """
-        cls._providers[name.lower()] = provider_class
+from .sina import SinaFinancialReport
+from .base import FinancialDataProvider
+
+
+class FinancialDataFactory:
+    """
+    Factory class for creating financial data providers
+    """
+
+    _providers = {
+        "sina": SinaFinancialReport,
+    }
+
+    @classmethod
+    def get_provider(cls, provider_name: str, **kwargs) -> FinancialDataProvider:
+        """
+        Get a financial data provider by name
+
+        Args:
+            provider_name: Name of the provider (e.g., 'sina')
+            **kwargs: Additional arguments to pass to the provider's constructor
+
+        Returns:
+            FinancialDataProvider: An instance of the requested provider
+
+        Raises:
+            ValueError: If the requested provider is not found
+        """
+        provider_class = cls._providers.get(provider_name.lower())
+        if not provider_class:
+            raise ValueError(f"Unknown financial data provider: {provider_name}")
+
+        return provider_class(**kwargs)
+
+    @classmethod
+    def register_provider(cls, name: str, provider_class: type):
+        """
+        Register a new financial data provider
+
+        Args:
+            name: Name to associate with this provider
+            provider_class: The provider class to register
+        """
+        cls._providers[name.lower()] = provider_class