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')