ezKit 1.8.2__py3-none-any.whl → 1.8.4__py3-none-any.whl

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/cls.py

@@ -88,3 +88,169 @@ def up_down_analysis(
         logger.error(f"{info} [失败]")
         logger.exception(e)
         return None
+
+
+# --------------------------------------------------------------------------------------------------
+
+
+def latest_data(
+    payload: str | dict,
+    data_type: str = "stock",
+    df: bool = False
+) -> list | pd.DataFrame | None:
+    """股票或板块的最新数据"""
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 判断参数类型
+    match True:
+        case True if True not in [isinstance(payload, str), isinstance(payload, dict)]:
+            logger.error("Incorrect function argument type: payload")
+            return None
+        case True if False in [isinstance(data_type, str), utils.v_true(data_type, str)]:
+            logger.error("Incorrect function argument type: data_type")
+            return None
+        case _:
+            pass
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 判断数据类型. 数据类型: 个股, 板块 (产业链: industry)
+    if data_type not in ["stock", "plate"]:
+        logger.error("data_type error")
+        return None
+
+    # ----------------------------------------------------------------------------------------------
+
+    # 日志信息
+
+    # 个股 (默认)
+    info: str = "获取股票最新数据"
+
+    # 板块
+    if data_type == "plate":
+        info = "获取板块最新数据"
+
+    # match True:
+    #     case True if data_type == "plate":
+    #         info = "获取板块最新数据"
+    #     case True if data_type == "industry":
+    #         info = "获取产业链最新数据"
+    #     case _:
+    #         pass
+
+    # ----------------------------------------------------------------------------------------------
+
+    try:
+
+        logger.info(f"{info} ......")
+
+        # ------------------------------------------------------------------------------------------
+
+        # HTTP User Agent
+        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"
+
+        # HTTP Headers
+        headers = {"User-Agent": user_agent}
+
+        # ------------------------------------------------------------------------------------------
+
+        # 请求参数
+        params: dict = {}
+
+        # 默认请求参数
+        if isinstance(payload, str) and utils.v_true(payload, str):
+            params = {"secu_code": payload}
+
+        # 请求参数
+        if isinstance(payload, dict) and utils.v_true(payload, dict):
+            params = payload
+
+        # ------------------------------------------------------------------------------------------
+
+        # 不直接在API后面跟参数, 使用 params 传递参数
+
+        # API: 股票
+        # api: str = f"https://x-quote.cls.cn/quote/stock/basic?secu_code={code}"
+        api: str = "https://x-quote.cls.cn/quote/stock/basic"
+
+        # API: 板块
+        if data_type == "plate":
+            # api = f"https://x-quote.cls.cn/web_quote/plate/stocks?secu_code={code}"
+            api = "https://x-quote.cls.cn/web_quote/plate/stocks"
+
+        # match True:
+        #     case True if data_type == "plate":
+        #         # 板块
+        #         # api = f"https://x-quote.cls.cn/web_quote/plate/stocks?secu_code={code}"
+        #         api = "https://x-quote.cls.cn/web_quote/plate/stocks"
+        #     case True if data_type == "industry":
+        #         # 产业链
+        #         # api = f"https://x-quote.cls.cn/web_quote/plate/industry?secu_code={code}"
+        #         api = "https://x-quote.cls.cn/web_quote/plate/industry"
+        #     case _:
+        #         pass
+
+        # ------------------------------------------------------------------------------------------
+
+        # 获取数据
+        # response = requests.get(api, headers=headers, timeout=10)
+        response = requests.get(api, headers=headers, params=params, timeout=10)
+
+        # 转换数据类型
+        response_dict: dict = response.json()
+
+        # ------------------------------------------------------------------------------------------
+
+        # 个股
+
+        if data_type == "stock":
+
+            # 停牌, 返回 None
+            if response_dict["data"]["trade_status"] == "STOPT":
+                logger.error(f"{info} [停牌]")
+                return None
+
+            # pd.DataFrame 数据
+            if utils.v_true(df, bool) is True:
+                df_data = {
+                    # "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=df_data)
+
+            # 默认返回的数据
+            logger.success(f"{info} [成功]")
+            return response_dict["data"]
+
+        # ------------------------------------------------------------------------------------------
+
+        # 板块
+
+        # 板块数据不能转换为 pd.DataFrame
+        if (data_type == "plate") and (utils.v_true(df, bool) is True):
+            logger.error(f"{info} [错误]")
+            return None
+
+        # 数据结果
+        result: list = []
+
+        # 筛选 主板, 非ST, 非银行, 非证券 的股票
+        for i in response_dict["data"]["stocks"]:
+            if (re.match(r"^(sz00|sh60)", i["secu_code"])) and (not re.search(r"ST|银行|证券", i["secu_name"])):
+                result.append(i)
+
+        # 返回数据
+        logger.success(f"{info} [成功]")
+        return result
+
+    except Exception as e:
+        logger.error(f"{info} [失败]")
+        logger.exception(e)
+        return None

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/http.py

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

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/stock.py

@@ -1,5 +1,4 @@
 """股票"""
-# pylint: disable=R0911
 from copy import deepcopy
 from typing import Any
 

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.2.dist-info → ezKit-1.8.4.dist-info}/METADATA

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

ezKit-1.8.4.dist-info/RECORD

@@ -0,0 +1,20 @@
+ezKit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ezKit/bottle.py,sha256=usKK1wVaZw4_D-4VwMYmOIc8jtz4TrpM30nck59HMFw,180178
+ezKit/bottle_extensions.py,sha256=LQikCbbYZBAa4AcihvrTvixWHHMY7OjCBsT02PqWMMM,1129
+ezKit/cipher.py,sha256=0T_StbjiNI4zgrjVgcfU-ffKgu1waBA9UDudAnqFcNM,2896
+ezKit/cls.py,sha256=_M6ymGiJVb0V5sK93rhkJnI3v9xwS0qN4VH0gwCWdSY,9260
+ezKit/database.py,sha256=kYb0ybACxrdFZnwJ40bTbG8Obh2oEF3GQWmCbPVgH-c,6878
+ezKit/http.py,sha256=N2x86HCmJ5ejwA3-gHowEgSpaFTFT9nCWFjPuQ1T_do,1801
+ezKit/mongo.py,sha256=P6WTwFRxaaHixJK_PyKlOfPHkeJRxxrNLV77xy5LVBQ,2048
+ezKit/qywx.py,sha256=rdhusXl6YZUbsYm4HJUzWqGPFFlGqwKA4ygzwxiiBl4,7171
+ezKit/redis.py,sha256=HVofsLdSBbBHAR-veumsrjTwZQspRDy2FXNR6MDCCXs,1972
+ezKit/sendemail.py,sha256=Qxu4XQkHRPeX6FSJdzj-MXND9NyKcgHljbafNmy34H0,8243
+ezKit/stock.py,sha256=p-l6Tv0oZdJ3juxkddScphm7BW4vH-mBozBgSH1Qu0w,1730
+ezKit/token.py,sha256=9CAZhPdXiRiWoOIeWmP0q6L3j1zQAv4YcVWH95Tjefs,1755
+ezKit/utils.py,sha256=an7joZy_EEpYfN8zBtEWAnhP0YXYfPieabsK_HAxXl4,48921
+ezKit/xftp.py,sha256=XyIdr_2rxRVLqPofG6fIYWhAMVsFwTyp46dg5P9FLW4,7774
+ezKit-1.8.4.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ezKit-1.8.4.dist-info/METADATA,sha256=XpWOxkE-UKyjOd9YDJ03BiumX3K2aLhvUolVanZQPjM,190
+ezKit-1.8.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ezKit-1.8.4.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
+ezKit-1.8.4.dist-info/RECORD,,

ezKit-1.8.2.dist-info/RECORD

@@ -1,20 +0,0 @@
-ezKit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ezKit/bottle.py,sha256=usKK1wVaZw4_D-4VwMYmOIc8jtz4TrpM30nck59HMFw,180178
-ezKit/bottle_extensions.py,sha256=LQikCbbYZBAa4AcihvrTvixWHHMY7OjCBsT02PqWMMM,1129
-ezKit/cipher.py,sha256=uVUzeoYg5ZPJLpqg2wLst-7vfdDGmqWKka9Oyi0WrCA,2883
-ezKit/cls.py,sha256=mkNSguUNocP8aKh1GEpkUMgL9xRXfMKdGs6tEeeV5X8,3266
-ezKit/database.py,sha256=awTEdD4aqevyJg5LPBHmUPrOKweqzJ2Ezqqx0_xgUDA,6859
-ezKit/http.py,sha256=cyS18-TW9f7p1OaiJ4nnVsKZT8Ghy-8OewkfYm8yesw,1790
-ezKit/mongo.py,sha256=P6WTwFRxaaHixJK_PyKlOfPHkeJRxxrNLV77xy5LVBQ,2048
-ezKit/qywx.py,sha256=ra2e3u8IJmNZXr569u1mik3KArAiMdC9x0SgXGXEJes,7003
-ezKit/redis.py,sha256=HVofsLdSBbBHAR-veumsrjTwZQspRDy2FXNR6MDCCXs,1972
-ezKit/sendemail.py,sha256=Qxu4XQkHRPeX6FSJdzj-MXND9NyKcgHljbafNmy34H0,8243
-ezKit/stock.py,sha256=gEy4N7S0GsZCdfEdYHfaUHMEHMFmNGfHm198hG5n5eI,1754
-ezKit/token.py,sha256=9CAZhPdXiRiWoOIeWmP0q6L3j1zQAv4YcVWH95Tjefs,1755
-ezKit/utils.py,sha256=an7joZy_EEpYfN8zBtEWAnhP0YXYfPieabsK_HAxXl4,48921
-ezKit/xftp.py,sha256=7BABr-gjxZxK2UXoW9XxN76i1HdubeaomYlYMqRurHE,7770
-ezKit-1.8.2.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.8.2.dist-info/METADATA,sha256=Vnx7icBFnyUYYAHZz3jh5hhDS11d3hHngGlBz-DkWwU,190
-ezKit-1.8.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.8.2.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.8.2.dist-info/RECORD,,