External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-17 12:41:19 -03:30
Code Issues Packages Projects Releases Wiki Activity

Fix problems with the package() function

Browse Source
This commit is contained in:
Jeff Bradberry 2021-03-12 13:52:42 -05:00
parent 1bf37266b4
commit 9cde10c93a
2 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
awx/main/analytics/core.py
View File

@ -85,20 +85,29 @@ def package(target, data, timestamp):
manifest = {}
with tarfile.open(target.joinpath(tarname), 'w:gz') as f:
for name, (item, version) in data.items():
manifest[name] = version
if isinstance(item, str):
info = f.gettarinfo(item, arcname=name)
f.addfile(info)
else:
info = tarfile.TarInfo(name)
fileobj = io.BytesIO(json.dumps(item).encode('utf-8'))
fileobj.size = len(fileobj.getvalue())
f.addfile(info, fileobj=fileobj)
try:
if isinstance(item, str):
info = f.gettarinfo(item, arcname=f'./{name}')
f.addfile(info)
else:
fileobj = io.BytesIO(json.dumps(item).encode('utf-8'))
info = tarfile.TarInfo(f'./{name}')
info.size = len(fileobj.getvalue())
info.mtime = timestamp.timestamp()
f.addfile(info, fileobj=fileobj)
manifest[name] = version
except Exception:
logger.exception(f"Could not generate metric {name}")
info = tarfile.TarInfo('manifest.json')
fileobj = io.BytesIO(json.dumps(manifest).encode('utf-8'))
fileobj.size = len(fileobj.getvalue())
f.addfile(info, fileobj=fileobj)
try:
fileobj = io.BytesIO(json.dumps(manifest).encode('utf-8'))
info = tarfile.TarInfo('./manifest.json')
info.size = len(fileobj.getvalue())
info.mtime = timestamp.timestamp()
f.addfile(info, fileobj=fileobj)
except Exception:
logger.exception("Could not generate manifest.json")
return None
return f.name
except Exception:

2
awx/main/tests/functional/analytics/test_core.py
View File

@ -39,7 +39,7 @@ def mock_valid_license():
def test_gather(mock_valid_license):
settings.INSIGHTS_TRACKING_STATE = True
tgzfiles = gather(module=importlib.import_module(__name__))
tgzfiles = gather(module=importlib.import_module(__name__), collection_type='dry-run')
files = {}
with tarfile.open(tgzfiles[0], "r:gz") as archive:
for member in archive.getmembers():