web2.0-backend/infoscreen/views/admin_views.py

"""File containing infoscreen views."""

from django.shortcuts import render
from django.http import HttpResponse, HttpResponseBadRequest
from django.views.decorators.csrf import ensure_csrf_cookie
from django.views.decorators.http import require_http_methods
from django.contrib.contenttypes.models import ContentType
from django.contrib.auth.decorators import permission_required, login_required
from django.db import DatabaseError
from infoscreen.models import UploadFileForm

import sikweb.settings as settings
import json
import logging
import threading
import requests

from infoscreen.models import (
    Rotation, InfoItem, InfoInstance, ABBInfoItem, ExternalImageInfoItem,
    ImageInfoItem, SossoInfoItem, HslInfoItem, EventInfoItem,
    ExternalWebsiteInfoItem, ImageUploadForm, ApyInfoItem, VideoInfoItem)


@login_required(login_url='/admin/login')
@permission_required('infoscreen.change_infoinstance', raise_exception=True)
def admin(request, *args, **kwargs):
    """Render infoscreen admin page."""
    return render(request, 'infoscreen:infoscreen_admin.html', {})


def create_item_generator(model):
    """Create Infoscreen item generator."""

    @ensure_csrf_cookie
    @require_http_methods(["POST"])
    @login_required(login_url='/admin/login')
    @permission_required('infoscreen.add_infoinstance', raise_exception=True)
    def create_item(request, *args, **kwargs):
        try:
            data = json.loads(request.body.decode("utf-8"))
        except json.JSONDecodeError:
            return HttpResponseBadRequest(
                '{"status":"failure","error":"invalid json supplied"}')
        try:
            model.create_from_dict(data)
            return HttpResponse('{"status":"success"}')
        except RuntimeError as e:
            return HttpResponseBadRequest(
                json.dumps({"status": "failure", "error": str(e)}))
    return create_item


def delete_item_generator(model):
    """Delete Infoscreen item generator."""

    @ensure_csrf_cookie
    @require_http_methods(["DELETE"])
    @login_required(login_url='/admin/login')
    @permission_required('infoscreen.delete_infoinstance', raise_exception=True)
    def delete_item(request, *args, **kwargs):
        idx = kwargs.pop("idx", 0)
        try:
            item = model.objects.get(pk=idx)
        except model.DoesNotExist:
            resp = HttpResponse('{"error":"object not found"}')
            resp.status_code = 404
            return resp
        try:
            item.delete()
            return HttpResponse('{"status":"success"}')
        except DatabaseError:
            resp = HttpResponse('{"error" : "could not delete item"}')
            resp.status_code = 500
            return resp
    return delete_item


# due to model structure this is little complicated
@ensure_csrf_cookie
@login_required(login_url='/admin/login')
@permission_required('infoscreen.delete_infoinstance', raise_exception=True)
@require_http_methods(["DELETE"])
def delete_info_item(request, *args, **kwargs):
    """Delete info item."""
    type_id = kwargs.pop("type_id", 0)
    idx = kwargs.pop("idx", 0)
    if True:
        ct = ContentType.objects.get_for_id(type_id)
        item = ct.get_object_for_this_type(pk=idx)
    else:
        resp = HttpResponse('{"error":"object not found"}')
        resp.status_code = 404
        return resp
    try:
        item.delete()
        return HttpResponse('{"status":"success"}')
    except DatabaseError:
        resp = HttpResponse('{"error" : "could not delete item"}')
        resp.status_code = 500
        return resp


@require_http_methods(["POST"])
@ensure_csrf_cookie
@login_required(login_url='/admin/login')
@permission_required('infoscreen.add_infoinstance', raise_exception=True)
def create_image_item(request, *args, **kwargs):
    """Create image Infoscreen item."""
    form = ImageUploadForm(request.POST, request.FILES)
    if not form.is_valid():
        return HttpResponseBadRequest('{"status": "failure",'
                                      '"error": "invalid data supplied"}')

    img = form.cleaned_data['image']
    name = form.cleaned_data['name']
    ImageInfoItem.objects.create(img=img, name=name)
    return HttpResponse('{"status":"success"}')


@require_http_methods(["POST"])
@ensure_csrf_cookie
@login_required(login_url='/admin/login')
@permission_required('infoscreen.add_infoinstance', raise_exception=True)
def create_video_item(request, *args, **kwargs):
    """Create video Infoscreen item."""
    form = UploadFileForm(request.POST, request.FILES)
    if not form.is_valid():
        return HttpResponseBadRequest('{"status": "failure",'
                                      '"error": "invalid data supplied"}')

    video = form.cleaned_data['video']
    name = form.cleaned_data['name']
    VideoInfoItem.objects.create(video=video, name=name)
    return HttpResponse('{"status": "success"}')


@require_http_methods(["POST"])
@ensure_csrf_cookie
@login_required(login_url='/admin/login')
@permission_required('infoscreen.add_rotation', raise_exception=True)
def create_rotation(request, *args, **kwargs):
    """Create rotation."""
    try:
        data = json.loads(request.body.decode("utf-8"))
    except json.JSONDecodeError:
        return HttpResponse('{"error": "bad post body!"}', status=400)

    try:
        name = data["name"]
        Rotation.objects.create(name=name)
        resp = HttpResponse(status=200)
    except DatabaseError:
        resp = HttpResponse(
            '{"error" : "could not create rotation!"}', status=400)

    return resp


@require_http_methods(["DELETE"])
@ensure_csrf_cookie
@login_required(login_url='/admin/login')
@permission_required('infoscreen.delete_rotation', raise_exception=True)
def delete_rotation(request, *args, **kwargs):
    """Delete rotation."""
    id = kwargs.pop("id", 0)
    logging.warning("Deleting rotation with id={}".format(id))

    try:
        Rotation.objects.filter(id=id).delete()
        resp = HttpResponse(status=200)
    except DatabaseError:
        resp = HttpResponse(
            '{"error" : "could not delete rotation!"}', status=400)

    return resp


createInstance = create_item_generator(InfoInstance)
deleteInstance = delete_item_generator(InfoInstance)
createABBItem = create_item_generator(ABBInfoItem)
createSossoItem = create_item_generator(SossoInfoItem)
createHslItem = create_item_generator(HslInfoItem)
createExternalImageInfoItem = create_item_generator(ExternalImageInfoItem)
createExternalWebsiteItem = create_item_generator(ExternalWebsiteInfoItem)
createEventItem = create_item_generator(EventInfoItem)
createApyItem = create_item_generator(ApyInfoItem)
createVideoItem = create_item_generator(VideoInfoItem)