summaryrefslogtreecommitdiff
path: root/mysql_utils.py
blob: e59c3d96de8c2a3aeb3a38e1b0743cfb000bd3a2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import mysqlhack
from secrets import *
from thread_utils import *
from com.ziclix.python.sql import zxJDBC
from traceback import format_exc as trace

class mysql_connect:
    def __init__(self):
        self.conn = zxJDBC.connect(mysql_database, mysql_user, mysql_pass, "com.mysql.jdbc.Driver")
        self.curs = self.conn.cursor()

    def execute(self, query, args=None):
        if args is None:
            return self.curs.execute(query)
        else:
            return self.curs.execute(query, args)

    def fetchall(self):
        return self.curs.fetchall()

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_inst, exc_tb):
        if exc_type is None:
            try:        
                self.conn.commit()
                self.curs.close()
                self.conn.close()
            except:
                print(trace())
        else:
            print(exc_tb)