tsshare 1.0.0__py3-none-any.whl

tsshare/__init__.py

@@ -0,0 +1,19 @@
+"""tsshare SDK
+
+用法示例（tinyshare 风格，只需改一行代码）：
+
+    import tsshare as ts
+    ts.set_token("your-auth-code")
+    pro = ts.pro_api()
+    df = pro.daily(ts_code="000001.SZ", start_date="20240101", end_date="20240201")
+
+也支持显式传参风格：
+
+    import tsshare as ts
+    pro = ts.pro_api("your-auth-code", base_url="http://localhost:8000")
+    df = pro.daily(ts_code="000001.SZ", start_date="20240101", end_date="20240201")
+"""
+
+from .client import MyShareClient, pro_api, set_token, set_base_url
+
+__all__ = ["MyShareClient", "pro_api", "set_token", "set_base_url"]

tsshare/client.py

@@ -0,0 +1,417 @@
+"""myshare 客户端 SDK
+
+提供与 tushare.pro_api 类似的调用方式：
+
+    import myshare as ts
+    ts.set_token("your-auth-code")
+    pro = ts.pro_api()
+    df = pro.daily(ts_code="000001.SZ", start_date="20240101", end_date="20240201")
+"""
+from __future__ import annotations
+
+import base64 as _b64
+import hashlib
+import json
+import logging
+import os
+import platform
+import subprocess
+from pathlib import Path
+from typing import Any, Callable, Optional
+
+import requests
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+import pandas as pd
+
+logger = logging.getLogger(__name__)
+
+# -------------------- 默认网关（混淆存储，防止字符串搜索直接提取）--------------------
+# 以下常量是 base64 编码的默认服务器地址，运行时解码使用。
+# 修改说明：将你的真实服务器地址做 base64 编码后替换下方常量即可：
+#   python -c "import base64; print(base64.b64encode(b'http://YOUR_IP:PORT').decode())"
+_E = b"aHR0cHM6Ly80Ny4xMTIuMTkxLjc1"  # base64("https://47.112.191.75")
+
+
+def _default_base_url() -> str:
+    """运行时解码默认网关地址。"""
+    return _b64.b64decode(_E).decode("utf-8")
+
+
+# -------------------- 全局配置（兼容 tinyshare 风格） --------------------
+_global_auth_code: Optional[str] = None
+_global_base_url: Optional[str] = None
+
+
+def set_token(token: str) -> None:
+    """全局设置 myshare auth_code（兼容 tinyshare 的 ts.set_token() 调用风格）
+
+    示例：
+        import myshare as ts
+        ts.set_token("msu_xxx")
+        pro = ts.pro_api()
+    """
+    global _global_auth_code
+    _global_auth_code = token
+    logger.debug("全局 auth_code 已设置")
+
+
+def set_base_url(url: str) -> None:
+    """全局设置 myshare 网关地址（可选，覆盖内置默认值）
+
+    示例：
+        ts.set_base_url("http://192.168.1.100:8000")
+    """
+    global _global_base_url
+    _global_base_url = url
+    logger.debug("全局 base_url 已设置")
+
+
+# -------------------- 设备指纹采集 --------------------
+
+def _get_device_id() -> str:
+    """获取本机设备唯一 ID，并缓存到 ~/.myshare/device_id.json。
+
+    采集逻辑：
+      - Windows : PowerShell Get-CimInstance 读取 CPU ProcessorId + 硬盘序列号 → SHA256
+      - macOS   : system_profiler 读取 Hardware UUID → SHA256
+      - Linux   : 读取 /etc/machine-id 或 /var/lib/dbus/machine-id → SHA256
+
+    如果所有方式都失败，退化为 hostname + platform 信息的 SHA256。
+    """
+    cache_dir = Path.home() / ".myshare"
+    cache_file = cache_dir / "device_id.json"
+
+    # 优先读取已缓存的 ID，并校验格式（SHA256 = 64位十六进制）
+    if cache_file.exists():
+        try:
+            data = json.loads(cache_file.read_text(encoding="utf-8"))
+            device_id = data.get("device_id", "")
+            if (
+                device_id
+                and len(device_id) == 64
+                and all(c in "0123456789abcdef" for c in device_id)
+            ):
+                return device_id
+        except Exception:
+            pass
+
+    raw_id = _collect_hardware_fingerprint()
+    device_id = hashlib.sha256(raw_id.encode("utf-8")).hexdigest()
+
+    # 持久化缓存
+    try:
+        cache_dir.mkdir(parents=True, exist_ok=True)
+        cache_file.write_text(
+            json.dumps({"device_id": device_id}, ensure_ascii=False),
+            encoding="utf-8",
+        )
+    except Exception as e:
+        logger.debug("写入设备指纹缓存失败: %s", e)
+
+    logger.debug("设备 ID 已生成: %s...", device_id[:8])
+    return device_id
+
+
+def _collect_hardware_fingerprint() -> str:
+    """根据操作系统收集硬件指纹原始字符串。"""
+    system = platform.system()
+
+    if system == "Windows":
+        return _fingerprint_windows()
+    elif system == "Darwin":
+        return _fingerprint_macos()
+    else:
+        return _fingerprint_linux()
+
+
+def _fingerprint_windows() -> str:
+    """Windows：使用 PowerShell Get-CimInstance 读取 CPU 序列号 + 硬盘序列号（替代已弃用的 wmic）。"""
+    parts = []
+
+    # CPU ProcessorId - 使用 PowerShell Get-CimInstance 替代 wmic
+    try:
+        result = subprocess.check_output(
+            ["powershell", "-Command", "Get-CimInstance -ClassName Win32_Processor | Select-Object -ExpandProperty ProcessorId"],
+            stderr=subprocess.DEVNULL,
+            timeout=5,
+        ).decode(errors="replace").strip()
+        if result:
+            parts.append(result)
+    except Exception:
+        pass
+
+    # 物理硬盘序列号 - 使用 PowerShell Get-CimInstance 替代 wmic
+    try:
+        result = subprocess.check_output(
+            ["powershell", "-Command", "Get-CimInstance -ClassName Win32_DiskDrive | Select-Object -ExpandProperty SerialNumber | Select-Object -First 1"],
+            stderr=subprocess.DEVNULL,
+            timeout=5,
+        ).decode(errors="replace").strip()
+        if result:
+            parts.append(result)
+    except Exception:
+        pass
+
+    if parts:
+        return "|".join(parts)
+
+    # 退化：hostname
+    return platform.node() + "|" + platform.machine()
+
+
+def _fingerprint_macos() -> str:
+    """macOS：system_profiler 读取 Hardware UUID。"""
+    try:
+        result = subprocess.check_output(
+            ["system_profiler", "SPHardwareDataType"],
+            stderr=subprocess.DEVNULL,
+            timeout=5,
+        ).decode(errors="replace")
+        for line in result.splitlines():
+            # 精确匹配 "Hardware UUID"，避免匹配到其他含 UUID 的行
+            if "Hardware UUID" in line:
+                return line.split(":")[-1].strip()
+    except Exception:
+        pass
+    return platform.node() + "|" + platform.machine()
+
+
+def _fingerprint_linux() -> str:
+    """Linux：读取 /etc/machine-id。"""
+    for path in ("/etc/machine-id", "/var/lib/dbus/machine-id"):
+        try:
+            machine_id = Path(path).read_text().strip()
+            if machine_id:
+                return machine_id
+        except Exception:
+            pass
+    return platform.node() + "|" + platform.machine()
+
+
+# -------------------- 客户端 --------------------
+
+class MyShareError(Exception):
+    """myshare 调用错误"""
+
+
+def _build_retry_session() -> requests.Session:
+    """构建带重试策略的 requests.Session。
+
+    重试条件：网络连接错误、502/503/504 状态码，最多重试 3 次，指数退避。
+
+    SSL 证书：
+      生产环境 Nginx 通常配置 Let's Encrypt 证书（自动验证通过）。
+      如果使用自签名证书，设置环境变量 MYSHARE_SSL_VERIFY=false 跳过验证。
+    """
+    session = requests.Session()
+    retry = Retry(
+        total=3,
+        backoff_factor=0.5,
+        status_forcelist=[502, 503, 504],
+        allowed_methods=["POST", "GET"],
+    )
+    adapter = HTTPAdapter(max_retries=retry)
+    session.mount("http://", adapter)
+    session.mount("https://", adapter)
+    # 通过环境变量控制 SSL 证书验证（默认开启）
+    session.verify = os.environ.get("MYSHARE_SSL_VERIFY", "true").lower() != "false"
+    return session
+
+
+class MyShareClient:
+    """myshare 客户端
+
+    参数：
+        auth_code: 由 myshare 服务端签发的用户访问 token
+        base_url: 网关地址（留空则使用内置默认值）
+        timeout: 请求超时时间（秒）
+    """
+
+    def __init__(
+        self,
+        auth_code: Optional[str] = None,
+        base_url: Optional[str] = None,
+        timeout: int = 30,
+    ) -> None:
+        self.auth_code = auth_code
+        # 优先级：显式传入 > 环境变量 > 内置混淆默认值
+        if base_url is not None:
+            self.base_url = base_url
+        else:
+            env_url = os.getenv("MYSHARE_BASE_URL")
+            self.base_url = env_url if env_url else _default_base_url()
+        self.timeout = timeout
+        self._session = _build_retry_session()
+        # 懒加载设备 ID（第一次调用时才采集，避免导入慢）
+        self._device_id: Optional[str] = None
+
+    def _get_device_id(self) -> str:
+        if self._device_id is None:
+            self._device_id = _get_device_id()
+        return self._device_id
+
+    # -------------------- 公共 API --------------------
+    def query(self, api_name: str, **params: Any) -> pd.DataFrame:
+        """通用查询接口，对应 tushare 的 pro.query(api_name, **params)"""
+        # 提前校验 auth_code，给出明确的错误提示
+        if not self.auth_code:
+            raise MyShareError(
+                "auth_code 未设置，请先调用 ts.set_token('your_code') 或在 pro_api() 中传入 auth_code 参数"
+            )
+
+        payload = {
+            "api_name": api_name,
+            "params": params or {},
+            "auth_code": self.auth_code,
+            "meta": {
+                "client": "python-sdk",
+                "device_id": self._get_device_id(),
+            },
+        }
+        url = self.base_url.rstrip("/") + "/api/v1/queue"
+        logger.debug("myshare request api=%s (async)", api_name)
+
+        try:
+            resp = self._session.post(url, json=payload, timeout=self.timeout)
+        except Exception as e:
+            logger.exception("请求 myshare 失败: %s", e)
+            raise MyShareError(f"request failed: {e}") from e
+
+        if resp.status_code != 200:
+            raise MyShareError(f"http status {resp.status_code}: {resp.text}")
+
+        try:
+            obj = resp.json()
+        except ValueError as e:
+            raise MyShareError(f"invalid json response: {e}") from e
+
+        code = obj.get("code", -1)
+        message = obj.get("message", "unknown error")
+        data = obj.get("data")
+
+        if code != 0:
+            raise MyShareError(f"error code {code}: {message}")
+
+        # 使用 is None 而非 not data，避免将空字典/空列表误判为"无数据"
+        if data is None:
+            return pd.DataFrame()
+
+        # 检查是否是异步任务响应（包含task_id）
+        if isinstance(data, dict) and "task_id" in data:
+            # 异步任务，需要轮询获取结果
+            task_id = data["task_id"]
+            return self._poll_async_result(task_id)
+        else:
+            # 同步响应或直接返回数据（兼容模式）
+            fields = data.get("fields") or []
+            items = data.get("items") or []
+            return pd.DataFrame(items, columns=fields)
+
+    def _poll_async_result(self, task_id: str, timeout: int = 60) -> pd.DataFrame:
+        """轮询异步任务结果
+        
+        Args:
+            task_id: 异步任务ID
+            timeout: 轮询超时时间（秒）
+        
+        Returns:
+            pandas DataFrame
+            
+        Raises:
+            MyShareError: 任务失败或超时
+        """
+        import time
+        
+        start_time = time.time()
+        poll_interval = 0.5  # 轮询间隔0.5秒
+        
+        while time.time() - start_time < timeout:
+            # 查询任务状态
+            result_url = self.base_url.rstrip("/") + f"/api/v1/task/{task_id}"
+            try:
+                result_resp = self._session.get(result_url, timeout=self.timeout)
+            except Exception as e:
+                logger.exception("查询异步任务结果失败: %s", e)
+                raise MyShareError(f"query async result failed: {e}") from e
+            
+            if result_resp.status_code != 200:
+                raise MyShareError(f"http status {result_resp.status_code}: {result_resp.text}")
+            
+            try:
+                result_obj = result_resp.json()
+            except ValueError as e:
+                raise MyShareError(f"invalid json response: {e}") from e
+            
+            result_code = result_obj.get("code", -1)
+            
+            if result_code != 0:
+                raise MyShareError(f"error code {result_code}: {result_obj.get('message')}")
+            
+            result_data = result_obj.get("data", {})
+            status = result_data.get("status")
+            
+            if status == "done":
+                # 任务完成，返回数据
+                final_result = result_data.get("result", {})
+                fields = final_result.get("fields") or []
+                items = final_result.get("items") or []
+                logger.debug("异步任务完成: %s, 耗时: %.2f秒", task_id, time.time() - start_time)
+                return pd.DataFrame(items, columns=fields)
+            elif status == "failed":
+                # 任务失败
+                error = result_data.get("error", "unknown error")
+                logger.error("异步任务失败: %s, 错误: %s", task_id, error)
+                raise MyShareError(f"async task failed: {error}")
+            elif status == "pending":
+                # 任务还在处理中，继续等待
+                pass
+            else:
+                # 未知状态
+                logger.error("未知的异步任务状态: %s, task_id: %s", status, task_id)
+                raise MyShareError(f"unknown task status: {status}")
+            
+            # 等待后继续轮询
+            time.sleep(poll_interval)
+        
+        # 超时
+        logger.error("异步任务超时: %s, 超时时间: %d秒", task_id, timeout)
+        raise MyShareError(f"async task timeout after {timeout} seconds")
+
+    # -------------------- 动态 API 映射 --------------------
+    def __getattr__(self, name: str) -> Callable[..., pd.DataFrame]:
+        """将未知属性当作 api_name 动态映射
+
+        例如：pro.daily(ts_code="000001.SZ") 等价于 query("daily", ts_code="000001.SZ")
+
+        注意：__getattr__ 只在普通属性查找失败时触发，无需再做 hasattr 检查
+        （hasattr 检查会导致递归调用 __getattr__ 进而引发 RecursionError）。
+        """
+        if name.startswith("_"):
+            raise AttributeError(name)
+
+        def _wrapper(**kwargs: Any) -> pd.DataFrame:
+            return self.query(name, **kwargs)
+
+        _wrapper.__name__ = name
+        _wrapper.__doc__ = f"Dynamic API proxy for '{name}' via myshare."
+        return _wrapper
+
+
+def pro_api(
+    auth_code: Optional[str] = None,
+    base_url: Optional[str] = None,
+    timeout: int = 30,
+) -> MyShareClient:
+    """创建 myshare 客户端实例（兼容 tushare.pro_api 接口风格）
+
+    auth_code 可省略，若已通过 set_token() 全局配置则自动读取。
+
+    示例：
+        ts.set_token("msu_xxx")
+        pro = ts.pro_api()
+        df = pro.daily(ts_code="000001.SZ", start_date="20240101", end_date="20240201")
+    """
+    final_auth_code = auth_code if auth_code is not None else _global_auth_code
+    final_base_url = base_url if base_url is not None else _global_base_url
+    return MyShareClient(auth_code=final_auth_code, base_url=final_base_url, timeout=timeout)

