From 10a444dcc374a7bb67d0d4384ab9e07383051fc6 Mon Sep 17 00:00:00 2001
From: Lukasz Mentel <lmmentel@users.noreply.github.com>
Date: Wed, 6 Nov 2024 18:07:26 +0100
Subject: [PATCH] improve export function (#205)

---
 tasks.py | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/tasks.py b/tasks.py
index 72e652ff..897d0949 100644
--- a/tasks.py
+++ b/tasks.py
@@ -3,9 +3,9 @@
 
 
 @task
-def export(c):
+def export(c, dest="data"):
     """Export data to a few formats files."""
-    tables = [
+    tables = {
         "elements",
         "groups",
         "ionicradii",
@@ -14,9 +14,11 @@ def export(c):
         "isotopes",
         "oxidationstates",
         "phasetransitions",
+        "propertymetadata",
+        "scattering_factors",
         "screeningconstants",
         "series",
-    ]
+    }
 
     formats = [
         "csv",
@@ -25,20 +27,26 @@ def export(c):
         "markdown",
     ]
 
+    root = Path(f"{dest}")
+    root.mkdir(exist_ok=True)
+
     for fmt in formats:
-        Path(f"data/{fmt}").mkdir(exist_ok=True)
+        path = root.joinpath(fmt)
+        path.mkdir(exist_ok=True)
         for table in tables:
             print(f"Exporting {table} to {fmt} ... ", end="")
             c.run(
-                f"sqlite3 -{fmt} mendeleev/elements.db 'SELECT * FROM {table};' > data/{fmt}/{table}.{fmt}",
+                f"sqlite3 -{fmt} mendeleev/elements.db 'SELECT * FROM {table};' > {path}/{table}.{fmt}",
                 echo=True,
             )
             print("done")
 
     # SQL dump
     print("Creating SQL files with the data ... ", end="")
+    path = root.joinpath("sql")
+    path.mkdir(exist_ok=True)
     c.run(
-        "sqlite3 mendeleev/elements.db .dump > data/sql/elements.sql",
+        f"sqlite3 mendeleev/elements.db .dump > {path}/mendeleev.sql",
         echo=True,
         pty=True,
     )