98 lines
2.7 KiB
Python
Executable File
98 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import os
|
|
import sys
|
|
import getpass
|
|
|
|
from ankisyncd.users import SqliteUserManager
|
|
|
|
DATAPREFIX = os.path.join(os.path.expanduser("~"), ".local", "share")
|
|
#DATADIR = os.path.join(DATAPREFIX, "ankisyncd")
|
|
DATADIR = "."
|
|
AUTHDBPATH = os.path.join(DATADIR, "auth.db")
|
|
COLLECTIONPATH = os.path.join(DATADIR, "collections")
|
|
|
|
def usage():
|
|
print("usage: "+sys.argv[0]+" <command> [<args>]")
|
|
print()
|
|
print("Commands:")
|
|
print(" adduser <username> - add a new user")
|
|
print(" deluser <username> - delete a user")
|
|
print(" lsuser - list users")
|
|
print(" passwd <username> - change password of a user")
|
|
|
|
def adduser(username):
|
|
if username:
|
|
password = getpass.getpass("Enter password for {}: ".format(username))
|
|
|
|
user_manager = SqliteUserManager(AUTHDBPATH, COLLECTIONPATH)
|
|
user_manager.add_user(username, password)
|
|
else:
|
|
usage()
|
|
|
|
def deluser(username):
|
|
if username and os.path.isfile(AUTHDBPATH):
|
|
user_manager = SqliteUserManager(AUTHDBPATH, COLLECTIONPATH)
|
|
|
|
try:
|
|
user_manager.del_user(username)
|
|
except ValueError as error:
|
|
print("Could not delete user {}: {}"
|
|
.format(username, error.message), file=sys.stderr)
|
|
elif not username:
|
|
usage()
|
|
else:
|
|
print("{}: Database file does not exist".format(sys.argv[0]),
|
|
file=sys.stderr)
|
|
|
|
def lsuser():
|
|
user_manager = SqliteUserManager(AUTHDBPATH, COLLECTIONPATH)
|
|
try:
|
|
users = user_manager.user_list()
|
|
for username in users:
|
|
print(username)
|
|
except ValueError as error:
|
|
print("Could not list users: {}".format(AUTHDBPATH, error.message),
|
|
file=sys.stderr)
|
|
|
|
def passwd(username):
|
|
if os.path.isfile(AUTHDBPATH):
|
|
password = getpass.getpass("Enter password for {}: ".format(username))
|
|
|
|
user_manager = SqliteUserManager(AUTHDBPATH, COLLECTIONPATH)
|
|
try:
|
|
user_manager.set_password_for_user(username, password)
|
|
except ValueError as error:
|
|
print("Could not set password for user {}: {}"
|
|
.format(username, error.message), file=sys.stderr)
|
|
else:
|
|
print("{}: Database file does not exist".format(sys.argv[0]),
|
|
file=sys.stderr)
|
|
|
|
def main():
|
|
argc = len(sys.argv)
|
|
|
|
cmds = {
|
|
"adduser": adduser,
|
|
"deluser": deluser,
|
|
"lsuser": lsuser,
|
|
"passwd": passwd,
|
|
}
|
|
|
|
if argc < 2:
|
|
usage()
|
|
exit(1)
|
|
|
|
c = sys.argv[1]
|
|
try:
|
|
if argc > 2:
|
|
for arg in sys.argv[2:]:
|
|
cmds[c](arg)
|
|
else:
|
|
cmds[c]()
|
|
except KeyError:
|
|
usage()
|
|
exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|