Firewall on Demand

from celery.task.sets import subtask

import logging

from celery.task.http import *

from flowspy.utils import beanstalkc

from django.conf import settings

FORMAT = '%(asctime)s %(levelname)s: %(message)s'

logging.basicConfig(format=FORMAT)

logger = logging.getLogger(__name__)

logger.setLevel(logging.DEBUG)

def add(route, callback=None):

    subtask(announce).delay("Route add: %s - Result: %s" %(route.name, response))

    route.save()

@task

def edit(route, callback=None):

    applier = PR.Applier(route_object=route)

    commit, response = applier.apply(operation="replace")

    if commit:

        is_online = True

    else:

        is_online = False

    route.is_active = True

    route.is_online = is_online

    route.response = response

    subtask(announce).delay("Route edit: %s - Result: %s" %(route.name, response))

def delete(route, callback=None):

    applier = PR.Applier(route_object=route)

    commit, response = applier.apply(operation="delete")

    if commit:

        is_online = False

        is_active = False

    else:

        is_online = route.is_online

        is_active = route.is_active

    route.is_online = is_online

    route.is_active = is_active

    route.response = response

    route.save()

    subtask(announce).delay("Route delete: %s - Result %s" %(route.name, response))

@task

def announce(messg):

    messg = str(messg)

    b = beanstalkc.Connection()

    b.use(settings.POLLS_TUBE)

    b.put(messg)

    b.close()