Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
6431f64
1
Parent(s):
8783eb5
自动迁移旧版设置文件,以后只使用config.json
Browse files- modules/config.py +24 -18
modules/config.py
CHANGED
@@ -28,6 +28,30 @@ if os.path.exists("config.json"):
|
|
28 |
else:
|
29 |
config = {}
|
30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
## 处理docker if we are running in Docker
|
32 |
dockerflag = config.get("dockerflag", False)
|
33 |
if os.environ.get("dockerrun") == "yes":
|
@@ -64,24 +88,6 @@ if dockerflag:
|
|
64 |
if not (isinstance(username, type(None)) or isinstance(password, type(None))):
|
65 |
auth_list.append((os.environ.get("USERNAME"), os.environ.get("PASSWORD")))
|
66 |
authflag = True
|
67 |
-
else:
|
68 |
-
if (
|
69 |
-
not my_api_key
|
70 |
-
and os.path.exists("api_key.txt")
|
71 |
-
and os.path.getsize("api_key.txt")
|
72 |
-
):
|
73 |
-
with open("api_key.txt", "r") as f:
|
74 |
-
my_api_key = f.read().strip()
|
75 |
-
if os.path.exists("auth.json"):
|
76 |
-
authflag = True
|
77 |
-
with open("auth.json", "r", encoding='utf-8') as f:
|
78 |
-
auth = json.load(f)
|
79 |
-
for _ in auth:
|
80 |
-
if auth[_]["username"] and auth[_]["password"]:
|
81 |
-
auth_list.append((auth[_]["username"], auth[_]["password"]))
|
82 |
-
else:
|
83 |
-
logging.error("请检查auth.json文件中的用户名和密码!")
|
84 |
-
sys.exit(1)
|
85 |
|
86 |
@contextmanager
|
87 |
def retrieve_openai_api(api_key = None):
|
|
|
28 |
else:
|
29 |
config = {}
|
30 |
|
31 |
+
if os.path.exists("api_key.txt"):
|
32 |
+
logging.info("检测到api_key.txt文件,正在进行迁移...")
|
33 |
+
with open("api_key.txt", "r") as f:
|
34 |
+
config["openai_api_key"] = f.read().strip()
|
35 |
+
os.rename("api_key.txt", "api_key(deprecated).txt")
|
36 |
+
with open("config.json", "w", encoding='utf-8') as f:
|
37 |
+
json.dump(config, f, indent=4)
|
38 |
+
|
39 |
+
if os.path.exists("auth.json"):
|
40 |
+
logging.info("检测到auth.json文件,正在进行迁移...")
|
41 |
+
auth_list = []
|
42 |
+
with open("auth.json", "r", encoding='utf-8') as f:
|
43 |
+
auth = json.load(f)
|
44 |
+
for _ in auth:
|
45 |
+
if auth[_]["username"] and auth[_]["password"]:
|
46 |
+
auth_list.append((auth[_]["username"], auth[_]["password"]))
|
47 |
+
else:
|
48 |
+
logging.error("请检查auth.json文件中的用户名和密码!")
|
49 |
+
sys.exit(1)
|
50 |
+
config["users"] = auth_list
|
51 |
+
os.rename("auth.json", "auth(deprecated).json")
|
52 |
+
with open("config.json", "w", encoding='utf-8') as f:
|
53 |
+
json.dump(config, f, indent=4)
|
54 |
+
|
55 |
## 处理docker if we are running in Docker
|
56 |
dockerflag = config.get("dockerflag", False)
|
57 |
if os.environ.get("dockerrun") == "yes":
|
|
|
88 |
if not (isinstance(username, type(None)) or isinstance(password, type(None))):
|
89 |
auth_list.append((os.environ.get("USERNAME"), os.environ.get("PASSWORD")))
|
90 |
authflag = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
91 |
|
92 |
@contextmanager
|
93 |
def retrieve_openai_api(api_key = None):
|