def __main__(submission):
submission.extract_tarball()
try:
submission.archive_metadata = export.Metadata(submission.tmpdir)
except Exception:
submission.exit_gracefully("ERROR_METADATA_PARSING")
if submission.archive_metadata.is_valid():
if submission.archive_metadata.export_method == "usb-test":
submission.check_usb_connected()
elif submission.archive_metadata.export_method == "disk":
# exports all documents in the archive to luks-encrypted volume
submission.unlock_luks_volume(
submission.archive_metadata.encryption_key)
submission.mount_volume()
submission.copy_submission()
elif submission.archive_metadata.export_method == "disk-test":
submission.check_luks_volume()
elif submission.archive_metadata.export_method == "printer":
# prints all documents in the archive
printer_uri = submission.get_printer_uri()
printer_ppd = submission.install_printer_ppd(printer_uri)
submission.setup_printer(printer_uri, printer_ppd)
submission.print_all_files()
elif submission.archive_metadata.export_method == "printer-test":
# Prints a test page to ensure the printer is functional
printer_uri = submission.get_printer_uri()
printer_ppd = submission.install_printer_ppd(printer_uri)
submission.setup_printer(printer_uri, printer_ppd)
submission.print_test_page()
else:
submission.exit_gracefully("ERROR_ARCHIVE_METADATA")
def test_empty_config(capsys):
export.SDExport("testfile", TEST_CONFIG)
temp_folder = tempfile.mkdtemp()
metadata = os.path.join(temp_folder, export.Metadata.METADATA_FILE)
with open(metadata, "w") as f:
f.write("{}")
config = export.Metadata(temp_folder)
assert not config.is_valid()
def test_valid_printer_config(capsys):
export.SDExport("", TEST_CONFIG)
temp_folder = tempfile.mkdtemp()
metadata = os.path.join(temp_folder, export.Metadata.METADATA_FILE)
with open(metadata, "w") as f:
f.write('{"device": "printer"}')
config = export.Metadata(temp_folder)
assert config.is_valid()
assert config.encryption_key is None
assert config.encryption_method is None
def test_valid_encryption_config(capsys):
export.SDExport("testfile", TEST_CONFIG)
temp_folder = tempfile.mkdtemp()
metadata = os.path.join(temp_folder, export.Metadata.METADATA_FILE)
with open(metadata, "w") as f:
f.write(
'{"device": "disk", "encryption_method": "luks", "encryption_key": "hunter1"}'
)
config = export.Metadata(temp_folder)
assert config.encryption_key == "hunter1"
assert config.encryption_method == "luks"
assert config.is_valid()
def __main__(submission):
submission.extract_tarball()
try:
submission.archive_metadata = export.Metadata(submission.tmpdir)
except Exception:
submission.exit_gracefully(ExportStatus.ERROR_METADATA_PARSING.value)
if submission.archive_metadata.is_valid():
if submission.archive_metadata.export_method == "usb-test":
logging.info('Export archive is usb-test')
submission.check_usb_connected()
elif submission.archive_metadata.export_method == "disk":
logging.info('Export archive is disk')
logging.info('Unlocking volume')
# exports all documents in the archive to luks-encrypted volume
submission.unlock_luks_volume(
submission.archive_metadata.encryption_key)
logging.info('Mounting volume')
submission.mount_volume()
logging.info('Copying submission to drive')
submission.copy_submission()
elif submission.archive_metadata.export_method == "disk-test":
logging.info('Export archive is disk-test')
submission.check_luks_volume()
elif submission.archive_metadata.export_method == "printer":
logging.info('Export archive is printer')
# prints all documents in the archive
logging.info('Searching for printer')
printer_uri = submission.get_printer_uri()
logging.info('Installing printer drivers')
printer_ppd = submission.install_printer_ppd(printer_uri)
logging.info('Setting up printer')
submission.setup_printer(printer_uri, printer_ppd)
logging.info('Printing files')
submission.print_all_files()
elif submission.archive_metadata.export_method == "printer-test":
# Prints a test page to ensure the printer is functional
printer_uri = submission.get_printer_uri()
printer_ppd = submission.install_printer_ppd(printer_uri)
submission.setup_printer(printer_uri, printer_ppd)
submission.print_test_page()
else:
submission.exit_gracefully(ExportStatus.ERROR_ARCHIVE_METADATA.value)