From 77d7c52f27ab25de13b58133155b3dc97bc178a3 Mon Sep 17 00:00:00 2001
From: Wojciech Janota <wojciech@janota.ovh>
Date: Wed, 9 Nov 2022 23:50:40 +0100
Subject: [PATCH] NXPR-1 Work increment

---
 test                       |  0
 utils/database/database.py | 28 ++++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 delete mode 100644 test

diff --git a/test b/test
deleted file mode 100644
index e69de29..0000000
diff --git a/utils/database/database.py b/utils/database/database.py
index a8994ea..e2d8a0f 100644
--- a/utils/database/database.py
+++ b/utils/database/database.py
@@ -102,3 +102,31 @@ class Database:
                 self.session.delete(client)
         except Exception as 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}")