def make_listing_profile():
"""add listing info to database"""
roommates = request.form.getlist("roommates")
laundry = True if request.form.get("laundry") == "True" else False
pets = True if request.form.get("pets") == "True" else False
living_there = True if request.form.get(
"living_there") == "True" else False
listing_id = request.form.get("listing_name")
user_id = session["current_user"]
main_photo = functions.save_photo("main_photo")
photos = []
photos.append(functions.save_photo("photo_1"))
photos.append(functions.save_photo("photo_2"))
photos.append(functions.save_photo("photo_3"))
kwargs = dict(listing_id=listing_id,
address=request.form.get("address"),
neighborhood=request.form.get("neighborhood"),
price=int(request.form.get("price")),
avail_as_of=request.form.get("avail_date"),
length_of_rental=request.form.get("duration"),
main_photo=main_photo,
bedrooms=request.form.get("bedrooms"),
bathrooms=request.form.get("bathrooms"),
laundry=laundry,
pets=pets,
description=request.form.get("description"),
primary_lister=user_id)
if Listing.query.get(listing_id):
flash("Listing Name Taken")
return redirect("/listings/{}".format(listing_id))
else:
for key in kwargs.keys():
if kwargs[key] == "":
del kwargs[key]
db.session.add(Listing(**kwargs))
db.session.commit()
if living_there:
functions.add_UserListing(user_id, listing_id)
for roommate in roommates:
functions.add_UserListing(roommate, listing_id)
for photo in photos:
db.session.add(Picture(listing_id=listing_id, photo=photo))
db.session.commit()
return redirect("/listings/{}".format(listing_id))
def make_profile():
"""add user info to database"""
friends = request.form.getlist("friends")
listings = request.form.getlist("listings")
looking = True if request.form.get("looking") == "True" else False
questions = Question.query.all()
user_id = request.form.get("user_name")
password = request.form.get("password").encode('utf-8')
hashed = bcrypt.hashpw(password, bcrypt.gensalt())
venmo = request.form.get("venmo")
photo = functions.save_photo("photo")
kwargs = dict(user_id=user_id,
name=request.form.get("full_name"),
email=request.form.get("email"),
password=hashed,
phone=request.form.get("phone"),
bio=request.form.get("bio"),
photo=photo,
state=request.form.get("state"),
looking_for_apt=looking,
venmo=venmo)
if User.query.get(user_id):
flash("User Name Taken")
return redirect("/users/{}".format(user_id))
else:
for key in kwargs.keys():
if kwargs[key] == "":
del kwargs[key]
db.session.add(User(**kwargs))
db.session.commit()
for friend in friends:
db.session.add(Friendship(friend_1_id=friend, friend_2_id=user_id))
db.session.add(Friendship(friend_2_id=friend, friend_1_id=user_id))
db.session.commit()
for listing in listings:
functions.add_UserListing(user_id, listing)
for question in questions:
answer_id = request.form.get("{}".format(question.question_id))
db.session.add(UserAnswer(user_id=user_id, answer_id=answer_id))
db.session.commit()
session["current_user"] = user_id
return redirect("/users/{}".format(user_id))
def change_main_photo():
print("in function")
photo = functions.save_photo("image")
print("saved photo")
user_id = session["current_user"]
user = User.query.get(user_id)
user.photo = photo
db.session.commit()
return redirect("/users/{}/my_homepage".format(user_id))
def add_user():
"""add a new user"""
name = request.form.get("name")
email = request.form.get("email")
password = request.form.get("password")
bio = request.form.get("bio")
photo = functions.save_photo("pic")
imdb = request.form.get("imdb")
current_link = request.form.get("current_link")
color_scheme = request.form.get("color")
print(name, email, password, bio, imdb, current_link, color_scheme)
kwargs = dict(name=name,
email=email,
password=password,
bio=bio,
photo=photo,
imdb=imdb,
current_link=current_link,
color_scheme_id=color_scheme)
print("test")
print(kwargs)
keys_to_remove = []
for key in kwargs.keys():
if kwargs[key] == "":
keys_to_remove.append(key)
for key in keys_to_remove:
del kwargs[key]
user = User(**kwargs)
#add data
db.session.add(user)
#commit
db.session.commit()
user = User.query.filter(User.email == email).first()
print(user)
print(user.user_id)
functions.login(user, password)
return redirect("/users/{}/my_homepage".format(user.user_id))
def add_video():
"""add video to database"""
user_id = session["current_user"]
category = request.form.get("category")
youtube = request.form.get("youtube")
print(category)
print(youtube)
if youtube:
kwargs = dict(user_id=user_id,
video=youtube,
category=category,
youtube=True)
db.session.add(Video(**kwargs))
db.session.commit()
print("added youtube")
else:
video = functions.save_photo("video")
print(video)
kwargs = dict(user_id=user_id,
video=video,
category=category,
youtube=False)
db.session.add(Video(**kwargs))
db.session.commit()
print("added vid")
user = User.query.get(user_id)
for vid in user.videos:
print(vid.video, vid.category, vid.youtube)
return redirect("/users/{}/my_videos".format(user_id))
def add_photo():
"""add photo to database"""
user_id = session["current_user"]
category = request.form.get("category").split("|")
photo = functions.save_photo("pic")
print(category)
print(category[0])
print(category[0] == "play")
if category[0] == "headshot":
kwargs = dict(user_id=user_id, headshot=photo)
keys_to_remove = []
for key in kwargs.keys():
if kwargs[key] == "":
keys_to_remove.append(key)
for key in keys_to_remove:
del kwargs[key]
db.session.add(Headshot(**kwargs))
db.session.commit()
elif category[0] == "misc":
kwargs = dict(user_id=user_id, misc_photo=photo)
keys_to_remove = []
for key in kwargs.keys():
if kwargs[key] == "":
keys_to_remove.append(key)
for key in keys_to_remove:
del kwargs[key]
db.session.add(Miscphoto(**kwargs))
db.session.commit()
elif category[0] == "play":
kwargs = dict(user_id=user_id, play_id=category[1], photo=photo)
keys_to_remove = []
for key in kwargs.keys():
if kwargs[key] == "":
keys_to_remove.append(key)
for key in keys_to_remove:
del kwargs[key]
db.session.add(Photo(**kwargs))
db.session.commit()
elif category[0] == "film":
kwargs = dict(user_id=user_id, film_id=category[1], photo=photo)
keys_to_remove = []
for key in kwargs.keys():
if kwargs[key] == "":
keys_to_remove.append(key)
for key in keys_to_remove:
del kwargs[key]
db.session.add(Photo(**kwargs))
db.session.commit()
return redirect("/users/{}/my_photos".format(user_id))