def add_data(): name, email, pw = 'Man', '*****@*****.**', 'demopass' pw_hash = generate_password_hash(pw) u = UserModel( email=email, name=name, password=pw_hash) u.save() label, addr, zone = 'Home', 'J-35, Rajouri Garden', 'n' address = AddressModel( label=label, _address=addr, zone=zone, user=u,default=True) address.save() label, addr, zone = 'Office', 'F-132, Rajouri Garden', 'n' address = AddressModel( label=label, _address=addr, zone=zone, user=u) address.save() home_addr = AddressModel.query.filter_by(label='Home').first() office_addr = AddressModel.query.filter_by(label='Office').first() dt = datetime(2019, 10, 3) p = PickupModel( scheduled_date=dt, user=u, address=home_addr, active=False, description='A fridge', completed=True ) db.session.add(p) dt = datetime(2019, 10, 10) p = PickupModel( scheduled_date=dt, user=u, address=office_addr, active=False, description='An old phone', completed=True ) db.session.add(p) dt = datetime(2019, 10, 22) p = PickupModel( scheduled_date=dt, user=u, address=home_addr, active=False, description='10 AA batteries', completed=True ) db.session.add(p) db.session.commit()
def add_address(): if request.method == 'POST': addr = request.form.get('addr') zone = request.form.get('zone') default = request.form.get('default') label = request.form.get('label') if addr == '' or 'zone' not in request.form or label == '': return redirect(url_for('dashboard.dash')) u = current_user if default == 'on': for address in u.addresses: address.default = False db.session.commit() address = AddressModel(label=label, _address=addr, zone=zone, default=True, user=current_user) else: if len(u.addresses) == 0: address = AddressModel(label=label, _address=addr, zone=zone, user=current_user, default=True) else: address = AddressModel(label=label, _address=addr, zone=zone, user=current_user) address.save() return redirect(url_for('dashboard.dash')) return render_template('user/add_address.html')