+"""Database management."""
+from os.path import isfile
+from sqlite3 import connect as sql_connect
+from plomtask.misc import HandledException
+
+PATH_DB_SCHEMA = 'scripts/init.sql'
+
+
+class DatabaseFile:
+ """Represents the sqlite3 database's file."""
+
+ def __init__(self, path):
+ self.path = path
+ self.check()
+
+ def check(self):
+ """Check file exists and is of proper schema."""
+ self.exists = isfile(self.path)
+ if self.exists:
+ self.validate_schema()
+
+ def remake(self):
+ """Create tables in self.path file as per PATH_DB_SCHEMA sql file."""
+ with sql_connect(self.path) as conn:
+ with open(PATH_DB_SCHEMA, 'r', encoding='utf-8') as f:
+ conn.executescript(f.read())
+ self.check()
+
+ def validate_schema(self):
+ """Compare found schema with what's stored at PATH_DB_SCHEMA."""
+ sql_for_schema = 'SELECT sql FROM sqlite_master ORDER BY sql'
+ msg_wrong_schema = 'Database has wrong tables schema.'
+ with sql_connect(self.path) as conn:
+ schema_rows = [r[0] for r in conn.execute(sql_for_schema) if r[0]]
+ retrieved_schema = ';\n'.join(schema_rows) + ';'
+ with open(PATH_DB_SCHEMA, 'r', encoding='utf-8') as f:
+ stored_schema = f.read().rstrip()
+ if stored_schema != retrieved_schema:
+ raise HandledException(msg_wrong_schema)