Ejemplo n.º 1
0
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)
Ejemplo n.º 2
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)