def db_populate_from_seed():
session.query(Category).delete()
category1 = Category(name="Bowling")
category2 = Category(name="Tennis")
session.add(category1)
session.add(category2)
session.flush()
session.refresh(category1)
session.refresh(category2)
item1 = Item(name="HeavyBall",
description="Bowling Ball",
category_id=category1.id)
item2 = Item(name="Racket",
description="Medium Size Racket",
category_id=category2.id)
session.add(item1)
session.add(item2)
try:
session.commit()
except Exception as e:
return "Error trying seed populate db"
return "db re-populated with seed"
def addCategory():
if request.method == 'POST':
newCategory = Category(name=request.form['name'],
description=request.form['description'])
db.session.add(newCategory)
db.session.flush()
if 'file' not in request.files:
photo = 'default.jpg'
else:
file = request.files['file']
extension = file.filename.rsplit('.', 1)[1].lower()
photo = "category%s.%s" % (newCategory.id, extension)
file.filename = photo
# if user does not select file, browser also
# submit a empty part without filename
if file.filename == '':
filename = 'default.jpg'
if file and allowed_file(file.filename):
uploadImage(
file,
os.path.join(app.config['UPLOAD_FOLDER'],
secure_filename(file.filename)))
flash('New category %s Successfully Created' % newCategory.name)
newCategory.picture = photo
db.session.commit()
return redirect(url_for('home'))
else:
return render_template('addcategory.html')
def populateCategory():
books = Category(name="Books")
instruments = Category(name="Instruments")
games = Category(name="Games")
session.add(books)
session.add(instruments)
session.add(games)
session.commit()
print "Added categories!"
def newCategory():
if request.method == 'POST':
name = request.form['name']
description = request.form['description']
if validate_category_name(name) and validate_category_descrip(
description):
newCate = Category(name=name,
description=description,
user_id=login_session['user_id'])
session.add(newCate)
flash('New Category %s Successfully Created' % newCate.name)
session.commit()
return redirect(url_for('showCategories'))
else:
error = "Some inputs are not valid"
return render_template('new_category.html',
name=name,
description=description,
error=error)
else:
return render_template('new_category.html')
def addCategory():
'''
GET show form for adding category
POST add category. As we are using Category.name for routes we need to
check for uniqueness
Returns:
For a successful POST we return a redirect object to the 'main' route.
For GET or unsuccessful POST we return the addcategory.html
render_template.
'''
if request.method == 'POST':
addedCategory = Category(name=request.form['category'])
if session.query(Category).filter_by(
name=addedCategory.name).one_or_none() is None:
session.add(addedCategory)
session.commit()
return redirect(url_for('main'))
else:
flash('Category exists!')
categories = session.query(Category).all()
return render_template('addcategory.html',
categories=categories,
login_session=login_session)
def create_category():
"""Creates a new category.
The GET endpoint displays a form to create the category.
The POST endpoint creates the category and shows it afterwards.
"""
if 'name' not in login_session:
flash('You must be logged in to do that!')
return redirect(url_for('login'))
elif request.method == 'GET':
return render_template('new_cat.html')
else:
if request.form['cat_name'] == '':
flash('Invalid category name!')
return redirect(url_for('create_category'))
# A valid category name was inserted but need to check if already exists
category = session.query(Category).filter_by(
name=request.form['cat_name'].lower()).first()
if category:
flash('This category already exists!')
return redirect(url_for('show_category', category_name=category.name))
# Creating new category
new_cat = Category(name=request.form['cat_name'].lower(),
user_id=login_session['user_id'])
session.add(new_cat)
session.commit()
flash('Category successfully created!')
return redirect(url_for('show_category', category_name=new_cat.name))
def register():
form = CategoryForm(request.form)
if request.method == 'POST' and form.validate():
# auto generate category id in betwen 10 - 20000
category = Category(randint(10, 20000), form.name.data)
db_session.add(category)
return redirect('/home')
return render_template('category_create.html', form=form)
def categoryCreate():
if request.method == 'POST':
category_name = request.form['name']
category = Category(name=category_name)
session.add(category)
session.commit()
return redirect(url_for('categoryCreate'))
elif request.method == 'GET':
categories = session.query(Category).order_by(asc(Category.name)).all()
return render_template('category-create.html', categories=categories)
def addCategory():
if request.method == 'POST':
newCategory = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCategory)
session.commit()
flash('new Category is Successfully Created..!')
return redirect(url_for('showcategory'))
else:
return render_template('addcategory.html', categories=category)
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
newCategory = Category(name=request.form['name'])
session.add(newCategory)
session.commit()
flash('Category Created: %s' % newCategory.name)
return redirect(url_for('showCategories'))
else:
return render_template('newCategory.html')
def categoryCheck(form): ''' Checks if parsed category name exists, if not it creates one ''' category = session.query(Category).filter_by(name=form.category.data).first() if category == None: newCategory = Category(name=form.category.data, user_id=login_session['user_id']) session.add(newCategory) session.commit() category = session.query(Category).filter_by(name=form.category.data).first() return category
def showNewCategory():
""" add new Category """
catquery = SESSION.query(Category).order_by(desc(Category.id)).all()
if request.method == 'POST':
req = request.form
newCategory = Category(name=req['name'],
user_id=login_session['user_id'])
SESSION.add(newCategory)
catquery = SESSION.query(Category).order_by(desc(Category.id)).all()
flash("New Record Added !!!!!")
return render_template('newcategory.html', category=catquery)
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
if request.form['name']:
newCategory = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCategory)
session.commit()
return redirect(url_for('showCategories'))
return render_template('newCategory.html')
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'GET':
return render_template('new_category.html')
elif request.method == 'POST':
nova_category = Category(name=request.form['name'],
description=request.form['description'],
user_id=login_session['user_id'])
session.add(nova_category)
flash('New Wallet %s Successfully Created' % nova_category.name)
session.commit()
return redirect(url_for('showCategories'))
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
newCat = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCat)
session.commit()
return redirect('/category/' + str(newCat.id))
else:
return render_template('newcategory.html',
is_user=isUser(login_session))
def newCategory():
"""API to create a new category"""
if request.method == 'POST':
print login_session
if 'user_id' not in login_session and 'email' in login_session:
login_session['user_id'] = getUserID(login_session['email'])
newCategory = Category(name=request.form['name'])
session.add(newCategory)
session.commit()
flash("New category created!", 'success')
return redirect(url_for('showCatalog'))
else:
return render_template('new_category.html')
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
newCategory = Category(name=request.form['name'],
url_name=request.form['urlname'],
user_id=login_session['user_id'])
session.add(newCategory)
flash('New Category %s was created Successfully' % newCategory.name)
session.commit()
return redirect(url_for('showCategories'))
else:
return render_template('newCategory.html', navitems=navitems)
def addNewCategory():
# Add a new category to the catalog
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
newCategory = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCategory)
session.commit()
flash('new category %s was added to the catalog' % (newCategory.name))
return redirect(url_for('showCategory', category_id=newCategory.id))
else:
return render_template('addcategory.html', categories=getCategories())
def init():
# Create categories
cat1 = Category(name='Sports & Entertainment')
cat2 = Category(name='Arts & Theatre')
cat3 = Category(name='Adventures & Outdoors')
cat4 = Category(name='Gaming & Puzzles')
cat5 = Category(name='Food & Drinks')
cat6 = Category(name='Competitions & Marathons')
cat7 = Category(name='Comedy')
cat8 = Category(name='Recreational')
cats = [cat1, cat2, cat3, cat4, cat5, cat6, cat7, cat8]
for cat in cats:
if not session.query(Category).filter_by(name=cat.name).all():
session.add(cat)
session.commit()
def new_category():
db_session = start()
if request.method == 'POST':
category = Category(name=request.form['name'], user_id=current_user.id)
print(category.name)
db_session.add(category)
db_session.commit()
db_session.close()
flash("New Category Added")
return redirect(url_for('home'))
db_session.close()
return render_template('new-category.html', user=current_user)
def add_category(category_name, user_id):
"""
Create a category in the database
Args: category_name - String(80)
Returns: category.id - Int
"""
category = Category(name=category_name, user_id=user_id)
ses.add(category)
ses.commit()
new_category = ses.query(Category).filter_by(name=category_name).one()
return new_category.id
def add_category():
if 'username' not in login_session:
flash('Please login before attempting to add a new category')
return redirect('/login')
else:
if request.method == 'GET':
return render_template('new_category.html')
else:
newCategory = Category(
name=request.form['name'], user_id=login_session['user_id'])
session.add(newCategory)
flash('Successfully added new category: %s' %
newCategory.name, 'success')
session.commit()
return redirect(url_for('index'))
def create_category():
if request.method == 'POST':
# retrieves the user_id if login_session doesn't have it
if 'user_id' not in login_session and 'email' in login_session:
login_session['user_id'] = get_user_id(login_session)
# makes sure name field is filled out
if request.form['name']:
new_category = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(new_category)
session.commit()
flash("New category created")
return redirect(url_for("show_home"))
if request.method == 'GET':
return render_template('new_category.html')
def new_category():
if request.method == 'POST':
if request.form['name'] and request.form['desc']:
new_category = Category(name=request.form['name'],
desc=request.form['desc'],
user_id=login_session['user_id'])
session.add(new_category)
session.commit()
flash('Category {} successfully added'.format(new_category.name))
return redirect(url_for('list_categories'))
else:
flash('Category not added, there is no name or description !')
return redirect(url_for('list_categories'))
else:
return render_template('category/new.html')
def newCategory():
if request.method == 'POST':
if not login_session['state']:
return jsonify(success=0, msg='Login session state not set.')
try:
name = request.json['name']
newCat = Category(name=name, user_id=login_session['user_id'])
session.add(newCat)
session.commit()
return jsonify(success=1,
id=newCat.id,
name=name,
msg="Category " + name + " added!")
except Exception as e:
return jsonify(success=0, msg=str(e))
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
user = session.query(User).filter_by(
email=login_session['email']).one()
newCategory = Category(name=request.form['name'], user_id=user.id)
session.add(newCategory)
print newCategory.user_id
print login_session['id']
flash('New Category %s Successfully Created' % newCategory.name)
session.commit()
return redirect(url_for('showCategories'))
else:
return render_template('newcategory.html')
def newCategory():
if request.method == 'POST':
if not request.form['name']:
flash('Category must have a name.')
if request.form['name']:
if session.query(Category).filter_by(
name=request.form['name']).count() > 0:
flash('This category already exists. Please create a new one.')
else:
newCategory = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCategory)
session.commit()
flash('New Category %s Successfully Created!' %
newCategory.name)
return redirect(url_for('catalog'))
return render_template('newcategory.html')
def newCategory():
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
if request.form['name']:
newCategory = Category(name=request.form['name'],
user_id=login_session['user_id'])
session.add(newCategory)
session.commit()
flash('Your new category has been created.')
return redirect(url_for('showMain'))
else:
flash('Please enter in a name.')
return render_template('newCategory.html',
login_session=login_session)
else:
return render_template('newCategory.html', login_session=login_session)
def categories_new():
"""Add a new category."""
global google_user_id
if 'credentials' not in flask.session:
return render_template('login_check.html')
if request.method == 'POST':
if google_user_id != "":
user = google_user_id
category_new = Category(name=request.form['name'],
description=request.form['description'],
user=user)
session.add(category_new)
session.commit()
return redirect(url_for('categories_list'))
else:
return render_template('categories_new.html')
def newcategories():
'''
this function is used to add the new category
if user not login it will goto the login page
the user fill the all the details
and submit it shows the Category was successfully added to the catalog
if this function is executed the newcategories.html page is opened
'''
if 'username' not in login_session:
return redirect('/login')
if request.method == 'POST':
newcategories = Category(name=request.form['name'],
description=request.form['description'],
user_id=login_session['user_id'])
session.add(newcategories)
session.commit()
flash('Category was successfully added to the catalog')
return redirect(url_for('showcategories'))
else:
return render_template('newcategories.html')
