Skip to content

Commit 0a16f17

Browse files
committed
feat:(multiproxy_tools, config): added confirmation prompt, test run OK
1 parent 54d963f commit 0a16f17

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

utils/fn_multiproxy_tools.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from utils.generator import assemble_docker_compose, generate_env_file
99
from utils.dumper import write_json
1010
from utils.loader import load_json_config
11+
from utils.prompt_helper import ask_question_yn
1112

1213

1314
def update_multiproxy_instances(proxies_file: str = 'proxies.txt', instances_dir: str = 'm4b_proxy_instances', user_config_path: str = './config/user-config.json', m4b_config_path: str = './config/m4b-config.json', app_config_path: str = './config/app-config.json', sleep_time: int = 3) -> None:
@@ -23,6 +24,13 @@ def update_multiproxy_instances(proxies_file: str = 'proxies.txt', instances_dir
2324
sleep_time (int): Time to wait between operations.
2425
"""
2526

27+
# Ask for confirmation using prompt_helper
28+
if not ask_question_yn("This will update all multiproxy instances with new proxies. Do you want to proceed?", default=False):
29+
print("Operation canceled.")
30+
logging.info("User canceled the update of multiproxy instances.")
31+
time.sleep(sleep_time)
32+
return
33+
2634
# Ensure proxies.txt exists
2735
if not os.path.isfile(proxies_file):
2836
logging.error(f"Proxies file '{proxies_file}' not found.")
@@ -156,7 +164,7 @@ def update_multiproxy_instances(proxies_file: str = 'proxies.txt', instances_dir
156164

157165
def submenu_multiproxy_tools():
158166
return [
159-
{"label": "Update Multiproxy Instances",
167+
{"label": "Update Proxies for Multiproxy Instances",
160168
"function": "update_multiproxy_instances"},
161169
{"label": "Exit", "function": "exit_submenu"}
162170
]
@@ -165,7 +173,6 @@ def submenu_multiproxy_tools():
165173
def exit_submenu(*args, **kwargs):
166174
print("Exiting Multiproxy Tools.")
167175
logging.info("User exited the Multiproxy Tools menu.")
168-
time.sleep(3)
169176
return False
170177

171178

0 commit comments

Comments
 (0)