def purchase_form_handler(form): purchase_data = Purchase(rstnumber=form.rst_number.data, weight=form.weight.data, moisture=form.moisture.data, rate=form.rate.data, variety_id=form.variety.data, agent_id=form.agent.data, timestamp=form.date.data, amount=form.amount.data) return write_to_db(purchase_data)
def user_form_handler(form, action): if action == "add": user = User(username=form.username.data, email=form.email.data, role_id=form.roles.data) user.set_password(form.password.data) user.active = form.active.data return write_to_db(user, db.session.add) elif action == "delete": user = User.query.filter_by(form.id.data).first() if user is None: abort(404) return write_to_db(user, db.session.delete) elif action == "update": user = User(email=form.email.data, role_id=form.roles.data, active=form.active.data) return write_to_db(user, db.session.add) return 500, Exception(f"Invalid action: {action}")
def sale_form_handler(form): quintol = form.quintol.data rate = form.rate.data gst = 5 amt = (quintol * rate) + (quintol * rate * gst / 100) sale_data = Sale(party_name=form.party_name.data, party_address=form.party_address.data, gst_number=form.gst_number.data, vehicle_number=form.vehicle_number.data, no_of_bags=form.no_of_bags.data, variety_id=form.variety.data, agent_id=form.agent.data, quintol=form.quintol.data, rate=form.rate.data, timestamp=form.date.data, amount=amt) return write_to_db(sale_data)
def agent_form_handler(form, action): agent_type_2_model = { "1": PurchaseAgent, "2": SaleAgent } # logger.info("data is coming") cur_model = agent_type_2_model.get(form.agent_type.data) if cur_model is None: abort(500) agent = cur_model( name=form.name.data, email=form.email.data, mobile=form.mobile.data, address=form.address.data ) return write_to_db(agent, db.session.add)
def change_password(): generic_data = {"title": "Change Password", "heading": "Change Password"} form = ChangePasswordForm() if form.validate_on_submit(): user = current_user user.set_password(form.new_password.data) status, e = write_to_db(user) db.session.commit() if status != 200: # logger.error(f"status: {status}, {e}") abort(status) return redirect( url_for('auth.logout', message=("Your password changed succesfully." " Please Login"))) return render_template(f"auth/change_password.html", form=form, data=generic_data)
def reset_user_password(): generic_data = { "title": "Change User Password", "heading": "Change User Password" } form = ChangeUserPasswordForm() if form.validate_on_submit(): user = User.query.filter_by(username=form.username.data).first() user.set_password(form.new_password.data) status, e = write_to_db(user) db.session.commit() if status != 200: # logger.error(f"status: {status}, {e}") abort(status) flash(f"Password for {form.username.data} changed succesfully.") return redirect(url_for('auth.reset_user_password')) return render_template(f"auth/reset_user_password.html", form=form, data=generic_data)
def variety_form_handler(form, action): variety = Variety(name=form.name.data) return write_to_db(variety, db.session.add)