mdblog/markdownblog/blog/views.py

import os
import random
import re

import markdown2

from django.contrib.auth.decorators import login_required
from django.db import IntegrityError
from django.http import HttpResponse
from django.shortcuts import render, redirect
from blog.factories import TopicFactory
from blog.models import Topic, Tag, Blogpost
from django.template import Template, Context
from django.views.decorators.csrf import csrf_exempt

from markdownblog import settings


def viewblog(request, title) -> HttpResponse:
    post = Blogpost.objects.get(title=title)
    filepath = os.path.join(os.environ.get("MD_FILE_PATH"), title + ".md")
    with open(filepath, 'r') as mdfile:
        md_file_content = mdfile.read()
        mdfile.close()

    html_form_md = markdown2.markdown(md_file_content,
                                      extras=['footnotes', 'task_list', 'strike', 'break-on-newline', 'tables',
                                              'fenced-code-blocks'])

    rendered_html = Template('{% load video %} '
                             '{% autoescape off %}' +
                             html_form_md +
                             '{% endautoescape %}').render(Context({}))
    context = {'post': post, 'html': rendered_html, 'roottopics': Topic.objects.all().filter(rootTopic=None),
               'allposts': Blogpost.objects.all()}

    return render(request, 'blog/viewpost.html', context)


def index(request) -> HttpResponse:
    context = {'roottopics': Topic.objects.all().filter(rootTopic=None), 'allposts': Blogpost.objects.all(),
               'debug': settings.DEBUG}
    return render(request, 'blog/index.html', context)


@login_required
def edit(request, id) -> HttpResponse:
    blogpost = Blogpost.objects.get(pk=id)
    mdfile_content = open(blogpost.mdfile, "r").read()
    context = {'alltopics': Topic.objects.all().order_by('name').values(), 'markdown': mdfile_content,
               'roottopics': Topic.objects.all().filter(rootTopic=None),
               'allposts': Blogpost.objects.all(), 'post': Blogpost.objects.get(pk=id)}

    if request.method == 'POST':
        title = request.POST['title']
        markdown = request.POST['markdown']

        tags = []

        if 'tags' in request.POST and request.POST['tags'] != '':
            tags_str = request.POST['tags'].split(" ")
            for tag in tags_str:
                tagobj = Tag.objects.get_or_create(name=tag)[0]
                tagobj.save()
                tags.append(tagobj.id)

        topicname = request.POST['topic']
        topic = None if topicname == "No topic" else Topic.objects.get(name=topicname)

        filepath = os.path.join(os.environ.get("MD_FILE_PATH"), title + ".md")

        blogpost = Blogpost.objects.get(pk=id)
        blogpost.topic = topic
        blogpost.title = title
        blogpost.tags.set(tags)
        blogpost.mdfile = filepath
        blogpost.save()

        with open(filepath, "w") as mdfile:
            mdfile.write(markdown)
            mdfile.close()
        context['error'] = "Post \"" + title + "\" edited."
    return render(request, 'blog/edit.html', context)


@login_required
@csrf_exempt
def order(request) -> HttpResponse:
    if request.method == "POST":
        root_id = int(request.POST['rootID']) if request.POST['rootID'] != 'root_list' else None
        child_id = int(request.POST['childID'])
        objtype = request.POST['relinkType']

        if objtype == 'topic':
            child_topic = Topic.objects.get(pk=child_id)
            child_topic.rootTopic = Topic.objects.get(pk=root_id) if root_id is not None else None
            child_topic.save()
        elif objtype == 'post':
            post = Blogpost.objects.get(pk=child_id)
            post.topic = Topic.objects.get(pk=root_id) if root_id is not None else None
            post.save()

    context = {'roottopics': Topic.objects.all().filter(rootTopic=None), 'allposts': Blogpost.objects.all()}
    return render(request, 'blog/order.html', context)


@login_required
def addpost(request) -> HttpResponse:
    context = {'alltopics': Topic.objects.all().order_by('name').values(), 'markdown': '',
               'roottopics': Topic.objects.all().filter(rootTopic=None),
               'allposts': Blogpost.objects.all()}
    if request.method == 'POST':
        title = request.POST['title']
        markdown = request.POST['markdown']

        tags = []

        if 'tags' in request.POST and request.POST['tags'] != '':
            tags_str = request.POST['tags'].split(" ")
            for tag in tags_str:
                tagobj = Tag.objects.get_or_create(name=tag)[0]
                tagobj.save()
                tags.append(tagobj.id)

        topicid = request.POST['topic']
        topic = None if topicid == "No topic" else Topic.objects.get(pk=topicid)

        filepath = os.path.join(os.environ.get("MD_FILE_PATH"), title + ".md")

        try:
            new_post = Blogpost.objects.create(title=title, topic=topic, mdfile=filepath)
            new_post.tags.set(tags)
        except IntegrityError:
            context = {'alltopics': Topic.objects.all().order_by('name').values(),
                       'error': 'Blogpost titles need to be unique. No post created.',
                       "markdown": markdown}
            return render(request, 'blog/addpost.html', context)

        with open(filepath, "w+") as mdfile:
            mdfile.write(markdown)
            mdfile.close()
        context['error'] = "Post \"" + title + "\" created."

    return render(request, 'blog/addpost.html', context)


@login_required
def createmocks(request, objtype, n) -> HttpResponse:
    topics = TopicFactory.create_batch(n)

    for topic in topics:
        topic.save()

    while len(topics) > 1:
        child = random.choice(topics)
        topics.remove(child)
        child.rootTopic = random.choice(topics)
        child.save()

    print('Created ' + str(n) + ' mock topics.')
    return redirect("index")


@login_required
def addtopic(request):
    context = {'roottopics': Topic.objects.all().filter(rootTopic=None), 'allposts': Blogpost.objects.all()}

    if request.method == 'POST':
        topictitle = request.POST['title']
        Topic.objects.get_or_create(name=topictitle)

    return render(request, 'blog/addtopic.html', context)