def test_skipping_rows_if_data_exist(self):
mysession = Session()
# write existing data
writer = SQLTableWriter(mysession,
[Pyexcel, self.data[0], None, None])
writer.write_array(self.data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(query_sets, self.data[0]).to_array()
assert list(results) == self.results
# update data using custom initializer
update_data = [
['birth', 'id', 'name', 'weight'],
[datetime.date(2014, 11, 11), 0, 'Adam_E', 12.25],
[datetime.date(2014, 11, 12), 1, 'Smith_E', 11.25]
]
def row_updater(row):
an_instance = mysession.query(Pyexcel).get(row['id'])
if an_instance is not None:
raise PyexcelSQLSkipRowException()
writer = SQLTableWriter(mysession,
[Pyexcel, update_data[0], None, row_updater])
writer.write_array(update_data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(query_sets, self.data[0]).to_array()
assert list(results) == self.results
mysession.close()
def test_one_table_using_mapdict_as_dict(self):
mysession = Session()
self.data = [
["Birth Date", "Id", "Name", "Weight"],
[datetime.date(2014, 11, 11), 0, 'Adam', 11.25],
[datetime.date(2014, 11, 12), 1, 'Smith', 12.25]
]
mapdict = {
"Birth Date": 'birth',
"Id": 'id',
"Name": 'name',
"Weight": 'weight'
}
writer = SQLTableWriter(
mysession,
[Pyexcel, self.data[0], mapdict, None])
writer.write_array(self.data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(
query_sets,
['birth', 'id', 'name', 'weight']).to_array()
assert list(results) == self.results
mysession.close()
def test_one_table(self):
mysession = Session()
writer = SQLTableWriter(mysession,
[Pyexcel, self.data[0], None, None])
writer.write_array(self.data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
reader = QuerysetsReader(query_sets, self.data[0])
results = reader.to_array()
assert list(results) == self.results
mysession.close()
def test_one_table(self):
mysession = Session()
writer = SQLTableWriter(mysession,
[Pyexcel, self.data[0], None, None],
auto_commit=False)
writer.write_array(self.data[1:])
writer.close()
mysession.close()
mysession2 = Session()
query_sets = mysession2.query(Pyexcel).all()
assert len(query_sets) == 0
mysession2.close()
def test_update_existing_row(self):
mysession = Session()
# write existing data
writer = SQLTableWriter(mysession,
[Pyexcel, self.data[0], None, None])
writer.write_array(self.data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(query_sets, self.data[0]).to_array()
assert list(results) == self.results
# update data using custom initializer
update_data = [
['birth', 'id', 'name', 'weight'],
[datetime.date(2014, 11, 11), 0, 'Adam_E', 12.25],
[datetime.date(2014, 11, 12), 1, 'Smith_E', 11.25]
]
updated_results = [
['birth', 'id', 'name', 'weight'],
['2014-11-11', 0, 'Adam_E', 12.25],
['2014-11-12', 1, 'Smith_E', 11.25]
]
def row_updater(row):
an_instance = mysession.query(Pyexcel).get(row['id'])
if an_instance is None:
an_instance = Pyexcel()
for name in row.keys():
setattr(an_instance, name, row[name])
return an_instance
writer = SQLTableWriter(mysession,
[Pyexcel, update_data[0], None, row_updater])
writer.write_array(update_data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(query_sets, self.data[0]).to_array()
assert list(results) == updated_results
mysession.close()
def test_one_table_with_empty_rows(self):
mysession = Session()
data = [
['birth', 'id', 'name', 'weight'],
['', '', ''],
[datetime.date(2014, 11, 11), 0, 'Adam', 11.25],
[datetime.date(2014, 11, 12), 1, 'Smith', 12.25]
]
writer = SQLTableWriter(mysession,
[Pyexcel, data[0], None, None])
writer.write_array(data[1:])
writer.close()
query_sets = mysession.query(Pyexcel).all()
results = QuerysetsReader(query_sets, data[0]).to_array()
assert list(results) == self.results
mysession.close()
def test_one_table(self):
if PY36:
# skip the test
# beause python 3.6 sqlite give segmentation fault
return
mysession = Session()
writer = SQLTableWriter(mysession,
[Pyexcel, self.data[0], None, None],
auto_commit=False)
writer.write_array(self.data[1:])
writer.close()
mysession.close()
mysession2 = Session()
query_sets = mysession2.query(Pyexcel).all()
eq_(len(query_sets), 0)
mysession2.close()
def test_invalid_parameters(self):
mysession = Session()
self.data = [
["Birth Date", "Id", "Name", "Weight"],
[datetime.date(2014, 11, 11), 0, 'Adam', 11.25],
[datetime.date(2014, 11, 12), 1, 'Smith', 12.25]
]
mapdict = {
"Birth Date": 'birth',
"Id": 'id',
"Name": 'name',
"Weight": 'weight'
}
SQLTableWriter(mysession,
[Pyexcel, self.data[0], mapdict, None, None])
mysession.close()