94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
import re
|
||
|
||
import pinyin
|
||
import pyfiglet
|
||
import toml
|
||
from fastapi import FastAPI
|
||
from pydantic.types import Callable
|
||
|
||
|
||
def extract_brand_name(input_str):
|
||
"""
|
||
提取品牌名称，规则如下：
|
||
- 中文：返回拼音首字母（大写）
|
||
- 英文：根据单词数量决定，少量单词返回大驼峰，大量单词返回纯大写
|
||
- 支持多种英文命名法（驼峰、中划线、下划线等）的解析
|
||
|
||
参数:
|
||
input_str: 输入的字符串（中英文均可）
|
||
返回:
|
||
处理后的品牌名称字符串
|
||
"""
|
||
# 判断是否为中文（包含中文字符）
|
||
if re.search(r"[\u4e00-\u9fff]", input_str):
|
||
# 提取中文并转换为拼音首字母
|
||
chinese_chars = re.findall(r"[\u4e00-\u9fff]", input_str)
|
||
pinyin_initials = "".join(
|
||
[
|
||
pinyin.get(c, format="strip", delimiter="")[0].upper()
|
||
for c in chinese_chars
|
||
]
|
||
)
|
||
return pinyin_initials
|
||
|
||
# 处理英文情况：拆分不同命名法的单词
|
||
# 处理驼峰命名（大小写转换处拆分）
|
||
s = re.sub(r"(?<=[a-z])(?=[A-Z])", " ", input_str)
|
||
# 处理中划线、下划线和其他非字母字符，统一替换为空格
|
||
s = re.sub(r"[-_^&%$#@!~`]", " ", s)
|
||
# 提取所有单词（忽略非字母字符）
|
||
words = re.findall(r"[A-Za-z]+", s)
|
||
# 统一转为小写后再处理（避免大小写影响）
|
||
words = [word.lower() for word in words if word]
|
||
|
||
if not words:
|
||
return "" # 空输入处理
|
||
|
||
# 定义少量单词与大量单词的阈值（这里使用3个作为分界）
|
||
threshold = 3
|
||
if len(words) <= threshold:
|
||
# 少量单词：返回大驼峰格式
|
||
return "".join([word.capitalize() for word in words])
|
||
else:
|
||
# 大量单词：返回纯大写字母（取每个单词首字母）
|
||
return "".join([word[0].upper() for word in words])
|
||
|
||
|
||
# Load the TOML file
|
||
config = toml.load("pyproject.toml")
|
||
|
||
# Access specific values
|
||
project_name = config.get("project", {}).get("name", None)
|
||
project_description = config.get("project", {}).get("description", None)
|
||
project_version = config.get("project", {}).get("version", None)
|
||
project_authors = config.get("project", {}).get("authors", [])
|
||
|
||
|
||
def start(startup: Callable | None = None) -> FastAPI:
|
||
# from pyfiglet import FigletFont
|
||
# fonts = FigletFont().getFonts()
|
||
# for font in fonts:
|
||
# try:
|
||
# result = pyfiglet.figlet_format(project_name, font="red_phoenix")
|
||
# print(f"Font: {font}")
|
||
# print(result)
|
||
# print("-" * 50)
|
||
# except Exception as e:
|
||
# logger.warning(f"Failed to render font {font}: {e}")
|
||
#
|
||
if not startup:
|
||
raise ValueError("startup function is required")
|
||
print("-" * 50)
|
||
print(pyfiglet.figlet_format(project_name, font="red_phoenix"))
|
||
print(f"Project Description: {project_description}")
|
||
print(f"Version: {project_version}")
|
||
print("Author: ", end="")
|
||
[
|
||
print(f"{author.get('name', '')} {author.get('email', '')}", end="")
|
||
for author in project_authors
|
||
]
|
||
print()
|
||
print("-" * 50)
|
||
|
||
return startup()
|