import falcon import simplejson as json import mysql.connector import config import uuid class CombinedEquipmentCollection: @staticmethod def __init__(): """ Initializes CombinedEquipmentCollection""" pass @staticmethod def on_options(req, resp): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp): cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) query = (" SELECT id, name, uuid " " FROM tbl_cost_centers ") cursor.execute(query) rows_cost_centers = cursor.fetchall() cost_center_dict = dict() if rows_cost_centers is not None and len(rows_cost_centers) > 0: for row in rows_cost_centers: cost_center_dict[row['id']] = {"id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid, " " is_input_counted, is_output_counted, " " cost_center_id, description " " FROM tbl_combined_equipments " " ORDER BY id ") cursor.execute(query) rows_combined_equipments = cursor.fetchall() result = list() if rows_combined_equipments is not None and len(rows_combined_equipments) > 0: for row in rows_combined_equipments: cost_center = cost_center_dict.get(row['cost_center_id'], None) meta_result = {"id": row['id'], "name": row['name'], "uuid": row['uuid'], "is_input_counted": bool(row['is_input_counted']), "is_output_counted": bool(row['is_output_counted']), "cost_center": cost_center, "description": row['description']} result.append(meta_result) cursor.close() cnx.disconnect() resp.body = json.dumps(result) @staticmethod def on_post(req, resp): """Handles POST requests""" try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description=ex) new_values = json.loads(raw_json) if 'name' not in new_values['data'].keys() or \ not isinstance(new_values['data']['name'], str) or \ len(str.strip(new_values['data']['name'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_NAME') name = str.strip(new_values['data']['name']) if 'is_input_counted' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_input_counted'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_INPUT_COUNTED_VALUE') is_input_counted = new_values['data']['is_input_counted'] if 'is_output_counted' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_output_counted'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_OUTPUT_COUNTED_VALUE') is_output_counted = new_values['data']['is_output_counted'] if 'cost_center_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['cost_center_id'], int) or \ new_values['data']['cost_center_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COST_CENTER_ID') cost_center_id = new_values['data']['cost_center_id'] if 'description' in new_values['data'].keys() and \ new_values['data']['description'] is not None and \ len(str(new_values['data']['description'])) > 0: description = str.strip(new_values['data']['description']) else: description = None cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE name = %s ", (name,)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.BAD_REQUEST', description='API.COMBINED_EQUIPMENT_NAME_IS_ALREADY_IN_USE') if cost_center_id is not None: cursor.execute(" SELECT name " " FROM tbl_cost_centers " " WHERE id = %s ", (new_values['data']['cost_center_id'],)) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COST_CENTER_NOT_FOUND') add_values = (" INSERT INTO tbl_combined_equipments " " (name, uuid, is_input_counted, is_output_counted, " " cost_center_id, description) " " VALUES (%s, %s, %s, %s, %s, %s) ") cursor.execute(add_values, (name, str(uuid.uuid4()), is_input_counted, is_output_counted, cost_center_id, description)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(new_id) class CombinedEquipmentItem: @staticmethod def __init__(): """Initializes CombinedEquipmentItem""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) query = (" SELECT id, name, uuid " " FROM tbl_cost_centers ") cursor.execute(query) rows_cost_centers = cursor.fetchall() cost_center_dict = dict() if rows_cost_centers is not None and len(rows_cost_centers) > 0: for row in rows_cost_centers: cost_center_dict[row['id']] = {"id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid, " " is_input_counted, is_output_counted, " " cost_center_id, description " " FROM tbl_combined_equipments " " WHERE id = %s ") cursor.execute(query, (id_,)) row = cursor.fetchone() cursor.close() cnx.disconnect() if row is None: raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') else: cost_center = cost_center_dict.get(row['cost_center_id'], None) meta_result = {"id": row['id'], "name": row['name'], "uuid": row['uuid'], "is_input_counted": bool(row['is_input_counted']), "is_output_counted": bool(row['is_output_counted']), "cost_center": cost_center, "description": row['description']} resp.body = json.dumps(meta_result) @staticmethod def on_delete(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() # check relation with space cursor.execute(" SELECT space_id " " FROM tbl_spaces_combined_equipments " " WHERE combined_equipment_id = %s ", (id_,)) rows_combined_equipments = cursor.fetchall() if rows_combined_equipments is not None and len(rows_combined_equipments) > 0: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.THERE_IS_RELATION_WITH_SPACES') # check relation with meter cursor.execute(" SELECT meter_id " " FROM tbl_combined_equipments_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_meters = cursor.fetchall() if rows_meters is not None and len(rows_meters) > 0: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.THERE_IS_RELATION_WITH_METER') # check relation with offline meter cursor.execute(" SELECT offline_meter_id " " FROM tbl_combined_equipments_offline_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_offline_meters = cursor.fetchall() if rows_offline_meters is not None and len(rows_offline_meters) > 0: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.THERE_IS_RELATION_WITH_OFFLINE_METER') # check relation with virtual meter cursor.execute(" SELECT virtual_meter_id " " FROM tbl_combined_equipments_virtual_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_virtual_meters = cursor.fetchall() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.THERE_IS_RELATION_WITH_VIRTUAL_METER') # delete all associated parameters cursor.execute(" DELETE FROM tbl_combined_equipments_parameters WHERE combined_equipment_id = %s ", (id_,)) cnx.commit() cursor.execute(" DELETE FROM tbl_combined_equipments WHERE id = %s ", (id_,)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204 @staticmethod def on_put(req, resp, id_): """Handles PUT requests""" if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) new_values = json.loads(raw_json) if 'name' not in new_values['data'].keys() or \ not isinstance(new_values['data']['name'], str) or \ len(str.strip(new_values['data']['name'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_NAME') name = str.strip(new_values['data']['name']) if 'is_input_counted' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_input_counted'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_INPUT_COUNTED_VALUE') is_input_counted = new_values['data']['is_input_counted'] if 'is_output_counted' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_output_counted'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_OUTPUT_COUNTED_VALUE') is_output_counted = new_values['data']['is_output_counted'] if 'cost_center_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['cost_center_id'], int) or \ new_values['data']['cost_center_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COST_CENTER_ID') cost_center_id = new_values['data']['cost_center_id'] if 'description' in new_values['data'].keys() and \ new_values['data']['description'] is not None and \ len(str(new_values['data']['description'])) > 0: description = str.strip(new_values['data']['description']) else: description = None cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE name = %s AND id != %s ", (name, id_)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.BAD_REQUEST', description='API.COMBINED_EQUIPMENT_NAME_IS_ALREADY_IN_USE') cursor.execute(" SELECT name " " FROM tbl_cost_centers " " WHERE id = %s ", (new_values['data']['cost_center_id'],)) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COST_CENTER_NOT_FOUND') update_row = (" UPDATE tbl_combined_equipments " " SET name = %s, is_input_counted = %s, is_output_counted = %s, " " cost_center_id = %s, description = %s " " WHERE id = %s ") cursor.execute(update_row, (name, is_input_counted, is_output_counted, cost_center_id, description, id_)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_200 # Clone a Combined Equipment @staticmethod def on_post(req, resp, id_): """Handles PUT requests""" if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) new_values = json.loads(raw_json) cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT name, is_input_counted, is_output_counted, " " cost_center_id, description " " FROM tbl_combined_equipments " " WHERE id = %s ") cursor.execute(query, (id_,)) row = cursor.fetchone() if row is None: raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') else: add_values = (" INSERT INTO tbl_combined_equipments " " (name, uuid, is_input_counted, is_output_counted, " " cost_center_id, description) " " VALUES (%s, %s, %s, %s, %s, %s) ") cursor.execute(add_values, (row['name'] + ' Copy', str(uuid.uuid4()), row['is_input_counted'], row['is_output_counted'], row['cost_center_id'], row['description'])) new_id = cursor.lastrowid cnx.commit() # clone relation with meter cursor.execute(" SELECT meter_id, is_output " " FROM tbl_combined_equipments_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_meters = cursor.fetchall() if rows_meters is not None and len(rows_meters) > 0: add_values = (" INSERT INTO tbl_combined_equipments_meters (combined_equipment_id, meter_id, is_output) " " VALUES ") for row in rows_meters: add_values += " (" + str(new_id) + "," add_values += str(row['meter_id']) + "," add_values += str(bool(row['is_output'])) + "), " # trim ", " at the end of string and then execute cursor.execute(add_values[:-2]) cnx.commit() # clone relation with offline meter cursor.execute(" SELECT offline_meter_id, is_output " " FROM tbl_combined_equipments_offline_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_offline_meters = cursor.fetchall() if rows_offline_meters is not None and len(rows_offline_meters) > 0: add_values = (" INSERT INTO tbl_combined_equipments_offline_meters " " (combined_equipment_id, offline_meter_id, is_output) " " VALUES ") for row in rows_offline_meters: add_values += " (" + str(new_id) + "," add_values += "'" + str(row['offline_meter_id']) + "'," add_values += str(bool(row['is_output'])) + "), " # trim ", " at the end of string and then execute cursor.execute(add_values[:-2]) cnx.commit() # clone relation with virtual meter cursor.execute(" SELECT virtual_meter_id, is_output " " FROM tbl_combined_equipments_virtual_meters " " WHERE combined_equipment_id = %s ", (id_,)) rows_virtual_meters = cursor.fetchall() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: add_values = (" INSERT INTO tbl_combined_equipments_virtual_meters " " (combined_equipment_id, virtual_meter_id, is_output) " " VALUES ") for row in rows_virtual_meters: add_values += " (" + str(new_id) + "," add_values += str(row['virtual_meter_id']) + "," add_values += str(bool(row['is_output'])) + "), " # trim ", " at the end of string and then execute cursor.execute(add_values[:-2]) cnx.commit() # clone parameters cursor.execute(" SELECT name, parameter_type, constant, point_id, numerator_meter_uuid, denominator_meter_uuid " " FROM tbl_combined_equipments_parameters " " WHERE combined_equipment_id = %s ", (id_,)) rows_parameters = cursor.fetchall() if rows_parameters is not None and len(rows_parameters) > 0: add_values = (" INSERT INTO tbl_combined_equipments_parameters" " (combined_equipment_id, name, parameter_type, constant, point_id, " " numerator_meter_uuid, denominator_meter_uuid) " " VALUES ") for row in rows_parameters: add_values += " (" + str(new_id) + "," add_values += "'" + str(row['name']) + "'," add_values += "'" + str(row['parameter_type']) + "'," if row['constant'] is not None: add_values += "'" + str(row['constant']) + "'," else: add_values += "null, " if row['point_id'] is not None: add_values += str(row['point_id']) + "," else: add_values += "null, " if row['numerator_meter_uuid'] is not None: add_values += "'" + row['numerator_meter_uuid'] + "'," else: add_values += "null, " if row['denominator_meter_uuid'] is not None: add_values += "'" + row['denominator_meter_uuid'] + "'), " else: add_values += "null), " # trim ", " at the end of string and then execute cursor.execute(add_values[:-2]) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(new_id) class CombinedEquipmentEquipmentCollection: @staticmethod def __init__(): """Initializes CombinedEquipmentEquipmentCollection""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT e.id, e.name, e.uuid " " FROM tbl_combined_equipments c, tbl_combined_equipments_equipments ce, tbl_equipments e " " WHERE ce.combined_equipment_id = c.id AND e.id = ce.equipment_id AND c.id = %s " " ORDER BY e.id ") cursor.execute(query, (id_,)) rows = cursor.fetchall() result = list() if rows is not None and len(rows) > 0: for row in rows: meta_result = {"id": row[0], "name": row[1], "uuid": row[2]} result.append(meta_result) resp.body = json.dumps(result) @staticmethod def on_post(req, resp, id_): """Handles POST requests""" try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') new_values = json.loads(raw_json) if 'equipment_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['equipment_id'], int) or \ new_values['data']['equipment_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_EQUIPMENT_ID') equipment_id = new_values['data']['equipment_id'] cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " from tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_equipments " " WHERE id = %s ", (equipment_id,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.EQUIPMENT_NOT_FOUND') query = (" SELECT id " " FROM tbl_combined_equipments_equipments " " WHERE combined_equipment_id = %s AND equipment_id = %s") cursor.execute(query, (id_, equipment_id,)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description='API.COMBINED_EQUIPMENT_EQUIPMENT_RELATION_EXISTS') add_row = (" INSERT INTO tbl_combined_equipments_equipments (combined_equipment_id, equipment_id) " " VALUES (%s, %s) ") cursor.execute(add_row, (id_, equipment_id,)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(id_) + '/equipments/' + str(equipment_id) class CombinedEquipmentEquipmentItem: @staticmethod def __init__(): """Initializes CombinedEquipmentEquipmentItem""" pass @staticmethod def on_options(req, resp, id_, eid): resp.status = falcon.HTTP_200 @staticmethod def on_delete(req, resp, id_, eid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not eid.isdigit() or int(eid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_equipments " " WHERE id = %s ", (eid,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT id " " FROM tbl_combined_equipments_equipments " " WHERE combined_equipment_id = %s AND equipment_id = %s ", (id_, eid)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_EQUIPMENT_RELATION_NOT_FOUND') cursor.execute(" DELETE FROM tbl_combined_equipments_equipments " " WHERE combined_equipment_id = %s AND equipment_id = %s ", (id_, eid)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204 class CombinedEquipmentParameterCollection: @staticmethod def __init__(): """Initializes CombinedEquipmentParameterCollection""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT id, name " " FROM tbl_points ") cursor.execute(query) rows_points = cursor.fetchall() point_dict = dict() if rows_points is not None and len(rows_points) > 0: for row in rows_points: point_dict[row['id']] = {"id": row['id'], "name": row['name']} query = (" SELECT id, name, uuid " " FROM tbl_meters ") cursor.execute(query) rows_meters = cursor.fetchall() meter_dict = dict() if rows_meters is not None and len(rows_meters) > 0: for row in rows_meters: meter_dict[row['uuid']] = {"type": 'meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_offline_meters ") cursor.execute(query) rows_offline_meters = cursor.fetchall() offline_meter_dict = dict() if rows_offline_meters is not None and len(rows_offline_meters) > 0: for row in rows_offline_meters: offline_meter_dict[row['uuid']] = {"type": 'offline_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_virtual_meters ") cursor.execute(query) rows_virtual_meters = cursor.fetchall() virtual_meter_dict = dict() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: for row in rows_virtual_meters: virtual_meter_dict[row['uuid']] = {"type": 'virtual_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, parameter_type, " " constant, point_id, numerator_meter_uuid, denominator_meter_uuid " " FROM tbl_combined_equipments_parameters " " WHERE combined_equipment_id = %s " " ORDER BY id ") cursor.execute(query, (id_, )) rows_parameters = cursor.fetchall() result = list() if rows_parameters is not None and len(rows_parameters) > 0: for row in rows_parameters: constant = None point = None numerator_meter = None denominator_meter = None if row['parameter_type'] == 'point': point = point_dict.get(row['point_id'], None) constant = None numerator_meter = None denominator_meter = None elif row['parameter_type'] == 'constant': constant = row['constant'] point = None numerator_meter = None denominator_meter = None elif row['parameter_type'] == 'fraction': constant = None point = None # find numerator meter by uuid numerator_meter = meter_dict.get(row['numerator_meter_uuid'], None) if numerator_meter is None: numerator_meter = virtual_meter_dict.get(row['numerator_meter_uuid'], None) if numerator_meter is None: numerator_meter = offline_meter_dict.get(row['numerator_meter_uuid'], None) # find denominator meter by uuid denominator_meter = meter_dict.get(row['denominator_meter_uuid'], None) if denominator_meter is None: denominator_meter = virtual_meter_dict.get(row['denominator_meter_uuid'], None) if denominator_meter is None: denominator_meter = offline_meter_dict.get(row['denominator_meter_uuid'], None) meta_result = {"id": row['id'], "name": row['name'], "parameter_type": row['parameter_type'], "constant": constant, "point": point, "numerator_meter": numerator_meter, "denominator_meter": denominator_meter} result.append(meta_result) cursor.close() cnx.disconnect() resp.body = json.dumps(result) @staticmethod def on_post(req, resp, id_): """Handles POST requests""" if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description=ex) new_values = json.loads(raw_json) if 'name' not in new_values['data'].keys() or \ not isinstance(new_values['data']['name'], str) or \ len(str.strip(new_values['data']['name'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_NAME') name = str.strip(new_values['data']['name']) if 'parameter_type' not in new_values['data'].keys() or \ not isinstance(new_values['data']['parameter_type'], str) or \ len(str.strip(new_values['data']['parameter_type'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_TYPE') parameter_type = str.strip(new_values['data']['parameter_type']) if parameter_type not in ('constant', 'point', 'fraction'): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_TYPE') constant = None if 'constant' in new_values['data'].keys(): if new_values['data']['constant'] is not None and \ isinstance(new_values['data']['constant'], str) and \ len(str.strip(new_values['data']['constant'])) > 0: constant = str.strip(new_values['data']['constant']) point_id = None if 'point_id' in new_values['data'].keys(): if new_values['data']['point_id'] is not None and \ new_values['data']['point_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_POINT_ID') point_id = new_values['data']['point_id'] numerator_meter_uuid = None if 'numerator_meter_uuid' in new_values['data'].keys(): if new_values['data']['numerator_meter_uuid'] is not None and \ isinstance(new_values['data']['numerator_meter_uuid'], str) and \ len(str.strip(new_values['data']['numerator_meter_uuid'])) > 0: numerator_meter_uuid = str.strip(new_values['data']['numerator_meter_uuid']) denominator_meter_uuid = None if 'denominator_meter_uuid' in new_values['data'].keys(): if new_values['data']['denominator_meter_uuid'] is not None and \ isinstance(new_values['data']['denominator_meter_uuid'], str) and \ len(str.strip(new_values['data']['denominator_meter_uuid'])) > 0: denominator_meter_uuid = str.strip(new_values['data']['denominator_meter_uuid']) cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_combined_equipments_parameters " " WHERE name = %s AND combined_equipment_id = %s ", (name, id_)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.COMBINED_EQUIPMENT_PARAMETER_NAME_IS_ALREADY_IN_USE') # validate by parameter type if parameter_type == 'point': if point_id is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_POINT_ID') query = (" SELECT id, name " " FROM tbl_points " " WHERE id = %s ") cursor.execute(query, (point_id, )) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.POINT_NOT_FOUND') elif parameter_type == 'constant': if constant is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_CONSTANT_VALUE') elif parameter_type == 'fraction': query = (" SELECT id, name, uuid " " FROM tbl_meters ") cursor.execute(query) rows_meters = cursor.fetchall() meter_dict = dict() if rows_meters is not None and len(rows_meters) > 0: for row in rows_meters: meter_dict[row['uuid']] = {"type": 'meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_offline_meters ") cursor.execute(query) rows_offline_meters = cursor.fetchall() offline_meter_dict = dict() if rows_offline_meters is not None and len(rows_offline_meters) > 0: for row in rows_offline_meters: offline_meter_dict[row['uuid']] = {"type": 'offline_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_virtual_meters ") cursor.execute(query) rows_virtual_meters = cursor.fetchall() virtual_meter_dict = dict() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: for row in rows_virtual_meters: virtual_meter_dict[row['uuid']] = {"type": 'virtual_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} # validate numerator meter uuid if meter_dict.get(numerator_meter_uuid) is None and \ virtual_meter_dict.get(numerator_meter_uuid) is None and \ offline_meter_dict.get(numerator_meter_uuid) is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_NUMERATOR_METER_UUID') # validate denominator meter uuid if denominator_meter_uuid == numerator_meter_uuid: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_DENOMINATOR_METER_UUID') if denominator_meter_uuid not in meter_dict and \ denominator_meter_uuid not in virtual_meter_dict and \ denominator_meter_uuid not in offline_meter_dict: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_DENOMINATOR_METER_UUID') add_values = (" INSERT INTO tbl_combined_equipments_parameters " " (combined_equipment_id, name, parameter_type, constant, " " point_id, numerator_meter_uuid, denominator_meter_uuid) " " VALUES (%s, %s, %s, %s, %s, %s, %s) ") cursor.execute(add_values, (id_, name, parameter_type, constant, point_id, numerator_meter_uuid, denominator_meter_uuid)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(id_) + 'parameters/' + str(new_id) class CombinedEquipmentParameterItem: @staticmethod def __init__(): """"Initializes CombinedEquipmentParameterItem""" pass @staticmethod def on_options(req, resp, id_, pid): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_, pid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not pid.isdigit() or int(pid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) query = (" SELECT id, name " " FROM tbl_points ") cursor.execute(query) rows_points = cursor.fetchall() point_dict = dict() if rows_points is not None and len(rows_points) > 0: for row in rows_points: point_dict[row['id']] = {"id": row['id'], "name": row['name']} query = (" SELECT id, name, uuid " " FROM tbl_meters ") cursor.execute(query) rows_meters = cursor.fetchall() meter_dict = dict() if rows_meters is not None and len(rows_meters) > 0: for row in rows_meters: meter_dict[row['uuid']] = {"type": 'meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_offline_meters ") cursor.execute(query) rows_offline_meters = cursor.fetchall() offline_meter_dict = dict() if rows_offline_meters is not None and len(rows_offline_meters) > 0: for row in rows_offline_meters: offline_meter_dict[row['uuid']] = {"type": 'offline_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_virtual_meters ") cursor.execute(query) rows_virtual_meters = cursor.fetchall() virtual_meter_dict = dict() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: for row in rows_virtual_meters: virtual_meter_dict[row['uuid']] = {"type": 'virtual_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, parameter_type, " " constant, point_id, numerator_meter_uuid, denominator_meter_uuid " " FROM tbl_combined_equipments_parameters " " WHERE combined_equipment_id = %s AND id = %s ") cursor.execute(query, (id_, pid)) row = cursor.fetchone() cursor.close() cnx.disconnect() if row is None: raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_PARAMETER_NOT_FOUND_OR_NOT_MATCH') else: constant = None point = None numerator_meter = None denominator_meter = None if row['parameter_type'] == 'point': point = point_dict.get(row['point_id'], None) constant = None numerator_meter = None denominator_meter = None elif row['parameter_type'] == 'constant': constant = row['constant'] point = None numerator_meter = None denominator_meter = None elif row['parameter_type'] == 'fraction': constant = None point = None # find numerator meter by uuid numerator_meter = meter_dict.get(row['numerator_meter_uuid'], None) if numerator_meter is None: numerator_meter = virtual_meter_dict.get(row['numerator_meter_uuid'], None) if numerator_meter is None: numerator_meter = offline_meter_dict.get(row['numerator_meter_uuid'], None) # find denominator meter by uuid denominator_meter = meter_dict.get(row['denominator_meter_uuid'], None) if denominator_meter is None: denominator_meter = virtual_meter_dict.get(row['denominator_meter_uuid'], None) if denominator_meter is None: denominator_meter = offline_meter_dict.get(row['denominator_meter_uuid'], None) meta_result = {"id": row['id'], "name": row['name'], "parameter_type": row['parameter_type'], "constant": constant, "point": point, "numerator_meter": numerator_meter, "denominator_meter": denominator_meter} resp.body = json.dumps(meta_result) @staticmethod def on_delete(req, resp, id_, pid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not pid.isdigit() or int(pid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_combined_equipments_parameters " " WHERE combined_equipment_id = %s AND id = %s ", (id_, pid,)) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_PARAMETER_NOT_FOUND_OR_NOT_MATCH') cursor.execute(" DELETE FROM tbl_combined_equipments_parameters " " WHERE id = %s ", (pid, )) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204 @staticmethod def on_put(req, resp, id_, pid): """Handles POST requests""" if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not pid.isdigit() or int(pid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_ID') try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description=ex) new_values = json.loads(raw_json) if 'name' not in new_values['data'].keys() or \ not isinstance(new_values['data']['name'], str) or \ len(str.strip(new_values['data']['name'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_NAME') name = str.strip(new_values['data']['name']) if 'parameter_type' not in new_values['data'].keys() or \ not isinstance(new_values['data']['parameter_type'], str) or \ len(str.strip(new_values['data']['parameter_type'])) == 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_TYPE') parameter_type = str.strip(new_values['data']['parameter_type']) if parameter_type not in ('constant', 'point', 'fraction'): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_PARAMETER_TYPE') constant = None if 'constant' in new_values['data'].keys(): if new_values['data']['constant'] is not None and \ isinstance(new_values['data']['constant'], str) and \ len(str.strip(new_values['data']['constant'])) > 0: constant = str.strip(new_values['data']['constant']) point_id = None if 'point_id' in new_values['data'].keys(): if new_values['data']['point_id'] is not None and \ new_values['data']['point_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_POINT_ID') point_id = new_values['data']['point_id'] numerator_meter_uuid = None if 'numerator_meter_uuid' in new_values['data'].keys(): if new_values['data']['numerator_meter_uuid'] is not None and \ isinstance(new_values['data']['numerator_meter_uuid'], str) and \ len(str.strip(new_values['data']['numerator_meter_uuid'])) > 0: numerator_meter_uuid = str.strip(new_values['data']['numerator_meter_uuid']) denominator_meter_uuid = None if 'denominator_meter_uuid' in new_values['data'].keys(): if new_values['data']['denominator_meter_uuid'] is not None and \ isinstance(new_values['data']['denominator_meter_uuid'], str) and \ len(str.strip(new_values['data']['denominator_meter_uuid'])) > 0: denominator_meter_uuid = str.strip(new_values['data']['denominator_meter_uuid']) cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_combined_equipments_parameters " " WHERE combined_equipment_id = %s AND id = %s ", (id_, pid,)) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_PARAMETER_NOT_FOUND_OR_NOT_MATCH') cursor.execute(" SELECT name " " FROM tbl_combined_equipments_parameters " " WHERE name = %s AND combined_equipment_id = %s AND id != %s ", (name, id_, pid)) row = cursor.fetchone() if row is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.COMBINED_EQUIPMENT_PARAMETER_NAME_IS_ALREADY_IN_USE') # validate by parameter type if parameter_type == 'point': if point_id is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_POINT_ID') query = (" SELECT id, name " " FROM tbl_points " " WHERE id = %s ") cursor.execute(query, (point_id, )) row = cursor.fetchone() if row is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.POINT_NOT_FOUND') elif parameter_type == 'constant': if constant is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_CONSTANT_VALUE') elif parameter_type == 'fraction': query = (" SELECT id, name, uuid " " FROM tbl_meters ") cursor.execute(query) rows_meters = cursor.fetchall() meter_dict = dict() if rows_meters is not None and len(rows_meters) > 0: for row in rows_meters: meter_dict[row['uuid']] = {"type": 'meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_offline_meters ") cursor.execute(query) rows_offline_meters = cursor.fetchall() offline_meter_dict = dict() if rows_offline_meters is not None and len(rows_offline_meters) > 0: for row in rows_offline_meters: offline_meter_dict[row['uuid']] = {"type": 'offline_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT id, name, uuid " " FROM tbl_virtual_meters ") cursor.execute(query) rows_virtual_meters = cursor.fetchall() virtual_meter_dict = dict() if rows_virtual_meters is not None and len(rows_virtual_meters) > 0: for row in rows_virtual_meters: virtual_meter_dict[row['uuid']] = {"type": 'virtual_meter', "id": row['id'], "name": row['name'], "uuid": row['uuid']} # validate numerator meter uuid if meter_dict.get(numerator_meter_uuid) is None and \ virtual_meter_dict.get(numerator_meter_uuid) is None and \ offline_meter_dict.get(numerator_meter_uuid) is None: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_NUMERATOR_METER_UUID') # validate denominator meter uuid if denominator_meter_uuid == numerator_meter_uuid: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_DENOMINATOR_METER_UUID') if denominator_meter_uuid not in meter_dict and \ denominator_meter_uuid not in virtual_meter_dict and \ denominator_meter_uuid not in offline_meter_dict: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_DENOMINATOR_METER_UUID') add_values = (" UPDATE tbl_combined_equipments_parameters " " SET name = %s , parameter_type = %s, constant = %s, " " point_id = %s, numerator_meter_uuid = %s, denominator_meter_uuid =%s " " WHERE id = %s ") cursor.execute(add_values, (name, parameter_type, constant, point_id, numerator_meter_uuid, denominator_meter_uuid, pid)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_200 class CombinedEquipmentMeterCollection: @staticmethod def __init__(): """"Initializes CombinedEquipmentMeterCollection""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT id, name, uuid " " FROM tbl_energy_categories ") cursor.execute(query) rows_energy_categories = cursor.fetchall() energy_category_dict = dict() if rows_energy_categories is not None and len(rows_energy_categories) > 0: for row in rows_energy_categories: energy_category_dict[row['id']] = {"id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT m.id, m.name, m.uuid, m.energy_category_id, em.is_output " " FROM tbl_combined_equipments e, tbl_combined_equipments_meters em, tbl_meters m " " WHERE em.combined_equipment_id = e.id AND m.id = em.meter_id AND e.id = %s " " ORDER BY m.id ") cursor.execute(query, (id_,)) rows = cursor.fetchall() result = list() if rows is not None and len(rows) > 0: for row in rows: energy_category = energy_category_dict.get(row['energy_category_id'], None) meta_result = {"id": row['id'], "name": row['name'], "uuid": row['uuid'], "energy_category": energy_category, "is_output": bool(row['is_output'])} result.append(meta_result) resp.body = json.dumps(result) @staticmethod def on_post(req, resp, id_): """Handles POST requests""" try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') new_values = json.loads(raw_json) if 'meter_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['meter_id'], int) or \ new_values['data']['meter_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_METER_ID') meter_id = new_values['data']['meter_id'] if 'is_output' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_output'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_OUTPUT_VALUE') is_output = new_values['data']['is_output'] cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " from tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_meters " " WHERE id = %s ", (meter_id,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.METER_NOT_FOUND') query = (" SELECT id " " FROM tbl_combined_equipments_meters " " WHERE combined_equipment_id = %s AND meter_id = %s") cursor.execute(query, (id_, meter_id,)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description='API.COMBINED_EQUIPMENT_METER_RELATION_EXISTS') add_row = (" INSERT INTO tbl_combined_equipments_meters (combined_equipment_id, meter_id, is_output ) " " VALUES (%s, %s, %s) ") cursor.execute(add_row, (id_, meter_id, is_output)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(id_) + '/meters/' + str(meter_id) class CombinedEquipmentMeterItem: @staticmethod def __init__(): """"Initializes CombinedEquipmentMeterItem""" pass @staticmethod def on_options(req, resp, id_, mid): resp.status = falcon.HTTP_200 @staticmethod def on_delete(req, resp, id_, mid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not mid.isdigit() or int(mid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_METER_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_meters " " WHERE id = %s ", (mid,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.METER_NOT_FOUND') cursor.execute(" SELECT id " " FROM tbl_combined_equipments_meters " " WHERE combined_equipment_id = %s AND meter_id = %s ", (id_, mid)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_METER_RELATION_NOT_FOUND') cursor.execute(" DELETE FROM tbl_combined_equipments_meters " " WHERE combined_equipment_id = %s AND meter_id = %s ", (id_, mid)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204 class CombinedEquipmentOfflineMeterCollection: @staticmethod def __init__(): """"Initializes CombinedEquipmentOfflineMeterCollection""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT id, name, uuid " " FROM tbl_energy_categories ") cursor.execute(query) rows_energy_categories = cursor.fetchall() energy_category_dict = dict() if rows_energy_categories is not None and len(rows_energy_categories) > 0: for row in rows_energy_categories: energy_category_dict[row['id']] = {"id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT m.id, m.name, m.uuid, m.energy_category_id, em.is_output " " FROM tbl_combined_equipments e, tbl_combined_equipments_offline_meters em, tbl_offline_meters m " " WHERE em.combined_equipment_id = e.id AND m.id = em.offline_meter_id AND e.id = %s " " ORDER BY m.id ") cursor.execute(query, (id_,)) rows = cursor.fetchall() result = list() if rows is not None and len(rows) > 0: for row in rows: energy_category = energy_category_dict.get(row['energy_category_id'], None) meta_result = {"id": row['id'], "name": row['name'], "uuid": row['uuid'], "energy_category": energy_category, "is_output": bool(row['is_output'])} result.append(meta_result) resp.body = json.dumps(result) @staticmethod def on_post(req, resp, id_): """Handles POST requests""" try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') new_values = json.loads(raw_json) if 'offline_meter_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['offline_meter_id'], int) or \ new_values['data']['offline_meter_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_OFFLINE_METER_ID') offline_meter_id = new_values['data']['offline_meter_id'] if 'is_output' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_output'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_OUTPUT_VALUE') is_output = new_values['data']['is_output'] cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " from tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_offline_meters " " WHERE id = %s ", (offline_meter_id,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.OFFLINE_METER_NOT_FOUND') query = (" SELECT id " " FROM tbl_combined_equipments_offline_meters " " WHERE combined_equipment_id = %s AND offline_meter_id = %s") cursor.execute(query, (id_, offline_meter_id,)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description='API.COMBINED_EQUIPMENT_OFFLINE_METER_RELATION_EXISTS') add_row = (" INSERT INTO tbl_combined_equipments_offline_meters " " (combined_equipment_id, offline_meter_id, is_output ) " " VALUES (%s, %s, %s) ") cursor.execute(add_row, (id_, offline_meter_id, is_output)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(id_) + '/offlinemeters/' + str(offline_meter_id) class CombinedEquipmentOfflineMeterItem: @staticmethod def __init__(): """"Initializes CombinedEquipmentOfflineMeterItem""" pass @staticmethod def on_options(req, resp, id_, mid): resp.status = falcon.HTTP_200 @staticmethod def on_delete(req, resp, id_, mid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not mid.isdigit() or int(mid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_OFFLINE_METER_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_offline_meters " " WHERE id = %s ", (mid,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.OFFLINE_METER_NOT_FOUND') cursor.execute(" SELECT id " " FROM tbl_combined_equipments_offline_meters " " WHERE combined_equipment_id = %s AND offline_meter_id = %s ", (id_, mid)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_OFFLINE_METER_RELATION_NOT_FOUND') cursor.execute(" DELETE FROM tbl_combined_equipments_offline_meters " " WHERE combined_equipment_id = %s AND offline_meter_id = %s ", (id_, mid)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204 class CombinedEquipmentVirtualMeterCollection: @staticmethod def __init__(): """"Initializes CombinedEquipmentVirtualMeterCollection""" pass @staticmethod def on_options(req, resp, id_): resp.status = falcon.HTTP_200 @staticmethod def on_get(req, resp, id_): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor(dictionary=True) cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') query = (" SELECT id, name, uuid " " FROM tbl_energy_categories ") cursor.execute(query) rows_energy_categories = cursor.fetchall() energy_category_dict = dict() if rows_energy_categories is not None and len(rows_energy_categories) > 0: for row in rows_energy_categories: energy_category_dict[row['id']] = {"id": row['id'], "name": row['name'], "uuid": row['uuid']} query = (" SELECT m.id, m.name, m.uuid, m.energy_category_id, em.is_output " " FROM tbl_combined_equipments e, tbl_combined_equipments_virtual_meters em, tbl_virtual_meters m " " WHERE em.combined_equipment_id = e.id AND m.id = em.virtual_meter_id AND e.id = %s " " ORDER BY m.id ") cursor.execute(query, (id_,)) rows = cursor.fetchall() result = list() if rows is not None and len(rows) > 0: for row in rows: energy_category = energy_category_dict.get(row['energy_category_id'], None) meta_result = {"id": row['id'], "name": row['name'], "uuid": row['uuid'], "energy_category": energy_category, "is_output": bool(row['is_output'])} result.append(meta_result) resp.body = json.dumps(result) @staticmethod def on_post(req, resp, id_): """Handles POST requests""" try: raw_json = req.stream.read().decode('utf-8') except Exception as ex: raise falcon.HTTPError(falcon.HTTP_400, title='API.EXCEPTION', description=ex) if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') new_values = json.loads(raw_json) if 'virtual_meter_id' not in new_values['data'].keys() or \ not isinstance(new_values['data']['virtual_meter_id'], int) or \ new_values['data']['virtual_meter_id'] <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_VIRTUAL_METER_ID') virtual_meter_id = new_values['data']['virtual_meter_id'] if 'is_output' not in new_values['data'].keys() or \ not isinstance(new_values['data']['is_output'], bool): raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_IS_OUTPUT_VALUE') is_output = new_values['data']['is_output'] cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " from tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_virtual_meters " " WHERE id = %s ", (virtual_meter_id,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.VIRTUAL_METER_NOT_FOUND') query = (" SELECT id " " FROM tbl_combined_equipments_virtual_meters " " WHERE combined_equipment_id = %s AND virtual_meter_id = %s") cursor.execute(query, (id_, virtual_meter_id,)) if cursor.fetchone() is not None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_400, title='API.ERROR', description='API.COMBINED_EQUIPMENT_VIRTUAL_METER_RELATION_EXISTS') add_row = (" INSERT INTO tbl_combined_equipments_virtual_meters " " (combined_equipment_id, virtual_meter_id, is_output ) " " VALUES (%s, %s, %s) ") cursor.execute(add_row, (id_, virtual_meter_id, is_output)) new_id = cursor.lastrowid cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_201 resp.location = '/combinedequipments/' + str(id_) + '/virtualmeters/' + str(virtual_meter_id) class CombinedEquipmentVirtualMeterItem: @staticmethod def __init__(): """"Initializes CombinedEquipmentVirtualMeterItem""" pass @staticmethod def on_options(req, resp, id_, mid): resp.status = falcon.HTTP_200 @staticmethod def on_delete(req, resp, id_, mid): if not id_.isdigit() or int(id_) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_COMBINED_EQUIPMENT_ID') if not mid.isdigit() or int(mid) <= 0: raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_VIRTUAL_METER_ID') cnx = mysql.connector.connect(**config.myems_system_db) cursor = cnx.cursor() cursor.execute(" SELECT name " " FROM tbl_combined_equipments " " WHERE id = %s ", (id_,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_NOT_FOUND') cursor.execute(" SELECT name " " FROM tbl_virtual_meters " " WHERE id = %s ", (mid,)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.VIRTUAL_METER_NOT_FOUND') cursor.execute(" SELECT id " " FROM tbl_combined_equipments_virtual_meters " " WHERE combined_equipment_id = %s AND virtual_meter_id = %s ", (id_, mid)) if cursor.fetchone() is None: cursor.close() cnx.disconnect() raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.COMBINED_EQUIPMENT_VIRTUAL_METER_RELATION_NOT_FOUND') cursor.execute(" DELETE FROM tbl_combined_equipments_virtual_meters " " WHERE combined_equipment_id = %s AND virtual_meter_id = %s ", (id_, mid)) cnx.commit() cursor.close() cnx.disconnect() resp.status = falcon.HTTP_204