#!/usr/bin/env python3
"""
InvMan
"""

# Imports
from flask import Flask, request, jsonify, abort
from flask_api import status

import storage

app = Flask(__name__)  # app is the Flask app


@app.route("/api/v1/locations", methods = ["GET"])
def api_get_locations():
    return jsonify(storage.get_locations())


@app.route("/<location>/api/v1/items", methods = ["GET"])
def api_get_items(location):
    try:
        items = storage.get_items(location)
        print("Got items at {0}".format(location))
        return jsonify(items)
    except KeyError:
        print("KeyError, returning 404")
        return jsonify({'status': 'error', 'error': 'NOT_FOUND'}), status.HTTP_404_NOT_FOUND


@app.route("/<location>/api/v1/item/<item>", methods = ["GET", "DELETE"])
def api_get_item(location, item):
    if request.method == "GET":
        try:
            itemresp = storage.get_items(location)[item.lower()]
            print("Got {0}".format(item))
            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_item(location, item)
            print("Delete item {0} at {1}".format(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("/<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("/<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("/<location>/api/v1/item/<item>/<brand>/new_brand", methods = ["POST"])
def api_new_brand(location, item, brand):
    try:
        storage.init_brand(location, item, brand)
        print("Created brand {0} under item {1} at {2}".format(brand, item, location))
        return jsonify({'status': 'success'})
    except storage.AlreadyExistsError:
        return jsonify({'status': 'error', 'error': 'BRAND_ALREADY_EXISTS'}), status.HTTP_403_FORBIDDEN


@app.route("/<location>/api/v1/item/<item>/new_item", methods = ["POST"])
def api_new_item(location, item):
    try:
        storage.init_item(location, item)
        print("Created item {0} at {1}".format(item, location))
        return jsonify({'status': 'success'})
    except storage.AlreadyExistsError:
        return jsonify({'status': 'error', 'error': 'ITEM_ALREADY_EXISTS'}), status.HTTP_403_FORBIDDEN


@app.route("/<location>/api/v1/new_location", methods = ["POST"])
def api_new_location(location):
    try:
        storage.init_location(location)
        print("Created location {0}".format(location))
        return jsonify({'status': 'success'})
    except storage.AlreadyExistsError:
        return jsonify({'status': 'error', 'error': 'ITEM_ALREADY_EXISTS'}), status.HTTP_403_FORBIDDEN


@app.route("/<location>/api/v1/item/<item>/<brand>/add_amount/<amount>", methods = ["POST"])
def api_add_amount(location, item, brand, amount):
    storage.increment_amount(location, item, brand, amount)
    return jsonify({'status': 'success'})


if __name__ == "__main__":
    print("Run with `flask` or a WSGI server!")