cloudoll 2.4.0__py3-none-any.whl

cloudoll/__init__.py

@@ -0,0 +1 @@
+__version__ = "2.4.0"

cloudoll/__main__.py

@@ -0,0 +1,4 @@
+from .cli import cli
+
+if __name__ == "__main__":
+    cli()

cloudoll/cli.py

@@ -0,0 +1,76 @@
+import click
+import asyncio
+import os
+import sys
+from .logging import print_error
+from . import __version__
+from typing import Any
+from .utils.cli_main import run_app, run_gen
+sys.path.append(os.path.abspath("."))
+
+
+@click.group()
+@click.version_option(__version__, "-V", "--version", prog_name="cloudoll")
+def cli() -> None:
+    pass
+
+
+life_cycle = ["on_startup", "on_shutdown", "on_cleanup", "cleanup_ctx"]
+
+
+@cli.command()
+@click.option("-p", "--path", help="Model's relative path,save models or create tables", default="models.py",)
+@click.option("-c", "--create", help="For create model or table, model / table", default="model")
+@click.option("-t", "--table", help="Table's name or Model's name, split by `,` or 'ALL'", required=True,)
+@click.option("-env", "--environment", help="Environment, local / test / prod", default="local")
+@click.option("-db", "--database", help="Database name, pick the database in conf.{env}.yaml", default="mysql",)
+def gen(**config: Any) -> None:
+    """Help to create models or tables."""
+
+    # click.echo(create)
+    async def start():
+        try:
+            sa = await run_gen(**config)
+        except Exception as e:
+            # traceback.print_exc()
+            print_error('Error: %s', e)
+            click.echo("input `cloudoll --help` for more helps.")
+        finally:
+            await sa.close()
+
+    asyncio.run(start())
+
+
+@cli.command()
+@click.option("-env", "--environment", help="Environment, local / test / prod", default="local")
+@click.option("-p", "--port", help="Server's port", default=None)
+@click.option("-h", "--host", help="Server's host", default=None)
+@click.option("-m", "--mode", help="development or production mode", default='development')
+@click.option("-path", "--path", help="Unix file system path to serve on. Specifying a path will cause hostname and port arguments to be ignored.",)
+@click.option("-e", "--entry", help="Entry point model name. delfault name app", default="app")
+def start(**config: Any) -> None:
+    """Start a services."""
+    try:
+        active_config = {k: v for k, v in config.items() if v is not None}
+        run_app(**active_config)
+    except Exception as e:
+        print_error('Error: %s', e)
+        # traceback.format_exc()
+        sys.exit(2)
+
+
+_project_existing = click.Path(dir_okay=True)
+
+
+@cli.command()
+@click.argument("project-name", type=_project_existing, required=True)
+def create(project_name) -> None:
+    full_path = os.path.join(os.path.abspath("."), project_name)
+    # log = logging.getLogger('cloudoll.create')
+    if os.path.exists(full_path):
+        print_error(f"Project name `{project_name}` already exist.")
+        # return None
+
+
+if __name__ == "__main__":
+    cli()

cloudoll/error/errors.py

