fastgenerateapi 0.0.27__py2.py3-none-any.whl → 1.1.6__py2.py3-none-any.whl

fastgenerateapi/settings/{settings.py → app_settings.py}

@@ -1,21 +1,25 @@
 from typing import Optional
 
-from pydantic import BaseSettings, Field, BaseModel
+from pydantic import Field
+from fastgenerateapi.pydantic_utils.base_settings import BaseSettings
 
 
 class AppSettings(BaseSettings):
+    # 字段配置 老版本(pydantic.utils.to_lower_camel)
+    ALIAS_GENERATOR: Optional[str] = Field(default="pydantic.alias_generators.to_snake", description="序列化参数命名方法路径")
+
     # 分页对应字段以及配置默认值
-    CURRENT_PAGE_FIELD: Optional[str] = Field(default='page', description="当前页字段")
-    PAGE_SIZE_FIELD: Optional[str] = Field(default='page_size', description="每页数量字段")
-    TOTAL_SIZE_FIELD: Optional[str] = Field(default='total', description="统计数量字段")
-    DETERMINE_WHETHER_PAGE_FIELD: Optional[str] = Field(default='no_page', description="判断是否分页字段")
+    CURRENT_PAGE_FIELD: Optional[str] = Field(default="page", description="当前页字段")
+    PAGE_SIZE_FIELD: Optional[str] = Field(default="page_size", description="每页数量字段")
+    TOTAL_SIZE_FIELD: Optional[str] = Field(default="total", description="统计数量字段")
+    DETERMINE_WHETHER_PAGE_FIELD: Optional[str] = Field(default="no_page", description="判断是否分页字段")
     DETERMINE_PAGE_BOOL_VALUE: Optional[bool] = Field(default=True, description="分页对应的布尔值")
     DEFAULT_WHETHER_PAGE: Optional[bool] = Field(default=False, description="默认是否分页")
     DEFAULT_PAGE_SIZE: Optional[int] = Field(default=10, description="默认每页数量")
     DEFAULT_MAX_PAGE_SIZE: Optional[int] = Field(default=200, description="默认最大每页数量")
 
     # 方法优化项
-    METHOD_TREE_CHOICE: Optional[str] = Field(default='map', description="树状查询方式{'sql': sql循环查询,'map': 内存查询}")
+    METHOD_TREE_CHOICE: Optional[str] = Field(default="map", description="树状查询方式{sql: sql循环查询,map: 内存查询}")
 
     # 缓存配置参数
     CACHE_GET_ONE_WHETHER_OPEN: Optional[bool] = Field(default=False, description="查询详情是否打开缓存")
@@ -28,24 +32,24 @@ class AppSettings(BaseSettings):
     # 路由后缀字段是否添加以及配置默认值
     ROUTER_WHETHER_UNDERLINE_TO_STRIKE: Optional[bool] = Field(default=False, description="路由是否下划线转中划线")
     ROUTER_WHETHER_ADD_SUFFIX: Optional[bool] = Field(default=True, description="增删改查路由是否添加后缀")
-    ROUTER_CREATE_SUFFIX_FIELD: Optional[str] = Field(default='create', description="创建路由后缀字段")
-    ROUTER_GET_ONE_SUFFIX_FIELD: Optional[str] = Field(default='get_one', description="获取一个路由后缀字段")
-    ROUTER_GET_ALL_SUFFIX_FIELD: Optional[str] = Field(default='get_all', description="获取列表路由后缀字段")
-    ROUTER_GET_TREE_SUFFIX_FIELD: Optional[str] = Field(default='get_tree', description="获取树状数据路由后缀字段")
-    ROUTER_UPDATE_SUFFIX_FIELD: Optional[str] = Field(default='update', description="修改路由后缀字段")
-    ROUTER_DELETE_SUFFIX_FIELD: Optional[str] = Field(default='delete', description="删除路由后缀字段")
-    ROUTER_RECURSION_DELETE_SUFFIX_FIELD: Optional[str] = Field(default='delete_tree', description="递归删除路由后缀字段")
-    ROUTER_FILTER_DELETE_SUFFIX_FIELD: Optional[str] = Field(default='delete_filter', description="递归删除路由后缀字段")
+    ROUTER_CREATE_SUFFIX_FIELD: Optional[str] = Field(default="create", description="创建路由后缀字段")
+    ROUTER_GET_ONE_SUFFIX_FIELD: Optional[str] = Field(default="detail", description="获取一个路由后缀字段")  # get_one
+    ROUTER_GET_ALL_SUFFIX_FIELD: Optional[str] = Field(default="list", description="获取列表路由后缀字段")   # get_all
+    ROUTER_GET_TREE_SUFFIX_FIELD: Optional[str] = Field(default="tree", description="获取树状数据路由后缀字段")  # get_tree
+    ROUTER_UPDATE_SUFFIX_FIELD: Optional[str] = Field(default="update", description="修改路由后缀字段")
+    ROUTER_DELETE_SUFFIX_FIELD: Optional[str] = Field(default="delete", description="删除路由后缀字段")
+    ROUTER_RECURSION_DELETE_SUFFIX_FIELD: Optional[str] = Field(default="delete_tree", description="递归删除路由后缀字段")
+    ROUTER_FILTER_DELETE_SUFFIX_FIELD: Optional[str] = Field(default="delete_filter", description="递归删除路由后缀字段")
 
     # 函数转换路由时，默认添加字段，（遵循restful规范时，get路由处理方案）
