ezKit 1.8.1__tar.gz → 1.8.3__tar.gz

{ezkit-1.8.1/ezKit.egg-info → ezkit-1.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ezKit
-Version: 1.8.1
+Version: 1.8.3
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit/cipher.py

@@ -1,9 +1,8 @@
-"""
-https://docs.python.org/3.10/library/hashlib.html
-https://www.pycrypto.org/
-https://stackoverflow.com/a/21928790
-pip install pycryptodome
-"""
+"""Cipher"""
+# https://docs.python.org/3.10/library/hashlib.html
+# https://www.pycrypto.org/
+# https://stackoverflow.com/a/21928790
+# pip install pycryptodome
 import base64
 import hashlib
 

ezkit-1.8.3/ezKit/cls.py

@@ -0,0 +1,170 @@
+"""财联社数据"""
+import re
+
+import pandas as pd
+import requests
+from loguru import logger
+
+from . import stock, utils
+
+
+def up_down_analysis(
+    target: str = "up_pool",
+    df: bool = False
+) -> list | pd.DataFrame | None:
+    """涨停跌停数据"""
+
+    if not utils.v_true(target, str):
+        logger.error(f"error type: {target}")
+        return None
+
+    info: str = "获取涨停池股票"
+    match True:
+        case True if target == "up_pool":
+            info = "获取涨停池股票"
+        case True if target == "continuous_up_pool":
+            info = "获取连板池股票"
+        case True if target == "up_open_pool":
+            info = "获取炸板池股票"
+        case True if target == "down_pool":
+            info = "获取跌停池股票"
+        case _:
+            pass
+
+    try:
+        logger.info(f"{info} ......")
+
+        user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
+        headers = {"User-Agent": user_agent}
+
+        # 涨停池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=up_pool
+        # 连板池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=continuous_up_pool
+        # 炸板池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=up_open_pool
+        # 跌停池: https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type=down_pool
+        api = f"https://x-quote.cls.cn/quote/index/up_down_analysis?rever=1&way=last_px&type={target}"
+
+        response = requests.get(api, headers=headers, timeout=10)
+
+        response_dict: dict = response.json()
+
+        result: list = []
+
+        for i in response_dict["data"]:
+
+            # if re.match(r"^(sz00|sh60)", i["secu_code"]):
+            #     print(i["secu_code"])
+
+            # if re.search(r"ST|银行", i["secu_name"]):
+            #     print(i["secu_name"])
+
+            # 主板, 非ST, 非银行, 非证券
+            if (not re.match(r"^(sz00|sh60)", i["secu_code"])) or re.search(r"ST|银行|证券", i["secu_name"]):
+                continue
+
+            if target in ["up_pool", "up_pool"]:
+                result.append({
+                    "code": stock.coderename(i["secu_code"], restore=True),
+                    "name": i["secu_name"],
+                    "up_days": i["limit_up_days"],
+                    "reason": i["up_reason"]
+                })
+
+            if target in ["up_open_pool", "down_pool"]:
+                result.append({
+                    "code": stock.coderename(i["secu_code"], restore=True),
+                    "name": i["secu_name"]
+                })
+
+        if utils.v_true(df, bool) is False:
+            logger.success(f"{info} [成功]")
+            return result
+
+        # data: pd.DataFrame = pd.DataFrame(response_dict["data"], columns=["secu_code", "secu_name", "limit_up_days", "up_reason"])
+        # data = data.rename(columns={"secu_code": "code", "secu_name": "name", "limit_up_days": "up_days", "up_reason": "reason"})
+
+        return pd.DataFrame(data=pd.DataFrame(result))
+
+    except Exception as e:
+        logger.error(f"{info} [失败]")
+        logger.exception(e)
+        return None
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+def latest_data(
+    payload: dict,
+    end: str = "basic",
+    data_type: str = "stock",
+    df: bool = False
+) -> dict | pd.DataFrame | None:
+    """股票或板块的最新数据"""
+
+    match True:
+        case True if isinstance(payload, dict) is False:
+            logger.error("Incorrect function argument type: payload")
+            return None
+        case True if isinstance(end, str) is False:
+            logger.error("Incorrect function argument type: end")
+            return None
+        case True if isinstance(data_type, str) is False:
+            logger.error("Incorrect function argument type: data_type")
+            return None
+        case _:
+            pass
+
+    info: str = "获取股票最新数据"
+
+    if utils.v_true(data_type, str) is True and data_type == "plate":
+        info = "获取板块最新数据"
+
+    try:
+
+        logger.info(f"{info} ......")
+
+        user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
+        headers = {"User-Agent": user_agent}
+
+        # https://x-quote.cls.cn/quote/stock/basic?secu_code={code}
+        api = f"https://x-quote.cls.cn/quote/stock/{end}"
+
+        if data_type == "plate":
+            # https://x-quote.cls.cn/web_quote/plate/stocks?secu_code=cls80382
+            # https://x-quote.cls.cn/web_quote/plate/stocks?secu_code={code}
+            # https://x-quote.cls.cn/web_quote/plate/industry?secu_code={code}
+            api = f"https://x-quote.cls.cn/web_quote/plate/{end}"
+
+        # response = requests.get(api, headers=headers, timeout=10)
+        response = requests.get(api, headers=headers, params=payload, timeout=10)
+
+        response_dict: dict = response.json()
+
+        if utils.v_true(df, bool) is False:
+            logger.success(f"{info} [成功]")
+            return response_dict
+
+        if data_type == "plate" and utils.v_true(df, bool) is True:
+            logger.error(f"{info} [错误]")
+            return None
+
+        if response_dict["data"]["trade_status"] == "STOPT":
+            logger.error(f"{info} [停牌]")
+            return None
+
+        data_object = {
+            # "date": [pd.to_datetime(date_today)],
+            "open": [float(response_dict["data"]["open_px"])],
+            "close": [float(response_dict["data"]["last_px"])],
+            "high": [float(response_dict["data"]["high_px"])],
+            "low": [float(response_dict["data"]["low_px"])],
+            "volume": [int(response_dict["data"]["business_amount"])],
+            "turnover": [float(response_dict["data"]["tr"])]
+        }
+        logger.success(f"{info} [成功]")
+        return pd.DataFrame(data=data_object)
+
+    except Exception as e:
+        logger.error(f"{info} [失败]")
+        logger.exception(e)
+        return None

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit/database.py

@@ -1,11 +1,10 @@
-"""
-Column, Table, MetaData API
-    https://docs.sqlalchemy.org/en/14/core/metadata.html#column-table-metadata-api
-CursorResult
-    https://docs.sqlalchemy.org/en/20/core/connections.html#sqlalchemy.engine.CursorResult
-PostgreSQL 14 Data Types
-    https://www.postgresql.org/docs/14/datatype.html
-"""
+"""Database"""
+# Column, Table, MetaData API
+#     https://docs.sqlalchemy.org/en/14/core/metadata.html#column-table-metadata-api
+# CursorResult
+#     https://docs.sqlalchemy.org/en/20/core/connections.html#sqlalchemy.engine.CursorResult
+# PostgreSQL 14 Data Types
+#     https://www.postgresql.org/docs/14/datatype.html
 import csv
 from typing import Any
 

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit/http.py

@@ -1,3 +1,4 @@
+"""HTTP"""
 import json
 from typing import Any
 

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit/qywx.py

@@ -1,14 +1,10 @@
-"""
-企业微信开发者中心
-
-    https://developer.work.weixin.qq.com/
-    https://developer.work.weixin.qq.com/document/path/90313 (全局错误码)
-
-参考文档:
-
-    https://www.gaoyuanqi.cn/python-yingyong-qiyewx/
-    https://www.jianshu.com/p/020709b130d3
-"""
+"""企业微信"""
+# 企业微信开发者中心
+#     https://developer.work.weixin.qq.com/
+#     https://developer.work.weixin.qq.com/document/path/90313 (全局错误码)
+# 参考文档:
+#     https://www.gaoyuanqi.cn/python-yingyong-qiyewx/
+#     https://www.jianshu.com/p/020709b130d3
 import json
 import time
 
@@ -39,12 +35,15 @@ class QYWX:
     def getaccess_token(self) -> str | None:
         try:
             response = requests.get(f"{self.url_prefix}/cgi-bin/gettoken?corpid={self.work_id}&corpsecret={self.agent_secret}", timeout=10)
-            if response.status_code == 200:
-                result: dict = response.json()
-                self.access_token = result.get('access_token')
-            else:
+
+            if response.status_code != 200:
                 self.access_token = None
+                return None
+
+            result: dict = response.json()
+            self.access_token = result.get('access_token')
             return result.get('access_token')
+
         except Exception as e:
             logger.exception(e)
             return None
@@ -151,15 +150,19 @@ class QYWX:
             users: list = []
 
             match True:
-                case True if utils.v_true(mobile, list):
+                case True if isinstance(mobile, list) and utils.v_true(mobile, list):
                     users = mobile
-                case True if utils.v_true(mobile, str):
+                case True if isinstance(mobile, str) and utils.v_true(mobile, str):
                     users.append(mobile)
                 case _:
-                    return None
+                    return False
 
             for user in users:
                 user_object = self.get_user_id_by_mobile(user)
+
+                if not isinstance(user_object, dict):
+                    continue
+
                 json_dict = {
                     'touser': user_object.get('userid'),
                     'msgtype': 'text',

ezkit-1.8.3/ezKit/stock.py

@@ -0,0 +1,60 @@
+"""股票"""
+from copy import deepcopy
+from typing import Any
+
+from loguru import logger
+
+from . import utils
+
+
+def coderename(target: str | dict, restore: bool = False) -> str | dict | None:
+    """
+    正向:
+        coderename('000001') => 'sz000001'
+        coderename({'code': '000001', 'name': '平安银行'}) => {'code': 'sz000001', 'name': '平安银行'}
+    反向:
+        coderename('sz000001', restore=True) => '000001'
+        coderename({'code': 'sz000001', 'name': '平安银行'}) => {'code': '000001', 'name': '平安银行'}
+    """
+
+    try:
+
+        _object: Any = None
+        _code_name: Any = None
+
+        # 判断 target 是 string 还是 dictionary
+        if isinstance(target, str) and utils.v_true(target, str):
+            _code_name = target
+        elif isinstance(target, dict) and utils.v_true(target, dict):
+            _object = deepcopy(target)
+            _code_name = str(deepcopy(target["code"]))
+        else:
+            return None
+
+        # 是否还原
+        if restore:
+            if len(_code_name) == 8 and ("sh" in _code_name or "sz" in _code_name):
+                _code_name = _code_name[2:8]
+            else:
+                return None
+        else:
+            if _code_name[0:2] == "00":
+                _code_name = "sz" + _code_name
+            elif _code_name[0:2] == "60":
+                _code_name = "sh" + _code_name
+            else:
+                return None
+
+        # 返回结果
+        if utils.v_true(target, str):
+            return _code_name
+
+        if utils.v_true(target, dict):
+            _object["code"] = _code_name
+            return _object
+
+        return None
+
+    except Exception as e:
+        logger.exception(e)
+        return None

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit/xftp.py

@@ -1,6 +1,5 @@
-"""
-ftplib: https://docs.python.org/3.11/library/ftplib.html
-"""
+"""FTP"""
+# ftplib: https://docs.python.org/3.11/library/ftplib.html
 import os
 from ftplib import FTP
 from pathlib import Path

{ezkit-1.8.1 → ezkit-1.8.3/ezKit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ezKit
-Version: 1.8.1
+Version: 1.8.3
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com

{ezkit-1.8.1 → ezkit-1.8.3}/ezKit.egg-info/SOURCES.txt

@@ -6,12 +6,14 @@ ezKit/__init__.py
 ezKit/bottle.py
 ezKit/bottle_extensions.py
 ezKit/cipher.py
+ezKit/cls.py
 ezKit/database.py
 ezKit/http.py
 ezKit/mongo.py
 ezKit/qywx.py
 ezKit/redis.py
 ezKit/sendemail.py
+ezKit/stock.py
 ezKit/token.py
 ezKit/utils.py
 ezKit/xftp.py

{ezkit-1.8.1 → ezkit-1.8.3}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages, setup
 
 setup(
     name='ezKit',
-    version='1.8.1',
+    version='1.8.3',
     author='septvean',
     author_email='septvean@gmail.com',
     description='Easy Kit',