gnomex.py

#!/usr/bin/env python

import pyodbc
import re
import os
import pipelineparams as params

class GNomExConnection:

  def CleanQuery( self, query ):
    query = re.sub("\t"," ",query)
    return query

  def GnConnect(self,db_user,db_password,asdict=False):
    '''
    Returns a database connection. Next step is to call its
    cursor() method, then cursor.execute(query) and cursor.fetchall()
    for example.
    '''

    try:
      host=os.environ['DBHOST']
    except KeyError:
      host='hci-db.hci.utah.edu,1433'
    database='GNomEx'
    return pyodbc.connect(DRIVER='{ODBC Driver 17 for SQL Server}',server=host,uid=db_user,pwd=db_password,database=database)

def test():
  g = GNomExConnection()
  connection = g.GnConnect(db_user=params.db_user,db_password=params.db_password,asdict=False)
  c = connection.cursor()
  c.execute("select count(*) n from AppUser")
  results = c.fetchall()
  try:
    #print "There are %d registered GNomEx users." % results[0][0]
    print("There are %d registered GNomEx users." % results[0][0])
  except:
    print("results=",results)
    raise

if __name__ == "__main__":
  test()