Added printing images

src/main.py

@@ -18,30 +18,39 @@ import time                             # To sleep
 import os                               # For VARS from the shell.
 # Variables
 
+app = Flask(__name__)
+
 # Load the configuration file
 try:
+    app.logger.debug("Loading config file...")
     configuration_file = toml.load("configuration/config.toml")
 except TypeError :
-    print("Unable to load the config file: invalid type or is a list containing invalid types")
+    app.logger.error("Unable to load the config file: invalid type or is a list containing invalid types")
     exit(-1)
 except toml.TomlDecodeError:
-    print("An error occured while decoding the file")
+    app.logger.error("An error occured while decoding the file")
     exit(-1)
 except Exception as e:
-    print("Error while loading file : " + str(e))
+    app.logger.error("Error while loading file : " + str(e))
+    exit(-1)
 
+app.logger.debug("Config file loaded !")
 # Define the USB connections here.
 vendor_id = configuration_file["printer"]["vendor_id"]
 device_id = configuration_file["printer"]["device_id"]
+UPLOAD_FOLDER = str(configuration_file["printer"]["upload_folder"])
+
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
 # Output the config file
 if os.getenv('LIPY_DEBUG') == True:
     pprint.pprint(configuration_file)
 
 # We define the app module used by Flask
-app = Flask(__name__)
 app.secret_key = configuration_file["secrets"]["flask_secret_key"]
-
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['ALLOWED_EXTENSIONS'] = ALLOWED_EXTENSIONS
+app.config['MAX_CONTENT_LENGTH'] = 3 * 1000 * 1000 # Maximum 3Mb for a file upload
 
 # Printer connection
 # Uses the class defined in the printer.py file
@@ -61,6 +70,7 @@ limiter = Limiter(
 @app.route('/')
 @limiter.limit("1/second", override_defaults=False)
 def index():
+    app.logger.debug("Loading index")
     return render_template('index.html')
 
 # API routes
@@ -74,50 +84,86 @@ def index():
 @app.route('/api/print')
 @limiter.limit("1/second", override_defaults=False)
 def api_index():
+    app.logger.debug("Loading API")
     return render_template("api.html")
 
 
 @app.route('/api/print/sms', methods=['POST'])
 @limiter.limit("2/minute", override_defaults=False)
 def api_print_sms():
+    app.logger.debug("Printing an sms")
     try:
         txt = request.form["txt"]
         sign = request.form["signature"]
     except Exception as e:
-        flash(e)
+        flash(e,'error')
         redirect(url_for('index'))
 
     try:
         web.print_sms(txt,sign)
     except Exception as e:
         pass
-        
+
     return redirect(url_for('index'))
 
-@app.route('/api/print/image', methods=['POST'])
+@app.route('/api/print/img', methods=['POST'])
 @limiter.limit("2/minute", override_defaults=False)
 def api_print_image():
-    flash(web.print_image())
-    return "print image"
+    app.logger.debug("Printing an image")
+
+    try:
+        sign = request.form["signature"]
+    except Exception as e:
+        flash(str(e),'error')
+        redirect(url_for('index'))
+
+    if request.method == 'POST':
+        # check if the post request has the file part
+
+        if 'img' not in request.files:
+            flash('No file found. Did you use the good form ?', 'error')
+            return redirect(url_for("index"))
+        else:
+            file = request.files['img']
+
+
+        # If the user does not select a file, the browser submits an
+        # empty file without a filename.
+        if file.filename == '':
+            app.logger.error("Submitted file has no filename !")
+            flash('No file submitted, please select a file','error')
+            return redirect(url_for("index"))
+
+        try:
+            app.logger.debug("Sending the image to the printer.")
+            web.print_image(file, sign)
+        except Exception as e:
+            pass
+    else:
+        flash('Cannot access to page with this method.','error')
+        app.logger.debug('Bad access type to this API.')
+
+    return redirect(url_for("index"))
 
 @app.route('/login')
 @limiter.limit("1/second", override_defaults=False)
 def login_page():
     # web.login(username,password)
-    return render_template("index.html")
+    return redirect(url_for("index"))
 
 @app.route('/logout')
 @limiter.limit("1/second", override_defaults=False)
 def logout_page():
     # web.logout(username, password)
-    return render_template("index.html")
+    return redirect(url_for("index"))
 
 @app.errorhandler(429)
 def ratelimit_handler(e):
     flash("Rate limit reached, please slow down :) ( Currently at "+ e.description + ")", 'error')
-    return render_template('index.html')
+    return redirect(url_for("index"))
 
 @app.route("/ping")
 @limiter.exempt
 def ping():
-    return "PONG"
+    flash("🏓 Pong !",'info')
+    return redirect(url_for("index"))