run_server.old.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-


# --------------------------------------------------------------------
# rDuino-Compiler-Uploader-Server : Mini server to compile and upload
# a code to the target board.
#
# This program serves HTTP requests on port 888
# Compile and upload the code using the arduino IDE (linux)
#
# Compile and upload example: arduino --board arduino:avr:nano:cpu=atmega168 --port /dev/ttyACM0 --upload /path/to/sketch/sketch.ino
# See here: https://github.com/arduino/Arduino/blob/ide-1.5.x/build/shared/manpage.adoc
#
# --------------------------------------------------------------------

import os
import subprocess
from datetime import datetime, date, time

# from http://flask.pocoo.org/ 
from flask import Flask, abort, redirect, url_for, request, render_template,  Markup,  make_response,  session,  escape,  jsonify

# Configuration data
#myPort = 888
#myPort = 5000
myPort = 5005
#myPort = 1020

#myArduinoExe = "arduino_debug.exe" # Windows
#myArduinoExe = "Arduino.app/Contents/MacOS/Arduino" # MAC
myArduinoExe = "export DISPLAY=:0.0 && arduino " # Linux

myCompileAndUploadOption = "--upload"
myVerify = "--verify"
myInstallLibrary = "--install-library"
myInstallBoard = "--install-boards"

myBoardOptions = "--board"
boardList = [ "arduino:avr:uno",  "arduino:avr:mega"]
#myBoard = "arduino:avr:nano:cpu=atmega168"
#myBoard = "arduino:avr:mega"
myBoard = "arduino:avr:uno"

myTargetOption = "--port"
#targetList = ["COM1","COM2","COM3","COM4","COM5","COM6"]
targetList = ["ttyUSB0", "ttyUSB1", "ttyUSB2", "ttyUSB3", "ttyACM0", "ttyACM1", "ttyACM2", "ttyACM3"]
#myTarget = "COM3"
myTargetRoot="/dev/"
myTarget = "ttyUSB0"
#myTarget = "ttyACM0"

#myOtherOptions = "--verbose-upload"
myOptionList = ["", "--verbose-upload",  "--verbose-build",  "--verbose",  "--preserve-temp-files"]
myOption = ""

#separator = "\"  # Windows
separator = "/"  # Linux
myTempDirectory = "/tmp/uploaded_file"
#myTempDirectory = "/tmp"
#myTempDirectory = "/home/admin"
#myTempDirectory = "~"

myCmd = ""
theResult = ""
theError = ""
theReturnCode = 0
compileTime = datetime.now()
myFileName = "uploaded_file.ino"

myProc = None


# Define the main application
app = Flask(__name__)

# Define the routes

