Add database.py
parent
c3d59e2e32
commit
78d45d045c
@ -0,0 +1,46 @@
|
||||
import sqlite3
|
||||
import sys
|
||||
from abc import ABC, abstractmethod
|
||||
from sqlite3 import Connection
|
||||
|
||||
from exclamation_mark_charity import DB_FILE, LOGGER
|
||||
|
||||
|
||||
class DatabaseInterface(ABC):
|
||||
|
||||
@abstractmethod
|
||||
def connect(self):
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def insert(self):
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def create(self):
|
||||
pass
|
||||
|
||||
|
||||
class Sqlite(DatabaseInterface):
|
||||
def __init__(self):
|
||||
self.conn: Connection = self.connect()
|
||||
|
||||
def connect(self) -> Connection:
|
||||
LOGGER.info(f"Connecting to {DB_FILE}...")
|
||||
|
||||
try:
|
||||
conn = sqlite3.connect(DB_FILE)
|
||||
conn.execute("PRAGMA foreign_keys = ON;")
|
||||
except sqlite3.Error as err:
|
||||
LOGGER.error(err)
|
||||
sys.exit(err)
|
||||
|
||||
LOGGER.info(f"Connection to {DB_FILE} successful!")
|
||||
|
||||
return conn
|
||||
|
||||
def create(self):
|
||||
pass
|
||||
|
||||
def insert(self):
|
||||
pass
|
Loading…
Reference in New Issue