From cace7a50a45c6d227bce52569fcfc2944fbcbe92 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 8 Jan 2024 04:50:42 +0300
Subject: Extract common database functions into a separate package.

---
 qc_app/db/species.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 qc_app/db/species.py

(limited to 'qc_app/db/species.py')

diff --git a/qc_app/db/species.py b/qc_app/db/species.py
new file mode 100644
index 0000000..7006d15
--- /dev/null
+++ b/qc_app/db/species.py
@@ -0,0 +1,24 @@
+"""Database functions for species."""
+import MySQLdb as mdb
+from MySQLdb.cursors import DictCursor
+
+def species(conn: mdb.Connection) -> tuple:
+    "Retrieve the species from the database."
+    with conn.cursor(cursorclass=DictCursor) as cursor:
+        cursor.execute(
+            "SELECT SpeciesId, SpeciesName, LOWER(Name) AS Name, MenuName "
+            "FROM Species")
+        return tuple(cursor.fetchall())
+
+    return tuple()
+
+def species_by_id(conn: mdb.Connection, speciesid) -> dict:
+    "Retrieve the species from the database by id."
+    with conn.cursor(cursorclass=DictCursor) as cursor:
+        cursor.execute(
+            (
+                "SELECT "
+                "SpeciesId, SpeciesName, LOWER(Name) AS Name, MenuName "
+                "FROM Species WHERE SpeciesId=%s"),
+            (speciesid,))
+        return cursor.fetchone()
-- 
cgit v1.2.3