# Install a new library in the Arduino IDE
@app.route('/install_library', methods=['GET', 'POST'])
def install_library():
    global theResult
    global theError
    global theReturnCode
    global myCmd

    if request.method == 'POST':
        myCmd = ""
        theResult = ""
        theError  =""
        theReturnCode = 0        
        theLibrary = request.form['library']
        # arduino --install-library "Bridge:1.0.0"
        myCmd = myArduinoExe+" "+myInstallLibrary+" "+theLibrary
        print("%s ..." % myCmd)
        #theResult = os.system(myCmd)
        proc = subprocess.Popen(myCmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
        (out, err) = proc.communicate()
        result = out.decode(encoding='UTF-8')
        theResult = result.replace("\n","<br/>")
        error = err.decode(encoding='UTF-8')
        theError = error.replace("\n","<br/>")
        
        print(" Done. Result:%s\n" % result)
    return render_template('install_library.html', cmd=myCmd, result=theResult, error=theError)


# Install a new baord in the Arduino IDE
@app.route('/install_board', methods=['GET', 'POST'])
def install_board():
    global theResult
    global theError
    global theReturnCode
    global myCmd

    if request.method == 'POST':
        myCmd = ""
        theResult = ""
        theError = ""
        theReturnCode = 0        
        theBoard = request.form['board']
        # arduino --install-boards "arduino:sam"
        myCmd = myArduinoExe+" "+myInstallBoard+" "+theBoard
        print("%s ..." % myCmd)
#        theResult = os.system(myCmd)
        proc = subprocess.Popen(myCmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
        (out, err) = proc.communicate()
        result = out.decode(encoding='UTF-8')
        theResult = result.replace("\n","<br/>")
        error = err.decode(encoding='UTF-8')
        theError = error.replace("\n","<br/>")
        
        print(" Done. Result:%s\n" % result)
    return render_template('install_board.html',  theBoardList=boardList, theBoard=myBoard, cmd=myCmd, result=theResult, error=theError)

# Define the target address
@app.route('/set_target', methods=['GET', 'POST'])
def set_target():
    global myTarget
    global theResult
    global theError
    global theReturnCode

    if request.method == 'POST':
        myTarget = request.form['target']
        print("Taget set to:[%s]\n" % (myTargetRoot + myTarget))
        theResult = ""
        theError = ""
        theReturnCode = 0        
    return redirect('/')
    
# Define the board
@app.route('/set_board', methods=['GET', 'POST'])
def set_board():
    global myBoard
    global theResult
    global theError
    global theReturnCode

    if request.method == 'POST':
        myBoard = request.form['board']
        print("Board set to:[%s]\n" % myBoard)
        theResult = ""
        theError = ""
        theReturnCode = 0        
    return redirect('/')
    
# Define the board
@app.route('/set_option', methods=['GET', 'POST'])
def set_option():
    global myOption
    global theResult
    global theError
    global theReturnCode

    if request.method == 'POST':
        myOption = request.form['option']
        print("Option set to:[%s]\n" % myOption)
        theResult = ""
        theError = ""
        theReturnCode = 0        
    return redirect('/')

@app.route('/get_result')
def get_result():
    global theResult
    global compileTime
    global theReturnCode
    global myProc
    
    if myProc:
        for line in myProc.stdout:
            #out = str(line.rstrip())
            result = line.decode('utf-8')
            theResult = theResult + result.replace("\n","<br/>")
            print("   ++ Result:[%s]\n" % result)
        #myProc.stdout.flush()
       
    theTime = datetime.now()
#    print("result: %s\n" % theTime.strftime("%A %d %B %Y %H:%M:%S") + theResult)
#    print("compile time : %s\n" % compileTime.strftime("%A %d %B %Y %H:%M:%S"))
    return jsonify(date=theTime.strftime("%A %d %B %Y %H:%M:%S"),  compileTime =  compileTime.strftime("%A %d %B %Y %H:%M:%S"), result=theResult,   returnCode=theReturnCode)
#    return time.strftime("%A %d %B %Y %H:%M:%S") + theResult

@app.route('/get_error')
def get_error():
    global theError
    global myProc
    global theReturnCode

    if myProc:
        for line in myProc.stderr:
            #err = str(line.rstrip())
            error = line.decode('utf-8')
            theError = theError + error.replace("\n","<br/>")
            print("   ++ Error:[%s]\n" % error)
        #myProc.stderr.flush()    

        status = myProc.poll()
        if status is not None: # End of subprocess
            if (theError.find("can't open") >= 0):
                theReturnCode = -2 # Error
            elif (theError.find("stk500_getsync") >= 0):
                theReturnCode = -2 # Error
            else:
                theReturnCode = 2 # Good
            
    return jsonify(error = theError,  returnCode=theReturnCode)
#    return theError
    
# Main page, and process code compile and upload requests
@app.route('/', methods=['GET', 'POST'])
def main_page():
    global theResult
    global theError
    global theReturnCode
    global myCmd
    global myFileName
    global compileTime
    global myProc
    
    if request.method == 'POST':

        theResult = ""
        theError = ""
        theReturnCode = 0
        theCode = ""
        myCmd = ""
        myFileName = "uploaded_file.ino"
        
        tmp = request.data
        theCode = tmp.decode(encoding='UTF-8')
        theFileName = myTempDirectory + separator + myFileName
        print("The code:\n%s\n" % theCode)
        print("Try to save the code to a local file %s\n" % theFileName)

        # Write the code to a temp file
        try:
            f = open(theFileName, "w")
            print("Trying...\n")
        except err:
            print("Unable to open file for writing: IOError: %s\n" % err)
        else:
            try:
                print("Writing...\n")
                f.write(theCode)
                print("Code written to %s\n" % theFileName)
#            except IOError, err:
            except err:
                print("Unable to write in the file : %s\n" % err)
            finally:
                f.close()
                print("File closed.\n")


        # arduino --board arduino:avr:nano:cpu=atmega168 --port /dev/ttyACM0 --upload /path/to/sketch/sketch.ino
        compileTime = datetime.now()
        myCmd = myArduinoExe+" "+myBoardOptions+" "+myBoard+" "+myTargetOption+" "+myTargetRoot+myTarget+" "+myOption+" "+myCompileAndUploadOption+" "+myTempDirectory+separator+myFileName
        print("\nThe shell command:\n%s\n" % myCmd)

#        theResult = os.system(myCmd)
        myProc = subprocess.Popen(myCmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
#        myProc = subprocess.Popen(myCmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        
        # Subprocess blocking
        #(out, err) = myProc.communicate()
        
        # Non blocking
        theResult = myCmd + "<br/><br/>"
            
        if myProc:
            for line in myProc.stdout:
                #out = str(line.rstrip())
                result = line.decode('utf-8')
                theResult = theResult + result.replace("\n","<br/>")
            print("-- Result:[%s]\n" % theResult)
            #myProc.stdout.flush()
            for line in myProc.stderr:
                #err = str(line.rstrip())
                error = line.decode('utf-8')
                theError = theError + error.replace("\n","<br/>")
            print("-- Error:[%s]\n" % theError)
            #myProc.stderr.flush()  
            
#        result = out.decode(encoding='UTF-8')
#        theResult = result.replace("\n","<br/>")
#        error = err.decode(encoding='UTF-8')
#        theError = error.replace("\n","<br/>")

        print("\nThe output of the compiler-linker-uploader:\n%s\n" % theError)
        print("\nThe errors :\n%s\n" % theError)
        print("\nThe Return Code :\n%s\n" % theReturnCode)
        print(" Done.\n")
    
    return render_template('main.html', thePort=myPort, theBoardList=boardList, theBoard=myBoard, theTargetList=targetList, theTarget=myTarget, theOptionList=myOptionList, theOption=myOption, theTempFile=myTempDirectory+separator+myFileName, result=theResult, error=theError)



if __name__ == '__main__':

    # Create the temp file directory if needed
    if not(os.path.isdir(myTempDirectory)):
        os.system("mkdir "+myTempDirectory)
        
    # Start the main server
    app.run(port=myPort)