Rewrite command handling

This commit is contained in:
flan 2017-10-28 17:54:58 +02:00
parent 91b5a6e052
commit 4a9ee4d40c

View File

@ -90,28 +90,28 @@ def passwd(username):
def main(): def main():
argc = len(sys.argv) argc = len(sys.argv)
exitcode = 0
cmds = {
"adduser": adduser,
"deluser": deluser,
"lsuser": lsuser,
"passwd": passwd,
}
if argc < 2: if argc < 2:
usage() usage()
exitcode = 1 exit(1)
else:
if argc < 3:
sys.argv.append(None)
if sys.argv[1] == "adduser": c = sys.argv[1]
adduser(sys.argv[2]) try:
elif sys.argv[1] == "deluser": if argc > 2:
deluser(sys.argv[2]) for arg in sys.argv[2:]:
elif sys.argv[1] == "lsuser": cmds[c](arg)
lsuser()
elif sys.argv[1] == "passwd":
passwd(sys.argv[2])
else: else:
cmds[c]()
except KeyError:
usage() usage()
exitcode = 1 exit(1)
sys.exit(exitcode)
if __name__ == "__main__": if __name__ == "__main__":
main() main()