Rewrite command handling
This commit is contained in:
parent
91b5a6e052
commit
4a9ee4d40c
@ -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()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user