diff --git a/server/Flask/main.py b/server/Flask/main.py index c434884..76362b7 100644 --- a/server/Flask/main.py +++ b/server/Flask/main.py @@ -10,42 +10,20 @@ def index(): 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) +@app.route('/updateData') +def updateDataIndex(): + node_id = request.args.get('node', None) + new_name = request.args.get('name', None) + new_location = request.args.get('location', None) + return updateData(node_id, new_name, new_location) + - 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): +def updateData(node, name, location): mydb = loginDB() - query = get_query(node, name, location, MAC) + query = update_query(node, name, location) cursor = mydb.cursor(dictionary=True) # Enable dictionary output cursor.execute(query) + mydb.commit() result = cursor.fetchall() # Fetch the results cursor.close() mydb.close() diff --git a/server/Flask/queries.py b/server/Flask/queries.py index c9aae44..e2da308 100644 --- a/server/Flask/queries.py +++ b/server/Flask/queries.py @@ -10,3 +10,14 @@ def get_query(node, dataType, MAC): else: query = "SELECT * FROM `Measurement`" return query + + + +def update_query(node, name, location): + if node and name and location: + query = f""" + UPDATE Node + SET Name = '{name}', Location = '{location}' + WHERE NodeID = {node}; + """ + return query \ No newline at end of file