-    RESTFUL_GET_ROUTER_ADD_PREFIX: Optional[str] = Field(default='', description="函数转换路由时：前缀添加字段")
-    RESTFUL_GET_ROUTER_ADD_SUFFIX: Optional[str] = Field(default='', description="函数转换路由时：后缀pk前添加字段")
-    RESTFUL_POST_ROUTER_ADD_PREFIX: Optional[str] = Field(default='', description="函数转换路由时：前缀添加字段")
-    RESTFUL_POST_ROUTER_ADD_SUFFIX: Optional[str] = Field(default='', description="函数转换路由时：后缀pk前添加字段")
-    RESTFUL_PUT_ROUTER_ADD_PREFIX: Optional[str] = Field(default='', description="函数转换路由时：前缀添加字段")
-    RESTFUL_PUT_ROUTER_ADD_SUFFIX: Optional[str] = Field(default='', description="函数转换路由时：后缀pk前添加字段")
-    RESTFUL_DELETE_ROUTER_ADD_PREFIX: Optional[str] = Field(default='', description="函数转换路由时：前缀添加字段")
-    RESTFUL_DELETE_ROUTER_ADD_SUFFIX: Optional[str] = Field(default='', description="函数转换路由时：后缀pk前添加字段")
+    RESTFUL_GET_ROUTER_ADD_PREFIX: Optional[str] = Field(default="", description="函数转换路由时：前缀添加字段")
+    RESTFUL_GET_ROUTER_ADD_SUFFIX: Optional[str] = Field(default="", description="函数转换路由时：后缀pk前添加字段")
+    RESTFUL_POST_ROUTER_ADD_PREFIX: Optional[str] = Field(default="", description="函数转换路由时：前缀添加字段")
+    RESTFUL_POST_ROUTER_ADD_SUFFIX: Optional[str] = Field(default="", description="函数转换路由时：后缀pk前添加字段")
+    RESTFUL_PUT_ROUTER_ADD_PREFIX: Optional[str] = Field(default="", description="函数转换路由时：前缀添加字段")
+    RESTFUL_PUT_ROUTER_ADD_SUFFIX: Optional[str] = Field(default="", description="函数转换路由时：后缀pk前添加字段")
+    RESTFUL_DELETE_ROUTER_ADD_PREFIX: Optional[str] = Field(default="", description="函数转换路由时：前缀添加字段")
+    RESTFUL_DELETE_ROUTER_ADD_SUFFIX: Optional[str] = Field(default="", description="函数转换路由时：后缀pk前添加字段")
 
     # 分布式id
     WORKER_ID: Optional[int] = Field(default=1, description="数据中心（机器区域）ID")
@@ -67,20 +71,19 @@ class AppSettings(BaseSettings):
     CODE_SUCCESS_DEFAULT_VALUE: Optional[int] = Field(default=200, description="code成功返回值")
     CODE_FAIL_DEFAULT_VALUE: Optional[int] = Field(default=-1, description="code失败返回值")
     SUCCESS_RESPONSE_FIELD: Optional[bool] = Field(default=True, description="success返回字段")
