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 124 of file data_sources.py.

Constructor & Destructor Documentation

def data_sources.sqlite_schema.__init__	(		self,
			sqlite_file_name
	)

Definition at line 126 of file data_sources.py.

    def __init__(self, sqlite_file_name):
        self._file_name = sqlite_file_name
        try:
            # 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(zip(table_to_columns[table], 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()

        except IOError as io:
            exit("Couldn't open the file '%s'" % self._file_name)