def delete(self):
try:
Website.get(url=self.url)
if Website.delete().where((Website.customer == self.customer)
and (Website.url == self.url)).execute():
return {"message": "Website deleted successfully"}
except DoesNotExist:
if DoesNotExist:
return {"message": "Website does not exist"}
def update(self):
try:
Website.get(url=self.url)
except DoesNotExist:
if DoesNotExist:
return {"message": "Website does not exist"}
else:
Website.update(url=self.url).where(
Website.customer == self.customer).returning(Website)
return {"message": "Wesite updated successfully"}
