monolith/commands/mod.py

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

            elif length == 6:
                if not spl[1] in main.network.keys():
                    failure("Network does not exist: %s" % spl[1])
                    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 e:
                    failure("Something went wrong.")
                    return

            else:
                incUsage("mod")
                return
        else:
            incUsage(None)