-    MESSAGE_RESPONSE_FIELD: Optional[str] = Field(default="message", description="消息返回字段")
+    MESSAGE_RESPONSE_FIELD: Optional[str] = Field(default="msg", description="消息返回字段")
     DATA_RESPONSE_FIELD: Optional[str] = Field(default="data", description="数据返回字段")
-    LIST_RESPONSE_FIELD: Optional[str] = Field(default='list', description="列表页返回字段")
+    LIST_RESPONSE_FIELD: Optional[str] = Field(default="list", description="列表页返回字段")
 
     # GetAll 筛选是否双下划线转单下划线
     FILTER_UNDERLINE_WHETHER_DOUBLE_TO_SINGLE: Optional[bool] = Field(default=True, description="筛选是否双下划线转单下划线")
     SCHEMAS_UNDERLINE_WHETHER_DOUBLE_TO_SINGLE: Optional[bool] = Field(default=True, description="序列化字段是否双下划线转单下划线")
 
     class Config:
-        env_prefix = 'APP_'
+        env_prefix = "APP_"
         env_file = "./.env"
         case_sensitive = True
+        extra = 'allow'
+
 
 
-class SettingsModel(BaseModel):
-    # 系统配置
-    app_settings: AppSettings = AppSettings()

fastgenerateapi/settings/db_settings.py

@@ -0,0 +1,69 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class DBSettings(BaseSettings):
+    """
+        Database Settings
+    """
+
+    TYPE: Optional[str] = Field(default='mysql', description="数据库类型")
+    HOST: Optional[str] = Field(default='127.0.0.1', description="数据库域名")
+    PORT: Optional[str] = Field(default='3306', description="数据库端口")
+    DATABASE: Optional[str] = Field(default='admin', description="数据库名")
+    USERNAME: Optional[str] = Field(default='root', description="数据库用户名")
+    PASSWORD: Optional[str] = Field(default='', description="数据库密码")
+
+    @property
+    def dsn(self):
+        return f"{self.TYPE.lower()}://{self.USERNAME}:{self.PASSWORD}@{self.HOST}:{self.PORT}/{self.DATABASE}"
+
+    class Config:
+        env_prefix = 'DB_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+class PostgresqlSettings(DBSettings):
+    """
+        Postgresql Settings
+    """
+    TYPE: Optional[str] = Field(default='postgres', description="数据库类型")
+
+    class Config:
+        env_prefix = 'Postgresql_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+class MySQLSettings(DBSettings):
+    """
+        MySQL Settings
+    """
+
+    class Config:
+        env_prefix = 'MYSQL_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+class LocalSettings(DBSettings):
+    """
+        MySQL Settings
+    """
+    TYPE: str = Field(..., description="数据库类型")
+
+    class Config:
+        env_prefix = 'LOCAL_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+
+

fastgenerateapi/settings/file_settings.py

@@ -0,0 +1,24 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class FileServerSettings(BaseSettings):
+    """
+    文件配置
+    """
+    FILE_SERVER: Optional[str] = Field(default='http://localhost:8001', description="文件服务域名", title="file Server Domain")
+    FILE_URL: Optional[str] = Field(default='/static/', description="文件路径前缀", title="file url prefix")
+    FILE_ROOT: Optional[str] = Field(default='static', description="文件储存路径", title="file storage path")
+
+    class Config:
+        env_prefix = 'FileServer_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+
+
+

fastgenerateapi/settings/jwt_settings.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class JWTSettings(BaseSettings):
+    """
+    jwt配置
+    """
+    SECRET_KEY: Optional[str] = Field(..., description="认证密钥", title="JWT SECRET KEY")
+    ALGORITHM: Optional[str] = Field(default='HS256', description="认证密钥", title="JWT ALGORITHM")
+    ACCESS_TOKEN_EXPIRE_MINUTES: Optional[int] = Field(default=30, description="过期时间 (分钟)", title="JWT Access Token")
+    REFRESH_TOKEN_EXPIRE_MINUTES: Optional[int] = Field(default=60, description="过期时间（分钟）", title="JWT Refresh Token")
+
+    class Config:
+        env_prefix = 'JWT_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+

fastgenerateapi/settings/otlp_settings.py

