This repository has been archived by the owner on Apr 17, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
seed.py
41 lines (31 loc) · 1.51 KB
/
seed.py
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
from runnershub import app
from runnershub.models import *
from flask_security.utils import hash_password
def create_roles(user_datastore):
user_datastore.create_role(name="Admin", description="Site Administrator")
user_datastore.create_role(name="Campaign Owner", description="The owner of the overall campaign, who gets the "
"final say in plot-related things.")
user_datastore.create_role(name="GM", description="Hub GM (may also be a player)")
user_datastore.create_role(name="Player", description="Hub Player")
user_datastore.commit()
def create_users(user_datastore):
user1 = user_datastore.create_user(email="sanchitsharma1@gmail.com",
password=hash_password("password"),
active=True)
user_datastore.add_role_to_user(user1, 'Admin')
user_datastore.add_role_to_user(user1, 'GM')
user_datastore.add_role_to_user(user1, 'Campaign Owner')
user2 = user_datastore.create_user(email="test@test.com",
password=hash_password("password"),
active=True)
user_datastore.add_role_to_user(user2, 'Player')
user_datastore.commit()
def seed():
user_datastore = app.security.datastore
with app.app_context():
db.drop_all()
db.create_all()
create_roles(user_datastore)
create_users(user_datastore)
if __name__ == '__main__':
seed()