def add_new_story(request):
allowed_objects = get_allowed_objects(
request.user, ArticleCategory, 'writer')
if len(allowed_objects) == 0:
return HttpResponseRedirect('/')
if request.method == 'POST':
form_story = ArticleStoryForm(request.POST, request.FILES, request=request)
fa_set = formset_factory(ArticleAttachmentsForm, extra=2)
form_attachments_set = fa_set(request.POST, request.FILES)
if form_story.is_valid():
if 'image' in request.FILES:
image_file = request.FILES['image']
mimetype = image_file.content_type
else:
image_file = None
mimetype = ''
if image_file:
story = form_story.save()
upload_path = get_upload_articles_images()
path_src = u"%s/%s" % (upload_path, story.image)
path_dest = u"%s/%s" % (upload_path, story.pk)
os.makedirs(path_dest)
shutil.move(path_src, path_dest)
s = os.path.splitext(image_file.name)
filename = u"%s%s" % (slugify(s[0]), s[1])
story.image = "%s/%s" % (story.pk, filename)
else:
story = form_story.save(commit=False)
story.mimetype = mimetype
story.owner = request.user
story.save()
story.categories.clear()
selected_categories = request.POST.getlist('_selected_action')
story.categories = selected_categories
if form_attachments_set.is_valid():
for i in range(0, form_attachments_set.total_form_count()):
form = form_attachments_set.forms[i]
try:
file = request.FILES['form-' + str(
i) + '-attached_file']
attachment = ArticleAttachments(description=form.cleaned_data[
'description'], mimetype=file.content_type)
attachment.article = story
attachment.save()
attachment.attached_file.save(file.name, file)
except:
pass
tags = map(int, request.POST.getlist('article_tags[]'))
tags = ArticleTags.objects.filter(pk__in=tags)
story.tags.add(*tags)
if story.categories:
notify_users(request, story, story.categories.all(), True)
return HttpResponseRedirect(reverse('writer_stories_list', args=[]))
else:
form_story = ArticleStoryForm(request=request)
fa_set = formset_factory(ArticleAttachmentsForm, extra=2)
form_attachments_set = fa_set()
tags = ArticleTags.objects.all()
context = RequestContext(request)
extra_context = {'form_story': form_story,
'form_attachments': form_attachments_set,
'tags': tags,
}
return render_to_response('articles/new_article.html', extra_context, context_instance=context)
