def JoinFamily(userID): session = Database.Session() familyID = session.query( Database.tUser.familyID).filter(Database.tUser.userID == userID) for family in familyID: familyID = family.familyID x = session.query(Database.tUser).get(current_user.id) x.familyID = familyID session.commit() dogIDs = session.query( Database.tDog.dogID).filter(Database.tDog.familyID == familyID) for dogID in dogIDs: dog = Database.tFollowers(dogID=dogID.dogID, userID=current_user.id) session.add(dog) session.commit() return redirect('/dashboard')
def FollowDogs(dogID): session = Database.Session() addFollower = Database.tFollowers(dogID = dogID, userID= current_user.id) session.add(addFollower) session.commit() return '', 204
def DogCreation(): FavoriteParkform = FavoriteParkForm() if FavoriteParkform.validate_on_submit(): #insert Breed cur, db = Tlbx.dbConnectDict() breedQuery = "INSERT INTO tBreed (breed) VALUES (%s);" data = (session.get('breed')) cur.execute(breedQuery, data) db.commit() breedID = cur.lastrowid #Insert Address then insert Favorite Park image = None addressQuery = "INSERT INTO tAddress (address, city, state) VALUES (%s, %s, %s);" try: address = request.form['desktopstreet_address'] city = request.form['desktopadministrative_area_level_1'] state = request.form['desktoplocality'] except: address = request.form['mobilestreet_address'] city = request.form['mobileadministrative_area_level_1'] state = request.form['mobilelocality'] data = (address, city, state) cur.execute(addressQuery, data) favParkQuery = "INSERT INTO tFavoritePark (parkName, AddressID, image) VALUES (%s, LAST_INSERT_ID(), %s);" data = (FavoriteParkform.parkName.data, image) cur.execute(favParkQuery, data) db.commit() favParkID = cur.lastrowid #Insert favorite toy. image = None favToyQuery = "INSERT INTO tFavoriteToy (ToyName, image) VALUES (%s, %s);" data = (session['favToy'], image) cur.execute(favToyQuery, data) db.commit() favToyID = cur.lastrowid cursor = Database.Session() familyID = session.get('familyID') if familyID is None: familyID = cursor.query(Database.tUser.familyID).filter( Database.tUser.userID == current_user.id) for familyID in familyID: familyID = familyID.familyID # commit all the dog data into the database. dogQuery = "Insert into tDog (name, gender, breedID, fixed, age, Size, Weight, bio, image, favToyID, favParkID, familyID) VALUES (%s, %s, %s, %s,%s, %s, %s, %s,%s, %s, %s, %s)" data = (session.get('dogName'), session.get('gender'), breedID, session.get('fixed'), session.get('age'), session.get('size'), session.get('weight'), session.get('bio'), session.get('image'), favToyID, favParkID, familyID) cur.execute(dogQuery, data) db.commit() #Follow the dog sqlalch = Database.Session() Query = Database.tFollowers(dogID=cur.lastrowid, userID=current_user.id) sqlalch.add(Query) sqlalch.commit() return redirect('/dashboard') flash("please input your dogs favorite park") return render_template('/Dog/NewPark.html', FavoriteParkform=FavoriteParkform) #finally commit all the dog data into the database. dogQuery = "Insert into tDog (name, gender, breedID, fixed, age, Size, Weight, bio, image, favToyID, favParkID, familyID) VALUES (%s, %s, %s, %s,%s, %s, %s, %s,%s, %s, %s, %s)" data = (session.get('dogName'), session.get('gender'), breedID, session.get('fixed'), session.get('age'), session.get('size'), session.get('weight'), session.get('bio'), image, favToyID, favParkID, session.get('familyID')) cur.execute(dogQuery, data) db.commit() #Default to follow the dog dogID = cur.lastrowid FollowDogs(dogID) #Redirect to the dashboard return redirect('/dashboard')