self-hosted/STARK
1
0
Fork 0
mirror of https://github.com/MarkParker5/STARK.git synced 2025-07-02 22:36:54 +02:00
Code Issues Releases Activity

replace this with self in all classes

Browse Source
This commit is contained in:
MarkParker5
2021-08-31 23:09:39 +03:00
parent 9ddc85c765
commit 090df2950c
14 changed files with 140 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
General/DataBase/DataBase.py
View File

@ -3,12 +3,12 @@ from sqlite3 import Error
import config
def sqlite(func):
def wrapper(this, *args):
def wrapper(self, *args):
result = None
try:
connect = sqlite3.connect(config.db_name)
cursor = connect.cursor()
result = func(this, cursor, *args)
result = func(self, cursor, *args)
connect.commit()
except Error as er:
print('SQLite error: %s' % (' '.join(er.args)))
@ -22,45 +22,45 @@ def sqlite(func):
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():
def __init__(self, cursor, table_name, columns = None):
self.table_name = table_name
if not cursor.execute(f'select * from sqlite_master WHERE type = "table" AND name = "{self.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
cursor.execute(f'create table if not exists {self.table_name}(id integer PRIMARY KEY, {", ".join(columns)})')
self.columns = ['id',]+columns
else:
this.columns = [properties[1] for properties in cursor.execute(f'pragma table_info({this.table_name})').fetchall()]
self.columns = [properties[1] for properties in cursor.execute(f'pragma table_info({self.table_name})').fetchall()]
@sqlite
def all(this, cursor):
rows = cursor.execute(f'select * from {this.table_name}').fetchall()
def all(self, cursor):
rows = cursor.execute(f'select * from {self.table_name}').fetchall()
data = []
for row in rows:
dict = {}
for i, value in enumerate(row):
dict[this.columns[i]] = value
dict[self.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}')
def where(self, cursor, condition):
return cursor.execute(f'select * from {self.table_name} where {condition}')
@sqlite
def count(this, count):
return len(this.all())
def count(self, count):
return len(self.all())
@sqlite
def update(this, cursor, values, where):
def update(self, 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}')
cursor.execute(f'update {self.table_name} set {updates} where {where}')
@sqlite
def insert(this, cursor, values):
def insert(self, cursor, values):
values = ['"'+v+'"' for v in values]
cursor.execute(f'insert into {this.table_name}({", ".join(this.columns[1:])}) values({", ".join(values)})')
cursor.execute(f'insert into {self.table_name}({", ".join(self.columns[1:])}) values({", ".join(values)})')
@sqlite
def drop(this, cursor):
cursor.execute(f'drop table if exists {this.table_name}')
def drop(self, cursor):
cursor.execute(f'drop table if exists {self.table_name}')