akshare-one 0.3.5__py3-none-any.whl → 0.3.6__py3-none-any.whl

akshare_one/eastmoney/client.py

@@ -0,0 +1,80 @@
+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()
+
+    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", "200")):
+                market = "0"
+            elif symbol.startswith(("600", "601", "603", "605", "688", "900")):
+                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/eastmoney/utils.py

@@ -0,0 +1,102 @@
+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")
+        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")
+    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/eastmoney_direct.py

@@ -0,0 +1,183 @@
+import pandas as pd
+import requests
+
+from akshare_one.modules.cache import cache
+from .base import FinancialDataProvider
+
+
+class EastMoneyDirectFinancialReport(FinancialDataProvider):
+    _balance_sheet_rename_map = {
+        "REPORT_DATE": "report_date",
+        "TOTAL_ASSETS": "total_assets",
+        "FIXED_ASSET": "fixed_assets_net",
+        "MONETARYFUNDS": "cash_and_equivalents",
+        "ACCOUNTS_RECE": "accounts_receivable",
+        "INVENTORY": "inventory",
+        "TOTAL_LIABILITIES": "total_liabilities",
+        "ACCOUNTS_PAYABLE": "trade_and_non_trade_payables",
+        "ADVANCE_RECEIVABLES": "deferred_revenue",
+        "TOTAL_EQUITY": "shareholders_equity",
+    }
+
+    _income_statement_rename_map = {
+        "REPORT_DATE": "report_date",
+        "TOTAL_OPERATE_INCOME": "revenue",
+        "TOTAL_OPERATE_COST": "total_operating_costs",
+        "OPERATE_PROFIT": "operating_profit",
+        "PARENT_NETPROFIT": "net_income_common_stock",
+    }
+
+    _cash_flow_rename_map = {
+        "REPORT_DATE": "report_date",
+        "NETCASH_OPERATE": "net_cash_flow_from_operations",
+        "NETCASH_INVEST": "net_cash_flow_from_investing",
+        "NETCASH_FINANCE": "net_cash_flow_from_financing",
+        "CCE_ADD": "change_in_cash_and_equivalents",
+    }
+
+    def __init__(self, symbol):
+        super().__init__(symbol)
+
+    def get_income_statement(self):
+        pass
+
+    def get_balance_sheet(self):
+        pass
+
+    def get_cash_flow(self):
+        pass
+
+    @cache(
+        "financial_cache",
+        key=lambda self, symbol=None: f"eastmoney_financial_metrics_{self.symbol}",
+    )
+    def get_financial_metrics(self) -> pd.DataFrame:
+        """获取三大财务报表关键指标"""
+        balance_sheet = self._fetch_balance_sheet()
+        income_statement = self._fetch_income_statement()
+        cash_flow = self._fetch_cash_flow()
+
+        if balance_sheet.empty and income_statement.empty and cash_flow.empty:
+            return pd.DataFrame()
+
+        merged = pd.merge(
+            balance_sheet, income_statement, on="report_date", how="outer"
+        )
+        merged = pd.merge(merged, cash_flow, on="report_date", how="outer")
+
+        # Convert report_date to datetime and format as YYYY-MM-DD
+        merged["report_date"] = pd.to_datetime(merged["report_date"]).dt.strftime(
+            "%Y-%m-%d"
+        )
+
+        # Sort by report_date in descending order (most recent first)
+        merged = merged.sort_values("report_date", ascending=False).reset_index(
+            drop=True
+        )
+
+        return merged
+
+    def _fetch_balance_sheet(self) -> pd.DataFrame:
+        """
+        Get stock balance sheet data from East Money API
+        """
+        try:
+            # API endpoint and parameters
+            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
+            params = {
+                "reportName": "RPT_DMSK_FN_BALANCE",
+                "filter": f'(SECURITY_CODE="{self.symbol}")',
+                "pageNumber": "1",
+                "pageSize": "1000",
+                "sortColumns": "REPORT_DATE",
+                "sortTypes": "-1",
+                "columns": ",".join(self._balance_sheet_rename_map.keys()),
+            }
+
+            # Fetch data from API
+            response = requests.get(api_url, params=params)
+            response.raise_for_status()
+            data = response.json()
+
+            # Extract the actual data
+            if data.get("result") and data["result"].get("data"):
+                df = pd.DataFrame(data["result"]["data"])
+                df.rename(columns=self._balance_sheet_rename_map, inplace=True)
+                return df
+            else:
+                print("No balance sheet data found in API response")
+                return pd.DataFrame()
+
+        except Exception as e:
+            print(f"Error occurred: {str(e)}")
+            return pd.DataFrame()
+
+    def _fetch_income_statement(self) -> pd.DataFrame:
+        """
+        Get stock income statement data from East Money API
+        """
+        try:
+            # API endpoint and parameters
+            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
+            params = {
+                "reportName": "RPT_DMSK_FN_INCOME",
+                "filter": f'(SECURITY_CODE="{self.symbol}")',
+                "pageNumber": "1",
+                "pageSize": "1000",
+                "sortColumns": "REPORT_DATE",
+                "sortTypes": "-1",
+                "columns": ",".join(self._income_statement_rename_map.keys()),
+            }
+
+            # Fetch data from API
+            response = requests.get(api_url, params=params)
+            response.raise_for_status()
+            data = response.json()
+
+            # Extract the actual data
+            if data.get("result") and data["result"].get("data"):
+                df = pd.DataFrame(data["result"]["data"])
+                df.rename(columns=self._income_statement_rename_map, inplace=True)
+                return df
+            else:
+                print("No income statement data found in API response")
+                return pd.DataFrame()
+
+        except Exception as e:
+            print(f"Error occurred: {str(e)}")
+            return pd.DataFrame()
+
+    def _fetch_cash_flow(self) -> pd.DataFrame:
+        """
+        Get stock cash flow statement data from East Money API
+        """
+        try:
+            # API endpoint and parameters
+            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
+            params = {
+                "reportName": "RPT_DMSK_FN_CASHFLOW",
+                "filter": f'(SECURITY_CODE="{self.symbol}")',
+                "pageNumber": "1",
+                "pageSize": "1000",
+                "sortColumns": "REPORT_DATE",
+                "sortTypes": "-1",
+                "columns": ",".join(self._cash_flow_rename_map.keys()),
+            }
+
+            # Fetch data from API
+            response = requests.get(api_url, params=params)
+            response.raise_for_status()
+            data = response.json()
+
+            # Extract the actual data
+            if data.get("result") and data["result"].get("data"):
+                df = pd.DataFrame(data["result"]["data"])
+                df.rename(columns=self._cash_flow_rename_map, inplace=True)
+                return df
+            else:
+                print("No cash flow statement data found in API response")
+                return pd.DataFrame()
+
+        except Exception as e:
+            print(f"Error occurred: {str(e)}")
+            return pd.DataFrame()

