Revize fa03de5c

Re #8476 - Auxiliary script for database initialization, FileAnchor

src/constants.py

@@ +1 @@
+from src.utils.file_anchor import FileAnchor
+
 DATABASE_FILE = "db/database_sqlite.db"
+DATABASE_FILE_LOCATION = FileAnchor("aswi2021jmsd", DATABASE_FILE)
+
 
 # Types of certificates
 ROOT_CA_ID = 1

src/db/setup_database.py

@@ +1 @@
+import sqlite3
+
+from src.constants import DATABASE_FILE_LOCATION
+from src.db.init_queries import SCHEMA_SQL, DEFAULT_VALUES_SQL
+
+
+def setup_database():
+    co = sqlite3.connect(DATABASE_FILE_LOCATION.shortest_relative_path())
+    cu = co.cursor()
+    cu.executescript(SCHEMA_SQL)
+    cu.executescript(DEFAULT_VALUES_SQL)
+
+
+if __name__ == '__main__':
+    setup_database()

src/utils/file_anchor.py

@@ +1 @@
+import os
+
+
+class FileAnchor:
+    def __init__(self, root_dir, tail, searching=False):
+        self.__root_dir = root_dir
+        self.__tail = tail
+        if searching:
+            self.shortest_relative_path = self.__srp_with_tree_search
+        else:
+            self.shortest_relative_path = self.__srp
+
+    def __srp_with_tree_search(self, src=None):
+        """
+        Shortest relative path handling the situation where src is not a descendant of root_dir.
+        Searches the subtree of src until it finds root_dir.
+        :param src: source node or cwd
+        :return: path from source to the anchored file
+        """
+        raise NotImplementedError("Not implemented yet.")
+
+    def __srp(self, src=None):
+        """
+        Shortest relative path from src to the anchored file. If src is not a descendant of root_dir, returns None.
+        :param src: source node or cwd
+        :return: path from source to the anchored file or None
+        """
+        if src is None:
+            src = os.getcwd()
+
+        ret = ""
+        h = src
+        while len(h) > 0 and h != "/":
+            h, t = os.path.split(h)
+            if t != self.__root_dir:
+                ret = os.path.join(ret, "..")
+            else:
+                return os.path.join(ret, self.__tail)
+
+        return None
+
+    def __str__(self):
+        return self.shortest_relative_path(None)