dealWithDBResults.py 文件源码

def getColumns(conn, connType, tableName, exampleNameForDB):
    retval = False
    if connType == DBTYPE_MYSQL:
        query = "SHOW COLUMNS FROM {}".format(tableName)
    else:
        query = "PRAGMA table_info({})".format(tableName)
    try:
        cursor = conn.cursor()
        cursor.execute(query)
        retval = cursor.fetchall()
        cursor.close()
    except (MySQLdb.ProgrammingError, sqlite3.Error,) as e:
        print "TABLE {} not found for example {}: {}".format(tableName,exampleNameForDB, e.args)
    if connType == DBTYPE_SQLITE:
        retval = [(x[1],) for x in retval] # weird unSELECTable tuple retval
    return retval