customer-engineering/wallarm
3
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: numbered mode selection, remove duplicate create_config

Browse source
This commit is contained in:
admin 2026-08-03 15:50:44 +00:00
parent f3c82258ac
commit 30c24da698
1 changed files with 7 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
sources/python/main.py
View file

@ -181,25 +181,6 @@ def main():
elif c.lower() == "q": elif c.lower() == "q":
break break
def create_config():
print("\n─── Create Node Config ───")
name = input("Node name: ").strip()
if not name: return
token = input("Wallarm token: ").strip()
cloud = input("Cloud [EU]: ").strip() or "EU"
port = input("Port [8081]: ").strip() or "8081"
upstream_ip = input("Upstream IP [127.0.0.1]: ").strip() or "127.0.0.1"
upstream_port = input("Upstream port [80]: ").strip() or "80"
mode = input("Mode [monitoring]: ").strip() or "monitoring"
cfg = load_config()
cfg[name] = {"token": token, "cloud": cloud, "port": port,
"upstream_ip": upstream_ip, "upstream_port": upstream_port,
"labels": f"group={name}", "mode": mode}
os.makedirs(APP, exist_ok=True)
with open(CONF, "w") as f:
json.dump(cfg, f, indent=2)
print(f"✅ {name} added to fw.conf")
def add_node(): def add_node():
print("\n─── Add New Node ───") print("\n─── Add New Node ───")
@ -211,7 +192,13 @@ def add_node():
port = input("Port [8081]: ").strip() or "8081" port = input("Port [8081]: ").strip() or "8081"
upstream_ip = input("Upstream IP [127.0.0.1]: ").strip() or "127.0.0.1" upstream_ip = input("Upstream IP [127.0.0.1]: ").strip() or "127.0.0.1"
upstream_port = input("Upstream port [80]: ").strip() or "80" upstream_port = input("Upstream port [80]: ").strip() or "80"
mode = input("Mode [monitoring]: ").strip() or "monitoring" print("\nTraffic mode:")
print(" [1] monitoring — detect only")
print(" [2] safe_blocking — block definitely malicious")
print(" [3] block — block all attacks")
print(" [4] off — disable")
m = input("Choose [1]: ").strip()
mode = {"1":"monitoring","2":"safe_blocking","3":"block","4":"off"}.get(m, "monitoring")
# Save to fw.conf # Save to fw.conf
cfg = load_config() cfg = load_config()