Connect to SQL and Select using Two Functions

[phillyfa]

This works, but I'd still like to make the attributes more dynamic by allowing user input for the values instead of inside the code execution.

import pyodbc
import csv
import logging
import sys

class SQLConnection(object):
    """Create a database connection and perform DML"""

    # Class attribute
    driver = ''
    server = ''
    database = ''
    username = ''
    password = ''
    table = ''

    def __init__(self, server, database, username, password):

        self.server = server
        self.database = database
        self.username = username
        self.password = password
        self.driver = '{SQL Server}'
        self.trustconn = 'yes'
        self.tablename = 'HumanResources.Department'

        connectstring = 'DRIVER=%s;SERVER=%s;PORT=1433;DATABASE=%s;UID=%s;PWD=%s;' % (self.driver, self.server, self.database, self.username, self.password)
        """Initialize database connection"""

        self.connection = pyodbc.connect(connectstring)

        return

    def getcursor(self):
        return self.connection.cursor()

    def select_from_database(self):

        cursor = self.getcursor()
        cursor.execute("SELECT * FROM %s " % self.tablename)
        row = cursor.fetchone()
        while row:
            print("%s %s" % (row[0], row[1]))
            row = cursor.fetchone()
        return row

def main():
    conn = SQLConnection('server', 'database', 'username', 'password' )
    conn.select_from_database()

if __name__ == "__main__":
    main()

How can I pass values to those variables/arguments inside the class and/or at execution?