def home():
if request.form and request.method == 'POST':
data = request.form
gender_mapping = {'Male': True, 'Female': False}
address = Address(country=data.get('country'), city=data.get('city'))
user = User(first_name=data.get('FirstName'),
last_name=data.get('LastName'),
email=data.get('Email'),
age=int(data.get('Age')),
gender=gender_mapping.get(data.get('gender')))
user.address = address
medical_record = MedicalRecord(
has_diabetes=bool(data.get('has_diabete')))
medical_record.user = user
try:
db.session.add(address)
db.session.add(user)
db.session.add(medical_record)
db.session.commit()
except Exception as exc:
return ('an error occurs when saving data')
elif request.method == 'GET':
records = MedicalRecord.query.all()
return render_template("home.html", records=records)
return render_template("home.html")
def register_process():
"""Registration page"""
login = request.form.get('login')
password = request.form.get('pwd')
first_name = request.form.get('first_name')
last_name = request.form.get('last_name')
email = request.form.get('email')
phone = request.form.get('phone')
add_line1 = request.form.get('add_line1')
add_line2 = request.form.get('add_line2')
city = request.form.get('city')
state = request.form.get('state')
zip_code = request.form.get('zip_code')
country = request.form.get('country')
user_type = request.form.get('user_type')
twitter_handle = request.form.get('twitter_handle')
# add user to db
user = User(first_name=first_name,
last_name=last_name,
email=email,
phone=phone,
login=login)
# set user address
user.address = Address(address_line1=add_line1,
address_line2=add_line2,
city=city,
state=state,
zip_code=zip_code,
country=country)
# add user login data to app_user
app_user = AppUser(login=login, password=password)
# write new user / app_user to database
db.session.add(app_user)
db.session.add(user)
db.session.commit()
# put user's email in flask session
session['email'] = email
if user_type == 'artist':
return render_template("artist_info.html", user_id=user.user_id)
if user_type == 'patron':
return render_template("patron_info.html", user_id=user.user_id)
if user_type == 'fan':
return render_template("fan_info.html", user_id=user.user_id)
flash('You were successfully registered %s.' % session['email'])
return redirect("/")
def example_data(self):
"""Create sample data"""
# populate users table
uk = User(first_name='Kushlani',
last_name='Jayasinha',
email='*****@*****.**',
address_id=1)
uk.app_user = AppUser(login='******', password='******')
uk.address = Address(address_line1='myhome')
uc = User(first_name='Chris',
last_name='Lane',
email='*****@*****.**',
address_id=2)
uc.app_user = AppUser(login='******', password='******')
uc.address = Address(address_line1='myhome')
uv = User(first_name='Vidharshi',
last_name='Dharmasens',
email='*****@*****.**',
address_id=3)
uv.app_user = AppUser(login='******', password='******')
uv.address = Address(address_line1='myhome')
ua = User(first_name='Alex',
last_name='Hall',
email='*****@*****.**',
address_id=4)
ua.app_user = AppUser(login='******', password='******')
ua.address = Address(address_line1='myhome')
# populate artists table
a1 = Artist(bio='I am an artist.',
statement='I love art!',
website='http://KushlaniFineArt.com')
# populate patrons table
p1 = Patron(patron_info='I love Chris....')
# populate fans table
f1 = Fan(fan_info='I love Alex.')
# populate artworks table
aw1 = Artwork(
title='Mendocino',
year_created='2015',
medium='oil',
substrate='canvas',
genre='abstracts',
length='40"',
height='30"',
depth='1.5"',
url=
'https://fasoimages-4cde.kxcdn.com/25287_1438386l+v=201609181617c201609181617error/mendocino.jpg'
)
aw2 = Artwork(
title='Autumn',
year_created='2015',
medium='oil',
substrate='canvas',
genre='abstracts',
length='8"',
height='8"',
depth='1.5"',
url=
'https://fasoimages-4cde.kxcdn.com/25287_1322110l+v=201609181617c201609181617error/autumn.jpg'
)
#db.session.add_all([uk, uk.app_user, uk.address, uc, uc.app_user, uc.address, ua, ua.app_user, ua.address, uv, uv.app_user, uv.address, a1, p1, f1, aw1, aw2])
db.session.add_all([uk, uc, ua, uv, a1, p1, f1, aw1, aw2])
db.session.commit()
def seed_db(session):
user = User(name='Michael', nickname='Mike')
user.address = Address(street="5th Avenue", city="New York")
session.add(user)
session.commit()
