Hexor/conf_bot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/conf_bot.git synced 2025-07-06 14:24:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add sql query support

Browse Source
This commit is contained in:
root
2018-02-26 15:50:32 +01:00
parent 0135fb7320
commit 9e69de3db1
2 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
database.py
View File

@ -125,6 +125,21 @@ class DataBase:
result = self.execute(sql)
return(result)
def command(self, sql):
if 'DELETE' in sql.upper() \
or 'INSERT' in sql.upper() \
or 'UPDATE' in sql.upper() \
or 'CREATE' in sql.upper() \
or 'ALTER' in sql.upper():
return('gtfo')
try:
if 'LIMIT' in sql.upper()[-9:]:
result = self.execute(sql)
else:
result = self.execute(sql + ' limit 20')
except Exception as err:
result = err
return(result)
def close(self):
self.conn.close()