pymecli 0.2.5__py3-none-any.whl

api/v1/__init__.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter
+
+from api.v1.clash import router as clash_router
+from api.v1.redis import router as redis_router
+
+api_router = APIRouter()
+
+api_router.include_router(clash_router, prefix="/clash")
+api_router.include_router(redis_router, prefix="/redis")

api/v1/clash.py

@@ -0,0 +1,34 @@
+import yaml
+from fastapi import APIRouter, Query, Response
+
+from core.clash import ClashYamlGenerator, get_generator_dependency
+
+router = APIRouter()
+
+
+@router.get(
+    "/sub",
+    summary="转换订阅(可以转换多个订阅)",
+    description="根据提供的URL、User-Agent和名称获取并处理订阅信息,返回YAML格式的Clash配置",
+    response_description="返回YAML格式的Clash配置文件",
+)
+async def sub(
+    urls: str = Query(..., description="订阅URL,逗号分隔"),
+    agents: str = Query(
+        "clash-verge/v2.4.3",
+        description="User-Agent,逗号分隔(根据客户段选择)",
+    ),
+    names: str = Query(
+        "订阅1", description="订阅名称,逗号分隔(可选,在客户端中的显示名)"
+    ),
+    generator: ClashYamlGenerator = get_generator_dependency(),
+):
+    # 将JSON字符串解析为Python对象
+    sub_list = generator.query2sub(urls, agents, names)
+    yaml_content, userinfo = generator.gen(sub_list)
+    yaml_string = yaml.dump(yaml_content, allow_unicode=True, default_flow_style=False)
+    return Response(
+        headers={"Subscription-Userinfo": userinfo},
+        content=yaml_string,
+        media_type="text/yaml",
+    )

api/v1/redis.py

@@ -0,0 +1,26 @@
+import json
+
+from fastapi import APIRouter, HTTPException, Query, Request
+
+from models.response import SuccessResponse
+
+router = APIRouter()
+
+
+@router.get(
+    "/get",
+    summary="根据redis.key获取数据",
+    response_description="返回json数据",
+)
+async def redis(
+    request: Request,
+    key: str = Query(..., description="redis.key"),
+):
+    value = await request.app.state.redis_client.get(key)
+
+    if value is None:
+        raise HTTPException(status_code=404, detail=f"Key '{key}' not found in Redis")
+
+    json_data = json.loads(value)
+
+    return SuccessResponse(data=json_data)

cli/bitget.py

@@ -0,0 +1,55 @@
+import typer
+
+from crypto.bitget import (
+    bitget_sf_close,
+    bitget_sf_open,
+    grid_close,
+    grid_open,
+    mix_tickers,
+    spot_tickers,
+)
+from utils.mysql import get_database_engine
+
+app = typer.Typer()
+
+
+@app.command()
+def sync(env_path: str = "d:/.env"):
+    """同步mysql中grid数据到csv文件"""
+    engine = get_database_engine(env_path)
+    grid_open(engine, "d:/github/meme2046/data/bitget_0.csv")
+    grid_close(engine, "d:/github/meme2046/data/bitget_0.csv")
+    bitget_sf_open(engine, "d:/github/meme2046/data/bitget_sf_0.csv")
+    bitget_sf_close(engine, "d:/github/meme2046/data/bitget_sf_0.csv")
+
+
+@app.command()
+def spot(
+    symbols: str,
+    proxy: str = typer.Option(
+        None, "--proxy", "-p", help="代理服务器地址，例如: http://127.0.0.1:7890"
+    ),
+):
+    """
+    从bitget获取加密货币现货价格.
+
+    参数:
+    symbols:加密货币符号,可以是多个,用逗号分隔,例如:"BTCUSDT,ETHUSDT"
+    """
+    spot_tickers(symbols.split(","), proxy)
+
+
+@app.command()
+def mix(
+    symbols: str,
+    proxy: str = typer.Option(
+        None, "--proxy", "-p", help="代理服务器地址，例如: http://127.0.0.1:7890"
+    ),
+):
+    """
+    从bitget获取加密货币合约价格.
+
+    参数:
+    symbols:加密货币符号,可以是多个,用逗号分隔,例如:"BTCUSDT,ETHUSDT"
+    """
+    mix_tickers(symbols.split(","), proxy)

cli/example.py

