def create_item():
"""Create Item with unique code"""
#if not request.json:
# abort(400)
parser = reqparse.RequestParser()
parser.add_argument('item_code', type=int, required=False, help="Item code missing")
parser.add_argument('item_name', type=str, required=True, help="Item name missing")
parser.add_argument('size', type=str, required=True, help="Size missing")
parser.add_argument('color', type=str, required=True, help="Color missing")
parser.add_argument('quality', type=str, required=True, help="Quality missing")
parser.add_argument('username', type=str, required=True, help="Username missing")
args = parser.parse_args(strict=True)
user_code = get_user_code(args['username'])
if user_code is None:
return make_response(jsonify({'error': 'User does not exists'}), 400)
new_item = dict(
item_code = args['item_code'],
item_name = args['item_name'],
size_code = get_size_code( args['size']),
color_code = get_color_code( args['color']),
quality_code = get_quality_code( args['quality'])
)
try:
u = models.Items(**new_item)
db.session.add(u)
db.session.commit()
except sqlalchemy.exc.IntegrityError, e:
return make_response(jsonify({'error': 'item code already exists.'}), 400)
def test_edititem_succesful(self):
'''Tests if edit functionality works'''
item = models.Items()
item.create_item('task')
item.edit_item('task', 'task1')
edit_not_working = False
if 'task' in models.items.get(models.currentbucket[0]):
edit_not_working = True
self.assertEqual(edit_not_working, False)
def test_deleteitem_succesful(self):
'''Tests if delete functionality works'''
item = models.Items()
item.create_item('do assignment')
item.delete_item('do assignment')
delete_not_working = False
if 'do assignment' in \
models.items.get(models.currentbucket[0]):
delete_not_working = True
self.assertEqual(delete_not_working, False)
def test_items_duplicateitems(self):
'''Test to check if duplicate list item can be added'''
item = models.Items()
item.create_item('swim')
item.create_item('swim')
duplicate_exists = True
if len(list(set(models.items.keys())))\
== len(models.items.keys()):
duplicate_exists = False
self.assertEqual(duplicate_exists, False)
def test_additem_succesful(self):
'''Tests if add functionality works'''
bkt = models.BucketLists()
bkt.current_bucket('bucket1')
item = models.Items()
item.create_item('raft')
add_item_works = False
if 'raft' in models.items.get(models.currentbucket[0]):
add_item_works = True
self.assertEqual(add_item_works, True)
from flask import Flask
from flask import render_template, request, redirect, flash, url_for
from app import forms
from app import models
app = Flask(__name__)
app.config.from_object('config')
USER = models.User()
BUCKETLIST = models.BucketLists()
ITEM = models.Items()
@app.route('/')
def home():
'''index page function'''
return redirect(url_for('register'))
@app.route('/logout')
def logout():
USER.logout()
return redirect(url_for('login'))
@app.route('/login', methods=['POST', 'GET'])
def login():
'''Login function'''
form = forms.LoginForm(request.form)
if request.method == 'POST':