46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
import pymssql
|
||
|
|
||
|
|
||
|
class FredDB(object):
|
||
|
conn = False
|
||
|
cursor = False
|
||
|
connected = False
|
||
|
|
||
|
def __init__(self, user, passwd, dbname):
|
||
|
self.user = user
|
||
|
self.passwd = passwd
|
||
|
self.dbname = dbname
|
||
|
self.host = 'NR-CORP-APP1'
|
||
|
|
||
|
def connect(self):
|
||
|
self.conn = pymssql.connect(self.host, self.user,
|
||
|
self.passwd, self.dbname)
|
||
|
self.connected = True
|
||
|
self.cursor = self.conn.cursor()
|
||
|
|
||
|
def close(self):
|
||
|
if self.cursor:
|
||
|
self.cursor.close()
|
||
|
self.cursor = False
|
||
|
if self.conn:
|
||
|
self.conn.close()
|
||
|
self.conn = False
|
||
|
|
||
|
def query(self, sql, data=[]):
|
||
|
if not self.conn:
|
||
|
self.connect()
|
||
|
self.conn.query(sql, data)
|
||
|
return self.conn.use_result()
|
||
|
|
||
|
def execute(self, sql, data=[]):
|
||
|
if not self.conn:
|
||
|
self.connect()
|
||
|
self.cursor.execute(sql, data)
|
||
|
return self.cursor.fetchall()
|
||
|
|
||
|
def executemany(self, sql, data=[]):
|
||
|
if not self.conn:
|
||
|
self.connect()
|
||
|
self.cursor.executemany(sql, data)
|
||
|
self.conn.commit()
|