def test_normal_text( self, tmpdir, csv_text, table_name, attr_names, expected_table_name, expected_attr_names, expected_data_matrix, ): p_db = tmpdir.join("tmp.db") con = SimpleSQLite(str(p_db), "w") try: con.create_table_from_csv(csv_text, table_name, attr_names) except ImportError: pytest.skip("requires pytablereader") assert con.fetch_table_names() == [expected_table_name] assert expected_attr_names == con.fetch_attr_names(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_file( self, tmpdir, csv_text, csv_filename, table_name, attr_names, expected_table_name, expected_attr_names, expected_data_matrix, ): p_db = tmpdir.join("tmp.db") p_csv = tmpdir.join(csv_filename) with open(str(p_csv), "w") as f: f.write(csv_text) con = SimpleSQLite(str(p_db), "w") con.create_table_from_csv(str(p_csv), table_name, attr_names) assert con.fetch_table_names() == [expected_table_name] assert expected_attr_names == con.fetch_attr_names(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_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 main(): with open("sample_data.csv", "w") as f: f.write("\n".join([ '"attr_a","attr_b","attr_c"', '1,4,"a"', '2,2.1,"bb"', '3,120.9,"ccc"' ])) # create table --- con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_csv("sample_data.csv") # output --- table_name = "sample_data" print(con.fetch_attr_names(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record)
#!/usr/bin/env python # encoding: utf-8 from __future__ import print_function from simplesqlite import SimpleSQLite with open("sample_data.csv", "w") as f: f.write("\n".join(['"attr_a","attr_b","attr_c"', '1,4,"a"', '2,2.1,"bb"', '3,120.9,"ccc"'])) # create table --- con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_csv("sample_data.csv") # output --- table_name = "sample_data" print(con.fetch_attr_names(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record)
#!/usr/bin/env python # encoding: utf-8 from simplesqlite import SimpleSQLite import six file_path = "sample_data.csv" with open(file_path, "w") as f: f.write("\n".join([ '"attr_a","attr_b","attr_c"', '1,4,"a"', '2,2.1,"bb"', '3,120.9,"ccc"', ])) # create table --- con = SimpleSQLite("sample.sqlite", "w") con.create_table_from_csv(file_path) # output --- table_name = "sample_data" 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)