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 infoscreen.models import UploadFileForm

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

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


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


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

    @ensure_csrf_cookie
    @require_http_methods(["POST"])
    @login_required(login_url='/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 ValueError:
            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='/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:
            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='/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:
        resp = HttpResponse('{"error" : "could not delete item"}')
        resp.status_code = 500
        return resp


@require_http_methods(["POST"])
@ensure_csrf_cookie
@login_required(login_url='/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='/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='/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:
        return HttpResponse('{"error": "bad post body!"}', status=400)

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

    return resp


@require_http_methods(["DELETE"])
@ensure_csrf_cookie
@login_required(login_url='/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:
        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)
createCoffeeStatsItem = create_item_generator(CoffeeStatsInfoItem)