lemoncurry/micropub/views.py

86 lines
2.5 KiB
Python

from django.http import HttpResponse
from django.urls import reverse
from django.utils.decorators import method_decorator
from django.views import View
from django.views.decorators.csrf import csrf_exempt
from urllib.parse import urljoin
from entries.jobs import ping_hub, send_mentions
from entries.models import Cat, Entry
from entries.kinds import Article, Note, Reply, Like, Repost
from lemoncurry import utils
from lemonauth import tokens
def form_to_mf2(post):
properties = {}
for key in post.keys():
if key.endswith('[]'):
key = key[:-2]
if key == 'access_token':
continue
properties[key] = post.getlist(key) + post.getlist(key + '[]')
type = []
if 'h' in properties:
type = ['h-' + p for p in properties['h']]
del properties['h']
return {'type': type, 'properties': properties}
@method_decorator(csrf_exempt, name='dispatch')
class MicropubView(View):
def post(self, request):
token = tokens.auth(request)
if hasattr(token, 'content'):
return token
post = request.POST
if post.get('h') != 'entry':
return utils.bad_req('only h=entry supported')
entry = Entry(author=token.user)
kind = Note
if 'name' in post:
entry.name = post['name']
kind = Article
if 'content' in post:
entry.content = post['content']
if 'in-reply-to' in post:
entry.in_reply_to = post['in-reply-to']
kind = Reply
if 'like-of' in post:
entry.like_of = post['like-of']
kind = Like
if 'repost-of' in post:
entry.repost_of = post['repost-of']
kind = Repost
cats = [
Cat.objects.from_name(c) for c in
post.getlist('category') + post.getlist('category[]')
]
entry.kind = kind.id
entry.save()
entry.cats = cats
entry.save()
base = utils.origin(request)
perma = urljoin(base, entry.url)
others = [urljoin(base, url) for url in (
reverse('home:index'),
reverse('entries:atom'),
reverse('entries:rss'),
reverse('entries:' + kind.index),
reverse('entries:' + kind.atom),
reverse('entries:' + kind.rss),
)] + [urljoin(base, cat.url) for cat in cats]
ping_hub.delay(perma, *others)
send_mentions.delay(perma)
res = HttpResponse(status=201)
res['Location'] = perma
return res
micropub = MicropubView.as_view()