##// END OF EJS Templates
Made quote button actually button so it is impossible to accidentally quote its text
Made quote button actually button so it is impossible to accidentally quote its text

File last commit:

r1874:aaf6b563 default
r1909:004a4ff8 default
Show More
api.py
315 lines | 9.3 KiB | text/x-python | PythonLexer
import json
import logging
from django.core import serializers
from django.db import transaction
from django.http import HttpResponse
from django.shortcuts import get_object_or_404
from django.views.decorators.csrf import csrf_protect
from boards.abstracts.settingsmanager import get_settings_manager
from boards.forms import PostForm, PlainErrorList
from boards.mdx_neboard import Parser
from boards.models import Post, Thread, Tag, Attachment, TagAlias
from boards.models.thread import STATUS_ARCHIVE
from boards.models.user import Notification
from boards.utils import datetime_to_epoch
from boards.views.thread import ThreadView
from boards.models.attachment.viewers import FILE_TYPES_IMAGE
__author__ = 'neko259'
PARAMETER_TRUNCATED = 'truncated'
PARAMETER_TAG = 'tag'
PARAMETER_OFFSET = 'offset'
PARAMETER_DIFF_TYPE = 'type'
PARAMETER_POST = 'post'
PARAMETER_UPDATED = 'updated'
PARAMETER_LAST_UPDATE = 'last_update'
PARAMETER_THREAD = 'thread'
PARAMETER_UIDS = 'uids'
PARAMETER_SUBSCRIBED = 'subscribed'
DIFF_TYPE_HTML = 'html'
DIFF_TYPE_JSON = 'json'
STATUS_OK = 'ok'
STATUS_ERROR = 'error'
logger = logging.getLogger(__name__)
@transaction.atomic
def api_get_threaddiff(request):
"""
Gets posts that were changed or added since time
"""
thread_id = request.POST.get(PARAMETER_THREAD)
uids_str = request.POST.get(PARAMETER_UIDS)
if not thread_id or not uids_str:
return HttpResponse(content='Invalid request.')
uids = uids_str.strip().split(' ')
opening_post = get_object_or_404(Post, id=thread_id)
thread = opening_post.get_thread()
json_data = {
PARAMETER_UPDATED: [],
PARAMETER_LAST_UPDATE: None, # TODO Maybe this can be removed already?
}
posts = Post.objects.filter(thread=thread).exclude(uid__in=uids)
diff_type = request.GET.get(PARAMETER_DIFF_TYPE, DIFF_TYPE_HTML)
for post in posts:
json_data[PARAMETER_UPDATED].append(post.get_post_data(
format_type=diff_type, request=request))
json_data[PARAMETER_LAST_UPDATE] = str(thread.last_edit_time)
settings_manager = get_settings_manager(request)
json_data[PARAMETER_SUBSCRIBED] = str(settings_manager.thread_is_fav(opening_post))
# If the tag is favorite, update the counter
settings_manager = get_settings_manager(request)
favorite = settings_manager.thread_is_fav(opening_post)
if favorite:
settings_manager.add_or_read_fav_thread(opening_post)
return HttpResponse(content=json.dumps(json_data))
@csrf_protect
def api_add_post(request, opening_post_id):
"""
Adds a post and return the JSON response for it
"""
opening_post = get_object_or_404(Post, id=opening_post_id)
logger.info('Adding post via api...')
status = STATUS_OK
errors = []
if request.method == 'POST':
form = PostForm(request.POST, request.FILES, error_class=PlainErrorList)
form.session = request.session
if form.need_to_ban:
# Ban user because he is suspected to be a bot
# _ban_current_user(request)
status = STATUS_ERROR
if form.is_valid():
post = ThreadView().new_post(request, form, opening_post,
html_response=False)
if not post:
status = STATUS_ERROR
else:
logger.info('Added post #%d via api.' % post.id)
else:
status = STATUS_ERROR
errors = form.as_json_errors()
response = {
'status': status,
'errors': errors,
}
return HttpResponse(content=json.dumps(response))
def get_post(request, post_id):
"""
Gets the html of a post. Used for popups. Post can be truncated if used
in threads list with 'truncated' get parameter.
"""
post = get_object_or_404(Post, id=post_id)
truncated = PARAMETER_TRUNCATED in request.GET
return HttpResponse(content=post.get_view(truncated=truncated, need_op_data=True))
def api_get_threads(request, count):
"""
Gets the JSON thread opening posts list.
Parameters that can be used for filtering:
tag, offset (from which thread to get results)
"""
if PARAMETER_TAG in request.GET:
tag_name = request.GET[PARAMETER_TAG]
if tag_name is not None:
tag = get_object_or_404(Tag, name=tag_name)
threads = tag.get_threads().exclude(status=STATUS_ARCHIVE)
else:
threads = Thread.objects.exclude(status=STATUS_ARCHIVE)
if PARAMETER_OFFSET in request.GET:
offset = request.GET[PARAMETER_OFFSET]
offset = int(offset) if offset is not None else 0
else:
offset = 0
threads = threads.order_by('-bump_time')
threads = threads[offset:offset + int(count)]
opening_posts = []
for thread in threads:
opening_post = thread.get_opening_post()
# TODO Add tags, replies and images count
post_data = opening_post.get_post_data(include_last_update=True)
post_data['status'] = thread.get_status()
opening_posts.append(post_data)
return HttpResponse(content=json.dumps(opening_posts))
# TODO Test this
def api_get_tags(request):
"""
Gets all tags or user tags.
"""
# TODO Get favorite tags for the given user ID
tags = TagAlias.objects.all()
term = request.GET.get('term')
if term is not None:
tags = tags.filter(name__contains=term)
tag_names = [tag.name for tag in tags]
return HttpResponse(content=json.dumps(tag_names))
def api_get_stickers(request):
attachments = Attachment.objects.filter(mimetype__in=FILE_TYPES_IMAGE)\
.exclude(alias='').exclude(alias=None)
term = request.GET.get('term')
if term:
attachments = attachments.filter(alias__contains=term)
image_dict = [{'thumb': attachment.get_thumb_url(),
'alias': attachment.alias}
for attachment in attachments]
return HttpResponse(content=json.dumps(image_dict))
# TODO The result can be cached by the thread last update time
# TODO Test this
def api_get_thread_posts(request, opening_post_id):
"""
Gets the JSON array of thread posts
"""
opening_post = get_object_or_404(Post, id=opening_post_id)
thread = opening_post.get_thread()
posts = thread.get_replies()
json_data = {
'posts': [],
'last_update': None,
}
json_post_list = []
for post in posts:
json_post_list.append(post.get_post_data())
json_data['last_update'] = datetime_to_epoch(thread.last_edit_time)
json_data['posts'] = json_post_list
return HttpResponse(content=json.dumps(json_data))
def api_get_notifications(request, username):
last_notification_id_str = request.GET.get('last', None)
last_id = int(last_notification_id_str) if last_notification_id_str is not None else None
posts = Notification.objects.get_notification_posts(usernames=[username],
last=last_id)
json_post_list = []
for post in posts:
json_post_list.append(post.get_post_data())
return HttpResponse(content=json.dumps(json_post_list))
def api_get_post(request, post_id):
"""
Gets the JSON of a post. This can be
used as and API for external clients.
"""
post = get_object_or_404(Post, id=post_id)
json = serializers.serialize("json", [post], fields=(
"pub_time", "_text_rendered", "title", "text", "image",
"image_width", "image_height", "replies", "tags"
))
return HttpResponse(content=json)
def api_get_preview(request):
raw_text = request.POST['raw_text']
parser = Parser()
return HttpResponse(content=parser.parse(parser.preparse(raw_text)))
def api_get_new_posts(request):
"""
Gets favorite threads and unread posts count.
"""
posts = list()
include_posts = 'include_posts' in request.GET
settings_manager = get_settings_manager(request)
fav_threads = settings_manager.get_fav_threads()
fav_thread_ops = Post.objects.filter(id__in=fav_threads.keys())\
.order_by('-pub_time').prefetch_related('thread')
ops = [{'op': op, 'last_id': fav_threads[str(op.id)]} for op in fav_thread_ops]
if include_posts:
new_post_threads = Thread.objects.get_new_posts(ops)
if new_post_threads:
thread_ids = {thread.id: thread for thread in new_post_threads}
else:
thread_ids = dict()
for op in fav_thread_ops:
fav_thread_dict = dict()
op_thread = op.get_thread()
if op_thread.id in thread_ids:
thread = thread_ids[op_thread.id]
new_post_count = thread.new_post_count
fav_thread_dict['newest_post_link'] = thread.get_replies()\
.filter(id__gt=fav_threads[str(op.id)])\
.first().get_absolute_url(thread=thread)
else:
new_post_count = 0
fav_thread_dict['new_post_count'] = new_post_count
fav_thread_dict['id'] = op.id
fav_thread_dict['post_url'] = op.get_link_view()
fav_thread_dict['title'] = op.title
posts.append(fav_thread_dict)
else:
fav_thread_dict = dict()
fav_thread_dict['new_post_count'] = \
Thread.objects.get_new_post_count(ops)
posts.append(fav_thread_dict)
return HttpResponse(content=json.dumps(posts))