@@ -0,0 +1,48 @@
+ERROR = {
+    CUSTOM: {code: -1, message: "%s"},
+    HSF_ERROR: {code: -6, message: "HSF 调用失败：%s。"},
+    SYSTEM_ERROR: {code: -10, message: "系统错误。"},
+    BAD_REQUEST: {code: -400, message: "错误的 http 请求。"},
+    NOT_FOUND: {code: -404, message: "您请求的 http 资源没有找到。"},
+    INTERNAL_SERVER_ERROR: {code: -500, message: "内部服务器错误。"},
+    DB: {code: -700, message: "%s"},
+    DATA_OCCUPIED: {code: -701, message: "数据被占用。"},
+    WHAT_REQUIRE: {code: -1001, message: "缺少参数 %s。"},
+    WHAT_WRONG_RANGE: {code: -1002, message: "%s 的取值范围错误。最小 %s，最大 %s"},
+    WHAT_WRONG_FORMAT: {code: -1003, message: "%s 格式不正确。"},
+    WHAT_NOT_SAME: {code: -1004, message: "输入的 %s 值不一致。"},
+    WHAT_NOT_EXISTS: {code: -1005, message: "%s 不存在。"},
+    WHAT_TOO_MUCH: {code: -1006, message: "%s 太多了。"},
+    WHAT_TOO_LITTLE: {code: -1007, message: "%s 太少了。"},
+    WHAT_NOT_BELONGS_TO_YOU: {code: -1008, message: "%s 不属于你。"},
+    WHAT_TOO_LONG: {code: -1009, message: "%s 太长了，请不要超过 %s。"},
+    WHAT_TOO_SHORT: {code: -1010, message: "%s 太短了，请不要少于 %s。"},
+    WHAT_EXISTED: {code: -1011, message: "%s 已经存在。"},
+    WHAT_OCCUPIED: {code: -1012, message: "%s 被占用。"},
+    WHAT_TIMEOUT: {code: -1013, message: "%s 已超时。"},
+    WHAT_EXPIRED: {code: -1014, message: "%s 已过期。"},
+    WHAT_ILLEGAL: {code: -1015, message: "%s 不合法。"},
+    WHAT_NOT_FOUND: {code: -1016, message: "%s 没找到。"},
+    WHAT_WRONG_LENGTH_RANGE: {code: -1017, message: "%s 的长度错误，最短是 %s，最长 %s。"},
+    WHAT_WRONG_TYPE: {code: -1018, message: "%s 的类型错误"},
+    ACCESS_TOKEN_NOT_FOUND: {code: -2001, message: "access_token 不存在。"},
+    ACCESS_TOKEN_EXPIRED: {code: -2002, message: "access_token 已经过期。"},
+    TICKET_EXPIRED: {code: -2050, message: "票据已经过期，请重新获取。"},
+    TICKET_VERIFY_FAILED: {code: -2051, message: "票据校验失败。篡改登录信息是违法行为！"},
+    TICKET_ILLEGAL: {code: -2052, message: "非法票据。篡改登录信息是违法行为！"},
+    SIGN_VERIFY_FAILED: {code: -2053, message: "签名验证失败。"},
+    PASSWORD_NOT_STRONG: {code: -3001, message: "密码太简单。%s"},
+    CHINA_MOBILE_ILLEGAL: {code: -3002, message: "不正确的手机号码。"},
+    EMAIL_ILLEGAL: {code: -3003, message: "不正确的Email。"},
+    CAPTCHA_VALIDATE_FAIL: {code: -3004, message: "验证码校验失败。"},
+    PASSPORT_ILLEGAL: {code: -3005, message: "不正确的登录凭据，必须是手机或者 Email。"},
+    MEMBER_ONLY: {code: -3006, message: "您还没有登录，请登录。"},
+    NO_RIGHTS: {code: -3007, message: "您没有权限，请联系管理员授权。"},
+    LOGIN_ERROR: {code: -3050, message: "登录失败"},
+    LOGIN_ERROR_BECAUSE: {code: -3051, message: "登录失败。%s"},
+    MEMBER_NOT_EXISTS: {code: -3052, message: "用户不存在。"},
+    MEMBER_NOT_APPROVED: {code: -3053, message: "尚未审批通过，请耐心等待。"},
+    MEMBER_FORBIDDEN: {code: -3054, message: "您已经被系统禁止。"},
+    MEMBER_LOGIN_TOO_MUCH_FAILURE: {code: -3055, message: "您登录失败的次数太多。暂时被系统锁定"},
+    IP_LOGIN_TOO_MUCH: {code: -3056, message: "您登录失败的次数太多。暴力破解他人密码是违法行为！"},
+}

cloudoll/logging/__init__.py

