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())
