def elems(elem_type): form = ElemForm() if elem_type == CONST.ELEM_ORG: elems = current_user.available_orgs() else: elems = current_user.available_agents() if request.method == "POST": if form.validate(elem_type, current_user): if elem_type == CONST.ELEM_ORG: newelem = Organization(form.name.data, form.inn.data, form.kpp.data, current_user.uid, form.is_ip.data) else: newelem = Agent(form.name.data, form.inn.data, form.kpp.data, current_user.uid, form.is_ip.data) db.session.add(newelem) db.session.commit() flash(u"Добавлен новый элемент", "alert-success") if elem_type == CONST.ELEM_ORG: elems = current_user.available_orgs() else: elems = current_user.available_agents() # clean fields lst_fields = ['name', 'inn', 'kpp', 'is_ip'] for fld in lst_fields: form[fld].raw_data = None form[fld].data = None return render_template("elems.html", form=form, elem_type=elem_type, elems=elems, edit=0)
def get_items(item_type='agent'): if item_type == CONST.ELEM_ORG: elems = current_user.available_orgs() elif item_type == CONST.ELEM_USER: if current_user.is_admin: elems = User.query.all() else: elems = None else: elems = current_user.available_agents() result = org_schema.dump(elems) return jsonify(result=result)