def get_attachments(self, obj):
attachments = itertools.chain(
itertools.chain(*map(
lambda a: itertools.chain(a.attachments.all(
), a.report_attachments.all()), obj.tpm_activities.all())),
obj.report_attachments.all())
return ', '.join(
map(
lambda a: '{} - {}'.format(a.file_type,
urljoin(site_url(), a.url)),
attachments))
def get(self, *args, **kwargs):
try:
attachment = Attachment.objects.get(pk=kwargs["pk"])
except Attachment.DoesNotExist:
return HttpResponseNotFound(
_("No Attachment matches the given query.")
)
if not attachment or (not attachment.file and not attachment.hyperlink):
return HttpResponseNotFound(
_("Attachment has no file or hyperlink")
)
url = urljoin(site_url(), attachment.url)
return HttpResponseRedirect(url)
def test_common_user_url(self):
self.assertIn(
site_url() + reverse('tokens:login'),
build_frontend_url('test', user=UserFactory(is_staff=False)))
def test_staff_user_url(self):
self.assertIn(
site_url() + reverse('main'),
build_frontend_url('test', user=UserFactory(is_staff=True)))
def get_token_auth_link(user):
return update_url_with_auth_token(
urljoin(site_url(), reverse('tokens:login')), user)