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 simplesqlite import SimpleSQLite import six file_path = "sample_data_single.json" # create sample data file --- with open(file_path, "w") as f: f.write("""[ {"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} ]""") # create table --- con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_json(file_path) # output --- for table_name in con.get_table_name_list(): six.print_("table: " + table_name) six.print_(con.get_attribute_name_list(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): six.print_(record) six.print_()
#!/usr/bin/env python # encoding: utf-8 from simplesqlite import SimpleSQLite import six con = SimpleSQLite("sample.sqlite", "w") con.create_table_with_data( table_name="hoge", attribute_name_list=["attr_a", "attr_b"], data_matrix=[[1, "a"], [2, "b"]]) six.print_(con.get_table_name_list())
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()
#!/usr/bin/env python # encoding: utf-8 from __future__ import print_function from simplesqlite import SimpleSQLite con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_data_matrix( table_name="hoge", attr_name_list=["attr_a", "attr_b"], data_matrix=[[1, "a"], [2, "b"]]) print(con.get_table_name_list())
#!/usr/bin/env python # encoding: utf-8 from __future__ import print_function from simplesqlite import SimpleSQLite con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_data_matrix(table_name="hoge", attr_name_list=["attr_a", "attr_b"], data_matrix=[[1, "a"], [2, "b"]]) print(con.get_table_name_list())