def delete(self, entity):
connection = ConnectorFactory().build_connector().get_connection()
cursor = connection.cursor()
cursor.execute(DELETE_ALARM, (entity.id, ))
connection.commit()
connection.close()
def update(self, entity) -> Alarm:
if type(entity) is not Alarm:
raise ValueError("'entity' must be of type 'Alarm'")
if entity.id <= 0:
return self.insert(entity)
else:
connection = ConnectorFactory().build_connector().get_connection()
cursor = connection.cursor()
cursor.execute(UPDATE_ALARM,
(entity.name, entity.hour, entity.minute,
entity.last_activated, entity.id))
connection.commit()
connection.close()
return self.find_by_id(entity.id)
def insert(self, entity) -> Alarm:
if type(entity) is not Alarm:
raise ValueError("'entity' must be of type 'Alarm'")
connection = ConnectorFactory().build_connector().get_connection()
cursor = connection.cursor()
cursor.execute(
INSERT_ALARM,
(entity.name, entity.hour, entity.minute, entity.last_activated))
entity.id = cursor.lastrowid
connection.commit()
connection.close()
return entity
def find_by_id(self, entity_id) -> Alarm:
connection = ConnectorFactory().build_connector().get_connection()
cursor = connection.cursor()
result = None
for row in cursor.execute(FIND_ALARM, (entity_id, )):
if result is not None:
raise RuntimeError(
"Entities with the same id are present in the sqlite table."
)
result = row
connection.commit()
connection.close()
if result is not None:
return AlarmResultConverter().convert_cursor_result_to_alarm(
result)
else:
return None