def test_normal_text(self, tmpdir, csv_text, table_name, attr_name_list, expected_table_name, expected_attr_name_list, expected_data_matrix): p_db = tmpdir.join("tmp.db") con = SimpleSQLite(str(p_db), "w") con.create_table_from_csv(csv_text, table_name, attr_name_list) assert con.get_table_name_list() == [expected_table_name] assert expected_attr_name_list == con.get_attr_name_list( expected_table_name) result = con.select(select="*", table_name=expected_table_name) result_matrix = result.fetchall() assert len(result_matrix) == 3 assert result_matrix == expected_data_matrix
def test_normal(self, tmpdir, value, expected): p_db = tmpdir.join("tmp.db") con = SimpleSQLite(str(p_db), "w") con.create_table_from_tabledata(value) assert con.get_table_name_list() == [value.table_name] assert con.get_attr_name_list(value.table_name) == value.header_list result = con.select(select="*", table_name=value.table_name) result_matrix = result.fetchall() assert result_matrix == expected actual = con.select_as_tabledata(column_list=value.header_list, table_name=value.table_name) assert actual == value
def to_table_data(self): from simplesqlite import SimpleSQLite from simplesqlite.sqlquery import SqlQuery con = SimpleSQLite(self._source_data, "r") for table in con.get_table_name_list(): self.__table_name = table attr_name_list = con.get_attr_name_list(table) data_matrix = con.select( select=",".join(SqlQuery.to_attr_str_list(attr_name_list)), table_name=table) yield TableData( table, attr_name_list, data_matrix)
def to_table_data(self): from simplesqlite import SimpleSQLite from simplesqlite.query import AttrList con = SimpleSQLite(self._source_data, "r") for table in con.get_table_name_list(): self.__table_name = table attr_name_list = con.get_attr_name_list(table) data_matrix = con.select(select=AttrList(attr_name_list), table_name=table).fetchall() yield TableData(table, attr_name_list, data_matrix, dp_extractor=self._loader.dp_extractor)
def test_normal_file(self, tmpdir, json_text, filename, table_name, expected_table_name, expected_attr_name_list, expected_data_matrix): p_db = tmpdir.join("tmp.db") p_json = tmpdir.join(filename) with open(str(p_json), "w") as f: f.write(json_text) con = SimpleSQLite(str(p_db), "w") con.create_table_from_json(str(p_json), table_name) assert con.get_table_name_list() == [expected_table_name] assert expected_attr_name_list == con.get_attr_name_list( expected_table_name) result = con.select(select="*", table_name=expected_table_name) result_matrix = result.fetchall() assert len(result_matrix) == 3 assert result_matrix == expected_data_matrix
#!/usr/bin/env python # encoding: utf-8 from __future__ import print_function import json from simplesqlite import SimpleSQLite table_name = "sample_table" con = SimpleSQLite("sample.sqlite", "w") # create table ----- data_matrix = [ [1, 1.1, "aaa", 1, 1], [2, 2.2, "bbb", 2.2, 2.2], [3, 3.3, "ccc", 3, "ccc"], ] con.create_table_from_data_matrix( table_name, attr_name_list=["attr_a", "attr_b", "attr_c", "attr_d", "attr_e"], data_matrix=data_matrix) # display values in the table ----- print(con.get_attr_name_list(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record) # display data type for each column in the table ----- print(json.dumps(con.get_attr_type(table_name), indent=4))
file_path = "sample_data_multi.json" # create sample data file --- with open(file_path, "w") as f: f.write("""{ "table_a" : [ {"attr_b": 4, "attr_c": "a", "attr_a": 1}, {"attr_b": 2.1, "attr_c": "bb", "attr_a": 2}, {"attr_b": 120.9, "attr_c": "ccc", "attr_a": 3} ], "table_b" : [ {"a": 1, "b": 4}, {"a": 2 }, {"a": 3, "b": 120.9} ] }""") # create table --- con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_json(file_path) # output --- for table_name in con.get_table_name_list(): print("table: " + table_name) print(con.get_attr_name_list(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record) print()