def update_supplier(self, supplier_id, supplier):
if not self.get_supplier_by_id(supplier_id):
return ErrorHandler().not_found()
try:
supplier_name = supplier["customer_first_name"]
supplier_city = supplier["customer_city"]
latitude = supplier["latitude"]
longitude = supplier["longitude"]
except KeyError:
ErrorHandler().bad_request()
if supplier_name and supplier_city and latitude and longitude:
supplier_id, location_id = SupplierDAO().update_supplier(
supplier_id,
supplier_name,
supplier_city,
)
LocationDAO().update_location(location_id, latitude, longitude)
return (
self.build_supplier((supplier_id, supplier_name,
supplier_city, location_id)),
200,
)
else:
return ErrorHandler().bad_request()
else:
return ErrorHandler().bad_request()
def search_suppliers(self, supplier):
try:
supplier_city = supplier["customer_city"]
except KeyError:
ErrorHandler().bad_request()
if supplier_city:
supplier_list = SupplierDAO().get_suppliers_by_city(
supplier_city)
result_list = []
for row in supplier_list:
result = self.build_supplier(row)
result_list.append(result)
return jsonify(suppliers=result_list)
else:
return ErrorHandler().bad_request()
def insert_supplier(self, supplier):
try:
username = supplier["username"]
password = supplier["password"]
phone = supplier["phone"]
supplier_name = supplier["supplier_name"]
supplier_city = supplier["supplier_city"]
latitude = supplier["latitude"]
longitude = supplier["longitude"]
except KeyError:
ErrorHandler().bad_request()
location_id = LocationDAO().insert_location(latitude, longitude)
supplier_id = SupplierDAO().insert_supplier(username, password, phone,
supplier_name,
supplier_city, location_id)
return (
self.build_supplier(
(supplier_id, supplier_name, supplier_city, location_id)),
201,
)
def insert_supplies_product_by_supplier_id(self, customer_id, product_id):
result = SupplierDAO().insert_supplies_product_by_supplier_id(
customer_id, product_id)
if not result:
return ErrorHandler().not_found()
return jsonify(SupplierProducts=result)
def get_supplier_location(self, supplier_id):
result = SupplierDAO().get_supplier_location(supplier_id)
if not result:
return ErrorHandler().not_found()
return jsonify(ProductsSupplier=result)
def get_supplier_by_id(self, supplier_id):
result = SupplierDAO().get_supplier_by_id(supplier_id)
if not result:
return ErrorHandler().not_found()
return jsonify(supplier=result), 200
def get_all_suppliers(self):
result = SupplierDAO().get_all_suppliers()
return jsonify(supplier=result), 200
def delete_supplier(self, supplier_id):
if not self.get_supplier_by_id(supplier_id):
return ErrorHandler().not_found()
else:
SupplierDAO().delete_supplier(supplier_id)
return jsonify(Deletion="OK"), 200