105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
import sys
|
||
import os
|
||
|
||
# .env文件路径：固定在api-key-vault根目录
|
||
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
ENV_FILE = os.path.join(BASE_DIR, ".env")
|
||
|
||
|
||
def load_keys():
|
||
"""读取.env文件，返回dict"""
|
||
keys = {}
|
||
if not os.path.exists(ENV_FILE):
|
||
return keys
|
||
with open(ENV_FILE, "r", encoding="utf-8") as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if not line or line.startswith("#"):
|
||
continue
|
||
if "=" in line:
|
||
name, _, value = line.partition("=")
|
||
keys[name.strip()] = value.strip()
|
||
return keys
|
||
|
||
|
||
def save_keys(keys):
|
||
"""把dict写回.env文件"""
|
||
with open(ENV_FILE, "w", encoding="utf-8") as f:
|
||
f.write("# API Key Vault - 自动生成，请勿手动乱改\n")
|
||
for name, value in keys.items():
|
||
f.write(f"{name}={value}\n")
|
||
|
||
|
||
def cmd_get(name):
|
||
keys = load_keys()
|
||
if name not in keys:
|
||
print("ERROR:KEY_NOT_FOUND")
|
||
else:
|
||
print(keys[name])
|
||
|
||
|
||
def cmd_set(name, value):
|
||
keys = load_keys()
|
||
keys[name] = value
|
||
save_keys(keys)
|
||
print(f"✅ 已保存：{name}")
|
||
|
||
|
||
def cmd_list():
|
||
keys = load_keys()
|
||
if not keys:
|
||
print("暂无已存储的Key")
|
||
return
|
||
print("已存储的Key列表：")
|
||
for name in keys:
|
||
value = keys[name]
|
||
# 只显示前4位和后4位，中间用*遮挡
|
||
masked = value[:4] + "****" + value[-4:] if len(value) > 8 else "****"
|
||
print(f" · {name} = {masked}")
|
||
|
||
|
||
def cmd_delete(name):
|
||
keys = load_keys()
|
||
if name not in keys:
|
||
print(f"❌ 未找到：{name}")
|
||
else:
|
||
del keys[name]
|
||
save_keys(keys)
|
||
print(f"🗑 已删除：{name}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) < 2:
|
||
print("用法：")
|
||
print(" python vault.py get <key名>")
|
||
print(" python vault.py set <key名> <key值>")
|
||
print(" python vault.py list")
|
||
print(" python vault.py delete <key名>")
|
||
sys.exit(1)
|
||
|
||
command = sys.argv[1].lower()
|
||
|
||
if command == "get":
|
||
if len(sys.argv) < 3:
|
||
print("错误：get命令需要提供key名")
|
||
sys.exit(1)
|
||
cmd_get(sys.argv[2])
|
||
|
||
elif command == "set":
|
||
if len(sys.argv) < 4:
|
||
print("错误：set命令需要提供key名和key值")
|
||
sys.exit(1)
|
||
cmd_set(sys.argv[2], sys.argv[3])
|
||
|
||
elif command == "list":
|
||
cmd_list()
|
||
|
||
elif command == "delete":
|
||
if len(sys.argv) < 3:
|
||
print("错误：delete命令需要提供key名")
|
||
sys.exit(1)
|
||
cmd_delete(sys.argv[2])
|
||
|
||
else:
|
||
print(f"错误：未知命令 {command}")
|
||
sys.exit(1) |