core.virtualmeter.VirtualMeterItem.on_get()

Complexity

Conditions

23

Size

Total Lines	131
Code Lines	100

Duplication

Lines	50
Ratio	38.17 %

Importance

Changes

0

import falcon
    @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_VIRTUAL_METER_ID')

        cnx = mysql.connector.connect(**config.myems_system_db)
        cursor = cnx.cursor(dictionary=True)

        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 id, name, uuid, energy_category_id "
                 " FROM tbl_energy_items ")
        cursor.execute(query)
        rows_energy_items = cursor.fetchall()

        energy_item_dict = dict()
        if rows_energy_items is not None and len(rows_energy_items) > 0:
            for row in rows_energy_items:
                energy_item_dict[row['id']] = {"id": row['id'],
                                               "name": row['name'],
                                               "uuid": row['uuid']}

        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, energy_category_id, is_counted, "
                 "        energy_item_id, cost_center_id, description "
                 " FROM tbl_virtual_meters "
                 " 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.VIRTUAL_METER_NOT_FOUND')
        else:
            energy_category = energy_category_dict.get(row['energy_category_id'], None)
            energy_item = energy_item_dict.get(row['energy_item_id'], None)
            cost_center = cost_center_dict.get(row['cost_center_id'], None)
            meta_result = {"id": row['id'],
                           "name": row['name'],
                           "uuid": row['uuid'],
                           "energy_category": energy_category,
                           "is_counted": True if row['is_counted'] else False,
                           "energy_item": energy_item,
                           "cost_center": cost_center,
                           "description": row['description'],
                           "expression": {}}

        expression = dict()
        query_expression = (" SELECT e.id, e.uuid, e.equation "
                            " FROM tbl_expressions e "
                            " WHERE e.virtual_meter_id = %s ")
        cursor.execute(query_expression, (id_,))
        row_expression = cursor.fetchone()

        if row_expression is not None:

This code seems to be duplicated in your project.

            expression = {'id': row_expression['id'],
                          'uuid': row_expression['uuid'],
                          'equation': row_expression['equation'],
                          'variables': []}

            query_variables = (" SELECT v.id, v.name, v.meter_type, v.meter_id "
                               " FROM tbl_expressions e, tbl_variables v "
                               " WHERE e.id = %s AND v.expression_id = e.id "
                               " ORDER BY v.name ")
            cursor.execute(query_variables, (row_expression['id'],))
            rows_variables = cursor.fetchall()
            if rows_variables is not None:
                for row_variable in rows_variables:
                    if row_variable['meter_type'].lower() == 'meter':
                        query_meter = (" SELECT m.name "
                                       " FROM tbl_meters m "
                                       " WHERE m.id = %s ")
                        cursor.execute(query_meter, (row_variable['meter_id'],))
                        row_meter = cursor.fetchone()
                        if row_meter is not None:
                            expression['variables'].append({'id': row_variable['id'],
                                                            'name': row_variable['name'],
                                                            'meter_type': row_variable['meter_type'],
                                                            'meter_id': row_variable['meter_id'],
                                                            'meter_name': row_meter['name']})
                    elif row_variable['meter_type'].lower() == 'offline_meter':
                        query_meter = (" SELECT m.name "
                                       " FROM tbl_offline_meters m "
                                       " WHERE m.id = %s ")
                        cursor.execute(query_meter, (row_variable['meter_id'],))
                        row_meter = cursor.fetchone()
                        if row_meter is not None:
                            expression['variables'].append({'id': row_variable['id'],
                                                            'name': row_variable['name'],
                                                            'meter_type': row_variable['meter_type'],
                                                            'meter_id': row_variable['meter_id'],
                                                            'meter_name': row_meter['name']})
                    elif row_variable['meter_type'].lower() == 'virtual_meter':
                        query_meter = (" SELECT m.name "
                                       " FROM tbl_virtual_meters m "
                                       " WHERE m.id = %s ")
                        cursor.execute(query_meter, (row_variable['meter_id'],))
                        row_meter = cursor.fetchone()
                        if row_meter is not None:
                            expression['variables'].append({'id': row_variable['id'],
                                                            'name': row_variable['name'],
                                                            'meter_type': row_variable['meter_type'],
                                                            'meter_id': row_variable['meter_id'],
                                                            'meter_name': row_meter['name']})

        meta_result['expression'] = expression

        cursor.close()
        cnx.disconnect()
        resp.body = json.dumps(meta_result)