Initial

images.py

@@ -0,0 +1,74 @@
+from datetime import datetime
+import typing
+from uuid import UUID, uuid4
+
+from database import Database
+
+class PetPicture:
+    uuid: str
+    title: str
+    contributor: str
+    created_at: int
+    posted: bool
+    filepath: str
+    description: str
+
+    def __init__(self, title, contributor, created_at, filepath, description, uuid=None, posted=False):
+        self.uuid = str(uuid4()) if uuid is None else uuid
+        self.title = title
+        self.contributor = contributor
+        self.created_at = created_at
+        self.posted = posted
+        self.filepath = filepath
+        self.description = description
+
+    def get_json(self):
+        return {
+                "uuid": self.uuid,
+                "title": self.title,
+                "contributor": self.contributor,
+                "description": self.description,
+                "created_at": self.created_at,
+                "posted": self.posted,
+                "filepath": self.filepath,
+        }
+
+    def upsert(self):
+        conn = Database.get_connection()
+        c = conn.cursor()
+        save_string = """INSERT INTO PetPictures (uuid, title, contributor, description, created_at, posted, filepath) 
+            VALUES (?, ?, ?, ?, ?, ?, ?) 
+            ON CONFLICT (uuid) DO UPDATE SET title = excluded.title, contributor = excluded.contributor, created_at = excluded.created_at, posted = excluded.posted, filepath = excluded.filepath, description = excluded.description"""
+        c.execute(save_string, (self.uuid, self.title, self.contributor, self.description, self.created_at, self.posted, self.filepath,))
+        conn.commit()
+        conn.close()
+
+    @staticmethod
+    def from_row(db_row):
+        return PetPicture(uuid=db_row[0], title=db_row[1], contributor=db_row[2], description=db_row[3], created_at=db_row[4], filepath=db_row[6], posted=db_row[5])
+
+    @staticmethod
+    def get(uuid: str):
+        conn = Database.get_connection()
+        c = conn.cursor()
+        FETCH_QUERY = "SELECT * FROM PetPictures WHERE UUID=?"
+        c.execute(FETCH_QUERY, (uuid,))
+        row = c.fetchone()
+        return PetPicture.from_row(row)
+
+    @staticmethod
+    def get_all():
+        conn =  Database.get_connection()
+        c = conn.cursor()
+        FETCH_QUERY = "SELECT * FROM PetPictures ORDER BY created_at DESC"
+        c.execute(FETCH_QUERY)
+        rows = c.fetchall()
+        conn.close()
+        return [
+                PetPicture.from_row(row) for row in rows
+        ]
+
+
+
+
+