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

ezKit/cls.py

@@ -0,0 +1,90 @@
+"""财联社数据"""
+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

ezKit/qywx.py

@@ -0,0 +1,186 @@
+"""
+企业微信开发者中心
+
+    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
+
+import requests
+from loguru import logger
+
+from . import utils
+
+
+class QYWX:
+    """企业微信"""
+
+    url_prefix = "https://qyapi.weixin.qq.com"
+    work_id: str | None = None
+    agent_id: str | None = None
+    agent_secret: str | None = None
+    access_token: str | None = None
+
+    def __init__(self, work_id: str | None, agent_id: str | None, agent_secret: str | None):
+        """Initiation"""
+        self.work_id = work_id
+        self.agent_id = agent_id
+        self.agent_secret = agent_secret
+
+        """获取 Token"""
+        self.getaccess_token()
+
+    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:
+                self.access_token = None
+            return result.get('access_token')
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def get_agent_list(self) -> dict | str | None:
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+            response = requests.get(f"{self.url_prefix}/cgi-bin/agent/list?access_token={self.access_token}", timeout=10)
+            if response.status_code == 200:
+                response_data: dict = response.json()
+                if response_data.get('errcode') == 42001:
+                    self.getaccess_token()
+                    time.sleep(1)
+                    self.get_agent_list()
+                return response_data
+            return response.text
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def get_department_list(self, eid: str | None = None) -> dict | str | None:
+        """eid: Enterprise ID"""
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+            response = requests.get(f"{self.url_prefix}/cgi-bin/department/list?access_token={self.access_token}&id={eid}", timeout=10)
+            if response.status_code == 200:
+                response_data: dict = response.json()
+                if response_data.get('errcode') == 42001:
+                    self.getaccess_token()
+                    time.sleep(1)
+                    self.get_department_list(eid)
+                return response_data
+            return response.text
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def get_user_list(self, eid: str | None = None) -> dict | str | None:
+        """eid: Enterprise ID"""
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+            response = requests.get(f"{self.url_prefix}/cgi-bin/user/list?access_token={self.access_token}&department_id={eid}", timeout=10)
+            if response.status_code == 200:
+                response_data: dict = response.json()
+                if response_data.get('errcode') == 42001:
+                    self.getaccess_token()
+                    time.sleep(1)
+                    self.get_user_list(eid)
+                return response_data
+            return response.text
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def get_user_id_by_mobile(self, mobile: str | None = None) -> dict | str | None:
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+            json_string = json.dumps({'mobile': mobile})
+            response = requests.post(f"{self.url_prefix}/cgi-bin/user/getuserid?access_token={self.access_token}", data=json_string, timeout=10)
+            if response.status_code == 200:
+                response_data: dict = response.json()
+                if response_data.get('errcode') == 42001:
+                    self.getaccess_token()
+                    time.sleep(1)
+                    self.get_user_id_by_mobile(mobile)
+                return response_data
+            return response.text
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def get_user_info(self, eid: str | None = None) -> dict | str | None:
+        """eid: Enterprise ID"""
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+            response = requests.get(f"{self.url_prefix}/cgi-bin/user/get?access_token={self.access_token}&userid={eid}", timeout=10)
+            if response.status_code == 200:
+                response_data: dict = response.json()
+                if response_data.get('errcode') == 42001:
+                    self.getaccess_token()
+                    time.sleep(1)
+                    self.get_user_info(eid)
+                return response_data
+            return response.text
+        except Exception as e:
+            logger.exception(e)
+            return None
+
+    def send_message_by_mobile(self, mobile: str | list, message: str) -> bool:
+        """发送消息"""
+
+        # 参考文档:
+        # https://developer.work.weixin.qq.com/document/path/90235
+
+        try:
+            if self.access_token is None:
+                self.getaccess_token()
+
+            users: list = []
+
+            match True:
+                case True if utils.v_true(mobile, list):
+                    users = mobile
+                case True if utils.v_true(mobile, str):
+                    users.append(mobile)
+                case _:
+                    return None
+
+            for user in users:
+                user_object = self.get_user_id_by_mobile(user)
+                json_dict = {
+                    'touser': user_object.get('userid'),
+                    'msgtype': 'text',
+                    'agentid': self.agent_id,
+                    'text': {'content': message},
+                    'safe': 0,
+                    'enable_id_trans': 0,
+                    'enable_duplicate_check': 0,
+                    'duplicate_check_interval': 1800
+                }
+                json_string = json.dumps(json_dict)
+                response = requests.post(f"{self.url_prefix}/cgi-bin/message/send?access_token={self.access_token}", data=json_string, timeout=10)
+                if response.status_code == 200:
+                    response_data: dict = response.json()
+                    if response_data.get('errcode') == 42001:
+                        self.getaccess_token()
+                        time.sleep(1)
+                        self.send_message_by_mobile(mobile, message)
+
+            return True
+
+        except Exception as e:
+            logger.exception(e)
+            return False

ezKit/stock.py

@@ -0,0 +1,61 @@
+"""股票"""
+# pylint: disable=R0911
+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.0.dist-info → ezKit-1.8.2.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: ezKit
-Version: 1.8.0
+Version: 1.8.2
 Summary: Easy Kit
 Author: septvean
 Author-email: septvean@gmail.com
 Requires-Python: >=3.11
 License-File: LICENSE
-Requires-Dist: loguru>=0.7.0
+Requires-Dist: loguru>=0.7

{ezKit-1.8.0.dist-info → ezKit-1.8.2.dist-info}/RECORD

@@ -2,16 +2,19 @@ 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.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ezKit-1.8.0.dist-info/METADATA,sha256=CNkyQ01iJgdaiUFUFq2n8aL9MBXtxduFVK6ZBpJrW68,192
-ezKit-1.8.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-ezKit-1.8.0.dist-info/top_level.txt,sha256=aYLB_1WODsqNTsTFWcKP-BN0KCTKcV-HZJ4zlHkCFw8,6
-ezKit-1.8.0.dist-info/RECORD,,
+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,,