engineering-degree/orchestrator-app
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

NXPR-1 Work increment

Browse Source
This commit is contained in:
Wojciech Janota 2022-11-09 23:50:40 +01:00
parent 26e7e4b2a8
commit 77d7c52f27
No known key found for this signature in database
GPG Key ID: E83FBD2850CC1F14
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
test
View File

28
utils/database/database.py
View File

@ -102,3 +102,31 @@ class Database:
self.session.delete(client) self.session.delete(client)
except Exception as ex: except Exception as ex:
self.logger.error(f"Error deleting client from database: {ex}") self.logger.error(f"Error deleting client from database: {ex}")
def get_image_by_id(self, image_id: int) -> VMImage:
try:
with self.session.begin():
response = self.session.query(
VMImage, image_id=image_id).first()
return response
except Exception as ex:
self.logger.error(f"Error getting image data from database: {ex}")
def get_images(self) -> list[VMImage]:
try:
with self.session.begin():
response = self.session.query(VMImage).all()
return response
except Exception as ex:
self.logger.error(
f"Error getting list of images from database: {ex}")
def get_image_by_name(self, image_name: str) -> list[VMImage]:
try:
with self.session.begin():
response = self.session.query(
VMImage, image_name=image_name).all()
return response
except Exception as ex:
self.logger.error(
f"Error getting list of images from database: {ex}")