def add_resource_for_account(request, pk): account = get_object_or_404(Account, pk=pk) if request.method == "POST": form = SiteAddForm(request.POST) if form.is_valid(): hostname = form.cleaned_data.get('hostname') website = get_object_or_None(Website, hostname=hostname) if not website: website = Website.objects.create(hostname=hostname) resource = get_object_or_None(Resource, account=account, website=website) if not resource: resource = Resource.objects.create(account=account, website=website, note=u"") messages.add_message(request, messages.SUCCESS, u"Ресурс %s был добавлен" % resource.website.hostname) else: messages.add_message(request, messages.WARNING, u"Ресурс %s уже существует" % resource.website.hostname) else: messages.add_message(request, messages.ERROR, u"Некорректное имя хоста", extra_tags='danger') return HttpResponseRedirect(reverse("account", kwargs={'pk': pk}) + '#sites')
def post(self, request, resource): form = self.form_class(request.POST) if form.is_valid(): querystring = form.cleaned_data['querystring'] query, q_created = Query.objects.get_or_create(querystring=querystring) for region_id in form.cleaned_data['regions']: region = get_object_or_404(Region, pk=region_id) subscription, s_created = Subscription.objects.get_or_create(query=query, region=region) resource_subscription = get_object_or_None( ResourceSubscription, resource=resource, subscription=subscription) if not resource_subscription: resource_subscription = ResourceSubscription.objects.create( resource=resource, subscription=subscription, search_depth=form.cleaned_data['search_depth'], datetime_subscribed = datetime.datetime.now(), subdomain_include = form.cleaned_data['subdomain_include']) #resource_subscription.search_depth = form.cleaned_data['search_depth'] #resource_subscription.datetime_subscribed = datetime.datetime.now() #resource_subscription.subdomain_include = form.cleaned_data['subdomain_include'] #resource_subscription.save() return HttpResponseRedirect(resource.get_absolute_url())