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()