@@ -0,0 +1,69 @@
+from typing import Optional
+
+import httpx
+from fastapi import FastAPI
+from opentelemetry import metrics, trace
+from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.tortoiseorm import TortoiseORMInstrumentor
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics._internal.export import PeriodicExportingMetricReader
+from opentelemetry.sdk.resources import Resource, SERVICE_NAME
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class OpenTelemetrySettings(BaseSettings):
+    """
+    跟踪和指标配置
+    """
+    ENDPOINT: Optional[str] = Field(default=None, description="访问接口", example="http://127.0.0.1")
+    AUTHORIZATION: Optional[str] = Field(default=None, description="认证字符串")
+
+    def setup_open_telemetry(self, app: FastAPI, service_name: str):
+        if not self.ENDPOINT:
+            return
+
+        with httpx.Client() as client:
+            health_check = client.get(f"{self.ENDPOINT}/healthz", timeout=3).json()
+            if health_check["status"] != "ok":
+                raise Exception("otlp服务异常")
+            resource = Resource(attributes={
+                SERVICE_NAME: service_name
+            })
+
+            tracer_provider = TracerProvider(resource=resource)
+            headers = {
+                "Authorization": self.AUTHORIZATION} if self.AUTHORIZATION else {}
+            exporter = OTLPMetricExporter(
+                endpoint=f"{self.ENDPOINT}/api/default/v1/metrics",
+                headers=headers,
+            )
+            meter_provider = MeterProvider([PeriodicExportingMetricReader(exporter)], resource=resource)
+            metrics.set_meter_provider(meter_provider)
+            processor = BatchSpanProcessor(
+                OTLPSpanExporter(
+                    endpoint=f"{self.ENDPOINT}/api/default/v1/traces",
+                    headers=headers
+                )
+            )
+            tracer_provider.add_span_processor(processor)
+            trace.set_tracer_provider(tracer_provider)
+            FastAPIInstrumentor.instrument_app(
+                app,
+                tracer_provider=tracer_provider,
+                meter_provider=meter_provider,
+            )
+            TortoiseORMInstrumentor().instrument(tracer_provider=tracer_provider)
+            HTTPXClientInstrumentor().instrument(tracer_provider=tracer_provider)
+
+    class Config:
+        env_prefix = 'OTLP_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'

fastgenerateapi/settings/redis_settings.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class RedisSettings(BaseSettings):
+    URL: Optional[str] = Field(default="redis://127.0.0.1", description="IP地址")
+    PORT: Optional[int] = Field(default=6379, description="映射端口")
+    PASSWORD: Optional[str] = Field(default="", description="密码")
+
+    class Config:
+        env_prefix = 'REDIS_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'

fastgenerateapi/settings/sms_settings.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class SmsSettings(BaseSettings):
+    """
+        短信配置
+        Sms Settings
+    """
+    CACHE_BACKEND: Optional[str] = Field(default="redis", description="缓存类型", title="cache_backend(redis, inmemory)")
+    CACHE_BACKEND_DSN: Optional[str] = Field(default="redis://localhost:6379/1", description="缓存dsn", title="cache_backend_dsn")
+    IS_LIMIT_CODE_FREQUENCY: Optional[bool] = Field(default=True, description="是否限制频率", title="code frequency check")
+    CHECK_CODE_RESEND_TIME: Optional[int] = Field(default=1, description="验证码重新发送时间", title="check code resend time (minute)")
+    DEFAULT_CODE: Optional[int] = Field(default="654123", description="默认验证码", title="sms default code")
+
+    class Config:
+        env_prefix = 'Sms_'
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+
+

fastgenerateapi/settings/system_settings.py

@@ -0,0 +1,30 @@
+from typing import Optional
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class SystemSettings(BaseSettings):
+    """
+        System Settings
+    """
+    NAME: str = Field(default='fastgenerateapi', description="项目名称")
+    HOST: Optional[str] = Field(default='127.0.0.1', description="本地运行host")
+    PORT: Optional[int] = Field(default=8001, description="本地运行port")
+    DEBUG: bool = Field(default=True, description="是否开启调试模式，本地修改自动重载")
+
+    # BASE_DIR: Union[PathType, str, None]
+    DOMAIN: Optional[str] = Field(default="http://127.0.0.1:8001", description="服务域名（对外暴露的域名地址包含协议）")
+    FIELD_SECRET: Optional[str] = Field(default=None, description="字段加密密钥")
+
+    # 分布式id
+    WORKER_ID: Optional[int] = Field(default='1', description="数据中心（机器区域）ID")
+    DATACENTER_ID: Optional[int] = Field(default='1', description="机器ID")
+
+    class Config:
+        env_prefix = ''
+        env_file = "./.env"
+        case_sensitive = True
+        extra = 'allow'
+
+

