Merge pull request #19 from fcabrera23/contoso-supermarket-source

Adding source code for Contoso Supermarket

Author: likamrat

contoso-supermarket/src/.vscode/launch.json

@@ -0,0 +1,50 @@
+{
+  // Use IntelliSense to learn about possible attributes.
+  // Hover to view descriptions of existing attributes.
+  // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "Python: Flask (0.11.x or later)",
+      "type": "python",
+      "request": "launch",
+      "module": "flask",
+      "env": {
+        "FLASK_APP": "app.py",
+        "SQL_HOST": "",
+        "SQL_USERNAME": "",
+        "SQL_PASSWORD": "",
+        "SQL_DATABASE": "",
+        "HEAD_TITLE" : "Hello World",
+        "CAMERAS_ENABLED" : "False",
+        "CAMERAS_BASEURL" : "",
+        "NEW_CATEGORY" : "True"
+      },
+      "args": [
+        "run",
+        "--no-debugger",
+        "--no-reload"
+      ]
+    },
+    {
+      "name": "Python: Current File",
+      "type": "python",
+      "request": "launch",
+      "program": "${file}",
+      "env": {
+        "FLASK_ENV": "development"
+      }
+    },
+    {
+      "name": "Python: Attach",
+      "type": "python",
+      "request": "attach",
+      "localRoot": "${workspaceFolder}",
+      "remoteRoot": "${workspaceFolder}",
+      "port": 3000,
+      "secret": "my_secret",
+      "host": "localhost"
+    }
+  ]
+}

contoso-supermarket/src/Dockerfile

@@ -0,0 +1,14 @@
+# syntax=docker/dockerfile:1
+FROM python:3.8-slim-buster
+
+RUN apt-get update
+RUN apt-get install ffmpeg libsm6 libxext6  -y
+
+WORKDIR /python-docker
+
+COPY requirements.txt requirements.txt
+RUN pip3 install -r requirements.txt
+
+COPY . .
+
+CMD [ "python3", "-m" , "flask", "run", "--host=0.0.0.0"]

contoso-supermarket/src/app.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+from importlib import import_module
+import os
+import cv2
+from flask import Flask, render_template, Response, request
+from sqlConnector import SqlConnector
+
+app = Flask(__name__)
+
+@app.route('/')
+def index():
+    """Contoso Supermarket home page."""
+    cameras_enabled = True
+    if os.environ.get('CAMERAS_ENABLED'):
+        cameras_enabled = os.environ.get('CAMERAS_ENABLED') == 'True'
+    
+    head_title = "Contoso Supermarket"
+    if os.environ.get('HEAD_TITLE'):
+        head_title = os.environ.get('HEAD_TITLE')
+
+    new_category = False
+    if os.environ.get('NEW_CATEGORY'):
+        new_category = os.environ.get('NEW_CATEGORY') == 'True'
+
+    return render_template('index2.html' if new_category else 'index.html', head_title = head_title, cameras_enabled = cameras_enabled)
+
+@app.route('/addPurchase',methods = ['POST'])
+def addPurchase():
+    content_type = request.headers.get('Content-Type')
+    if (content_type == 'application/json; charset=UTF-8'):
+        json = request.get_json()
+        sqlDb = SqlConnector()
+        successful = sqlDb.addPurchase(json['ProductId'])
+        if(successful):
+            return "Ok"
+        else:
+            return "Error processing request"
+    else:
+        return 'Content-Type not supported!'
+
+@app.route('/video_feed/<feed>')
+def video_feed(feed):
+    return Response(gen_frames(feed),
+                    mimetype='multipart/x-mixed-replace; boundary=frame')
+
+
+def gen_frames(source):
+    """Video streaming frame capture function."""
+    baseUrl = "rtsp://localhost:554/media/" 
+    if os.environ.get('CAMERAS_BASEURL'):
+        baseUrl = str(os.environ['CAMERAS_BASEURL'])
+
+    cap = cv2.VideoCapture(baseUrl + source)  # capture the video from the live feed
+
+    while True:
+        # # Capture frame-by-frame. Return boolean(True=frame read correctly. )
+        success, frame = cap.read()  # read the camera frame
+        if not success:
+            break
+        else:
+            ret, buffer = cv2.imencode('.jpg', frame)
+            frame = buffer.tobytes()
+            yield (b'--frame\r\n'
+                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')  # concat frame one by one and show result
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', threaded=True, debug=True)

contoso-supermarket/src/initDatabase.sql

@@ -0,0 +1,21 @@
+USE contososupermarket;
+
+CREATE TABLE `Stocks` (
+	`ProductID` INT NOT NULL AUTO_INCREMENT,
+	`Name` TEXT,
+	`Sold` INT DEFAULT '10000',
+	`Stock` INT zerofill,
+	PRIMARY KEY (`ProductID`)
+);
+
+INSERT INTO `Stocks`
+VALUES
+(1, "Red apple", 10000, 0),
+(2, "Banana", 10000, 0),
+(3, "Avocado", 10000, 0),
+(4, "Bread", 10000, 0),
+(5, "Milk", 10000, 0),
+(6, "Orange juice", 10000, 0),
+(7, "Chips", 10000, 0),
+(8, "Red bell pepper", 10000, 0),
+(9, "Lettuce", 10000, 0);

contoso-supermarket/src/requirements.txt

@@ -0,0 +1,9 @@
+click==8.1.3
+Flask==2.2.2
+itsdangerous==2.1.2
+Jinja2==3.1.2
+MarkupSafe==2.1.1
+Werkzeug==2.2.2
+numpy==1.22.0
+opencv-python==4.6.0.66
+mysql-connector-python==8.0.30

contoso-supermarket/src/sqlConnector.py

@@ -0,0 +1,59 @@
+import os
+import mysql.connector
+
+class SqlConnector:
+    sqlHost = ""
+    sqlUsername = ""
+    sqlPassword = ""
+    sqlDatabase = ""
+
+    def __init__(self):
+        if os.environ.get('SQL_HOST'):
+            self.sqlHost = str(os.environ['SQL_HOST'])
+
+        if os.environ.get('SQL_USERNAME'):
+            self.sqlUsername = str(os.environ['SQL_USERNAME'])
+        
+        if os.environ.get('SQL_PASSWORD'):
+            self.sqlPassword = str(os.environ['SQL_PASSWORD'])
+
+        if os.environ.get('SQL_DATABASE'):
+            self.sqlDatabase = str(os.environ['SQL_DATABASE']) 
+
+        self.connection = self.createServerConnection()
+        if(self.connection != None):
+            self.cursor = self.connection.cursor()
+        else:
+            print("Fail to establish MySQL database connection")
+
+    def createServerConnection(self):
+        connection = None
+        try:
+            connection = mysql.connector.connect(
+                user= self.sqlUsername, 
+                password= self.sqlPassword,
+                host= self.sqlHost, 
+                port=3306, 
+                database= self.sqlDatabase, 
+                ssl_disabled=True
+            )
+
+            print("MySQL Database connection successful")
+        except Exception as ex:
+            print(f"Error: '{ex}'")
+
+        return connection
+
+    def addPurchase(self, productId):
+        try:
+            query = "UPDATE `Stocks` SET `Sold` = `Sold` + 1, `Stock` = `Stock` - 1 WHERE `ProductID` = {}".format(productId)
+            self.cursor.execute(query)
+            self.connection.commit()
+            if(self.cursor.rowcount == 1):
+                return True
+            else:
+                return False
+        except Exception as ex:
+            print(f"Error: '{ex}'")
+            return False
+