flask-rest-api-product-demo-app-1

from flask import Flask, request, jsonify
from flask_sqlalchemy import SQLAlchemy
from flask_marshmallow import Marshmallow
import os


app = Flask(__name__)
basedir = os.path.abspath(os.path.dirname(__file__))

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'db.sqlite')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)

ma= Marshmallow(app)

# @app.route('/', methods=['GET'])
# def get():
#   return jsonify({'msg': 'hello world!'})


class Product(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), unique=True)
    description = db.Column(db.String(200))
    price = db.Column(db.Float)
    qty = db.Column(db.Integer)

    def __init__(self, name, description, price, qty):
        self.name = name
        self.description = description
        self.price = price
        self.qty = qty

    def __repr__(self):
        return "Product, Name: %s and desc: %s" % (self.name, self.description)


class ProductSchema(ma.Schema):
    class Meta:
          fields = ('id','name','description','price','qty')


product_schema = ProductSchema()
products_schema = ProductSchema(many=True)

@app.route('/product', methods=['POST'])
def add_product():

    name=request.json.get('name')
    description = request.json.get('description')
    price=request.json.get('price')
    qty=request.json.get('qty')

    new_product = Product(name=name, description=description, price=price, qty=qty)

    db.session.add(new_product)
    db.session.commit()

    return product_schema.jsonify(new_product)

@app.route('/product', methods=['GET'])
def get_products():
    all_products = Product.query.all()
    result = products_schema.dump(all_products)
    print(result)
    return jsonify(result)


@app.route('/product/<string:id>', methods=['GET'])
def get_product(id):
    product = Product.query.get(id)
    return product_schema.jsonify(product)


@app.route('/product/<string:id>', methods=['PUT'])
def update_product(id):

    product = Product.query.get(id)

    name=request.json.get('name')
    description = request.json.get('description')
    price=request.json.get('price')
    qty=request.json.get('qty')

    product.name = name
    product.description = description
    product.price = price
    product.qty = qty

    db.session.commit()

    return product_schema.jsonify(product)

@app.route('/product/<string:id>', methods=['DELETE'])
def delete_product(id):
    product = Product.query.get(id)
    db.session.delete(product)
    db.session.commit()

    product = Product.query.get(id)
    return product_schema.jsonify(product)

if __name__ == '__main__':
    app.run(debug=True)