78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
from flask import Flask, request, jsonify
|
|
import mysql.connector
|
|
from queries import *
|
|
app = Flask(__name__)
|
|
|
|
@app.route('/')
|
|
def index():
|
|
node = request.args.get('node', default = None)
|
|
dataType = request.args.get('dataType', default = None)
|
|
MAC = request.args.get('MAC', default = None)
|
|
return getData(node, dataType, MAC)
|
|
|
|
@app.route('/putData', methods=['PUT'])
|
|
def putData():
|
|
node_id = request.json.get('node_id', None)
|
|
new_name = request.json.get('new_name', None)
|
|
new_location = request.json.get('new_location', None)
|
|
|
|
if node_id is None:
|
|
return jsonify({"error": "node_id is required"}), 400
|
|
|
|
mydb = mysql.connector.connect(
|
|
host="localhost",
|
|
user="root",
|
|
password="Dingleberries69!",
|
|
database="NodeData"
|
|
)
|
|
|
|
cursor = mydb.cursor()
|
|
|
|
if new_name is not None:
|
|
cursor.execute("UPDATE Node SET Name = %s WHERE NodeID = %s", (new_name, node_id))
|
|
mydb.commit()
|
|
|
|
if new_location is not None:
|
|
cursor.execute("UPDATE Node SET Location = %s WHERE NodeID = %s", (new_location, node_id))
|
|
mydb.commit()
|
|
|
|
cursor.close()
|
|
mydb.close()
|
|
|
|
return jsonify({"message": "Node updated successfully"}), 200
|
|
|
|
def putData(node, name, location, MAC):
|
|
mydb = loginDB()
|
|
query = get_query(node, name, location, MAC)
|
|
cursor = mydb.cursor(dictionary=True) # Enable dictionary output
|
|
cursor.execute(query)
|
|
result = cursor.fetchall() # Fetch the results
|
|
cursor.close()
|
|
mydb.close()
|
|
|
|
return result
|
|
|
|
def loginDB():
|
|
mydb = mysql.connector.connect(
|
|
host="localhost",
|
|
user="root",
|
|
password="Dingleberries69!",
|
|
database="NodeData"
|
|
)
|
|
return mydb
|
|
|
|
def getData(node, dataType, MAC):
|
|
mydb = loginDB()
|
|
query = get_query(node, dataType, MAC)
|
|
cursor = mydb.cursor(dictionary=True) # Enable dictionary output
|
|
cursor.execute(query)
|
|
result = cursor.fetchall() # Fetch the results
|
|
cursor.close()
|
|
mydb.close()
|
|
|
|
return result
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True, host='localhost')
|