@@ -0,0 +1,32 @@
+import typer
+
+app = typer.Typer()
+
+
+@app.command()
+def hello(
+    name: str = typer.Argument(
+        "from My CLI!",
+        help="Name of the person to greet",
+    ),
+):
+    print(f"Hello {name}!")
+
+
+@app.command()
+def goodbye(
+    name: str = typer.Argument(
+        "from My CLI!",
+        help="Name of the person to goodbye",
+    ),
+    formal: bool = typer.Option(
+        False,
+        "--formal",
+        "-f",
+        help="是否为正式场合回答",
+    ),
+):
+    if formal:
+        print(f"Goodbye Ms. {name}. Have a good day.")
+    else:
+        print(f"Bye {name}!")

cli/fast.py

@@ -0,0 +1,169 @@
+import os
+from contextlib import asynccontextmanager
+
+import redis.asyncio as redis
+import typer
+import uvicorn
+from fastapi import FastAPI
+from fastapi.exceptions import RequestValidationError
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, PlainTextResponse
+from starlette.exceptions import HTTPException as StarletteHTTPException
+
+from api.v1 import api_router
+from core.clash import ClashConfig, init_generator
+from core.config import settings
+from models.response import SuccessResponse
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """管理应用生命周期的上下文管理器"""
+    redis_pool = redis.ConnectionPool(
+        host=os.getenv("REDIS_HOST", "192.168.123.7"),
+        port=int(os.getenv("REDIS_PORT", 6379)),
+        db=int(os.getenv("REDIS_DB", 0)),
+        password=os.getenv("REDIS_PASSWORD"),
+        max_connections=20,  # 根据需要调整最大连接数
+        decode_responses=True,
+    )
+    redis_client = redis.Redis(connection_pool=redis_pool)
+
+    # 启动时
+    app.state.redis_client = redis_client
+    yield
+    # 关闭时
+    if redis_client:
+        await redis_client.aclose()
+
+
+typer_app = typer.Typer()
+
+
+app = FastAPI(
+    title=settings.NAME,
+    description=settings.DESCRIPTION,
+    version=settings.VERSION,
+    lifespan=lifespan,
+)
+
+# 注册 v1 版本的所有路由
+app.include_router(api_router, prefix=settings.API_V1_STR)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 允许所有源，生产环境建议设置具体的源
+    allow_credentials=True,
+    allow_methods=["*"],  # 允许所有 HTTP 方法
+    allow_headers=["*"],  # 允许所有 headers
+)
+
+
+@app.exception_handler(StarletteHTTPException)
+async def http_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"error": exc.detail, "success": False, "data": None},
+    )
+
+
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request, exc):
+    # 将错误信息转换为可序列化的格式
+    errors = []
+    for error in exc.errors():
+        errors.append(
+            {
+                "type": error.get("type"),
+                "loc": error.get("loc"),
+                "msg": error.get("msg"),
+                "input": error.get("input"),
+            }
+        )
+
+    return JSONResponse(
+        status_code=422,
+        content={
+            "error": errors,
+            "success": False,
+            "data": None,
+        },
+    )
+
+
+@app.exception_handler(Exception)
+async def general_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=500,
+        content={
+            "error": "Internal server error",
+            "success": False,
+            "data": None,
+        },
+    )
+
+
+@app.get("/")
+async def root():
+    return SuccessResponse(data=f"Welcome to {settings.DESCRIPTION}")
+
+
+@app.get("/ping", response_class=PlainTextResponse)
+async def pingpong():
+    return "pong"
+
+
+@typer_app.command()
+def run_app(
+    host: str = typer.Argument(
+        "0.0.0.0",
+        help="fastapi监听的<ip>地址",
+    ),
+    port: int = typer.Option(
+        80,
+        "--port",
+        help="fastapi监听的端口号",
+    ),
+    ssl_keyfile: str = typer.Option(
+        None,
+        "--ssl-keyfile",
+        "-sk",
+        help="ssl keyfile",
+    ),
+    ssl_certfile: str = typer.Option(
+        None,
+        "--ssl-certfile",
+        "-sc",
+        help="ssl certfile",
+    ),
+    rule: str = typer.Option(
+        "https://cdn.jsdelivr.net/gh/Loyalsoldier/clash-rules@release",
+        "--rule",
+        "-r",
+        help="clash Rule base URL",
+    ),
+    my_rule: str = typer.Option(
+        "https://raw.githubusercontent.com/meme2046/data/refs/heads/main/clash",
+        "--my-rule",
+        "-mr",
+        help="my clash rule base URL(自定义规则)",
+    ),
+    proxy: str = typer.Option(
+        None,
+        "--proxy",
+        "-p",
+        help="服务器代理,传入则通过代理转换Clash订阅,比如:socks5://127.0.0.1:7890",
+    ),
+):
+    settings.reload()
+
+    clash_config = ClashConfig(rule, my_rule, proxy)
+    init_generator(clash_config)
+
+    uvicorn.run(
+        "cli.fast:app",
+        host=host,
+        port=port,
+        reload=False,
+        ssl_keyfile=ssl_keyfile,
+        ssl_certfile=ssl_certfile,
+    )

