Refactor get_query function to handle additional conditions

This commit is contained in:
sietse jonker
2024-04-01 11:50:47 +02:00
parent 5e2daa445a
commit 69ef7013bb

View File

@@ -1,5 +1,13 @@
def get_query(node, dataType, MAC, questionID, replies, dateStart, dateEnd):
if node and dataType:
if dateStart and dateEnd and node:
query = f'''SELECT *
FROM Measurement
WHERE TimeStamp BETWEEN '{dateStart}' AND '{dateEnd}' AND NodeID = {node};'''
elif dateStart and dateEnd:
query = f'''SELECT *
FROM Measurement
WHERE TimeStamp BETWEEN '{dateStart}' AND '{dateEnd}';'''
elif node and dataType:
query = f"SELECT * FROM Measurement WHERE NodeID = {node} AND Type = '{dataType}'"
elif node:
query = f"SELECT * FROM Measurement WHERE NodeID = {node}"
@@ -13,7 +21,7 @@ def get_query(node, dataType, MAC, questionID, replies, dateStart, dateEnd):
query = f"SELECT * FROM Question"
elif replies:
query = f"SELECT * FROM Reply"
elif dateStart and dateEnd:
elif dateStart and dateEnd and node:
query = f'''SELECT *
FROM Measurement
WHERE TimeStamp BETWEEN '{dateStart}' AND '{dateEnd}';'''