def test_export_csv(self):
"Export metadata to CSV"
with temp_directory_with_files(TESTDATA / 'shapes.inselect',
TESTDATA / 'shapes.png') as tempdir:
# Create an empty CSV file
csv = tempdir / 'shapes.csv'
with csv.open('w'):
pass
main([str(tempdir), '--overwrite'])
csv = tempdir / 'shapes.csv'
self.assertTrue(csv.is_file())
# Check CSV contents
doc = InselectDocument.load(tempdir / 'shapes.inselect')
with csv.open('rb') as f:
res = unicodecsv.DictReader(f, encoding='utf-8')
for index, item, row in zip(count(), doc.items, res):
expected = item['fields']
expected.update({
'ItemNumber': str(1+index),
'Cropped_image_name': '{0:04}.jpg'.format(1+index)
})
actual = {
k: v for k, v in row.items()
if v and k not in BOUNDING_BOX_FIELD_NAMES
}
self.assertEqual(expected, actual)
def test_export_csv_with_template(self):
"Export metadata to CSV using a metadata template"
with temp_directory_with_files(TESTDATA / 'shapes.inselect',
TESTDATA / 'shapes.png') as tempdir:
main([str(tempdir),
'--template={0}'.format(TESTDATA / 'test.inselect_template')])
# nose hooks up stdout to a file-like object
stdout = sys.stdout.getvalue()
self.assertIn('because there are validation problems', stdout)
self.assertIn('Box [1] [0001] lacks mandatory field [Taxonomy]', stdout)
self.assertIn('Box [1] [0001] lacks mandatory field [Location]', stdout)
self.assertIn(
'Could not parse value of [catalogNumber] [1] for box [1] [0001]',
stdout
)
csv = tempdir / 'shapes.csv'
self.assertFalse(csv.is_file())
def test_export_csv_with_existing(self):
"Attempt to export metadata over an existing CSV file"
with temp_directory_with_files(TESTDATA / 'shapes.inselect',
TESTDATA / 'shapes.png') as tempdir:
# Create CSV file
csv = tempdir / 'shapes.csv'
with csv.open('w') as outfile:
outfile.write('This is only a test\n')
main([str(tempdir)])
# nose hooks up stdout to a file-like object
stdout = sys.stdout.getvalue()
self.assertIn('exists - skipping', stdout)
# File should not have been altered
with csv.open('r') as infile:
res = infile.read()
self.assertEqual('This is only a test\n', res)