fastgenerateapi/utils/auto_discover.py

@@ -0,0 +1,61 @@
+import importlib
+import os
+from pathlib import Path
+from typing import Optional, List
+
+
+def get_path_list(path: str, match_prefix: str, prefix: str = "") -> List[str]:
+    files = []
+    prefix = prefix + "." + os.path.basename(path) if prefix else os.path.basename(path)
+    for item in Path(path).iterdir():
+        if item.is_file():
+            if item.name.startswith(match_prefix) and item.name.endswith(".py"):
+                files.append(prefix+"."+item.name[:-3])
+        elif item.is_dir():
+            if item.name.startswith(match_prefix):
+                for s_item in Path(os.path.join(path, item.name)).iterdir():
+                    if s_item.is_file() and s_item.name.startswith(match_prefix):
+                        files.append(prefix + "." + item.name + "." + s_item.name[:-3])
+            else:
+                files += get_path_list(os.path.join(path, item.name), match_prefix, prefix)
+
+    return files
+
+
+def discover_models(
+        base_path="apps",  # 模块路径
+        include: Optional[List[str]] = None,  # 包含的模型
+        exclude: Optional[List[str]] = None,  # 排除的模型
+) -> List[str]:
+    """
+    自动发现models，并生成tortoise-orm所需要的字符串
+    会自动添加 aerich.models，如不需要请在 exclude 参数排除
+    """
+    models_list = ["aerich.models"] + get_path_list(base_path, "models")
+    return include + [model for model in models_list if model not in exclude] if exclude else models_list
+
+
+def discover_routes(
+        app,  # Fastapi 生成的实体类
+        base_path="apps",  # 模块路径
+        prefix="",  # 路由前缀
+        include: Optional[List[str]] = None,  # 包含的路由
+        exclude: Optional[List[str]] = None,  # 排除的路由
+):
+    """自动发现路由，并注册"""
+    paths = include if include else []
+
+    for path in paths + get_path_list(base_path, "router"):
+        if exclude and path in exclude:
+            continue
+        module = importlib.import_module(path)
+        if hasattr(module, "router"):
+            app.include_router(prefix=prefix, router=getattr(module, "router"))
+    return app
+
+
+if __name__ == '__main__':
+    # print(discover_models("../../modules"))
+    print(get_path_list("../../modules", "router"))
+
+

fastgenerateapi/utils/file_utils.py

@@ -0,0 +1,76 @@
+import struct
+
+type_dict = {
+    # '424D': 'bmp',
+    'FFD8FF': 'jpg',
+    # '2E524D46': 'rm',
+    # '4D546864': 'mid',
+    '89504E47': 'png',
+    '47494638': 'gif',
+    '49492A00': 'tif',
+    # '41433130': 'dwg',
+    # '38425053': 'psd',
+    # '2142444E': 'pst',
+    # 'FF575043': 'wpd',
+    # 'AC9EBD8F': 'qdf',
+    # 'E3828596': 'pwl',
+    '504B0304': 'zip',
+    '52617221': 'rar',
+    '57415645': 'wav',
+    '41564920': 'avi',
+    '2E7261FD': 'ram',
+    '000001BA': 'mpg',
+    '000001B3': 'mpg',
+    '6D6F6F76': 'mov',
+    # '7B5C727466': 'rtf',
+    '3C3F786D6C': 'xml',
+    '68746D6C3E': 'html',
+    'D0CF11E0': 'doc/xls',
+    '255044462D312E': 'pdf',
+    'CFAD12FEC5FD746F': 'dbx',
+    # '3026B2758E66CF11': 'asf',
+    '5374616E64617264204A': 'mdb',
+    # '252150532D41646F6265': 'ps/eps',
+    # '44656C69766572792D646174653A': 'eml'
+}
+max_len = len(max(type_dict, key=len)) // 2
+
+
+async def get_filetype(file):
+    # 读取二进制文件开头一定的长度
+    if isinstance(file, str):
+        filename = file
+        with open(filename, 'rb') as f:
+            byte = f.read(max_len)
+    else:
+        filename = file.filename
+        byte = await file.read(max_len)
+        await file.seek(0)
+    # 解析为元组
+    byte_list = struct.unpack('B' * max_len, byte)
+    # 转为16进制
+    code = ''.join([('%X' % each).zfill(2) for each in byte_list])
+    # 根据标识符筛选判断文件格式
+    result = list(filter(lambda x: code.startswith(x), type_dict))
+    nametype = filename.split('.')[-1]
+    if result:
+        filetype = type_dict[result[0]]
+        if filetype == "zip":
+            filetype = nametype if nametype in ['xlsx', 'docx'] else "zip"
+        elif filetype == "doc/xls":
+            filetype = nametype if nametype in ['doc', 'xls'] else None
+        elif filetype == "jpg":
+            filetype = nametype if nametype in ['jpg', 'jpeg'] else None
+        else:
+            filetype = filetype if filetype == nametype else None
+    else:
+        filetype = nametype if nametype in ['mp3', 'mp4', 'txt'] else None
+    return filetype
+
+file_type = {
+    "image": ["jpg", "png", "gif", "tif"],
+    "document": ["zip", "rar", "wav", "xml", "html", "doc", "xls", "pdf", "dbx", "mdb"],
+    "voice": ["mp3", "amr"],
+    "video": ["wav", "avi", "ram", "mpg", "mov", "mp4"],
+}
+

