def create_dummy_data():
default_user_type = ["Admin", "Staff", "Client"]
for user_type in default_user_type:
db.session.add(UserType(name=user_type))
db.session.commit()
# Create product area
default_product_area = ["Policies", "Billing", "Claims", "Reports"]
for product_area in default_product_area:
db.session.add(ProductArea(name=product_area))
db.session.commit()
# Create dummy user
admin = User(username="******", email="*****@*****.**", userTypeId=get_user_type("Admin"), fullname="Administrator")
staff = User(username="******", email="*****@*****.**", userTypeId=get_user_type("Staff"), fullname="Staff 1")
clienta = User(username="******", email="*****@*****.**", userTypeId=get_user_type("Client"), fullname="Client A")
clientb = User(username="******", email="*****@*****.**", userTypeId=get_user_type("Client"), fullname="Client B")
clientc = User(username="******", email="*****@*****.**", userTypeId=get_user_type("Client"), fullname="Client C")
users = [admin, staff, clienta, clientb, clientc]
for user in users:
db.session.add(user)
db.session.commit()
for client_username in ["clienta", "clientb", "clientc"]:
create_dummy_feature_request(client_username)
db.session.commit()
def setup_users(users):
user_types = open('{}/initial/user_types.txt'.format(dir_path),
'r').read().splitlines()
for user_type_name in user_types:
# print 'checked {}'.format(user_type_name)
# check user type
try:
user_type = session.query(UserType) \
.filter(UserType.name == user_type_name) \
.one()
# make new entry if not found
except NoResultFound as e:
user_type = add(UserType(name=user_type_name))
for u in users:
try:
session.query(User) \
.filter(User.name == u['name']) \
.one()
except NoResultFound as e:
add(
User(name=u['name'],
email=u['email'],
password=bcrypt.hashpw(u['password'],
bcrypt.gensalt()),
user_type_id=u['user_type_id']))
for k, v in answers.iteritems():
user = session.query(User) \
.filter(User.user_type_id == k) \
.one()
for element_id, text in v.iteritems():
try:
session.query(Answer) \
.filter(Answer.user_id == user.id) \
.filter(Answer.element_id == element_id) \
.one()
except:
add(
Answer(name=text,
element_id=element_id,
user_id=user.id))