From f91d49d987065df43cbaa8264d349885f001aa17 Mon Sep 17 00:00:00 2001
From: scragly <29337040+scragly@users.noreply.github.com>
Date: Thu, 19 Sep 2019 22:59:46 +1000
Subject: Add persistent datafile utils.

---
 bot/utils/persist.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 bot/utils/persist.py

(limited to 'bot/utils/persist.py')

diff --git a/bot/utils/persist.py b/bot/utils/persist.py
new file mode 100644
index 00000000..ec6f306a
--- /dev/null
+++ b/bot/utils/persist.py
@@ -0,0 +1,24 @@
+import sqlite3
+from pathlib import Path
+from shutil import copyfile
+
+DIRECTORY = Path("data")  # directory that has a persistent volume mapped to it
+
+
+def datafile(file_path: Path) -> Path:
+    """Copy datafile at the provided file_path to the persistent data directory."""
+    if not file_path.exists():
+        raise OSError(f"File not found at {file_path}.")
+
+    persistant_path = Path(DIRECTORY, file_path.name)
+
+    if not persistant_path.exists():
+        copyfile(file_path, persistant_path)
+
+    return persistant_path
+
+
+def sqlite(db_path: Path) -> sqlite3.Connection:
+    """Copy sqlite file to the persistent data directory and return an open connection."""
+    persistant_path = datafile(db_path)
+    return sqlite3.connect(persistant_path)
-- 
cgit v1.2.3