monolith/commands/auto.py

50 lines
2.3 KiB
Python

import main
from modules import provision
class AutoCommand:
def __init__(self, *args):
self.auto(*args)
def auto(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length):
if authed:
if length == 3:
if not spl[1] in main.network.keys():
failure("No such network: %s" % spl[1])
return
if not spl[2].isdigit():
failure("Must be a number, not %s" % spl[2])
return
relayNum = int(spl[2])
num, alias = main.network[spl[1]].add_relay(relayNum)
success("Successfully created relay %i on network %s with alias %s" % (num, spl[1], alias))
main.saveConf("network")
rtrn = provision.provisionRelay(relayNum, spl[1])
success("Started provisioning network %s on relay %s for alias %s" % (spl[1], spl[2], rtrn))
return
elif length == 2:
if not spl[1] in main.network.keys():
failure("No such network: %s" % spl[1])
return
for i in main.alias.keys():
print("num", i)
num, alias = main.network[spl[1]].add_relay(i)
success("Successfully created relay %i on network %s with alias %s" % (num, spl[1], alias))
rtrn = provision.provisionRelay(num, spl[1])
success("Started provisioning network %s on relay %s for alias %s" % (spl[1], num, rtrn))
main.saveConf("network")
return
elif length == 1:
for i in main.network.keys():
for x in main.alias.keys():
num, alias = main.network[i].add_relay(x)
success("Successfully created relay %i on network %s with alias %s" % (num, i, alias))
rtrn = provision.provisionRelay(num, i)
success("Started provisioning network %s on relay %s for alias %s" % (i, num, rtrn))
main.saveConf("network")
return
else:
incUsage("auto")
return
else:
incUsage(None)