46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
import redis
|
||
|
|
import json
|
||
|
|
import secrets
|
||
|
|
import sys
|
||
|
|
|
||
|
|
REDIS_PWD = "WKzl7jgKTkeWxFWGIBXfzokm59Gvfr76"
|
||
|
|
|
||
|
|
try:
|
||
|
|
r = redis.Redis(host="redis", port=6379, password=REDIS_PWD, db=0)
|
||
|
|
print("Redis connect OK")
|
||
|
|
|
||
|
|
# Ping
|
||
|
|
print(f"PING: {r.ping()}")
|
||
|
|
|
||
|
|
# Test write
|
||
|
|
r.setex("test:debug_token", 60, "test_value")
|
||
|
|
print(f"GET: {r.get('test:debug_token')}")
|
||
|
|
|
||
|
|
# 写 token
|
||
|
|
token = secrets.token_urlsafe(32)
|
||
|
|
user_info = {
|
||
|
|
"employee_id": "sxn",
|
||
|
|
"username": "admin",
|
||
|
|
"name": "宋献",
|
||
|
|
"role": "admin",
|
||
|
|
"department": "IT支持组",
|
||
|
|
"login_source": "test",
|
||
|
|
"login_method": "test_script",
|
||
|
|
"last_active": "2026-07-28T02:00:00",
|
||
|
|
}
|
||
|
|
r.setex(f"user:token:{token}", 8 * 3600, json.dumps(user_info, ensure_ascii=False))
|
||
|
|
print(f"Token written: {token}")
|
||
|
|
|
||
|
|
# 验证
|
||
|
|
val = r.get(f"user:token:{token}")
|
||
|
|
print(f"Stored: {val[:80] if val else 'None'}")
|
||
|
|
|
||
|
|
# 输出token到文件
|
||
|
|
with open("/tmp/test_token.txt", "w") as f:
|
||
|
|
f.write(token)
|
||
|
|
print(f"Token saved to /tmp/test_token.txt")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"FAIL: {e}", file=sys.stderr)
|
||
|
|
sys.exit(1)
|