From 12f5079381a45e9886ee02a1f26bd5bcfa3726e8 Mon Sep 17 00:00:00 2001 From: brmbar Date: Mon, 1 Jul 2013 14:55:54 +0200 Subject: [PATCH] Catching exception when sql query gets invalid data like too large number --- brmbar3/brmbar/Database.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/brmbar3/brmbar/Database.py b/brmbar3/brmbar/Database.py index e5962ab..d0202d4 100644 --- a/brmbar3/brmbar/Database.py +++ b/brmbar3/brmbar/Database.py @@ -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")