dr.botzo/dr_botzo/dispatch/views.py

"""Handle dispatcher API requests."""

from __future__ import unicode_literals

import copy
import logging
import os
import xmlrpclib

from django.conf import settings
from django.shortcuts import get_object_or_404
from rest_framework import generics, status
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response

from dispatch.models import Dispatcher, DispatcherAction
from dispatch.serializers import DispatchMessageSerializer, DispatcherSerializer, DispatcherActionSerializer


log = logging.getLogger('dispatch.views')


class HasSendMessagePermission(IsAuthenticated):

    def has_permission(self, request, view):
        if request.user.has_perm('dispatch.send_message'):
            return True

        return False


class DispatcherList(generics.ListAPIView):

    """List all dispatchers."""

    queryset = Dispatcher.objects.all()
    serializer_class = DispatcherSerializer


class DispatcherDetail(generics.RetrieveAPIView):

    """Detail the given dispatcher."""

    queryset = Dispatcher.objects.all()
    serializer_class = DispatcherSerializer


class DispatcherDetailByKey(DispatcherDetail):

    lookup_field = 'key'


class DispatchMessage(generics.GenericAPIView):

    """Send a message to the given dispatcher."""

    permission_classes = (HasSendMessagePermission,)

    queryset = Dispatcher.objects.all()
    serializer_class = DispatchMessageSerializer

    def get(self, request, *args, **kwargs):
        dispatcher = self.get_object()
        data = {'message': "", 'status': "READY"}
        message = self.serializer_class(data=data)
        return Response(message.initial_data)

    def post(self, request, *args, **kwargs):
        dispatcher = self.get_object()
        message = self.serializer_class(data=request.data)
        if message.is_valid():
            for action in dispatcher.actions.all():
                if action.type == DispatcherAction.PRIVMSG_TYPE:
                    bot_url = 'http://{0:s}:{1:d}/'.format(settings.IRCBOT_XMLRPC_HOST, settings.IRCBOT_XMLRPC_PORT)
                    bot = xmlrpclib.ServerProxy(bot_url)
                    log.debug("sending '%s' to channel %s", message.data['message'], action.destination)
                    bot.privmsg(action.destination, message.data['message'])
                elif action.type == DispatcherAction.FILE_TYPE:
                    filename = os.path.abspath(action.destination)
                    log.debug("sending '%s' to file %s", message.data['message'], filename)
                    with open(filename, 'w') as f:
                        f.write(message.data['message'])
                        f.write(b'\n')

            new_data = copy.deepcopy(message.data)
            new_data['status'] = "OK"
            new_message = self.serializer_class(data=new_data)
            return Response(new_message.initial_data)

        return Response(message.errors, status=status.HTTP_400_BAD_REQUEST)


class DispatchMessageByKey(DispatchMessage):

    lookup_field = 'key'


class DispatcherActionList(generics.ListAPIView):

    """List all dispatchers."""

    queryset = DispatcherAction.objects.all()
    serializer_class = DispatcherActionSerializer


class DispatcherActionDetail(generics.RetrieveAPIView):

    """Detail the given dispatcher."""

    queryset = DispatcherAction.objects.all()
    serializer_class = DispatcherActionSerializer