You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
import sqlite3
|
|
import sys
|
|
from contextlib import closing
|
|
from sqlite3 import Connection
|
|
|
|
from exclamation_mark_charity.constants import DB_FILE
|
|
from exclamation_mark_charity.logger_factory import LoggerFactory
|
|
|
|
logger = LoggerFactory.get_logger(__name__)
|
|
|
|
|
|
class Database:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@staticmethod
|
|
def __connect() -> 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.critical(err)
|
|
sys.exit(err)
|
|
|
|
logger.info(f"Connection to '{DB_FILE}' Successful!")
|
|
return conn
|
|
|
|
@staticmethod
|
|
def create_tournament(name: str) -> (bool, int):
|
|
sql = '''INSERT INTO Tournament (name) VALUES (?)'''
|
|
tournament_id = -1
|
|
|
|
try:
|
|
with closing(Database.__connect()) as conn:
|
|
with conn:
|
|
with closing(conn.cursor()) as cur:
|
|
cur.execute(sql, [name])
|
|
tournament_id = cur.lastrowid
|
|
except sqlite3.Error as err:
|
|
logger.error(err)
|
|
return False, tournament_id
|
|
|
|
return True, tournament_id
|
|
|
|
@staticmethod
|
|
def view_all_tournaments() -> (bool, list):
|
|
sql = '''SELECT * FROM Tournament'''
|
|
results = []
|
|
|
|
try:
|
|
with closing(Database.__connect()) as conn:
|
|
with conn:
|
|
with closing(conn.cursor()) as cur:
|
|
cursor = cur.execute(sql)
|
|
results.append([desc[0] for desc in cursor.description[1:]])
|
|
for row in cursor:
|
|
results.append(row[1:])
|
|
except sqlite3.Error as err:
|
|
logger.error(err)
|
|
return False, results
|
|
|
|
return True, results
|
|
|
|
@staticmethod
|
|
def delete_many_tournaments(tables: []) -> bool:
|
|
sql = '''DELETE FROM Tournament WHERE name = (?)'''
|
|
|
|
try:
|
|
with closing(Database.__connect()) as conn:
|
|
with conn:
|
|
conn.executemany(sql, tables)
|
|
except sqlite3.Error as err:
|
|
logger.error(err)
|
|
return False
|
|
|
|
return True
|
|
|
|
def insert(self):
|
|
pass
|