aiohttp-three-template/adminuser.py at master · RobertoPrevato/aiohttp-three-template · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
"""
* Copyright 2016, Roberto Prevato roberto.prevato@gmail.com
* https://github.com/RobertoPrevato/aiohttp-three-template
*
* Licensed under the MIT license:
* http://www.opensource.org/licenses/MIT
*
* Utility script to administer the application users.
"""
import argparse

separator = "******************************************************\n"

parser = argparse.ArgumentParser(description= "Adds users to the application, using the configured DAL project.",
                                 epilog = "{}\n{}".format("author: Roberto Prevato roberto.prevato@gmail.com", separator))

parser.add_argument("-k", "--userkey", dest= "userkey",
                    required=True, help="User key, when performing login (e.g. email address)")

parser.add_argument("-p", "--password", dest= "password",
                    required=False, help="User password")

parser.add_argument("-o", "--operation", dest="operation",
                    required=True, choices=["add", "delete", "setroles", "setpassword"],
                    help="Operation to perform: add user, delete user, set roles, set password")

parser.add_argument("-r", "--roles", nargs='+', dest= "roles",
                    required=False, help="User roles")

args = parser.parse_args()

from dal.membership import MembershipStore
from bll.membership import MembershipProvider

# instantiate the membership provider
membership = MembershipProvider(store=MembershipStore())

def main(options):
    if not options.operation:
        options.operation = "add"

    success = False
    result = ""

    if options.operation == "add":
        # password is required
        if not options.password:
            print("error: argument -p/--password is required")
            return

        success, result = membership.create_account(options.userkey, options.password, None, options.roles)
        if success:
            print("Account created successfully")

    elif options.operation == "delete":
        success, result = membership.delete_account(operations.userkey)
        if success:
            print("Account deleted successfully")

    elif options.operation == "setroles":
        # roles are required
        if not options.roles:
            print("error: argument -r/--roles is required")
            return

        success, result = membership.update_account(options.userkey, {
            "roles": options.roles
        })
        if success:
            print("Account updated successfully")

    elif options.operation == "setpassword":
        # password is required
        if not options.password:
            print("error: argument -p/--password is required")
            return

        success, result = membership.update_password(options.userkey, options.password)
        if success:
            print("Account password updated successfully")

    if not success:
        print("ERROR: " + result)

main(args)