data_sources.sqlite_schema Class Reference

Inheritance diagram for data_sources.sqlite_schema:

Public Member Functions
def	__init__
def	data
Public Member Functions inherited from data_sources.data_source
def	__init__
def	__repr__
def	get_data
Public Member Functions inherited from data_sources.node
def	__init__
def	__str__
def	add_child
def	child
def	children
def	data
def	is_leaf
def	left_child
def	right_child

Private Attributes
	_data
	_file_name

Detailed Description

Definition at line 79 of file data_sources.py.

Constructor & Destructor Documentation

def data_sources.sqlite_schema.__init__	(		self,
			sqlite_file_name
	)

Definition at line 81 of file data_sources.py.

    def __init__(self, sqlite_file_name):
        self._file_name = sqlite_file_name
        # import sqlite3 and connect to the database file
        import sqlite3
        connection = sqlite3.connect(self._file_name)
        cursor = connection.cursor()
        if query_object == None:
            # try to query the file to get table and column data
            tables = cursor.execute("select name from sqlite_master where type = 'table'")

            # now build a mapping of tables to columns - with a dictionary
            table_to_columns = {}
            for table in tables.fetchall():
                table_to_columns[table[0]] = []
                # now query columns for this table
                columns = cursor.execute("pragma table_info(%s)" % table[0])
                for column in columns.fetchall():
                    table_to_columns[table[0]].append(str(column[1]))

            # now query with the mapping
            table_to_data = {}
            for table in table_to_columns:
                # query with all columns
                column_string = ",".join(table_to_columns[table])
                sql_query = "select %s from %s" % (column_string, table)
                results = cursor.execute(sql_query).fetchall()
                for n in range(0, len(results)):
                    results[n] = dict(list(zip(table_to_columns[table], list(map(str, results[n])))))
                table_to_data[str(table)] = results
            self._data = json_data_node.make(table_to_data)
        else:
            sql_query = query_object.to_sql()