@@ -0,0 +1,188 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+__author__ = "chuchur/chuchur.com"
+
+import logging as loger
+import os
+from datetime import datetime
+from logging.handlers import RotatingFileHandler
+from logging import INFO, DEBUG, ERROR, CRITICAL, DEBUG
+import colorlog
+
+__ALL__ = (
+    "debug",
+    "info",
+    "warning",
+    "error",
+    "exception",
+    "critical",
+    "print_info",
+    "print_error",
+    "print_warn",
+)
+_COLORS = {
+    # 终端输出日志颜色配置
+    "DEBUG": "white",
+    "INFO": "green",
+    "WARNING": "yellow",
+    "ERROR": "red",
+    "CRITICAL": "bold_red",
+}
+
+
+class ColorAnsi:
+    RED = "\033[91m"
+    GREEN = "\033[92m"
+    YELLOW = "\033[93m"
+    BLUE = "\033[94m"
+    END = "\033[0m"
+
+
+_FORMATS = {
+    # 终端输出格式
+    # "console_format": "%(log_color)s[%(levelname)s]-%(asctime)s-%(pathname)s-%(module)s-%(funcName)s-[line:%(lineno)d]: %(message)s",
+    "console_format": "%(log_color)s%(asctime)s %(levelname)s %(message)s",
+    # 日志输出格式
+    # "log_format": "[%(levelname)s]-%(asctime)s-%(pathname)s-%(module)s-%(funcName)s-[line:%(lineno)d]: %(message)s",
+    "log_format": "%(asctime)s %(levelname)s %(message)s",
+}
+
+_LOG_SIZE = 1024 * 1024 * 1  # 1MB 日志最大1MB
+_LOG_FILES_COUNT = 3  # 最多保留3个日志文件
+
+
+class HandleLog:
+    """
+    建日志记录器(logging.getLogger)
+    设置日志级别(logger.setLevel)
+    创建日志文件(logging.FileHandler)
+    设置日志格式(logging.Formatter)
+    日志处理程序记录到记录器(addHandler)
+    """
+
+    def __init__(self):
+        # cur_path = os.path.dirname(os.path.realpath(__file__))  # 当前项目路径
+        # log_path = os.path.join(os.path.normpath(os.getcwd() + os.sep + os.pardir), 'logs')
+        # self.__logger = logging.getLogger()
+        self.getLogger()
+        log_path = os.path.join(os.path.normpath(os.getcwd() + os.sep), "logs")
+        # now_time = datetime.now().strftime("%Y-%m-%d")  # 当前日期格式化
+        if not os.path.exists(log_path):
+            os.mkdir(log_path)  # 若不存在logs文件夹，则自动创建
+        self.log_path = log_path
+        self.log_path_prefix = None  # 当前日期格式化
+        self.all_handle = None
+        self.error_handle = None
+
+        # console
+        # if __debug__:
+        console_handle = colorlog.StreamHandler()
+        console_fmt = colorlog.ColoredFormatter(
+            _FORMATS["console_format"], log_colors=_COLORS
+        )
+        console_handle.setFormatter(console_fmt)
+        console_handle.setLevel(DEBUG)
+        self.__logger.addHandler(console_handle)
+
+    def setLevel(self, level):
+        self.__logger.setLevel(level)
+
+    def getLogger(self, name=None):
+        self.__logger = loger.getLogger("cloudoll")  # 创建日志记录器
+        self.__logger.setLevel(INFO)
+        # self._set_handle()
+        return self.__logger
+
+    def __create_handler(self, log_path, level):
+        handler = RotatingFileHandler(
+            log_path,
+            maxBytes=_LOG_SIZE,
+            backupCount=_LOG_FILES_COUNT,
+            encoding="utf-8",
+        )
+        formatter = loger.Formatter(_FORMATS["log_format"], datefmt="%Y-%m-%d %H:%M-%S")
+        handler.setFormatter(formatter)
+        handler.setLevel(level=level)
+        self.__logger.addHandler(handler)
+        return handler
+
+    def __free(self):
+        if self.all_handle:
+            self.__logger.removeHandler(self.all_handle)
+            self.all_handle.close()
+        if self.error_handle:
+            self.__logger.removeHandler(self.error_handle)
+            self.error_handle.close()
+
+    def log(self, method):
+        logger = self.__logger
+        log_path_prefix = datetime.now().strftime("%Y-%m-%d")  # 当前日期格式化
+        if self.log_path_prefix != log_path_prefix:
+            self.log_path_prefix = log_path_prefix
+            self.__free()
+            # 收集所有日志信息文件
+            all_log_path = os.path.join(self.log_path, "%s-all.log" % log_path_prefix)
+            # 收集错误日志信息文件
+            error_log_path = os.path.join(
+                self.log_path, "%s-error.log" % log_path_prefix
+            )
+
+            # set handle
+            # self.__logger.setLevel(logging.INFO)  # 设置默认日志记录器记录级别
+            # all
+            self.all_handle = self.__create_handler(all_log_path, INFO)
+            # error
+            self.error_handle = self.__create_handler(error_log_path, ERROR)
+
+        return getattr(logger, method, None)
+
+
+_handler = None
+
+
+def print_format(color_ansi, type, *args):
+    ft = color_ansi + "{}" + ColorAnsi.END
+    timestamp = ft.format(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " " + type)
+    text = " ".join(ft.format(str(arg)) for arg in args)
+    print(timestamp, text)
+
+
+def print_info(*args):
+    print_format(ColorAnsi.GREEN, "INFO", *args)
+
+
+def print_error(*args):
+    print_format(ColorAnsi.RED, "ERROR", *args)
+
+
+def print_warn(*args):
+    print_format(ColorAnsi.YELLOW, "WARN", *args)
+
+
+def _get_handle():
+    global _handler
+    handler = _handler if _handler else HandleLog()
+    _handler = handler
+    return handler
+
+
+def setLevel(level: int):
+    _get_handle().setLevel(level)
+
+
+def getLogger(name=None):
+    return _get_handle().getLogger(name=name)
+
+
+debug = _get_handle().log("debug")
+
+info = _get_handle().log("info")
+
+warning = _get_handle().log("warning")
+
+error = _get_handle().log("error")
+
+exception = _get_handle().log("exception")
+
+critical = _get_handle().log("critical")

