Merge branch 'refactor'

1 files changed, 22 insertions, 0 deletions

diff --git a/jfr_playoff/db.py b/jfr_playoff/db.py
new file mode 100644
index 0000000..b94f3d5
--- /dev/null
+++ b/jfr_playoff/db.py
@@ -0,0 +1,22 @@
+import mysql.connector
+
+
+class PlayoffDB(object):
+
+    db_cursor = None
+
+    def __init__(self, settings):
+        self.database = mysql.connector.connect(
+            user=settings['user'],
+            password=settings['pass'],
+            host=settings['host'],
+            port=settings['port'])
+        self.db_cursor = self.database.cursor(buffered=True)
+
+    def get_cursor(self):
+        return self.db_cursor
+
+    def fetch(self, db_name, sql, params):
+        self.db_cursor.execute(sql.replace('#db#', db_name), params)
+        row = self.db_cursor.fetchone()
+        return row