1
0
mirror of https://github.com/MarkParker5/STARK.git synced 2025-07-02 22:36:54 +02:00
Files
STARK/DataBase/DataBase.py

66 lines
2.2 KiB
Python
Raw Normal View History

2020-10-04 21:18:36 +03:00
import sqlite3
from sqlite3 import Error
def sqlite(func):
def wrapper(this, *args):
result = None
try:
connect = sqlite3.connect('archie.db')
cursor = connect.cursor()
result = func(this, cursor, *args)
connect.commit()
except Error as er:
print('SQLite error: %s' % (' '.join(er.args)))
print("Exception class is: ", er.__class__)
print('SQLite traceback: ')
finally:
connect.close()
return result
return wrapper
class DataBase:
@sqlite
def __init__(this, cursor, table_name, columns = None):
this.table_name = table_name
if not cursor.execute(f'select * from sqlite_master WHERE type = "table" AND name = "{this.table_name}"').fetchall():
if columns:
cursor.execute(f'create table if not exists {this.table_name}(id integer PRIMARY KEY, {", ".join(columns)})')
this.columns = ['id',]+columns
else:
this.columns = [properties[1] for properties in cursor.execute(f'pragma table_info({this.table_name})').fetchall()]
@sqlite
def all(this, cursor):
rows = cursor.execute(f'select * from {this.table_name}').fetchall()
data = []
for row in rows:
dict = {}
for i, value in enumerate(row):
dict[this.columns[i]] = value
data.append(dict)
return data
@sqlite
def where(this, cursor, condition):
return cursor.execute(f'select * from {this.table_name} where {condition}')
@sqlite
def count(this, count):
return len(this.all())
@sqlite
def update(this, cursor, values, where):
updates = " ".join([f'{key} = "{value}"' for key, value in values.items()])
cursor.execute(f'update {this.table_name} set {updates} where {where}')
@sqlite
def insert(this, cursor, values):
values = ['"'+v+'"' for v in values]
cursor.execute(f'insert into {this.table_name}({", ".join(this.columns[1:])}) values({", ".join(values)})')
@sqlite
def drop(this, cursor):
cursor.execute(f'drop table if exists {this.table_name}')