pymecli 0.3.1__tar.gz

pymecli-0.3.1/.gitignore

@@ -0,0 +1,17 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+
+tmp/
+node_modules/
+pnpm-lock.yaml
+tmp.json
+tmp.yaml
+cloudflare.ini

pymecli-0.3.1/PKG-INFO

@@ -0,0 +1,45 @@
+Metadata-Version: 2.4
+Name: pymecli
+Version: 0.3.1
+Summary: My CLI
+Project-URL: Homepage, https://pypi.org/project/pymecli
+Project-URL: Repository, https://github.com/meme2046/pymecli
+License: MIT
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.10
+Requires-Dist: arrow>=1.4.0
+Requires-Dist: babel>=2.17.0
+Requires-Dist: cowsay>=6.1
+Requires-Dist: dotenv>=0.9.9
+Requires-Dist: fastapi>=0.127.0
+Requires-Dist: numpy<2.0.0
+Requires-Dist: pandas>=2.3.3
+Requires-Dist: pymysql>=1.1.2
+Requires-Dist: pytz>=2025.2
+Requires-Dist: pyyaml>=6.0.3
+Requires-Dist: redis[hiredis]>=7.1.0
+Requires-Dist: requests[socks]>=2.32.5
+Requires-Dist: sqlalchemy>=2.0.45
+Requires-Dist: tabulate>=0.9.0
+Requires-Dist: toml>=0.10.2
+Requires-Dist: typer>=0.20.1
+Requires-Dist: uvicorn>=0.40.0
+Description-Content-Type: text/markdown
+
+# typer example
+
+```shell
+uv run example hello
+uv run example hello Xiaoming
+uv run example goodbye
+uv run example goodbye Xiaoming
+uv run example goodbye Xiaoming -f
+```
+
+# fastapi server
+
+```shell
+uv run fast --port 8877
+```

pymecli-0.3.1/README.md

@@ -0,0 +1,15 @@
+# typer example
+
+```shell
+uv run example hello
+uv run example hello Xiaoming
+uv run example goodbye
+uv run example goodbye Xiaoming
+uv run example goodbye Xiaoming -f
+```
+
+# fastapi server
+
+```shell
+uv run fast --port 8877
+```

pymecli-0.3.1/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")

pymecli-0.3.1/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",
+    )

pymecli-0.3.1/api/v1/redis.py

@@ -0,0 +1,98 @@
+import json
+
+from fastapi import APIRouter, HTTPException, Query, Request
+from fastapi.responses import PlainTextResponse
+
+from models.response import SuccessResponse
+
+router = APIRouter()
+
+
+@router.get(
+    "/json",
+    summary="根据redis.key获取数据",
+    response_description="返回json数据",
+)
+async def json_by_key(
+    request: Request,
+    key: str = Query(..., description="redis.key"),
+):
+    r = request.app.state.redis_client
+    key_type = await r.type(key)
+
+    if key_type == "string":  # 或 'string' 取决于客户端
+        value = await r.get(key)
+        value = json.loads(value)
+    elif key_type == "hash":
+        value = await r.hgetall(key)
+    elif key_type == "list":
+        value = await r.lrange(key, 0, -1)
+    elif key_type == "set":
+        value = await r.smembers(key)
+    elif key_type == "zset":
+        value = await r.zrange(key, 0, -1, withscores=True)
+    else:
+        raise HTTPException(status_code=404, detail=f"Key '{key}' not found in Redis")
+
+    return SuccessResponse(data=value)
+
+
+@router.get(
+    "/plaintext",
+    summary="根据redis.key获取数据",
+    response_description="返回plaintext数据",
+    # response_class=PlainTextResponse,
+)
+async def plaintext_by_key(
+    request: Request,
+    key: str = Query(..., description="redis.key"),
+):
+    r = request.app.state.redis_client
+    key_type = await r.type(key)
+    # print(f"Key type: {key_type}")
+
+    if key_type == "string":  # 或 'string' 取决于客户端
+        value = await r.get(key)
+        return PlainTextResponse(
+            content=value, headers={"Content-Type": "text/plain; charset=utf-8"}
+        )
+    elif key_type == "none":
+        raise HTTPException(status_code=404, detail=f"Key '{key}' not found in Redis")
+    else:
+        raise HTTPException(
+            status_code=400, detail=f"key:<{key}>,value type not string"
+        )
+
+
+@router.get(
+    "/byset",
+    summary="根据redis.key获取数据",
+)
+async def list_by_set(
+    request: Request,
+    key: str = Query(..., description="redis.key"),
+    cursor: int = Query(0, description="从第几条开始取数据"),
+    size: int = Query(5000, description="取多少数据"),
+):
+    r = request.app.state.redis_client
+    key_type = await r.type(f"by_time:{key}")
+
+    if key_type == "none":
+        raise HTTPException(status_code=404, detail=f"Key '{key}' not found in Redis")
+
+    if key_type not in ["zset", "set"]:
+        raise HTTPException(
+            status_code=400, detail=f"key:<{key}>,value type not in [zset,set]"
+        )
+
+    ids = await r.zrevrange(f"by_time:{key}", cursor, cursor + size - 1)
+    if not ids:
+        return SuccessResponse(data=[])
+
+    pipe = r.pipeline()
+    for id in ids:
+        pipe.hgetall(f"{key}:{id}")
+    value = await pipe.execute()
+    filtered_value = [v for v in value if v]
+
+    return SuccessResponse(data=filtered_value)

pymecli-0.3.1/cli/bitget.py

@@ -0,0 +1,59 @@
+import asyncio
+
+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_csv_fp = "d:/github/meme2046/data/bitget_grid_0.csv"
+    sf_csv_fp = "d:/github/meme2046/data/bitget_sf_0.csv"
+    asyncio.run(grid_open(engine, grid_csv_fp))
+    asyncio.run(grid_close(engine, grid_csv_fp))
+    asyncio.run(bitget_sf_open(engine, sf_csv_fp))
+    asyncio.run(bitget_sf_close(engine, sf_csv_fp))
+
+
+@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)

pymecli-0.3.1/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}!")

pymecli-0.3.1/cli/fast.py

@@ -0,0 +1,167 @@
+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 = 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",
+    ),
+):
+    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,
+    )

pymecli-0.3.1/cli/gate.py

@@ -0,0 +1,27 @@
+import asyncio
+
+import typer
+
+from crypto.gate import grid_close, grid_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)
+
+
+@app.command()
+def rsync(env_path: str = "d:/.env"):
+    """同步mysql中grid数据到csv文件"""
+    engine = get_database_engine(env_path)
+    grid_csv_fp = "d:/github/meme2046/data/gate_grid_0.csv"
+    asyncio.run(grid_open(engine, grid_csv_fp))
+    asyncio.run(grid_close(engine, grid_csv_fp))