main.py

# -*- coding: utf-8 -*-
"""
Created on Thu Apr 16 13:32:08 2020

@author: ingvord
"""


from flask import Flask, request, g
from flask_restful import Api
from picontroller import PiController, PiControllerServoMode, PiControllerReference, PiControllerPosition, \
    PiControllerStop, PiControllerReboot

from pi_device import create_pi_device as internal_create_pi_device
from elasticapm.contrib.flask import ElasticAPM


app = Flask(__name__)

#TODO move to env
app.config['ELASTIC_APM'] = {
    # Set the required service name. Allowed characters:
    # a-z, A-Z, 0-9, -, _, and space
    'SERVICE_NAME': 'axsis-rest',

    # Use if APM Server requires a secret token
    'SECRET_TOKEN': '',

    # Set the custom APM Server URL (default: http://localhost:8200)
    'SERVER_URL': 'http://apm-server:8200',

    # Set the service environment
    'ENVIRONMENT': 'production',
}
apm = ElasticAPM(app, logging=True)

api = Api(app)

api.add_resource(PiController, '/axsis/controllers/<int:id>')
api.add_resource(PiControllerServoMode, '/axsis/controllers/<int:id>/servo')
api.add_resource(PiControllerReference, '/axsis/controllers/<int:id>/reference')
api.add_resource(PiControllerPosition, '/axsis/controllers/<int:id>/position')
api.add_resource(PiControllerReboot, '/axsis/controllers/<int:id>/reboot')
api.add_resource(PiControllerStop, '/axsis/controllers/<int:id>/stop')

@app.route('/')
def iamok():
    return "iamok"

@app.before_request
def create_pi_device():
    host = request.args.get('ip', default="127.0.0.1")
    port = int(request.args.get('port', default=50000))

    request.pi_device = internal_create_pi_device(host, port)


@app.teardown_request
def destroy_pi_device(response):
    # request.pi_device.CloseConnection()
    return response