from flask import Flask, request
from flask_restful import Resource, Api
from sqlalchemy import create_engine
from flask_jsonpify import jsonify

db_connect = create_engine('sqlite:///database.db')
app = Flask(__name__)
api = Api(app)

class Sellers(Resource):
    def get(self):
        conn = db_connect.connect()
        query = conn.execute("select * from sellers")
        # fetch first column (product IDs)
        return {'products': [i[0] for i in query.cursor.fetchall()]}

class Products(Resource):
    def get(self):
        conn = db_connect.connect()
        query = conn.execute("select * from products")
        result = {'products': [dict(zip(tuple(query.keys()), i)) for i in query.cursor]}
        return jsonify(result)

class OneSeller(Resource):
    def get(self, seller_id):
        conn = db_connect.connect()
        query = conn.execute("select * from sellers where id=%d" %int(seller_id))
        result = {'product': [dict(zip(tuple(query.keys()), i)) for i in query.cursor]}
        return jsonify(result)

api.add_resource(Sellers, '/api/sellers')
api.add_resource(Products, '/api/products')
api.add_resource(OneSeller, '/api/sellers/<seller_id>')

if __name__ == '__main__':
    app.run(port=4000)