myems-api/core/distributioncircuit.py
import uuid
import falcon
import mysql.connector
import simplejson as json
from core.useractivity import user_logger, admin_control, access_control, api_key_control
import config
class DistributionCircuitCollection:
def __init__(self):
"""Initializes DistributionCircuitCollection"""
pass
@staticmethod
def on_options(req, resp):
resp.status = falcon.HTTP_200
@staticmethod
def on_get(req, resp):
if 'API-KEY' not in req.headers or \
not isinstance(req.headers['API-KEY'], str) or \
len(str.strip(req.headers['API-KEY'])) == 0:
access_control(req)
else:
api_key_control(req)
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
query = (" SELECT id, name, uuid "
" FROM tbl_distribution_systems ")
cursor.execute(query)
rows_distribution_systems = cursor.fetchall()
distribution_system_dict = dict()
if rows_distribution_systems is not None and len(rows_distribution_systems) > 0:
for row in rows_distribution_systems:
distribution_system_dict[row[0]] = {"id": row[0],
"name": row[1],
"uuid": row[2]}
query = (" SELECT id, name, uuid, distribution_system_id, "
" distribution_room, switchgear, peak_load, peak_current, customers, meters "
" FROM tbl_distribution_circuits "
" ORDER BY id ")
cursor.execute(query)
rows_distribution_circuits = cursor.fetchall()
result = list()
if rows_distribution_circuits is not None and len(rows_distribution_circuits) > 0:
for row in rows_distribution_circuits:
meta_result = {"id": row[0],
"name": row[1],
"uuid": row[2],
"distribution_system": distribution_system_dict.get(row[3]),
"distribution_room": row[4],
"switchgear": row[5],
"peak_load": row[6],
"peak_current": row[7],
"customers": row[8],
"meters": row[9]}
result.append(meta_result)
cursor.close()
cnx.close()
resp.text = json.dumps(result)
@staticmethod
@user_logger
def on_post(req, resp):
"""Handles POST requests"""
admin_control(req)
try:
raw_json = req.stream.read().decode('utf-8')
except Exception as ex:
raise falcon.HTTPError(status=falcon.HTTP_400,
title='API.BAD_REQUEST',
description='API.FAILED_TO_READ_REQUEST_STREAM')
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(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_NAME')
name = str.strip(new_values['data']['name'])
if 'distribution_system_id' not in new_values['data'].keys() or \
not isinstance(new_values['data']['distribution_system_id'], int) or \
new_values['data']['distribution_system_id'] <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_SYSTEM_ID')
distribution_system_id = new_values['data']['distribution_system_id']
if 'distribution_room' not in new_values['data'].keys() or \
not isinstance(new_values['data']['distribution_room'], str) or \
len(str.strip(new_values['data']['distribution_room'])) == 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_ROOM')
distribution_room = str.strip(new_values['data']['distribution_room'])
if 'switchgear' not in new_values['data'].keys() or \
not isinstance(new_values['data']['switchgear'], str) or \
len(str.strip(new_values['data']['switchgear'])) == 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_SWITCHGEAR')
switchgear = str.strip(new_values['data']['switchgear'])
if 'peak_load' not in new_values['data'].keys() or \
not (isinstance(new_values['data']['peak_load'], float) or
isinstance(new_values['data']['peak_load'], int)):
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_PEAK_LOAD')
peak_load = float(new_values['data']['peak_load'])
if 'peak_current' not in new_values['data'].keys() or \
not (isinstance(new_values['data']['peak_current'], float) or
isinstance(new_values['data']['peak_current'], int)):
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_PEAK_CURRENT')
peak_current = float(new_values['data']['peak_current'])
if 'customers' in new_values['data'].keys() and \
new_values['data']['customers'] is not None and \
len(str(new_values['data']['customers'])) > 0:
customers = str.strip(new_values['data']['customers'])
else:
customers = None
if 'meters' in new_values['data'].keys() and \
new_values['data']['meters'] is not None and \
len(str(new_values['data']['meters'])) > 0:
meters = str.strip(new_values['data']['meters'])
else:
meters = None
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
cursor.execute(" SELECT name "
" FROM tbl_distribution_systems "
" WHERE id = %s ",
(distribution_system_id,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_SYSTEM_NOT_FOUND')
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE distribution_system_id = %s AND name = %s ",
(distribution_system_id, name,))
if cursor.fetchone() is not None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.DISTRIBUTION_CIRCUIT_NAME_IS_ALREADY_IN_USE')
add_values = (" INSERT INTO tbl_distribution_circuits "
" (name, uuid, distribution_system_id,"
" distribution_room, switchgear, peak_load, peak_current, customers, meters) "
" VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s) ")
cursor.execute(add_values, (name,
str(uuid.uuid4()),
distribution_system_id,
distribution_room,
switchgear,
peak_load,
peak_current,
customers,
meters))
new_id = cursor.lastrowid
cnx.commit()
cursor.close()
cnx.close()
resp.status = falcon.HTTP_201
resp.location = '/distributioncircuits/' + str(new_id)
class DistributionCircuitItem:
def __init__(self):
"""Initializes DistributionCircuitItem"""
pass
@staticmethod
def on_options(req, resp, id_):
resp.status = falcon.HTTP_200
@staticmethod
def on_get(req, resp, id_):
if 'API-KEY' not in req.headers or \
not isinstance(req.headers['API-KEY'], str) or \
len(str.strip(req.headers['API-KEY'])) == 0:
access_control(req)
else:
api_key_control(req)
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
query = (" SELECT id, name, uuid "
" FROM tbl_distribution_systems ")
cursor.execute(query)
rows_distribution_systems = cursor.fetchall()
distribution_system_dict = dict()
if rows_distribution_systems is not None and len(rows_distribution_systems) > 0:
for row in rows_distribution_systems:
distribution_system_dict[row[0]] = {"id": row[0],
"name": row[1],
"uuid": row[2]}
query = (" SELECT id, name, uuid, distribution_system_id, "
" distribution_room, switchgear, peak_load, peak_current, customers, meters "
" FROM tbl_distribution_circuits "
" WHERE id = %s ")
cursor.execute(query, (id_,))
row = cursor.fetchone()
cursor.close()
cnx.close()
if row is None:
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
else:
meta_result = {"id": row[0],
"name": row[1],
"uuid": row[2],
"distribution_system": distribution_system_dict.get(row[3]),
"distribution_room": row[4],
"switchgear": row[5],
"peak_load": row[6],
"peak_current": row[7],
"customers": row[8],
"meters": row[9]}
resp.text = json.dumps(meta_result)
@staticmethod
@user_logger
def on_delete(req, resp, id_):
admin_control(req)
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
# delete relation with points
cursor.execute(" DELETE FROM tbl_distribution_circuits_points "
" WHERE distribution_circuit_id = %s ", (id_,))
# delete distribution circuit itself
cursor.execute(" DELETE FROM tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
cnx.commit()
cursor.close()
cnx.close()
resp.status = falcon.HTTP_204
@staticmethod
@user_logger
def on_put(req, resp, id_):
"""Handles PUT requests"""
admin_control(req)
try:
raw_json = req.stream.read().decode('utf-8')
except Exception as ex:
raise falcon.HTTPError(status=falcon.HTTP_400,
title='API.BAD_REQUEST',
description='API.FAILED_TO_READ_REQUEST_STREAM')
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
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(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_NAME')
name = str.strip(new_values['data']['name'])
if 'distribution_system_id' not in new_values['data'].keys() or \
not isinstance(new_values['data']['distribution_system_id'], int) or \
new_values['data']['distribution_system_id'] <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_SYSTEM_ID')
distribution_system_id = new_values['data']['distribution_system_id']
if 'distribution_room' not in new_values['data'].keys() or \
not isinstance(new_values['data']['distribution_room'], str) or \
len(str.strip(new_values['data']['distribution_room'])) == 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_ROOM')
distribution_room = str.strip(new_values['data']['distribution_room'])
if 'switchgear' not in new_values['data'].keys() or \
not isinstance(new_values['data']['switchgear'], str) or \
len(str.strip(new_values['data']['switchgear'])) == 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_SWITCHGEAR')
switchgear = str.strip(new_values['data']['switchgear'])
if 'peak_load' not in new_values['data'].keys() or \
not (isinstance(new_values['data']['peak_load'], float) or
isinstance(new_values['data']['peak_load'], int)):
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_PEAK_LOAD')
peak_load = float(new_values['data']['peak_load'])
if 'peak_current' not in new_values['data'].keys() or \
not (isinstance(new_values['data']['peak_current'], float) or
isinstance(new_values['data']['peak_current'], int)):
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_PEAK_CURRENT')
peak_current = float(new_values['data']['peak_current'])
if 'customers' in new_values['data'].keys() and \
new_values['data']['customers'] is not None and \
len(str(new_values['data']['customers'])) > 0:
customers = str.strip(new_values['data']['customers'])
else:
customers = None
if 'meters' in new_values['data'].keys() and \
new_values['data']['meters'] is not None and \
len(str(new_values['data']['meters'])) > 0:
meters = str.strip(new_values['data']['meters'])
else:
meters = None
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
cursor.execute(" SELECT name "
" FROM tbl_distribution_systems "
" WHERE id = %s ",
(distribution_system_id,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_SYSTEM_NOT_FOUND')
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE distribution_system_id = %s AND name = %s AND id != %s ",
(distribution_system_id, name, id_))
if cursor.fetchone() is not None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.DISTRIBUTION_CIRCUIT_NAME_IS_ALREADY_IN_USE')
update_row = (" UPDATE tbl_distribution_circuits "
" SET name = %s, distribution_system_id = %s, distribution_room = %s, switchgear = %s, "
" peak_load = %s, peak_current = %s, customers = %s, meters = %s "
" WHERE id = %s ")
cursor.execute(update_row, (name,
distribution_system_id,
distribution_room,
switchgear,
peak_load,
peak_current,
customers,
meters,
id_))
cnx.commit()
cursor.close()
cnx.close()
resp.status = falcon.HTTP_200
class DistributionCircuitPointCollection:
def __init__(self):
"""Initializes DistributionCircuitPointCollection"""
pass
@staticmethod
def on_options(req, resp, id_):
resp.status = falcon.HTTP_200
@staticmethod
def on_get(req, resp, id_):
if 'API-KEY' not in req.headers or \
not isinstance(req.headers['API-KEY'], str) or \
len(str.strip(req.headers['API-KEY'])) == 0:
access_control(req)
else:
api_key_control(req)
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
query = (" SELECT id, name, uuid "
" FROM tbl_distribution_systems ")
cursor.execute(query)
rows_distribution_systems = cursor.fetchall()
distribution_system_dict = dict()
if rows_distribution_systems is not None and len(rows_distribution_systems) > 0:
for row in rows_distribution_systems:
distribution_system_dict[row[2]] = {"id": row[0],
"name": row[1],
"uuid": row[2]}
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
query = (" SELECT p.id AS point_id, p.name AS point_name, p.address AS point_address, "
" dc.id AS distribution_circuit_id, dc.name AS distribution_circuit_name, "
" dc.uuid AS distribution_circuit_uuid "
" FROM tbl_points p, tbl_distribution_circuits_points dcp, tbl_distribution_circuits dc "
" WHERE dcp.distribution_circuit_id = %s AND p.id = dcp.point_id "
" AND dcp.distribution_circuit_id = dc.id "
" ORDER BY p.name ")
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], "address": row[2],
"distribution_circuit": {"id": row[3],
"name": row[4],
"uuid": row[5]}}
result.append(meta_result)
resp.text = json.dumps(result)
@staticmethod
@user_logger
def on_post(req, resp, id_):
"""Handles POST requests"""
admin_control(req)
try:
raw_json = req.stream.read().decode('utf-8')
except Exception as ex:
raise falcon.HTTPError(status=falcon.HTTP_400,
title='API.BAD_REQUEST',
description='API.FAILED_TO_READ_REQUEST_STREAM')
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
new_values = json.loads(raw_json)
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
cursor.execute(" SELECT name "
" from tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
cursor.execute(" SELECT name "
" FROM tbl_points "
" WHERE id = %s ", (new_values['data']['point_id'],))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.POINT_NOT_FOUND')
query = (" SELECT id "
" FROM tbl_distribution_circuits_points "
" WHERE distribution_circuit_id = %s AND point_id = %s")
cursor.execute(query, (id_, new_values['data']['point_id'],))
if cursor.fetchone() is not None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.ERROR',
description='API.DISTRIBUTION_CIRCUIT_POINT_RELATION_EXISTS')
add_row = (" INSERT INTO tbl_distribution_circuits_points (distribution_circuit_id, point_id) "
" VALUES (%s, %s) ")
cursor.execute(add_row, (id_, new_values['data']['point_id'],))
cnx.commit()
cursor.close()
cnx.close()
resp.status = falcon.HTTP_201
resp.location = '/distributioncircuits/' + str(id_) + '/points/' + str(new_values['data']['point_id'])
class DistributionCircuitPointItem:
def __init__(self):
"""Initializes DistributionCircuitPointItem"""
pass
@staticmethod
def on_options(req, resp, id_, pid):
resp.status = falcon.HTTP_200
@staticmethod
@user_logger
def on_delete(req, resp, id_, pid):
admin_control(req)
if not id_.isdigit() or int(id_) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_DISTRIBUTION_CIRCUIT_ID')
if not pid.isdigit() or int(pid) <= 0:
raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_POINT_ID')
cnx = mysql.connector.connect(**config.myems_system_db)
cursor = cnx.cursor()
cursor.execute(" SELECT name "
" FROM tbl_distribution_circuits "
" WHERE id = %s ", (id_,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_NOT_FOUND')
cursor.execute(" SELECT name "
" FROM tbl_points "
" WHERE id = %s ", (pid,))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.POINT_NOT_FOUND')
cursor.execute(" SELECT id "
" FROM tbl_distribution_circuits_points "
" WHERE distribution_circuit_id = %s AND point_id = %s ", (id_, pid))
if cursor.fetchone() is None:
cursor.close()
cnx.close()
raise falcon.HTTPError(status=falcon.HTTP_404, title='API.NOT_FOUND',
description='API.DISTRIBUTION_CIRCUIT_POINT_RELATION_NOT_FOUND')
cursor.execute(" DELETE FROM tbl_distribution_circuits_points "
" WHERE distribution_circuit_id = %s AND point_id = %s ", (id_, pid))
cnx.commit()
cursor.close()
cnx.close()
resp.status = falcon.HTTP_204