def user1(): # return new user named one one = User( name="test_one", nickname="test_one", avatar_url="/static/pic/monkey-32-32px.png", is_super=False) one.set_password("test_password") one.save() return one
def user2(): # return new user named two two = User( name="test_two", nickname="test_two", avatar_url="/static/pic/monkey-32-32px.png", is_super=False) two.set_password("test_password") two.save() return two
def admin1(): # return new admin named one admin_one = User( name="admin_one", nickname="admin_one", avatar_url="/static/pic/monkey-32-32px.png", is_super=True) admin_one.set_password("test_password") admin_one.save() return admin_one
def add_super_user(name, nickname, password, is_super=True): admin = User( name=name, nickname=nickname, avatar_url="/static/pic/monkey-32-32px.png", is_super=is_super) admin.set_password(password) User.objects(name=name).update_one(__raw__={"$set": admin.to_mongo().to_dict()}, upsert=True) return admin