{akshare_one-0.3.5.dist-info → akshare_one-0.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.3.5
+Version: 0.3.6
 Summary: Standardized interface for Chinese financial market data, built on AKShare with unified data formats and simplified APIs
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/zwldarren/akshare-one

{akshare_one-0.3.5.dist-info → akshare_one-0.3.6.dist-info}/RECORD

@@ -1,8 +1,11 @@
 akshare_one/__init__.py,sha256=htgTumncxiGHqXMgqmBuIQXoPBcOnwcTjCg02Ydr73I,6524
 akshare_one/indicators.py,sha256=x3Amff9CG_GvQpA-sqGfFwEAIvaaXlBxDfzTxD05taQ,12533
+akshare_one/eastmoney/client.py,sha256=SSMB4oupaCns5hxtSwxaX-UE_uOrxZEGxqUiC3BT-4k,2794
+akshare_one/eastmoney/utils.py,sha256=fATw0L5SW14wHWXlJ4IFEqnSsSBMT8MYGevxo7Kf1nY,2935
 akshare_one/modules/cache.py,sha256=_3n35rt9xJfQzZSV6JZ6bGzf2VnqTmLfe49WXk4c9K8,867
 akshare_one/modules/utils.py,sha256=msHqsjWSRULbX-3Bnit1p26a4a7MOEuNfkPSaECXr4k,333
 akshare_one/modules/financial/base.py,sha256=TG3ncf3rXfgWCk4qUORN01uxT1SgLWiyjkt5Jb9eoxo,688
+akshare_one/modules/financial/eastmoney_direct.py,sha256=BwiUWi3X24qbtrQv5ht2Lj6crgFrhdcgW4Aa53ZFEOg,6725
 akshare_one/modules/financial/factory.py,sha256=9xR_uKt7n8dndYUxEGDDL65LXnbm-edtTLdhF0Bfyro,1468
 akshare_one/modules/financial/sina.py,sha256=c6rSxCVNU6h-7XWSiqPHDN_XAhRdGHdqI9Haruy3mDs,12801
 akshare_one/modules/historical/base.py,sha256=kDy76OJUp-LIddsC23YAQdf4Q_YGCrnZ8AvU4xRzQsI,1286
@@ -29,8 +32,8 @@ akshare_one/modules/realtime/eastmoney.py,sha256=6acJeIdrvkW4ZqM9CALithlx85QSogr
 akshare_one/modules/realtime/eastmoney_direct.py,sha256=A2ScBRfIP6n_BxQ6muB26AEykIvTG7Mt3BDAZMyugkg,1236
 akshare_one/modules/realtime/factory.py,sha256=_7jBDgqWqkt5xTTT1SpZoUHM9IpMRpcUQeyyCglM5z0,1528
 akshare_one/modules/realtime/xueqiu.py,sha256=CHTN5VUwo24H-2EGKQkN8oqr3MWjDi-7DpvQEDyPlls,2196
-akshare_one-0.3.5.dist-info/licenses/LICENSE,sha256=3bqxoD7aU4QS7kpNtQmRd4MikxXe6Gtm_DrojyFHGAc,1087
-akshare_one-0.3.5.dist-info/METADATA,sha256=4z4aTCn71In_R93ysoloGZM2aDRZCk0C-WfcXEJJIww,2272
-akshare_one-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-akshare_one-0.3.5.dist-info/top_level.txt,sha256=kNiucyLVAGa89wmUSpXbBLWD7pF_RuahuiaOfLHZSyw,12
-akshare_one-0.3.5.dist-info/RECORD,,
+akshare_one-0.3.6.dist-info/licenses/LICENSE,sha256=3bqxoD7aU4QS7kpNtQmRd4MikxXe6Gtm_DrojyFHGAc,1087
+akshare_one-0.3.6.dist-info/METADATA,sha256=OfOSB9aFz49VnqjJpyHwocEmCAbSLMeC3YoPTNpkwzo,2272
+akshare_one-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+akshare_one-0.3.6.dist-info/top_level.txt,sha256=kNiucyLVAGa89wmUSpXbBLWD7pF_RuahuiaOfLHZSyw,12
+akshare_one-0.3.6.dist-info/RECORD,,