fastgenerateapi/utils/pwd_utils.py

@@ -0,0 +1,49 @@
+import random
+
+from passlib.hash import pbkdf2_sha256, md5_crypt
+
+
+def make_random_salt(length=64):
+    """
+        生成随机salt字符串，可指定长度
+    :param length: int
+    :return: salt: str
+    """
+    seed = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!#$&*="
+    sa = []
+    for i in range(length):
+        sa.append(random.choice(seed))
+    salt = ''.join(sa)
+    return salt
+
+
+def create_password_hash(password: str, salt: str) -> str:
+    """
+    根据传入的password与salt参数生成hash值
+    :param password:
+    :param salt:
+    :return:
+    """
+    custom_pbkdf2 = pbkdf2_sha256.using(salt=salt.encode('utf-8'), rounds=1000)
+    return custom_pbkdf2.hash(password)
+
+
+def validate_password(password: str, salt: str, password_hash: str) -> bool:
+    """
+    password 校验
+    :param password:
+    :param salt:
+    :param password_hash:
+    :return:
+    """
+    validating_password_hash = create_password_hash(password, salt)
+    return validating_password_hash == password_hash
+
+
+# if __name__ == '__main__':
+#     # salt = make_random_salt(length=60)
+#     salt = "lpiZRQwWoj&*wh29Jovk$uA3YGP$TSQYpM4ooy1z3zGuNT=vmLpb&Ves"
+#     hash_str = create_password_hash('123456132312312312312321', salt)
+#     print(hash_str)
+#     print(len(hash_str))
+#     print(validate_password(password='123456132312312312312321', salt=salt, password_hash=hash_str))

fastgenerateapi/utils/ramdom_utils.py

@@ -0,0 +1,48 @@
+import hashlib
+import random
+import uuid
+
+
+def get_code(n=4):
+    """
+    生成随机验证码，数字和字母的概率各一半
+    :param n: 验证码长度
+    :return: str
+    """
+    s = ''
+    for i in range(n):
+        ret_num = random.randint(0, 9)
+        ret_alpha = chr(random.randint(65, 90))
+        result = random.choice([ret_num, ret_alpha])
+        s += str(result)
+    return s
+
+
+def random_str() -> str:
+    """
+    唯一随机字符串
+    :return: str
+    """
+    only = hashlib.md5(str(uuid.uuid1()).encode(encoding='UTF-8')).hexdigest()
+    return str(only)
+
+
+def get_code_number(n: int = 6) -> str:
+    """
+    随机数字
+    :param n: 长度
+    :return: str
+    """
+    code = ""
+    for i in range(n):
+        ch = chr(random.randrange(ord('0'), ord('9') + 1))
+        code += ch
+
+    return code
+
+
+if __name__ == '__main__':
+    print(get_code())
+    print(random_str())
+    print(get_code_number(6))
+