def lineage(self):
if self._use_vo_naming:
if '_th.jpg' in self._file_name:
# thumbnail
return mc.get_lineage(self._v_collection, self.product_id,
self._file_name, self._v_scheme)
else:
return mc.get_lineage(self._v_collection, self.product_id,
self._file_name, SCHEME)
else:
return mc.get_lineage(ARCHIVE, self.product_id, self._file_name,
self.scheme)
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
product_id = NEOSSatName.extract_product_id(ii)
fits = mc.get_lineage(ARCHIVE, product_id, '{}.fits'.format(ii))
result = '{} {}'.format(result, fits)
return result
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
fits = mc.get_lineage(cfht_name.ARCHIVE,
cfht_name.CFHTName.remove_extensions(ii), ii)
result = f'{result } {fits}'
return result
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
storage_name = PHANGSName(file_name=ii)
fits = mc.get_lineage(ARCHIVE, storage_name.product_id, ii)
result = f'{result } {fits}'
return result
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
product_id = VliteName.extract_product_id(ii)
fits = mc.get_lineage(COLLECTION, product_id, '{}.fits'.format(ii))
result = '{} {}'.format(result, fits)
return result
def get_args(self, obs_id):
if obs_id in self.repaired_ids:
result = []
for repaired_id in self.repaired_ids[obs_id]:
lineage = ''
urls = ''
for file_id in self.repaired_names[repaired_id]:
# works because the file id == product id
lineage += mc.get_lineage(
gem_name.ARCHIVE, file_id, '{}.fits'.format(file_id),
gem_name.SCHEME)
urls += '{}{}.fits'.format(HEADER_URL, file_id)
if file_id != self.repaired_names[repaired_id][-1]:
lineage += ' '
urls += ' '
c = CommandLineBits(
'{} {}'.format(gem_name.COLLECTION, repaired_id),
lineage, urls)
result.append(c)
return result
else:
logging.warning(
'Could not find observation ID {} in Gemini-provided '
'list.'.format(obs_id))
return []
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
fits = mc.get_lineage(COLLECTION, mc.StorageName.remove_extensions(ii),
ii)
result = f'{result} {fits}'
return result
def test_main_app(test_name):
basename = os.path.basename(test_name)
obs_id = AskapName.get_obs_id(basename)
product_id = AskapName.get_product_id(basename)
lineage = mc.get_lineage(COLLECTION, product_id,
basename.replace('.header', ''),
AskapName.scheme())
output_file = '{}.actual.xml'.format(test_name)
local = test_name
plugin = PLUGIN
with patch('caom2utils.fits2caom2.CadcDataClient') as data_client_mock:
def get_file_info(archive, file_id):
if '_prev' in file_id:
return {
'size': 10290,
'md5sum': md5('-37'.encode()).hexdigest(),
'type': 'image/jpeg'
}
else:
return {
'size': 37,
'md5sum': md5('-37'.encode()).hexdigest(),
'type': 'application/fits'
}
data_client_mock.return_value.get_file_info.side_effect = \
get_file_info
sys.argv = \
('{} --no_validate --local {} '
'--plugin {} --module {} --observation {} {} -o {} --lineage {}'.
format(APPLICATION, local, plugin, plugin, COLLECTION, obs_id,
output_file, lineage)).split()
print(sys.argv)
caom_main()
obs_path = _get_expected_name(test_name)
expected = mc.read_obs_from_file(obs_path)
actual = mc.read_obs_from_file(output_file)
result = get_differences(expected, actual, 'Observation')
if result:
msg = 'Differences found in observation {}\n{}'. \
format(expected.observation_id, '\n'.join(
[r for r in result]))
raise AssertionError(msg)
def _get_lineage(blank_name):
result = mc.get_lineage(
COLLECTION, blank_name.product_id, f'{blank_name.file_name}', 'cadc'
)
return result
def _get_lineage(blank_name):
result = mc.get_lineage('GEMINI', blank_name.product_id,
f'{blank_name.file_name}')
return result
def test_get_lineage():
result = mc.get_lineage('TEST_COLLECTION', 'TEST_PRODUCT_ID',
'TEST_FILE_NAME.fits')
assert result == 'TEST_PRODUCT_ID/ad:TEST_COLLECTION/TEST_FILE_NAME.fits'
def test_get_lineage():
result = mc.get_lineage('TEST_COLLECTION', 'TEST_PRODUCT_ID',
'TEST_FILE_NAME.fits')
assert result == 'TEST_PRODUCT_ID/ad:TEST_COLLECTION/TEST_FILE_NAME.fits'
def _get_lineage(blank_name):
result = mc.get_lineage(ARCHIVE, blank_name.product_id,
f'{blank_name.file_name}')
return result
def _get_lineage(cfht_name):
result = mc.get_lineage(ARCHIVE, cfht_name.product_id,
f'{cfht_name.file_name}')
return result
def _get_lineage(obs_id):
result = ''
for ii in LOOKUP[obs_id]:
fits = mc.get_lineage(ARCHIVE, ii, '{}.fits'.format(ii), SCHEME)
result = '{} {}'.format(result, fits)
return result
