def execute_query(self, query, use_parameter="false", param="0"):
cursor = None
try:
cursor = self.get_connection().cursor()
Log.debug("\tExecuting the query: " + query)
if use_parameter == "true":
cursor.execute(query, param)
else:
cursor.execute(query)
Log.debug('\tQuery was executed successfully')
except Exception as ex:
Log.error(ex)
return cursor
def poll(self):
self.ensure_server_available()
# Loop through all messages in the socket till it's empty
# If the lock is active, then the queue is not empty
requestcounter = 0
while self.requestLock:
self.requestLock = False
badsockets = []
inputready = None
outputready = None
try:
inputready, outputready, exceptready = select.select(
self.inputSockets.keys(), self.outputSockets.keys(),
badsockets, 0)
except socket.error, e:
if e[0] == NetworkErrors.EBADF:
Log.error(
"Bad file descriptor! Probably a dead socket passed to select"
)
Log.debug(self.inputSockets.keys())
Log.debug(self.outputSockets.keys())
if badsockets:
Log.error("Bad sockets: %s" % badsockets)
if inputready:
for s in inputready:
endpoint = self.inputSockets[s]
try:
endpoint.recv_msg()
except (ReadError, RuntimeError), e:
Log.error(
"Socket receive error! Closing %s. Reason: %s" %
(endpoint, e))
endpoint.close()
try:
del self.inputSockets[endpoint.socket]
del self.outputSockets[endpoint.socket]
try:
outputready.remove(endpoint.socket)
except ValueError:
pass
except KeyError:
Log.network("Socket missing. In input hangup")
continue
def poll(self):
self.ensure_server_available()
# Loop through all messages in the socket till it's empty
# If the lock is active, then the queue is not empty
requestcounter = 0
while self.requestLock:
self.requestLock = False
badsockets = []
inputready = None
outputready = None
try:
inputready, outputready, exceptready = select.select(
self.inputSockets.keys(),
self.outputSockets.keys(),
badsockets, 0)
except socket.error, e:
if e[0] == NetworkErrors.EBADF:
Log.error("Bad file descriptor! Probably a dead socket passed to select")
Log.debug(self.inputSockets.keys())
Log.debug(self.outputSockets.keys())
if badsockets:
Log.error("Bad sockets: %s" % badsockets)
if inputready:
for s in inputready:
endpoint = self.inputSockets[s]
try:
endpoint.recv_msg()
except (ReadError, RuntimeError), e:
Log.error("Socket receive error! Closing %s. Reason: %s" % (endpoint, e))
endpoint.close()
try:
del self.inputSockets[endpoint.socket]
del self.outputSockets[endpoint.socket]
try:
outputready.remove(endpoint.socket)
except ValueError:
pass
except KeyError:
Log.network("Socket missing. In input hangup")
continue
def get_connection(self):
if self.db_connection is not None:
Log.debug('\tDB Connection Successful')
return self.db_connection
else:
try:
driver = pyodbc.drivers()[0]
Log.debug("\tConnecting to DB Using Driver: " + driver)
username = os.getenv("DB_USERNAME", "dev")
password = os.getenv("DB_PASSWORD", "dera@54321")
db_connection = pyodbc.connect('DRIVER={' + driver + '};' +
'SERVER=' + DBConstants.SERVER +
';' + 'PORT=' +
DBConstants.DEFAULT_PORT + ';' +
'DATABASE=' +
DBConstants.DATABASE + ';' +
'UID=' + username + ';' +
'PWD=' + password)
Log.debug('\tDB Connection Successful')
return db_connection
except Exception as ex:
Log.error(ex)
raise Exception('ERROR: Not able to establish a DB Connection')
