Add sql query feature.

This commit is contained in:
Alexandr
2018-02-26 19:13:04 +01:00
parent 9e69de3db1
commit 51e2fe48db
2 changed files with 15 additions and 0 deletions

View File

@ -3,6 +3,7 @@ import datetime as dt
class DataBase:
def __init__(self, basefile, scheme):
import sqlite3
self.scheme = ''
try:
self.conn = sqlite3.connect(basefile, check_same_thread=False)
except:
@ -10,6 +11,7 @@ class DataBase:
return None
with open(scheme, 'r') as scheme_sql:
sql = scheme_sql.read()
self.scheme = sql
if self.conn is not None:
try:
cursor = self.conn.cursor()
@ -129,6 +131,7 @@ class DataBase:
if 'DELETE' in sql.upper() \
or 'INSERT' in sql.upper() \
or 'UPDATE' in sql.upper() \
or 'DROP' in sql.upper() \
or 'CREATE' in sql.upper() \
or 'ALTER' in sql.upper():
return('gtfo')