fazt · July 18, 2019 01:24 · Jul 18, 2019
diff --git a/python-flask-sqlite3-restapi.py b/python-flask-sqlite3-restapi.py
@@ -0,0 +1,36 @@
+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)