tsshare-1.0.0.dist-info/METADATA

@@ -0,0 +1,69 @@
+Metadata-Version: 2.4
+Name: tsshare
+Version: 1.0.0
+Summary: tsshare — Tushare data proxy SDK, drop-in replacement for tushare pro_api
+Author: 
+Author-email: tsshare team <tsshare@example.com>
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/LinMason168/tsshare
+Project-URL: Documentation, https://github.com/LinMason168/tsshare#readme
+Project-URL: Repository, https://github.com/LinMason168/tsshare
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Financial and Insurance Industry
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Office/Business :: Financial
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.28.0
+Requires-Dist: pandas>=1.5.0
+Requires-Dist: urllib3>=1.26.0
+Dynamic: license-file
+Dynamic: requires-python
+
+# myshare SDK
+
+A drop-in replacement for Tushare's `pro_api` — same API, no quota limits.
+
+## Installation
+
+```bash
+pip install myshare
+```
+
+## Quick Start
+
+```python
+import myshare as ts
+
+# One-line setup: paste your auth_code
+ts.set_token("your-auth-code")
+
+# Use exactly like tushare pro_api
+pro = ts.pro_api()
+df = pro.daily(ts_code="000001.SZ", start_date="20240101", end_date="20240201")
+print(df)
+```
+
+## Supported APIs
+
+All Tushare APIs are supported: `daily`, `stock_basic`, `trade_cal`, `income`, `balancesheet`, `moneyflow`, and more.
+
+## Platform Support
+
+- Windows (x86_64)
+- macOS (Apple Silicon / Intel)
+- Linux (x86_64 / aarch64)
+
+Python 3.8 or later required.
+
+## License
+
+MIT

tsshare-1.0.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+tsshare/__init__.py,sha256=04Rs88HupzmrOHkE9FypDtRgGF4nF2BtCCrhXSfzZoU,622
+tsshare/client.py,sha256=Ah1fEdJjk2uZfAxn2VtiK8b1I2O4jY9B_qFVmlHFYzQ,15286
+tsshare-1.0.0.dist-info/licenses/LICENSE,sha256=mfZBsenkJrxFQ7Gidx0tef34i03o8ectL7JUSypwsYk,1064
+tsshare-1.0.0.dist-info/METADATA,sha256=fFMYGPo4-Pmn7X8B23yaZ5Lk89qKPO7EU1rg2QZmlzA,1979
+tsshare-1.0.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+tsshare-1.0.0.dist-info/top_level.txt,sha256=y3Fcx9aqp9-L9yGQ2v8kEFexuqRRIfOKwHKDQeAMwn4,8
+tsshare-1.0.0.dist-info/RECORD,,

tsshare-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

tsshare-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 myshare
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

tsshare-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+tsshare