47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import sqlite3
|
|
import os.path
|
|
import logging
|
|
|
|
latest_db_version = 0
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Storage(object):
|
|
def __init__(self, db_path):
|
|
"""Setup the database
|
|
|
|
Runs an initial setup or migrations depending on whether a database file has already
|
|
been created
|
|
|
|
Args:
|
|
db_path (str): The name of the database file
|
|
"""
|
|
self.db_path = db_path
|
|
|
|
# Check if a database has already been connected
|
|
if os.path.isfile(self.db_path):
|
|
self._run_migrations()
|
|
else:
|
|
self._initial_setup()
|
|
|
|
def _initial_setup(self):
|
|
"""Initial setup of the database"""
|
|
logger.info("Performing initial database setup...")
|
|
|
|
# Initialize a connection to the database
|
|
conn = sqlite3.connect(self.db_path)
|
|
self.cursor = conn.cursor()
|
|
|
|
# Sync token table
|
|
self.cursor.execute("CREATE TABLE sync_token ("
|
|
"token TEXT PRIMARY KEY"
|
|
")")
|
|
|
|
logger.info("Database setup complete")
|
|
|
|
def _run_migrations(self):
|
|
"""Execute database migrations"""
|
|
# Initialize a connection to the database
|
|
conn = sqlite3.connect(self.db_path)
|
|
self.cursor = conn.cursor()
|