cloudoll/mail/smtp.py

@@ -0,0 +1,197 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+__author__ = "chuchur"
+
+"""
+MAIL = {
+    "smtp_server": "smtp.qq.com",
+    "account": "123456789@qq.com",
+    "account_name": "chuchur",
+    "password": "abcdefg",
+    "prot": 465,  # 587
+    "debug_level": 1,
+}
+
+m = smtp.Mail(**MAIL)
+# 标题
+m.subject = "test title"
+# 正文
+m.content = "long long ago..."
+
+# 嵌入html 和 html 调用附件
+m.addfile("/home/chuchur/img/a.jpg") # cid 0
+m.addfile("/home/chuchur/img/b.jpg") # cid 1
+m.addhtml("<html><body><h1>Hello</h1>" + '<p><img src="cid:0"><img src="cid:1"></p>' + "</body></html>")
+
+# 多个收件人
+m.add_to_addr("李彦宏", "liyanhong@baidu.com")
+m.add_to_addr("马云", "jackma@alibaba.com")
+
+# 附件
+m.addfile(filepathA)
+m.addfile(filepathB)
+
+# 发送
+m.send()
+
+"""
+
+import smtplib, os
+from email import encoders
+from email.header import Header
+from email.mime.text import MIMEText
+from email.mime.base import MIMEBase
+from email.mime.multipart import MIMEMultipart
+
+from email.utils import parseaddr, formataddr
+import mimetypes
+from ..logging import print_info, print_error
+
+
+def _format_addr(s):
+    name, addr = parseaddr(s)
+    return formataddr((Header(name, "utf-8").encode(), addr))
+
+
+class Client(object):
+    def __init__(self, **config):
+        self._content = None
+        self._subject = None
+        smtp_server = config.get("smtp_server")
+        debug_level = config.get("debug_level", 1)
+        port = config.get("port", 25)
+        self._account = config.get("account", "")
+        self._account_name = config.get("account_name", "")
+        self._password = config.get("password", "")
+        self._server = smtplib.SMTP_SSL(smtp_server, port)
+        # self._server.starttls()  # 调用starttls()方法加密
+        self._server.set_debuglevel(debug_level)  # 打印出和SMTP服务器交互的所有信息
+        self._msg = MIMEMultipart()
+        self._to_addr = []
+        self._mime_type = "plain"  # 'html'
+        self._file_index = 0
+
+    def _login(self):
+        try:
+            account = self._account
+            password = self._password
+            if account is None or password is None:
+                raise KeyError("请设置账号密码")
+            print_info("登录中...")
+            self._server.login(account, password)
+        except BaseException as e:
+            print_error(e)
+            raise
+
+    def send(self):
+        """
+        发送邮件
+        """
+        try:
+            self._login()
+            # msg = MIMEText(self._content, "plain", "uft-8")
+            msg = self._msg
+            msg["From"] = _format_addr("%s <%s>" % (self._account_name, self._account))
+            to = []
+            for t in self._to_addr:
+                to.append("%s <%s>" % (t["name"], t["addr"]))
+            msg["To"] = _format_addr(",".join(to))
+
+            # 邮件标题
+            msg["Subject"] = Header(self._subject, "utf-8").encode()
+
+            # 邮件正文
+            msg.attach(MIMEText(self._content, self._mime_type, "utf-8"))
+            to_addr = list(map(lambda x: x["addr"], self._to_addr))
+            self._server.sendmail(self._account, to_addr, msg.as_string())
+            self._server.quit()
+        except BaseException as er:
+            print_error(er)
+
+    def add_to_addr(self, nick, addr):
+        """
+        添加收件人
+        :params nick 收件人昵称
+        :params addr 收件人地址
+        """
+        if addr is None:
+            raise KeyError("请输入邮箱地址")
+        obj = {"name": nick, "addr": addr}
+        self._to_addr.append(obj)
+
+    def addhtml(self, htmltext):
+        """
+        添加html正文
+        :params htmltext html正文内容
+        """
+        self._msg.attach(MIMEText(htmltext, "html", "utf-8"))
+
+    def addfile(self, filepath):
+        """
+        添加附件
+        :params filepath 附件绝对路径
+        """
+        index = self._file_index
+        with open(filepath, "rb") as f:
+            mime_type = ""
+            mime_types = mimetypes.guess_type(filepath)  # image/png
+            if len(mime_types) > 0:
+                mime_type = mime_types[0]
+            else:
+                print_error("无法匹配附件类型,可以尝试安装httpd服务")
+            ## 这里如果拿不到type 需要安装httpd ,dnf install httpd
+            [t, n] = mime_type.split("/")
+            filename = os.path.basename(filepath)  # 'a.txt'
+            # 设置附件的MIME和文件名
+            mime = MIMEBase(t, n, filename=filename)
+            # 加上头信息
+            mime.add_header("Content-Disposition", "attachment", filename=filename)
+            mime.add_header("Content-ID", "<%s>" % index)
+            mime.add_header("X-Attachment-Id", "%s" % index)
+            # 把附件内容读入
+            mime.set_payload(f.read())
+            # 用Base64编码
+            encoders.encode_base64(mime)
+
+            # 添加到MIMEMultipart
+            self._msg.attach(mime)
+            self._file_index += 1
+
+    @property
+    def mime_type(self):
+        return self._mime_type
+
+    @mime_type.setter
+    def mime_type(self, value):
+        self._mime_type = value
+
+    @property
+    def subject(self):
+        """
+        邮件标题内容
+        """
+        return self._subject
+
+    @subject.setter
+    def subject(self, value):
+        """
+        邮件标题内容
+        :params value 标题内容
+        """
+        self._subject = value
+
+    @property
+    def content(self):
+        """
+        邮件正文内容
+        """
+        return self._content
+
+    @content.setter
+    def content(self, value):
+        """
+        邮件正文内容
+        :params value 正文内容
+        """
+        self._content = value