2020-10-02 night

This commit is contained in:
BBaoVanC 2020-10-02 22:16:01 -05:00
parent 1ce7954476
commit 5a44b65bf3
No known key found for this signature in database
GPG Key ID: 6D74C8B0E7D791C2

189
webapi.py
View File

@ -55,7 +55,11 @@ def get_locations():
with db.cursor() as cur:
cur.execute("SELECT name,description FROM location")
data = cur.fetchall()
print(data)
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
print("ran SELECT name,description FROM location")
data2 = {}
for row in data:
data2[row[0]] = {}
@ -63,49 +67,104 @@ def get_locations():
# data2 = []
# for row in data:
# data2.append({'name': row[0], 'description': row[1]})
print("processed data:")
print(data2)
return jsonify(data2)
@app.route("/api/v1/location/<location>/current_quantities", methods = ["GET"])
@app.route("/api/v1/location/name/<search>", methods = ["GET"])
def api_get_location_information(search):
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT name,description FROM location WHERE name = %s", (search,))
data = cur.fetchall()
print("ran SELECT name,description FROM location WHERE name = %s")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
data2 = {}
row = data[0]
data2['name'] = row[0]
data2['description'] = row[1]
print("processed data:")
print(data2)
return jsonify(data2)
@app.route("/api/v1/location/<location>/quantities", methods = ["GET"])
def api_get_current_quantities(location):
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT product_upc,quantity FROM product_quantity WHERE location = %s", (location,))
data = cur.fetchall()
print(data)
print("ran SELECT product_upc,quantity FROM product_quantity WHERE location = %s")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
data2 = {}
for row in data:
data2[row[0]] = row[1]
print("processed data:")
print(data2)
return jsonify(data2)
@app.route("/api/v1/location/<location>/current_quantity/<upc>", methods = ["GET"])
def api_get_quantity_of_product_in_location(location, upc):
@app.route("/api/v1/location/<location>/quantity/<searchmethod>/<search>", methods = ["GET"])
def api_get_quantity_of_product_in_location(location, searchmethod, search):
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT quantity FROM product_quantity WHERE location = %s AND product_upc = %s", (location, upc))
methodmap = {'upc': 'product_upc', 'name': 'name'}
if searchmethod in methodmap.keys():
method = methodmap[searchmethod]
cur.execute("SELECT quantity FROM product_quantity WHERE location = %s AND {0} = %s".format(method), (location, search))
data = cur.fetchall()
print("ran SELECT quantity FROM product_quantity WHERE location = %s AND {0} = %s".format(method))
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
return jsonify(data[0][0])
data2 = data[0][0]
print("processed data:")
print(data2)
return jsonify(data2)
else:
print("invalid search method, sending 400")
return jsonify({'error': 'INVALID_SEARCH_METHOD'}), status.HTTP_400_BAD_REQUEST
@app.route("/api/v1/product/<product_upc>", methods = ["GET"])
def api_get_product_information(product_upc):
@app.route("/api/v1/products", methods = ["GET"])
def api_get_products():
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT upc,brand,name,size,sizeunit,description FROM product WHERE upc = %s", (product_upc,))
cur.execute("SELECT upc,brand,name,size,sizeunit,description FROM product")
data = cur.fetchall()
print("ran SELECT upc,brand,name,size,sizeunit,description FROM product")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
data2 = {}
for row in data:
data2[row[0]] = {'brand': row[1], 'name': row[2], 'size': row[3], 'sizeunit': row[4], 'description': row[5]}
print("processed data:")
print(data2)
return jsonify(data2)
@app.route("/api/v1/product/<searchmethod>/<search>", methods = ["GET"])
def api_get_product_information(searchmethod, search):
db = get_db()
with db.cursor() as cur:
if searchmethod in ['name', 'upc']:
cur.execute("SELECT upc,brand,name,size,sizeunit,description FROM product WHERE {0} = %s".format(searchmethod), (search,))
data = cur.fetchall()
print("ran SELECT upc,brand,name,size,sizeunit,description FROM product WHERE {0} = %s")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
row = data[0]
print(row)
data2 = {}
data2['upc'] = row[0]
@ -115,40 +174,92 @@ def api_get_product_information(product_upc):
data2['sizeunit'] = row[4]
data2['description'] = row[5]
print("processed data:")
print(data2)
return jsonify(data2)
else:
print("invalid search method, sending 400")
return jsonify({'error': 'INVALID_SEARCH_METHOD'}), status.HTTP_400_BAD_REQUEST
@app.route("/api/v1/brands", methods = ["GET"])
def api_list_brands():
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT name,description FROM brand")
data = cur.fetchall()
print("ran SELECT name,description FROM brand")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
data2 = {}
for row in data:
data2[row[0]] = {'description': row[1]}
print("processed data:")
print(data2)
return jsonify(data2)
# @app.route("/<location>/api/v1/item/<item>/<brand>", methods = ["GET", "DELETE"])
# def api_item_brand(location, item, brand):
# if request.method == "GET":
# try:
# itemresp = storage.get_items(location)[item.lower()][brand.lower()]
# print("Got {0} of brand {1} at {2}".format(item, brand, location))
# return jsonify(itemresp)
# except KeyError:
# print("KeyError, returning 404")
# return jsonify({'status': 'error', 'error': 'NOT_FOUND'}), status.HTTP_404_NOT_FOUND
# elif request.method == "DELETE":
# try:
# storage.rm_brand(location, item, brand)
# print("Deleted {0} of brand {1} at {2}".format(item, brand, location))
# return jsonify({'status': 'success'})
# except KeyError:
# print("KeyError, returning 404")
# return jsonify({'status': 'error', 'error': 'NOT_FOUND'}), status.HTTP_404_NOT_FOUND
@app.route("/api/v1/brand/name/<search>", methods = ["GET"])
def api_get_brand_by_name(search):
db = get_db()
with db.cursor() as cur:
cur.execute("SELECT name,description FROM brand WHERE name = %s", (search,))
data = cur.fetchall()
print("ran SELECT name,description FROM brand WHERE name = %s")
if len(data) <= 0:
print("no results returned, sending 404")
return jsonify({'error': 'NO_RESULTS_RETURNED'}), status.HTTP_404_NOT_FOUND
else:
data2 = {}
for row in data:
data2[row[0]] = {'description': row[1]}
print("processed data:")
print(data2)
return jsonify(data2)
# @app.route("/<location>/api/v1/item/<item>/<brand>/<field>", methods = ["DELETE"])
# def api_rm_brand_field(location, item, brand, field):
# try:
# storage.rm_brand_field(location, item, brand, field)
# print("Deleted field {0} from brand {1} of item {2} at {3}".format(field, brand, item, location))
# return jsonify({'status': 'success'})
# except KeyError:
# print("KeyError, returning 404")
# return jsonify({'status': 'error', 'error': 'NOT_FOUND'}), status.HTTP_404_NOT_FOUND
@app.route("/api/v1/create_location", methods = ["POST"])
def api_create_location():
db = get_db()
with db.cursor() as cur:
if "name" not in request.form:
return jsonify({'error': 'NO_NAME_PROVIDED'}), status.HTTP_400_BAD_REQUEST
else:
locname = request.form['name']
if "description" in request.form:
locdesc = request.form['description']
else:
locdesc = None
print(locname, locdesc)
cur.execute("INSERT INTO location (name, description) VALUES (%s, %s)", (locname, locdesc))
print("ran INSERT INTO location (name, description) VALUES (%s, %s)")
db.commit()
print("committed changes to database")
return jsonify({'api_endpoint': '/api/v1/location/name/{0}'.format(locname)})
@app.route("/api/v1/create_brand", methods = ["POST"])
def api_create_brand():
db = get_db()
with db.cursor() as cur:
if "name" not in request.form:
return jsonify({'error': 'NO_NAME_PROVIDED'}), status.HTTP_400_BAD_REQUEST
else:
locname = request.form['name']
if "description" in request.form:
locdesc = request.form['description']
else:
locdesc = None
print(locname, locdesc)
cur.execute("INSERT INTO brand (name, description) VALUES (%s, %s)", (locname, locdesc))
print("ran INSERT INTO brand (name, description) VALUES (%s, %s)")
db.commit()
print("committed changes to database")
return jsonify({'api_endpoint': '/api/v1/brand/name/{0}'.format(locname)})
# @app.route("/<location>/api/v1/item/<item>/<brand>/new_brand", methods = ["POST"])