import main from yaml import dump class ModCommand: # This could be greatly improved, but not really important right now def __init__(self, *args): self.mod(*args) def mod(self, addr, authed, data, obj, spl, success, failure, info, incUsage, length): if authed: if length == 4: if not spl[1] in main.network.keys(): failure("Network does not exist: %s" % spl[1]) return try: setattr(main.network[spl[1]], spl[2], spl[3]) except e: failure("Something went wrong.") return main.saveConf("network") success("Successfully set key %s to %s on %s" % (spl[2], spl[3], spl[1])) return # Find a better way to do this #elif length == 6: # if not spl[1] in main.network.keys(): # failure("Network does not exist: %s" % spl[1]) # return # if not spl[3].isdigit(): # failure("Must be a number, not %s" % spl[3]) # return # if not int(spl[3]) in main.network[spl[1]].relays.keys(): # failure("Relay/alias does not exist: %s" % spl[3]) # return # try: # x = getattr(main.network[spl[1]], spl[2]) # x[spl[3]] = spl[4] # except Exception as err: # failure("Error: %s" % err) # return else: incUsage("mod") return else: incUsage(None)