Save a species to the database.

1 files changed, 42 insertions, 0 deletions

diff --git a/uploader/species/models.py b/uploader/species/models.py
index 98337a3..bde38ce 100644
--- a/uploader/species/models.py
+++ b/uploader/species/models.py
@@ -1,4 +1,6 @@
 """Database functions for species."""
+from typing import Optional
+
 import MySQLdb as mdb
 from MySQLdb.cursors import DictCursor
 
@@ -20,3 +22,43 @@ def species_by_id(conn: mdb.Connection, speciesid) -> dict:
             "FullName FROM Species WHERE SpeciesId=%s",
             (speciesid,))
         return cursor.fetchone()
+
+
+def save_species(conn: mdb.Connection,
+                 common_name: str,
+                 scientific_name: str,
+                 taxon_id: Optional[str] = None) -> int:
+    """
+    Save a new species to the database.
+
+    Parameters
+    ----------
+    conn: A connection to the MariaDB database.
+    taxon_id: The taxonomy identifier for the new species.
+    common_name: The species' common name.
+    scientific_name; The species' scientific name.
+    """
+    genus, species = scientific_name.split(" ")
+    species = {
+        "common_name": common_name,
+        "common_name_lower": common_name.lower(),
+        "menu_name": f"{common_name} ({genus[0]}. {species.lower()})",
+        "scientific_name": scientific_name,
+        "taxon_id": taxon_id
+    }
+    with conn.cursor() as cursor:
+        cursor.execute(
+            "INSERT INTO Species("
+            "SpeciesName, Name, MenuName, FullName, TaxonomyId"
+            ") VALUES ("
+            "%(common_name)s, %(common_name_lower)s, %(menu_name)s, "
+            "%(scientific_name)s, %(taxon_id)s"
+            ")",
+            species)
+        species_id = cursor.lastrowid
+        cursor.execute("UPDATE Species SET SpeciesId=%s WHERE Id=%s",
+                       (species_id, species_id))
+        return {
+            **species,
+            "species_id": species_id
+        }