def initialize_service_path_path_db(
self, service_path_ip, service_path_name, service_path_description,
service_path_enabled, created_by):
date_created = datetime.datetime.now()
self.service_path_ip = service_path_ip
self.service_path_name = service_path_name
self.service_path_description = service_path_description
self.service_path_enabled = service_path_enabled
self.created_by = created_by
try:
ServicePathPaths.create(
service_path_ip=service_path_ip,
service_path_name=service_path_name,
service_path_description=service_path_description,
service_path_enabled=service_path_enabled,
created_by=created_by, date_created=date_created)
self.status = 'Service Path Added: {}'.format(service_path_name)
self.error = 0
return self.status
except IntegrityError as out_error:
output = str(out_error)
if "Duplicate" in output and "service_path_ip" in output:
self.status = 'IP is already in use.'
self.error = True
return self.status
if "Duplicate" in output and "service_path_name" in output:
self.status = 'Name is already in use.'
self.error = True
return self.status
def get_service_paths(self, enabled=True):
"""
Get the current list of service Paths from the Service Path Path's table
in the DB.
Enabled Arg:
False = All service paths not enabled
True = All service paths enabled
"""
self.enabled = enabled
if self.enabled == False:
route_list = ServicePathPaths.select()
return route_list
elif self.enabled:
route_list = ServicePathPaths.select() \
.where(ServicePathPaths.service_path_enabled == 1)
return route_list
else:
raise ValueError(
'Enabled argument must be set to either True or False')