cli/gate.py

@@ -0,0 +1,16 @@
+import typer
+
+from crypto.gate import gate_close, gate_open
+from utils.mysql import get_database_engine
+
+app = typer.Typer()
+
+
+@app.command()
+def sync(
+    env_path: str = "d:/.env", csv_path: str = "d:/github/meme2046/data/gate_0.csv"
+):
+    """同步mysql中grid数据到csv文件"""
+    engine = get_database_engine(env_path)
+    gate_open(engine, csv_path)
+    gate_close(engine, csv_path)

cli/util.py

@@ -0,0 +1,112 @@
+import platform
+import secrets
+import string
+import sys
+import uuid as u
+from datetime import datetime
+
+import arrow
+import pytz
+import typer
+from babel.dates import format_datetime
+from cowsay.__main__ import cli
+
+app = typer.Typer()
+
+
+@app.command()
+def sid(
+    length: int = typer.Argument(30, help="生成secure_id的长度"),
+):
+    chars = string.ascii_letters + string.digits
+    id = "".join(secrets.choice(chars) for _ in range(length))
+    print(id)
+
+
+@app.command()
+def uuid():
+    print(u.uuid4())
+
+
+@app.command()
+def os():
+    print(platform.system())
+
+
+@app.command()
+def ts():
+    timestamp = arrow.now().timestamp()
+    print(int(timestamp))
+
+
+@app.command()
+def ms():
+    timestamp = arrow.now().timestamp()
+    print(int(timestamp * 1000))
+
+
+@app.command()
+def v():
+    print(f"🧊 python:{sys.version}")
+
+
+def strf_time(zone: str):
+    tz = pytz.timezone(zone)
+    now = datetime.now(tz)
+    # locale="zh_CN" 会使月份和星期的名称显示为中文
+    # locale="en_US" 则会显示为英文
+    return format_datetime(
+        now, "yyyy年MM月dd日 HH:mm:ss EEEE ZZZZ zzzz", locale="zh_CN"
+    )
+
+
+@app.command()
+def st():
+    t0 = strf_time("UTC")
+    t1 = strf_time("America/New_York")
+    t2 = strf_time("Asia/Shanghai")
+
+    print(t0)
+    print(t1)
+    print(t2)
+
+
+@app.command()
+def stoken(
+    a: str = typer.Argument(..., help="第一个Token地址"),
+    b: str = typer.Argument(..., help="第二个Token地址"),
+):
+    """
+    返回按照 Uniswap 规则排序后的 token0 和 token1
+    """
+    # 移除可能存在的 "0x" 前缀
+    addr_a = a.lower().replace("0x", "")
+    addr_b = b.lower().replace("0x", "")
+
+    # 验证地址格式是否正确
+    if len(addr_a) != 40 or len(addr_b) != 40:
+        print("错误: 地址必须是40位十六进制字符串")
+        return
+
+    # 按照Uniswap的方式进行比较（基于数值大小而不是字符串排序）
+    if int(addr_a, 16) < int(addr_b, 16):
+        token0, token1 = a, b
+    else:
+        token0, token1 = b, a
+
+    print(token0, token1)
+    return token0, token1
+
+
+def say():
+    cli()
+
+
+if __name__ == "__main__":
+    usdc = "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48"
+    weth = "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"
+    usdt = "0xdac17f958d2ee523a2206206994597c13d831ec7"
+    xaut = "0x68749665ff8d2d112fa859aa293f07a622782f38"
+    uni = "0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"
+    aave = "0x7fc66500c84a76ad7e9c93437bfc5ac33e2ddae9"
+    stoken(usdt, xaut)