brmlab/brmbar
4
0
Fork 0
forked from brmlab/brmbar-github
Code Issues Pull requests Projects 1 Wiki Activity

Catching exception when sql query gets invalid data like too large number

Browse source
This commit is contained in:
brmbar 2013-07-01 14:55:54 +02:00
parent 3ecbf192e2
commit 12f5079381
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
brmbar3/brmbar/Database.py
View file

@ -36,6 +36,12 @@ class Database:
else:
cur.execute(query, attrs)
return cur
except psycopg2.DataError as error: # when biitr comes and enters '99999999999999999999' for amount
print("We have invalid input data (SQLi?): level %s (%s) @%s" % (
level, error, time.strftime("%Y%m%d %a %I:%m %p")
))
self.db_conn.rollback()
raise RuntimeError("Unsanitized data entered again... BOBBY TABLES")
except psycopg2.OperationalError as error:
print("Sleeping: level %s (%s) @%s" % (
level, error, time.strftime("%Y%m%d %a %I:%m %p")