MyMonTools/traps/views.py

# -*- encoding: utf-8 -*-

from django.core.urlresolvers import reverse
from django.db import connection
from django.http import HttpResponse, HttpResponseRedirect
from django.shortcuts import render

from mymontools.paginate import paginate

from .models import *
from .forms import *
from .tools import get_filtered_traps_for_querydict, get_filtered_cfgtraps_for_querydict
from .translate import *

from pprint import pprint

##############################################################################

def trap_index(Request):
	env = {}

	page = Request.GET.get('page')

	pprint( Request.GET )
	traps = get_filtered_traps_for_querydict(Request.GET)
	env['traps'] = paginate(traps, page, 50)
	env['trapcount'] = traps.count()

	tsf = TrapSimpleFilter(Request.GET)
	tsf.fields['hostname'].choices = [ (x, x) for x in Trap.objects.exclude(hostname='').values_list('hostname', flat=True).order_by('hostname').distinct() ]
	tsf.fields['trapoid'].choices = [ (oid, trapname or oid) for oid, trapname in Trap.objects.exclude(trapoid='').values_list('trapoid','trapname').order_by('trapname').distinct() ]
	tsf.fields['severity'].choices = [ (x, x) for x in Trap.objects.exclude(severity='').values_list('severity', flat=True).order_by('severity').distinct() ] + [ ('__', '<EMPTY>'), ]
	tsf.fields['category'].choices = [ (x, x) for x in Trap.objects.exclude(category='').values_list('category', flat=True).order_by('category').distinct() ] + [ ('__', '<EMPTY>'), ]
	tsf.fields['trapread'].choices = [ ('False', 'Unread'), ('True', 'Read') ]
	env['trapfilter'] = tsf

	env['stats_traps_wo_trapname'] = Trap.objects.filter(trapname='').count()
	env['stats_traps_wo_category'] = Trap.objects.filter(category='').count()
	env['stats_traps_wo_severity'] = Trap.objects.filter(severity='').count()
	env['stats_cfgtraps_wo_category'] = CfgTrap.objects.filter(category='').count()
	env['stats_cfgtraps_wo_severity'] = CfgTrap.objects.filter(severity='').count()

	return render(Request, 'traps/trap_index.html', env)



def trap_modify(Request):
	action = Request.POST.get('action') or Request.GET.get('action')
	trapID = Request.GET.get('trapID')
	trapIDs= Request.POST.getlist('trapIDs')

	# FIXME:#45 'archive']:
	if not action in ['read', 'delete', ]:
		return HttpResponse(u'Action "%s" not implemented' % action )

	if trapID:
		trapIDs.append(trapID)

	if action == 'read':
		Trap.objects.filter(id__in=trapIDs).update(trapread=True)
	elif action == 'delete':
		Trap.objects.filter(id__in=trapIDs).delete()

	params = Request.GET.copy()
	if 'action' in params:
		params.pop('action')
	if trapID in params:
		params.pop('trapID')
	return HttpResponseRedirect( reverse('trap_index') + '?%s' % params.urlencode() )



def config_index(Request):
	env = {}

	page = Request.GET.get('page')
	cfgtraps = get_filtered_cfgtraps_for_querydict(Request.GET)
	env['cfgtraps'] = paginate(cfgtraps, page, 50)
	env['cfgtrapcount'] = cfgtraps.count()

	ctsf = CfgTrapSimpleFilter(Request.GET)
	ctsf.fields['severity'].choices = [ (x, x) for x in CfgTrap.objects.exclude(severity='').values_list('severity', flat=True).order_by('severity').distinct() ] + [ ('__', '<EMPTY>'), ]
	ctsf.fields['category'].choices = [ (x, x) for x in CfgTrap.objects.exclude(category='').values_list('category', flat=True).order_by('category').distinct() ] + [ ('__', '<EMPTY>'), ]
	env['cfgtrapfilter'] = ctsf

	env['stats_cfgtraps_wo_category'] = CfgTrap.objects.filter(category='').count()
	env['stats_cfgtraps_wo_severity'] = CfgTrap.objects.filter(severity='').count()

	return render(Request, 'traps/trap_config_index.html', env)



def config_trapoid(Request, trapoid):
	env = {}
	try:
		# FIXME:#56 Mehrere, aktiv/inaktiv
		cfgtrap = CfgTrap.objects.get(trapoid=trapoid)
	except CfgTrap.DoesNotExist:
		cfgtrap = CfgTrap(trapoid=trapoid)
	env['snmptranslate_trapoid'] = external_snmptranslate(trapoid)

	if Request.method == 'POST':
		form = CfgTrapForm(Request.POST, instance=cfgtrap)
		if form.is_valid():
			form.save()
			return HttpResponseRedirect( reverse('trap_config_index') )
	else:
		form = CfgTrapForm(instance=cfgtrap)

	form.fields['trapname'].label = 'Trap name (in DB)'
	env['form'] = form
	return render(Request, 'traps/config_trapoid.html', env)



def resolve_trapoid(Request, trapoid, db=False):
	env = {}
	env['trapoid'] = trapoid
	env['trapname'] = external_snmptranslate(trapoid)
	env['trapcount'] = Trap.objects.filter(trapoid=trapoid).count()

	if db:
		try:
			# FIXME: Mehrere, aktiv/inaktiv
			cfgtrap = CfgTrap.objects.get(trapoid=trapoid)
		except CfgTrap.DoesNotExist:
			cfgtrap = CfgTrap()
			cfgtrap.trapoid = trapoid
		cfgtrap.trapname = env['trapname']
		cfgtrap.save()

		with connection.cursor() as c:
			c.execute('UPDATE ' + Trap._meta.db_table + ' SET trapname=%s WHERE trapoid=%s', [env['trapname'], trapoid,])
		return HttpResponseRedirect( reverse('trap_index') + '?trapoid=%s' % trapoid )

	return render(Request, 'traps/resolv_trapoid.html', env)