def init_db():
Department.objects(name="Engineering").update_one(set__name="Engineering",
upsert=True)
engineering = Department.objects.get(name="Engineering")
print(engineering)
Department.objects(name="Human Resources").update_one(
set__name="Human Resources", upsert=True)
hr = Department.objects.get(name="Human Resources")
print(hr)
Role.objects(name="manager").update_one(set__name="manager", upsert=True)
manager = Role.objects.get(name="manager")
print(manager)
engineer = Role.objects(name="engineer").update_one(set__name="engineer",
upsert=True)
engineer = Role.objects.get(name="engineer")
print(engineer)
task1 = Task(name="Build apps")
# task1.save()
task2 = Task(name="Graphql tutorials")
# task2.save()
task3 = Task(name="Mongodb tutorials")
# task3.save()
task4 = Task(name="Graphql Relay")
# task4.save()
aman = Employee(
name="Aman",
department=engineering,
roles=[engineer],
tasks=[task2, task3, task4],
)
richa = Employee(
name="Richa",
department=engineering,
roles=[engineer],
leader=aman,
tasks=[task1, task2],
)
rian = Employee(name="Rian",
department=hr,
roles=[manager],
tasks=[task2, task3])
try:
aman.save()
richa.save()
rian.save()
except NotUniqueError:
pass
except ValidationError:
pass
def get_role_by_id(_id=1, lang="en"):
item = []
if lang == 'en':
item = Role.objects(_id=_id).exclude("translation")
item = json.loads(item.to_json())[0]
elif lang is not None:
item = Role.objects(_id=_id, translation__language=lang).only("translation")[0]
item = item2dict(item)
item = translation_unify(item, lang)
else:
item = []
return item
def role(id):
if request.method == 'GET':
rewards = Role.objects(name=id)
return get_items(request, rewards)
if request.method == 'POST':
item = Role.objects(name=id).first()
print 'Fetched Item'
if request_has_json():
item.description = request.json['description']
item.save()
return 'Updated'
else:
form = RoleForm(request.form)
form.populate_obj(item)
print 'Populated Item from Form'
print item.description
item.save()
return render_template('list_roles.html',items=[item])
def list_roles(): rewards = Role.objects() return get_items(request, rewards)
def edit_reward_form(id):
item = Role.objects(name=id).first()
form = RoleForm(request.form, obj=item)
return render_template('add_roles.html',form=form, edit=True)
