api.py
248 lines
| 7.1 KiB
| text/x-python
|
PythonLexer
|
r450 | import json | ||
|
r639 | import logging | ||
|
r1117 | |||
|
r450 | from django.db import transaction | ||
from django.http import HttpResponse | ||||
|
r1117 | from django.shortcuts import get_object_or_404 | ||
|
r564 | from django.core import serializers | ||
|
r1323 | from boards.abstracts.settingsmanager import get_settings_manager | ||
|
r564 | |||
|
r533 | from boards.forms import PostForm, PlainErrorList | ||
|
r499 | from boards.models import Post, Thread, Tag | ||
|
r542 | from boards.utils import datetime_to_epoch | ||
from boards.views.thread import ThreadView | ||||
|
r994 | from boards.models.user import Notification | ||
|
r1217 | from boards.mdx_neboard import Parser | ||
|
r450 | |||
|
r1117 | |||
|
r450 | __author__ = 'neko259' | ||
|
r501 | PARAMETER_TRUNCATED = 'truncated' | ||
PARAMETER_TAG = 'tag' | ||||
PARAMETER_OFFSET = 'offset' | ||||
|
r524 | PARAMETER_DIFF_TYPE = 'type' | ||
|
r1075 | PARAMETER_POST = 'post' | ||
|
r1076 | PARAMETER_UPDATED = 'updated' | ||
|
r1086 | PARAMETER_LAST_UPDATE = 'last_update' | ||
|
r1119 | PARAMETER_THREAD = 'thread' | ||
PARAMETER_UIDS = 'uids' | ||||
|
r524 | |||
DIFF_TYPE_HTML = 'html' | ||||
DIFF_TYPE_JSON = 'json' | ||||
|
r501 | |||
|
r533 | STATUS_OK = 'ok' | ||
STATUS_ERROR = 'error' | ||||
|
r639 | logger = logging.getLogger(__name__) | ||
|
r450 | |||
@transaction.atomic | ||||
|
r1086 | def api_get_threaddiff(request): | ||
|
r627 | """ | ||
Gets posts that were changed or added since time | ||||
""" | ||||
|
r450 | |||
|
r1191 | thread_id = request.POST.get(PARAMETER_THREAD) | ||
|
r1119 | uids_str = request.POST.get(PARAMETER_UIDS).strip() | ||
|
r1118 | uids = uids_str.split(' ') | ||
|
r639 | |||
|
r1323 | opening_post = get_object_or_404(Post, id=thread_id) | ||
thread = opening_post.get_thread() | ||||
|
r450 | |||
json_data = { | ||||
|
r1076 | PARAMETER_UPDATED: [], | ||
|
r1119 | PARAMETER_LAST_UPDATE: None, # TODO Maybe this can be removed already? | ||
|
r504 | } | ||
|
r1118 | posts = Post.objects.filter(threads__in=[thread]).exclude(uid__in=uids) | ||
|
r524 | |||
|
r917 | diff_type = request.GET.get(PARAMETER_DIFF_TYPE, DIFF_TYPE_HTML) | ||
|
r524 | |||
|
r1118 | for post in posts: | ||
|
r1191 | json_data[PARAMETER_UPDATED].append(get_post_data(post.id, diff_type, | ||
request)) | ||||
|
r1086 | json_data[PARAMETER_LAST_UPDATE] = str(thread.last_edit_time) | ||
|
r450 | |||
|
r1323 | # 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) | ||||
|
r450 | return HttpResponse(content=json.dumps(json_data)) | ||
|
r533 | def api_add_post(request, opening_post_id): | ||
""" | ||||
|
r627 | Adds a post and return the JSON response for it | ||
|
r533 | """ | ||
opening_post = get_object_or_404(Post, id=opening_post_id) | ||||
|
r640 | logger.info('Adding post via api...') | ||
|
r533 | status = STATUS_OK | ||
errors = [] | ||||
if request.method == 'POST': | ||||
|
r640 | form = PostForm(request.POST, request.FILES, error_class=PlainErrorList) | ||
|
r533 | form.session = request.session | ||
|
r638 | if form.need_to_ban: | ||
# Ban user because he is suspected to be a bot | ||||
# _ban_current_user(request) | ||||
status = STATUS_ERROR | ||||
|
r533 | if form.is_valid(): | ||
|
r640 | post = ThreadView().new_post(request, form, opening_post, | ||
|
r690 | html_response=False) | ||
|
r640 | if not post: | ||
|
r638 | status = STATUS_ERROR | ||
|
r640 | else: | ||
logger.info('Added post #%d via api.' % post.id) | ||||
|
r533 | else: | ||
status = STATUS_ERROR | ||||
errors = form.as_json_errors() | ||||
response = { | ||||
'status': status, | ||||
'errors': errors, | ||||
} | ||||
return HttpResponse(content=json.dumps(response)) | ||||
|
r499 | |||
|
r491 | def get_post(request, post_id): | ||
|
r499 | """ | ||
|
r627 | Gets the html of a post. Used for popups. Post can be truncated if used | ||
|
r499 | in threads list with 'truncated' get parameter. | ||
""" | ||||
|
r491 | |||
post = get_object_or_404(Post, id=post_id) | ||||
|
r1110 | truncated = PARAMETER_TRUNCATED in request.GET | ||
|
r491 | |||
|
r1110 | return HttpResponse(content=post.get_view(truncated=truncated)) | ||
|
r499 | |||
def api_get_threads(request, count): | ||||
""" | ||||
|
r627 | Gets the JSON thread opening posts list. | ||
|
r499 | Parameters that can be used for filtering: | ||
tag, offset (from which thread to get results) | ||||
""" | ||||
|
r501 | if PARAMETER_TAG in request.GET: | ||
tag_name = request.GET[PARAMETER_TAG] | ||||
|
r499 | if tag_name is not None: | ||
tag = get_object_or_404(Tag, name=tag_name) | ||||
|
r908 | threads = tag.get_threads().filter(archived=False) | ||
|
r499 | else: | ||
threads = Thread.objects.filter(archived=False) | ||||
|
r501 | if PARAMETER_OFFSET in request.GET: | ||
offset = request.GET[PARAMETER_OFFSET] | ||||
|
r499 | 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() | ||||
|
r524 | # TODO Add tags, replies and images count | ||
|
r976 | post_data = get_post_data(opening_post.id, include_last_update=True) | ||
post_data['bumpable'] = thread.can_bump() | ||||
post_data['archived'] = thread.archived | ||||
opening_posts.append(post_data) | ||||
|
r499 | |||
return HttpResponse(content=json.dumps(opening_posts)) | ||||
|
r500 | |||
|
r502 | # TODO Test this | ||
|
r500 | def api_get_tags(request): | ||
""" | ||||
|
r627 | Gets all tags or user tags. | ||
|
r500 | """ | ||
# TODO Get favorite tags for the given user ID | ||||
tags = Tag.objects.get_not_empty_tags() | ||||
|
r1115 | |||
term = request.GET.get('term') | ||||
if term is not None: | ||||
tags = tags.filter(name__contains=term) | ||||
tag_names = [tag.name for tag in tags] | ||||
|
r500 | |||
return HttpResponse(content=json.dumps(tag_names)) | ||||
|
r502 | |||
# TODO The result can be cached by the thread last update time | ||||
# TODO Test this | ||||
def api_get_thread_posts(request, opening_post_id): | ||||
""" | ||||
|
r627 | Gets the JSON array of thread posts | ||
|
r502 | """ | ||
opening_post = get_object_or_404(Post, id=opening_post_id) | ||||
|
r627 | thread = opening_post.get_thread() | ||
|
r502 | posts = thread.get_replies() | ||
|
r525 | json_data = { | ||
'posts': [], | ||||
'last_update': None, | ||||
} | ||||
|
r502 | json_post_list = [] | ||
for post in posts: | ||||
|
r853 | json_post_list.append(get_post_data(post.id)) | ||
|
r542 | json_data['last_update'] = datetime_to_epoch(thread.last_edit_time) | ||
|
r525 | json_data['posts'] = json_post_list | ||
|
r524 | |||
|
r525 | return HttpResponse(content=json.dumps(json_data)) | ||
|
r524 | |||
|
r994 | 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(username=username, | ||||
|
r1117 | last=last_id) | ||
|
r994 | |||
json_post_list = [] | ||||
for post in posts: | ||||
json_post_list.append(get_post_data(post.id)) | ||||
return HttpResponse(content=json.dumps(json_post_list)) | ||||
|
r564 | def api_get_post(request, post_id): | ||
""" | ||||
|
r627 | Gets the JSON of a post. This can be | ||
|
r564 | 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) | ||||
|
r853 | # TODO Remove this method and use post method directly | ||
def get_post_data(post_id, format_type=DIFF_TYPE_JSON, request=None, | ||||
include_last_update=False): | ||||
post = get_object_or_404(Post, id=post_id) | ||||
return post.get_post_data(format_type=format_type, request=request, | ||||
include_last_update=include_last_update) | ||||
|
r1217 | |||
def api_get_preview(request): | ||||
raw_text = request.POST['raw_text'] | ||||
parser = Parser() | ||||
return HttpResponse(content=parser.parse(parser.preparse(raw_text))) | ||||