from yaml import dump import main class EmailCommand: def __init__(self, *args): self.email(*args) def email( self, addr, authed, data, obj, spl, success, failure, info, incUsage, length ): if authed: if length == 4: if spl[1] == "add": if not spl[2].isdigit(): # failure("Must be a number, not %s" % spl[2]) if spl[2] == "domain": domain = spl[3] if "@" in domain: failure("Not a domain: %s" % domain) return if not domain in main.irc["_"]["domains"]: main.irc["_"]["domains"].append(domain) success( "Successfully added domain %s to default config" % domain ) main.saveConf("irc") else: failure( "Domain already exists in default config: %s" % domain ) return elif spl[1] == "del": if not spl[2].isdigit(): # failure("Must be a number, not %s" % spl[2]) if spl[2] == "domain": domain = spl[3] if domain in main.irc["_"]["domains"]: main.irc["_"]["domains"].remove(domain) success( "Successfully removed domain %s to default config" % domain ) main.saveConf("irc") else: failure( "Domain does not exist in default config: %s" % domain ) return else: num = int(spl[2]) if spl[1] == "add": if not num in main.alias.keys(): failure("No such alias: %i" % num) return if not spl[3] in main.alias[num]["emails"]: main.alias[num]["emails"].append(spl[3]) main.saveConf("alias") success( "Successfully added email %s to alias %i" % (spl[3], num) ) return else: failure( "Email already exists in alias %i: %s" % (num, spl[3]) ) return elif spl[1] == "del": if not num in main.alias.keys(): failure("No such alias: %i" % num) return if spl[3] in main.alias[num]["emails"]: main.alias[num]["emails"].remove(spl[3]) main.saveConf("alias") success( "Successfully removed email %s from alias %i" % (spl[3], num) ) return else: failure( "Email does not exist in alias %i: %s" % (spl[3], num) ) return elif length == 2: if spl[1] == "list": info(dump(main.alias)) return else: incUsage("email") return elif length == 3: if spl[1] == "list": if spl[2] == "domain": filtered = { f"{k}:{k2}": v2 for k, v in main.irc.items() for k2, v2 in v.items() if k2 == "domains" } info(dump(filtered)) return else: incUsage("email") return else: incUsage("email") return else: incUsage("email") return else: incUsage(None)