Commander1024/lostplaces-backend
Commander1024/lostplaces-backend
1
0
Fork 0
Code Issues 14 Pull Requests Releases Wiki Activity

#50 Added logic to set link_type based on url and LINK_DOMAINS dict.

Browse Source
This commit is contained in:
Marcus Scholz 2021-10-02 05:56:16 +02:00
parent 86f04c08ff
commit 056d937f15
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
django_lostplaces/lostplaces/views/views.py
View File

@ -12,7 +12,7 @@ from django.http import HttpResponseForbidden
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
from lostplaces.forms import SignupVoucherForm, TagSubmitForm from lostplaces.forms import SignupVoucherForm, TagSubmitForm
from lostplaces.models import Place, ExternalLink from lostplaces.models import Place, ExternalLink, external_links
from lostplaces.views.base_views import IsAuthenticatedMixin from lostplaces.views.base_views import IsAuthenticatedMixin
from lostplaces.common import redirect_referer_or from lostplaces.common import redirect_referer_or
@ -54,6 +54,18 @@ class ExternalLinkCreateView(PlaceAssetCreateView):
template_name = 'external_link/external_link_create.html' template_name = 'external_link/external_link_create.html'
success_message = _('External link submitted') success_message = _('External link submitted')
def post(self, request, place_id, *args, **kwargs):
response = super().post(request, place_id, *args, **kwargs)
if not self.object.linktype:
for domain, link_type in external_links.LINK_DOMAINS.items():
if domain in self.object.url:
self.object.linktype = link_type
self.object.save()
break
else:
self.object.linktype = None
return response
class ExternalLinkDeleteView(PlaceAssetDeleteView): class ExternalLinkDeleteView(PlaceAssetDeleteView):
model = ExternalLink model = ExternalLink
pk_url_kwarg = 'pk' pk_url_kwarg = 'pk'