dbmgr.py

import glob
import os
import re
import sqlite3


class DbMgr:
	def __init__(self, dbpath="db"):
		self.db_handles = {}
		self.dbpath = dbpath

	def open(self, guild):
		guild = str(guild)
		if guild in self.db_handles:
			return

		self.db_handles[guild] = self.create_new_conn(guild)
		self.upgrade_db(self.db_handles[guild])

	def get(self, guild):
		self.open(str(guild))

		return self.db_handles[str(guild)]

	def get_all(self):
		for file in glob.glob(f"{self.dbpath}/*.db"):
			search = re.search('/([0-9]+?)\.db', file)
			if search is None or search.group(1) is None:
				continue

			self.open(search.group(1))

		return list(self.db_handles.values())

	def create_new_conn(self, guild):
		connection = sqlite3.connect(f"{self.dbpath}/{guild}.db", check_same_thread=False)
		connection.row_factory = sqlite3.Row
		return connection

	def upgrade_db(self, connection):
		user_version = connection.execute("PRAGMA user_version").fetchone()[0]

		while os.path.isfile(f"db/db-patterns/schema_{user_version + 1}.sql"):
			with open(f"db/db-patterns/schema_{user_version + 1}.sql") as file:
				connection.executescript(file.read())
			user_version = connection.execute("PRAGMA user_version").fetchone()[0]

	def close(self, guild, commit=True):
		guild = str(guild)
		if guild in self.db_handles:
			if commit:
				self.db_handles[guild].commit()
			self.db_handles[guild].close()
			self.db_handles.pop(guild, None)

	def close_all(self):
		while len(self.db_handles) > 0:
			self.close(list(self.db_handles.keys())[0])