pymecli 0.5.4__tar.gz → 0.5.6__tar.gz

{pymecli-0.5.4 → pymecli-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: pymecli
-Version: 0.5.4
+Version: 0.5.6
 Summary: My CLI
 Project-URL: Homepage, https://pypi.org/project/pymecli
 Project-URL: Repository, https://github.com/meme2046/pymecli

{pymecli-0.5.4 → pymecli-0.5.6}/cli/bitget.py

@@ -20,7 +20,10 @@ app = typer.Typer()
 
 @app.command()
 def sync(
-    env_path: str = "d:/.env",
+    env_path: str = typer.Argument(
+        ".env",
+        help="dotenv环境变量路径",
+    ),
 ):
     """同步mysql中grid数据到redis"""
     engine = get_database_engine(env_path)

{pymecli-0.5.4 → pymecli-0.5.6}/cli/fast.py

@@ -18,11 +18,8 @@ from models.response import SuccessResponse
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """管理应用生命周期的上下文管理器"""
-    redis_pool = redis.ConnectionPool(
-        host=settings.REDIS_HOST,
-        port=settings.REDIS_PORT,
-        db=settings.REDIS_DB,
-        password=settings.REDIS_PASSWORD,
+    redis_pool = redis.ConnectionPool.from_url(
+        settings.REDIS_URL,
         max_connections=20,  # 根据需要调整最大连接数
         decode_responses=True,
     )
@@ -152,16 +149,15 @@ def run_app(
         "-p",
         help="服务器代理,传入则通过代理转换Clash订阅,比如:socks5://127.0.0.1:7897",
     ),
-    redis_host: str = typer.Option(
+    redis_url: str = typer.Option(
         None,
-        "--redis-host",
-        "-r",
-        help="redis host",
+        "--redis-url",
+        help="redis url, 格式: redis://[user:password@]host[:port][/db]",
     ),
 ):
     settings.reload()
-    if redis_host is not None:
-        settings.REDIS_HOST = redis_host
+    if redis_url is not None:
+        settings.REDIS_URL = redis_url
 
     clash_config = ClashConfig(rule, my_rule, proxy)
     init_generator(clash_config)

pymecli-0.5.6/cli/gate.py

@@ -0,0 +1,20 @@
+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 rsync(
+    env_path: str = typer.Argument(
+        ".env",
+        help="dotenv环境变量路径",
+    ),
+):
+    """同步mysql中grid数据到redis"""
+    engine = get_database_engine(env_path)
+    asyncio.run(grid_open(engine))
+    asyncio.run(grid_close(engine))

pymecli-0.5.6/cli/okx.py

@@ -0,0 +1,22 @@
+import asyncio
+
+import typer
+
+from crypto.okx import grid_close, grid_open
+from utils.mysql import get_database_engine
+
+app = typer.Typer()
+
+
+@app.command()
+def sync(
+    env_path: str = typer.Argument(
+        ".env",
+        help="dotenv环境变量路径",
+    ),
+):
+    """同步mysql中grid数据到redis"""
+    engine = get_database_engine(env_path)
+
+    asyncio.run(grid_open(engine))
+    asyncio.run(grid_close(engine))

{pymecli-0.5.4 → pymecli-0.5.6}/cli/tele.py

@@ -12,10 +12,10 @@ app = typer.Typer()
 @app.command()
 def info(
     dotenv_path: str = typer.Argument(
-        help="dotenv file contains TELE_SEESION_STRING", default="d:/.env"
+        help="dotenv file contains TELE_SEESION_STRING", default=".env"
     ),
 ):
-    load_dotenv("d:/.env")
+    load_dotenv(dotenv_path)
     logger.info(os.getenv("TELE_SESSION_STRING"))
 
 
@@ -23,7 +23,7 @@ def info(
 def check(
     dotenv_path: str = typer.Argument(
         help="dotenv file contains TELE_SESSION_STRING / TELE_API_ID / TELE_API_HASH",
-        default="d:/.env",
+        default=".env",
     ),
 ):
     """验证 TELE_SESSION_STRING 是否有效。
@@ -87,7 +87,7 @@ def check(
 def login(
     dotenv_path: str = typer.Argument(
         help="dotenv file containing TELE_API_ID / TELE_API_HASH",
-        default="d:/.env",
+        default=".env",
     ),
 ):
     """交互式登录并打印 TELE_SESSION_STRING。
@@ -141,7 +141,7 @@ def upload(
         help="要上传的图片文件路径",
     ),
     dotenv_path: str = typer.Option(
-        help="dotenv file path", default="d:/.env"
+        help="dotenv file path", default=".env"
     ),
 ):
     """上传图片到指定 Telegram 频道。
@@ -289,7 +289,7 @@ def upload(
 @app.command()
 def list_channels(
     dotenv_path: str = typer.Option(
-        help="dotenv file path", default="d:/.env"
+        help="dotenv file path", default=".env"
     ),
 ):
     """列出当前账号加入的所有频道及其 ID。

{pymecli-0.5.4 → pymecli-0.5.6}/core/config.py

@@ -13,10 +13,7 @@ class Settings(BaseSettings):
     API_V1_STR: str = "/api/v1"
     PROJECT_NAME: str = metadata["Name"]
 
-    REDIS_HOST: str = "192.168.123.7"
-    REDIS_PORT: int = 6379
-    REDIS_DB: int = 0
-    REDIS_PASSWORD: str = ""
+    REDIS_URL: str = "redis://192.168.123.7:6379/0"
 
     PROJECT_DESCRIPTION: str = (
         f"{metadata['Summary']}, FastAPI提供: clash订阅转换、redis API"

{pymecli-0.5.4 → pymecli-0.5.6}/core/redis_client.py

@@ -8,26 +8,17 @@ import redis
 class RedisClient:
     def __init__(
         self,
-        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"),
-        decode_responses=True,
+        url: str = os.getenv("REDIS_URL", "redis://192.168.123.7:6379/0"),
+        decode_responses: bool = True,
     ):
         """
         初始化Redis客户端
-        :param host: Redis服务器地址
-        :param port: Redis服务器端口
-        :param db: 数据库编号
-        :param password: 密码（如果需要）
+        :param url: Redis连接URL, 格式 redis://[user:password@]host[:port][/db]
         :param decode_responses: 是否自动解码响应（将字节转换为字符串）
         """
 
-        self.client = redis.Redis(
-            host=host,
-            port=port,
-            db=db,
-            password=password,
+        self.client = redis.Redis.from_url(
+            url,
             decode_responses=decode_responses,
             socket_connect_timeout=5,
             socket_timeout=5,

pymecli-0.5.6/crypto/okx.py

@@ -0,0 +1,52 @@
+from sqlalchemy import Engine
+
+from utils import logger
+from utils.mysql import mysql_to_redis
+
+
+async def grid_open(engine: Engine):
+    query = "select * from okx_spot where ((cost is not null or benefit is not null) and profit is null) and up_status = 0 and order_id is not null and deleted_at is null;"
+    key_prefix = "okx_grid"
+    table = "okx_spot"
+
+    row_count = await mysql_to_redis(
+        engine,
+        key_prefix,
+        table,
+        query,
+        update_status=1,
+        d_column_names=["order_id", "client_order_id"],
+        pd_dtype={
+            "order_id": str,
+            "fx_order_id": str,
+            "created_at": "datetime64[ns]",
+            "open_at": "datetime64[ns]",
+            "close_at": "datetime64[ns]",
+        },
+    )
+
+    logger.info(f"🧮 okx grid open count:({row_count})")
+
+
+async def grid_close(engine: Engine):
+    query = "select * from okx_spot where profit is not null and up_status in (0,1) and deleted_at is null;"
+    key_prefix = "okx_grid"
+    table = "okx_spot"
+
+    row_count = await mysql_to_redis(
+        engine,
+        key_prefix,
+        table,
+        query,
+        update_status=2,
+        d_column_names=["order_id", "client_order_id"],
+        pd_dtype={
+            "order_id": str,
+            "fx_order_id": str,
+            "created_at": "datetime64[ns]",
+            "open_at": "datetime64[ns]",
+            "close_at": "datetime64[ns]",
+        },
+    )
+
+    logger.info(f"🧮 okx grid close count:({row_count})")

{pymecli-0.5.4 → pymecli-0.5.6}/pyproject.toml

@@ -36,7 +36,7 @@
   name = "pymecli"
   readme = "README.md"
   requires-python = ">=3.10"
-  version = "0.5.4"
+  version = "0.5.6"
 
 [project.urls]
   Homepage = "https://pypi.org/project/pymecli"
@@ -49,6 +49,7 @@
   example = "cli.example:app"
   fast = "cli.fast:typer_app"
   gate = "cli.gate:app"
+  okx = "cli.okx:app"
   util = "cli.util:app"
   # 临时命令
   rd = "cli.redis_csv:app"

{pymecli-0.5.4 → pymecli-0.5.6}/utils/mysql.py

@@ -1,5 +1,6 @@
 import os
-import time
+import re
+from urllib.parse import parse_qsl, urlencode, urlsplit, urlunsplit
 
 import pandas as pd
 from dotenv import load_dotenv
@@ -9,25 +10,62 @@ from utils import logger
 from utils.pd import deduplicated, dt_to_timestamp
 from utils.pyredis import get_redis_client
 
-
-def get_database_engine(env_path: str) -> Engine:
+# Go go-sql-driver/mysql 专有参数，Python pymysql / SQLAlchemy 不支持
+_GO_DSN_PARAMS = {
+    "parseTime", "loc", "timeout", "readTimeout", "writeTimeout",
+    "allowNativePasswords", "tls", "serverCert", "clientCert",
+    "clientKey", "multiStatements", "columnsWithAlias", "interpolateParams"}
+
+
+def _strip_go_params(query: str) -> str:
+    """从 query string 中移除 Go DSN 专有参数"""
+    if not query:
+        return query
+    kept = [(k, v) for k, v in parse_qsl(query) if k not in _GO_DSN_PARAMS]
+    return urlencode(kept)
+
+
+def _to_sqlalchemy_url(dsn: str) -> str:
+    """
+    将 Go/Node 通用 MySQL DSN 转为 SQLAlchemy URL。
+
+    输入格式(Go go-sql-driver/mysql DSN 或 Node mysql URI):
+        root:password@192.168.123.7:3366/bot_tx?charset=utf8mb4&parseTime=True&loc=Local
+        root:password@tcp(192.168.123.7:3366)/bot_tx?charset=utf8mb4
+        mysql://root:password@192.168.123.7:3306/mydb
+
+    输出格式(SQLAlchemy):
+        mysql+pymysql://root:password@192.168.123.7:3366/bot_tx?charset=utf8mb4
+    """
+    # 如果已经是完整的 scheme:// URL，直接处理
+    if "://" in dsn:
+        parts = urlsplit(dsn)
+        # mysql:// → mysql+pymysql://
+        scheme = parts.scheme
+        if scheme == "mysql":
+            scheme = "mysql+pymysql"
+        return urlunsplit((scheme, parts.netloc, parts.path, _strip_go_params(parts.query), ""))
+
+    # Go DSN 格式：[user[:password]@][tcp(]host[:port][)]/db[?params]
+    # 去掉 tcp() 包装
+    dsn = re.sub(r"tcp\(([^)]+)\)", r"\1", dsn)
+
+    # user:pass@host:port/db?query → mysql+pymysql://user:pass@host:port/db?query
+    url = f"mysql+pymysql://{dsn}"
+    parts = urlsplit(url)
+    return urlunsplit((parts.scheme, parts.netloc, parts.path, _strip_go_params(parts.query), ""))
+
+
+def get_database_engine(env_path: str = ".env") -> Engine:
     """创建数据库引擎"""
     load_dotenv(env_path)
-    host = os.getenv("MYSQL_HOST")
-    port = os.getenv("MYSQL_PORT")
-    database = os.getenv("MYSQL_DATABASE")
-    user = os.getenv("MYSQL_USER")
-    password = os.getenv("MYSQL_PASSWORD")
-
-    engine = create_engine(
-        f"mysql+pymysql://{user}:{password}@{host}:{port}/{database}",
-        connect_args={
-            # 对于MySQL，可以在连接参数中指定时区
-            # "init_command": "SET time_zone='+00:00'",  # UTC时间
-            # 或者使用本地时区，例如：
-            # "init_command": "SET time_zone='+08:00'",  # 中国标准时间
-        },
+    dsn = os.getenv(
+        "MYSQL_URL",
+        "root:@127.0.0.1:3306/?charset=utf8mb4",
     )
+    url = _to_sqlalchemy_url(dsn)
+
+    engine = create_engine(url, connect_args={})
 
     try:
         with engine.connect() as connection:
@@ -124,13 +162,16 @@ async def mysql_to_redis_and_csv(
         id = f"{idx1}_{idx2}"
         key = f"{key_prefix}:{id}"
 
-        # 转换行数据为字典（处理 NaN 为 None 或空字符串）
+        # 转换行数据为字典(处理 NaN 为 None 或空字符串)
         row_dict = row.where(pd.notna(row), "").to_dict()
 
-        # 1. 写入完整数据到 Hash（自动覆盖）
+        # 1. 写入完整数据到 Hash(自动覆盖)
         pipe.hset(key, mapping=row_dict)
-        # 2. 写入 ZSet 索引：score = Unix 时间戳
-        pipe.zadd(f"by_time:{key_prefix}", {id: time.time()})
+        # 2. 写入 ZSet 索引：score = 开仓时间戳(空值用 0 兜底)
+        score = row["open_at"]
+        if score is None or pd.isna(score):
+            score = 0
+        pipe.zadd(f"by_time:{key_prefix}", {id: score})
         count += 1
 
     await pipe.execute()
@@ -221,7 +262,10 @@ async def mysql_to_redis(
         row_dict = row.where(pd.notna(row), "").to_dict()
 
         pipe.hset(key, mapping=row_dict)
-        pipe.zadd(f"by_time:{key_prefix}", {id: time.time()})
+        score = row["open_at"]
+        if score is None or pd.isna(score):
+            score = 0
+        pipe.zadd(f"by_time:{key_prefix}", {id: score})
         count += 1
 
     await pipe.execute()

pymecli-0.5.6/utils/pyredis.py

@@ -0,0 +1,18 @@
+import os
+
+import redis as redis_sync
+import redis.asyncio as redis
+
+
+def get_redis_client(
+    url: str = os.getenv("REDIS_URL", "redis://192.168.123.7:6379/0"),
+    decode_responses: bool = True,
+) -> redis.Redis:
+    return redis.Redis.from_url(url, decode_responses=decode_responses)
+
+
+def get_redis_client_sync(
+    url: str = os.getenv("REDIS_URL", "redis://192.168.123.7:6379/0"),
+    decode_responses: bool = True,
+) -> redis_sync.Redis:
+    return redis_sync.Redis.from_url(url, decode_responses=decode_responses)

pymecli-0.5.4/cli/gate.py

@@ -1,28 +0,0 @@
-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数据到redis"""
-    engine = get_database_engine(env_path)
-    asyncio.run(grid_open(engine))
-    asyncio.run(grid_close(engine))

pymecli-0.5.4/utils/pyredis.py

@@ -1,36 +0,0 @@
-import os
-
-import redis as redis_sync
-import redis.asyncio as redis
-
-
-def get_redis_client(
-    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"),
-    decode_responses=True,
-) -> redis.Redis:
-    return redis.Redis(
-        host=host,
-        port=port,
-        db=db,
-        password=password,
-        decode_responses=decode_responses,
-    )
-
-
-def get_redis_client_sync(
-    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"),
-    decode_responses=True,
-) -> redis_sync.Redis:
-    return redis_sync.Redis(
-        host=host,
-        port=port,
-        db=db,
-        password=password,
-        decode_responses=decode_responses,
-    )