Projet

fastprod_backend/fastprod/api.py

@@ -0,0 +1,106 @@
+from flask import Flask,jsonify,request,abort, make_response
+from werkzeug.exceptions import HTTPException
+from nornir import InitNornir
+
+
+import json
+
+ALLOWED_EXTENSIONS = {'conf'}
+UPLOAD_FOLDER = 'fastprod/upload_files/'
+
+
+def init_nornir():
+    app.config['nr'] = InitNornir(config_file="fastprod/inventory/config.yaml")
+    app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+def allowed_file(filename):
+    return '.' in filename and \
+    filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+app = Flask(__name__)
+from services.devices import ( get_inventory, add_device,get_device_by_name,delete_device,get_device_interfaces,get_device_interfaces_ip,get_device_technical_info)
+from services.config import (get_config_by_device)
+from services.tasks import (run_show_commands_by_device,run_config_commands_by_device)
+init_nornir()
+
+@app.route("/")
+def hello_world():
+    return jsonify({
+        "env": "DEV",
+        "name": "fastprod_backend",
+        "version": 1.0
+    })
+
+@app.route("/devices", methods=['GET', 'POST'])
+def devices():
+    if request.method == 'GET':
+        init_nornir()
+        devices = get_inventory()
+        return jsonify(devices=devices, total_count=len(devices))
+    if request.method == 'POST':
+        data = request.get_json()
+        new_device = add_device(data)
+        return jsonify(device=new_device)
+
+@app.route("/devices/<device_name>", methods=['GET', 'DELETE'])
+def device_by_name(device_name):
+    if request.method == 'GET':
+        device = get_device_by_name(device_name)
+        return jsonify(device=device)
+    if request.method == 'DELETE':
+        device = get_device_by_name(device_name)
+        delete_device(device)
+        return jsonify(message="Device deleted")
+
+
+@app.route("/devices/<device_name>/interfaces", methods=['GET'])
+def get_interfaces(device_name):
+    device = get_device_by_name(device_name)
+    interfaces = get_device_interfaces(device)
+    return jsonify(interfaces=interfaces)
+
+@app.route("/devices/<device_name>/interfaces/ip", methods=['GET'])
+def get_interfaces_ip(device_name):
+    if request.method == 'GET':
+        device = get_device_by_name(device_name)
+        interfaces_ip = get_device_interfaces_ip(device)
+        return jsonify(interfaces_ip=interfaces_ip)
+
+@app.route("/devices/<device_name>/facts", methods=['GET'])
+def get_technical_info(device_name):
+    if request.method == 'GET':
+        device = get_device_by_name(device_name)
+        technical_info = get_device_technical_info(device)
+        return jsonify(interfaces_ip=technical_info)
+
+@app.route("/devices/<device_name>/config", methods=['GET','POST'])
+def get_config(device_name):
+    device = get_device_by_name(device_name)
+    if request.method == 'GET':
+        config = get_config_by_device(device)
+        return jsonify(interfaces_ip=config)
+    if request.method == 'POST':
+        if request.get_json().get('mode') == 'enable':
+            commands = request.get_json().get('commands')
+            result = run_show_commands_by_device(device, commands=commands)
+            return jsonify(result=result)
+        if request.get_json().get('mode') == 'config':
+            commands = request.get_json().get('commands')
+            result = run_config_commands_by_device(device, commands=commands)
+            return jsonify(result=result, commands=commands)
+
+
+
+@app.errorhandler(HTTPException)
+def handle_exception(e):
+    """Return JSON instead of HTML for HTTP errors."""
+    # start with the correct headers and status code from the error
+    response = e.get_response()
+    # replace the body with JSON
+    response.data = json.dumps({
+    "code": e.code,
+    "name": e.name,
+    "description": e.description,
+    })
+    response.content_type = "application/json"
+    return response