def insert_rows_from_pg(self, model, columns, transformer=None):
table_dumper = TableDumper(model, columns, transformer)
with tempfile.NamedTemporaryFile() as f:
table_dumper.dump_to_file(f)
f.seek(0)
self.insert_rows_from_csv(f.name)
def test_dump_to_file_without_transformer(self):
dumper = TableDumper(PCT, ["code", "name"])
with tempfile.TemporaryFile("w+t") as f:
dumper.dump_to_file(f)
f.seek(0)
records = list(csv.reader(f))
self.assertEqual(records, [["ABC", "CCG 1"], ["XYZ", "CCG 2"]])
def test_dump_to_file_without_transformer(self):
dumper = TableDumper(PCT, ['code', 'name'])
with tempfile.TemporaryFile() as f:
dumper.dump_to_file(f)
f.seek(0)
records = list(csv.reader(f))
self.assertEqual(records, [['ABC', 'CCG 1'], ['XYZ', 'CCG 2']])
def test_dump_to_file_with_transformer(self):
def transformer(row):
return [row[0][::-1], row[1].lower()]
dumper = TableDumper(PCT, ["code", "name"], transformer)
with tempfile.TemporaryFile("w+t") as f:
dumper.dump_to_file(f)
f.seek(0)
records = list(csv.reader(f))
self.assertEqual(records, [["CBA", "ccg 1"], ["ZYX", "ccg 2"]])
def test_dump_to_file_with_transformer(self):
def transformer(row):
return [row[0][::-1], row[1].lower()]
dumper = TableDumper(PCT, ['code', 'name'], transformer)
with tempfile.TemporaryFile() as f:
dumper.dump_to_file(f)
f.seek(0)
records = list(csv.reader(f))
self.assertEqual(records, [['CBA', 'ccg 1'], ['ZYX', 'ccg 2']])
def insert_rows_from_pg(self,
model,
schema,
columns=None,
transformer=None):
if columns is None:
columns = [field.name for field in schema]
table_dumper = TableDumper(model, columns, transformer)
with tempfile.NamedTemporaryFile("w+t") as f:
table_dumper.dump_to_file(f)
f.seek(0)
self.insert_rows_from_csv(f.name, schema)