def setUp(self):
super(TestMigrateGuids, self).setUp()
self.clear_guids()
self.project = ProjectFactory()
self.paths = ['peppers', 'sausage', 'pepperoni']
self.legacy_objs = [
OsfGuidFile(node=self.project, name=path) for path in self.paths
]
for obj in self.legacy_objs:
obj.save()
def view_file(auth, **kwargs):
node_settings = kwargs['node_addon']
node = kwargs['node'] or kwargs['project']
file_name = kwargs['fid']
file_name_clean = file_name.replace('.', '_')
try:
guid = OsfGuidFile.find_one(
Q('node', 'eq', node) &
Q('name', 'eq', file_name)
)
except:
guid = OsfGuidFile(
node=node,
name=file_name,
)
guid.save()
redirect_url = check_file_guid(guid)
if redirect_url:
return redirect(redirect_url)
# Throw 404 and log error if file not found in files_versions
try:
file_id = node.files_versions[file_name_clean][-1]
except KeyError:
logger.error('File {} not found in files_versions of component {}.'.format(
file_name_clean, node._id
))
raise HTTPError(http.NOT_FOUND)
file_object = NodeFile.load(file_id)
# Ensure NodeFile is attached to Node; should be fixed by actions or
# improved data modeling in future
if not file_object.node:
file_object.node = node
file_object.save()
download_url = file_object.download_url(node)
render_url = file_object.render_url(node)
info_url = file_object.info_url(node)
file_path = os.path.join(
settings.UPLOADS_PATH,
node._primary_key,
file_name
)
# Throw 404 and log error if file not found on disk
if not os.path.isfile(file_path):
logger.error('File {} not found on disk.'.format(file_path))
raise HTTPError(http.NOT_FOUND)
_, file_ext = os.path.splitext(file_path.lower())
# Get or create rendered file
cache_file = get_cache_file(
file_object.filename,
file_object.latest_version_number(node)
)
rendered = get_cache_content(
node_settings, cache_file, start_render=True, file_path=file_path,
file_content=None, download_path=download_url,
)
rv = {
'file_name': file_name,
'render_url': render_url,
'rendered': rendered,
'info_url': info_url,
}
rv.update(_view_project(node, auth))
return rv
