shop-java/tools/seed_store_user.py

# -*- coding: utf-8 -*-
"""创建/删除 store 端临时测试账号（配合 test_token_user.py 使用）。

用法：
    python tools/seed_store_user.py add      # 建账号 camel_test_admin / 123456
    python tools/seed_store_user.py remove   # 跑完清理
"""
import re
import sys

import pymysql

YML = r"F:/shop/java-rewrite/yoshop-java/src/main/resources/application.yml"
PWD = re.findall(r"password:\s*(.+)", open(YML, encoding="utf-8").read())[0].strip().strip("\"'")
HASH = "$2a$10$sh3uRv6KiZaab8973cVxXex533fU5oYUQlOMy97k1YmrM.YO3sJDK"  # 123456
NAME = "camel_test_admin"

conn = pymysql.connect(host="192.168.1.177", user="root", password=PWD,
                       database="shop", charset="utf8mb4", autocommit=True)
cur = conn.cursor()
cur.execute("delete from yoshop_store_user where user_name=%s", (NAME,))
if sys.argv[1] == "add":
    # 可选第二参数：自定义口令哈希（默认 123456 的 $2a$；传 PHP password_hash 的 $2y$ 可验证兼容性）
    pwd_hash = sys.argv[2] if len(sys.argv) > 2 else HASH
    cur.execute(
        "insert into yoshop_store_user (user_name,password,real_name,is_super,is_delete,"
        "sort,store_id,create_time,update_time) values (%s,%s,%s,0,0,100,10001,"
        "unix_timestamp(),unix_timestamp())", (NAME, pwd_hash, "temp"))
    print("seeded", NAME, cur.lastrowid, pwd_hash[:7])
else:
    print("removed", NAME)
conn.close()