codosdk 1.0.0__py2.py3-none-any.whl

websdk2/utils/cc_crypto.py

@@ -0,0 +1,85 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Version : 0.0.1
+Contact : 191715030@qq.com
+Author  : shenshuo
+Date    : 2023/4/17 17:07
+Desc    : 加密
+"""
+
+import base64
+from cryptography.fernet import Fernet
+from ..consts import const
+from ..configs import configs
+
+
+class AESCryptoV3:
+    """
+    usage:  mc = AESCryptoV3()                  实例化
+            mc.my_encrypt('ceshi')          对字符串ceshi进行加密
+            mc.my_decrypt('')               对密文进行解密
+    """
+
+    def __init__(self, key: str = 'W1zFCF-pnUXi1zRtfgNkHmM3qv_3zvCkVSx68vXqks4='):
+        # 这里密钥key 长度必须为16（AES-128）、24（AES-192）、或32（AES-256）Bytes 长度
+        if not isinstance(key, bytes): key = key.encode('utf-8')
+        if len(key) > 32:
+            key = key[0:32]
+        else:
+            key = key.rjust(32, b'0')
+
+        self.key = base64.urlsafe_b64encode(key)
+        self.f = Fernet(self.key)
+
+    @property
+    def create_key(self):
+        return Fernet.generate_key()
+
+    def my_encrypt(self, text: str):
+        if isinstance(text, str): text = text.encode('utf-8')
+        return self.f.encrypt(text).decode('utf-8')
+
+    def my_decrypt(self, text: str):
+        if isinstance(text, str): text = text.encode('utf-8')
+        return self.f.decrypt(text).decode('utf-8')
+
+
+class AESCryptoV4:
+    """
+    usage:  mc = AESCryptoV4()                  实例化
+            mc.my_encrypt('ceshi')          对字符串ceshi进行加密
+            mc.my_decrypt('')               对密文进行解密
+    """
+
+    def __init__(self, key: str = None):
+        # 如果没有提供密钥，则使用默认密钥
+        if key is None:
+            # 若没有提供密钥，则生成一个新密钥
+            key = configs.get(const.AES_CRYPTO_KEY, 'W1zFCF-pnUXi1zRtfgNkHmM3qv_3zvCkVSx68vXqks4=')
+
+        # 确保密钥是字节类型
+        if not isinstance(key, bytes):
+            key = key.encode('utf-8')
+
+        if len(key) > 32:
+            key = key[0:32]
+        else:
+            key = key.rjust(32, b'0')
+        # 创建Fernet对象
+        self.f = Fernet(base64.urlsafe_b64encode(key))
+
+    @property
+    def create_key(self):
+        return Fernet.generate_key()
+
+    def my_encrypt(self, text: str):
+        if isinstance(text, str): text = text.encode('utf-8')
+        return self.f.encrypt(text).decode('utf-8')
+
+    def my_decrypt(self, text: str):
+        if isinstance(text, str): text = text.encode('utf-8')
+        return self.f.decrypt(text).decode('utf-8')
+
+
+mcv4 = AESCryptoV4()

websdk2/utils/date_format.py

@@ -0,0 +1,35 @@
+#!/usr/bin/env python
+# -*-coding:utf-8-*-
+""""
+Author  : shenshuo
+Date    : 2023年2月5日13:37:54
+Desc    : 日期时间格式化处理
+"""
+
+from datetime import datetime, timedelta
+
+
+def date_format_to8(start_date: str = None, end_date: str = None) -> tuple:
+    """
+    # iview 前端日期时间段数据格式化
+    start_time_tuple, end_time_tuple = date_format_to8(start_date, end_date)
+    # 查询语句中使用
+    session.query(dbA).filter(dbA.create_time.between(start_time_tuple, end_time_tuple).all()
+    """
+    date_format_1 = "%Y-%m-%dT%H:%M:%S.%fZ"
+    date_format_2 = "%Y-%m-%d"
+
+    if not start_date:
+        start_date = (datetime.now() - timedelta(days=30)).strftime(date_format_2)
+    if not end_date:
+        end_date = (datetime.now() + timedelta(days=1)).strftime(date_format_2)
+
+    for date_format in [date_format_1, date_format_2]:
+        try:
+            start_time_tuple = datetime.strptime(start_date, date_format) + timedelta(hours=8)
+            end_time_tuple = datetime.strptime(end_date, date_format) + timedelta(hours=8)
+            return start_time_tuple, end_time_tuple
+        except ValueError:
+            pass
+
+    raise ValueError(f"Unable to parse the dates. Expected formats are {date_format_1} and {date_format_2}.")

websdk2/utils/pydantic_utils.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Version : 0.0.8
+Contact : 191715030@qq.com
+Author  : shenshuo
+Date    : 2021/1/26 20:28
+Desc    :  https://github.com/tiangolo/pydantic-sqlalchemy
+"""
+
+####
+from typing import Container, Optional, Type
+from pydantic import BaseConfig, BaseModel, create_model, ValidationError
+from sqlalchemy.inspection import inspect
+from sqlalchemy.orm.properties import ColumnProperty
+
+
+### 删除的时候一般只有id
+class PydanticDel(BaseModel):
+    id: int
+
+class PydanticDelList(BaseModel):
+    id_list: list[int]
+
+
+class OrmConfig(BaseConfig):
+    orm_mode = True
+
+
+def sqlalchemy_to_pydantic(db_model: Type, *, config: Type = OrmConfig, exclude: Container[str] = []) -> Type[
+    BaseModel]:
+    mapper = inspect(db_model)
+    fields = {}
+    for attr in mapper.attrs:
+        if isinstance(attr, ColumnProperty):
+            if attr.columns:
+                name = attr.key
+                if name in exclude:
+                    continue
+                column = attr.columns[0]
+                python_type: Optional[type] = None
+                if hasattr(column.type, "impl"):
+                    if hasattr(column.type.impl, "python_type"):
+                        python_type = column.type.impl.python_type
+                elif hasattr(column.type, "python_type"):
+                    python_type = column.type.python_type
+                assert python_type, f"Could not infer python_type for {column}"
+                default = None
+                if column.default is None and not column.nullable:
+                    default = ...
+                fields[name] = (python_type, default)
+    pydantic_model = create_model(
+        db_model.__name__, __config__=config, **fields  # type: ignore
+    )
+    return pydantic_model

websdk2/web_logs.py

@@ -0,0 +1,127 @@
+#!/usr/bin/env python
+# -*-coding:utf-8-*-
+'''
+Author : ss
+date   : 2018-3-19
+role   : web  log
+'''
+
+import logging
+import os
+import sys
+import time
+import tornado.log
+from shortuuid import uuid
+
+log_fmt = ''.join(('PROGRESS:%(progress_id) -5s %(levelname) ', '-10s %(asctime)s %(name) -25s %(funcName) '
+                                                                '-30s LINE.NO:%(lineno) -5d : %(message)s'))
+log_key = 'logger_key'
+
+
+def singleton(class_):
+    instances = {}
+
+    def getinstance(*args, **kwargs):
+        if class_ not in instances:
+            instances[class_] = class_(*args, **kwargs)
+        return instances[class_]
+
+    return getinstance
+
+
+class ProgressLogFilter(logging.Filter):
+    def filter(self, record):
+        record.progress_id = Logger().progress_id
+        return True
+
+
+@singleton
+class Logger(object):
+    def __init__(self, progress_id='', log_file='/tmp/xxx.log'):
+        self.__log_key = log_key
+        self.progress_id = progress_id
+        self.log_file = log_file
+
+    def read_log(self, log_level, log_message):
+        ###创建一个logger
+        if self.progress_id == '':
+            Logger().progress_id = str(uuid())
+        else:
+            Logger().progress_id = self.progress_id
+        logger = logging.getLogger(self.__log_key)
+        logger.addFilter(ProgressLogFilter())
+        logger.setLevel(logging.DEBUG)
+
+        ###创建一个handler用于输出到终端
+        th = logging.StreamHandler()
+        th.setLevel(logging.DEBUG)
+
+        ###定义handler的输出格式
+        formatter = logging.Formatter(log_fmt)
+        th.setFormatter(formatter)
+
+        ###给logger添加handler
+        logger.addHandler(th)
+
+        ###记录日志
+        level_dic = {'debug': logger.debug, 'info': logger.info, 'warning': logger.warning, 'error': logger.error,
+                     'critical': logger.critical}
+        level_dic[log_level](log_message)
+
+        th.flush()
+        logger.removeHandler(th)
+
+    def write_log(self, log_level, log_message):
+        ###创建一个logger
+        ###创建一个logger
+        if self.progress_id == '':
+            Logger().progress_id = str(uuid())
+        else:
+            Logger().progress_id = self.progress_id
+        logger = logging.getLogger(self.__log_key)
+        logger.addFilter(ProgressLogFilter())
+        logger.setLevel(logging.DEBUG)
+
+        ###建立日志目录
+        log_dir = os.path.dirname(self.log_file)
+        if not os.path.isdir(log_dir):
+            os.makedirs(log_dir)
+
+        ###创建一个handler用于写入日志文件
+        fh = logging.FileHandler(self.log_file)
+        fh.setLevel(logging.DEBUG)
+
+        ###定义handler的输出格式
+        formatter = logging.Formatter(log_fmt)
+        fh.setFormatter(formatter)
+
+        ###给logger添加handler
+        logger.addHandler(fh)
+
+        ###记录日志
+        level_dic = {'debug': logger.debug, 'info': logger.info, 'warning': logger.warning, 'error': logger.error,
+                     'critical': logger.critical}
+        level_dic[log_level](log_message)
+
+        ###删除重复记录
+        fh.flush()
+        logger.removeHandler(fh)
+
+
+ins_log = Logger()
+
+
+def timeit(func):
+    def wrapper(*args, **kwargs):
+        start_time = time.time()
+        result = func(*args, **kwargs)
+        end_time = time.time()
+        duration = end_time - start_time
+        ins_log.read_log('info', '%s execute duration :%.3f second' % (str(func), duration))
+        return result
+
+    return wrapper
+
+
+ins_log.write_log('info', 'xxxx')
+ins_log.read_log('info', 'xxxx')