def setUp(self):
db.create_all()
user1 = User(username='******', password='******')
user2 = User(username='******', password='******')
user3 = User(username='******', password='******')
puppy1 = Puppy('whiskey', 2)
puppy2 = Puppy('moxie', 3)
db.session.add_all([user1, user2, user3, puppy1, puppy2])
db.session.commit()
def post(self, user_id):
parser = reqparse.RequestParser()
parser.add_argument('name', type=str, help='Name')
args = parser.parse_args()
new_puppy = Puppy(args['name'], user_id)
db.session.add(new_puppy)
db.session.commit()
return new_puppy
def add():
form = Addform()
if form.validate_on_submit():
new_puppy = Puppy(form.name.data)
db.session.add(new_puppy)
db.session.commit()
return redirect(url_for('puppy.list'))
return render_template('add.html', form=form)
def addpuppy():
form = AddForm()
if form.validate_on_submit():
name = form.name.data
# No ID checker since there can be multiple puppies with the same name.
add_pup = Puppy(name)
db.session.add(add_pup)
db.session.commit()
flash("Successfully added new puppy!", 'success')
return redirect(url_for('puppies.listpuppy'))
return render_template('addpuppy.html', form=form)
def add():
form = AddForm()
if form.validate_on_submit():
name = form.name.data
if name != None and name != "":
exists = Puppy.query.filter_by(name=name).first()
if exists == None:
new_pup = Puppy(name)
db.session.add(new_pup)
db.session.commit()
return redirect(url_for('puppies.list'))
else:
return render_template("error.html",
error="Puppy is already in system.")
else:
return render_template("error.html",
error="Please enter puppy name.")
return render_template("add.html", form=form)