Commit 88a1f0e8 authored by Diederik van der Boor's avatar Diederik van der Boor
Browse files

Fix PrivateStorageDetailView.model_file_field usage.

parent 67a99996
...@@ -76,7 +76,11 @@ class PrivateStorageDetailView(SingleObjectMixin, PrivateStorageView): ...@@ -76,7 +76,11 @@ class PrivateStorageDetailView(SingleObjectMixin, PrivateStorageView):
Implement access controls by overriding :meth`get_queryset` or redefining :meth:`can_access_file`. Implement access controls by overriding :meth`get_queryset` or redefining :meth:`can_access_file`.
""" """
model = None #
#: Define the model to fetch.
model = None
#: Define which field the file name is stored at.
model_file_field = 'file' model_file_field = 'file'
def get(self, request, *args, **kwargs): def get(self, request, *args, **kwargs):
...@@ -84,7 +88,7 @@ class PrivateStorageDetailView(SingleObjectMixin, PrivateStorageView): ...@@ -84,7 +88,7 @@ class PrivateStorageDetailView(SingleObjectMixin, PrivateStorageView):
return super().get(request, *args, **kwargs) return super().get(request, *args, **kwargs)
def get_path(self): def get_path(self):
file = getattr(self.object, 'file') file = getattr(self.object, self.model_file_field)
return file.name return file.name
def can_access_file(self, private_file): def can_access_file(self, private_file):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment