# Re2o est un logiciel d'administration développé initiallement au rezometz. Il # se veut agnostique au réseau considéré, de manière à être installable en # quelques clics. # # Copyright © 2017 Gabriel Détraz # Copyright © 2017 Goulven Kermarec # Copyright © 2017 Augustin Lemesle # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License along # with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. # App de recherche pour re2o # Augustin lemesle, Gabriel Détraz, Goulven Kermarec # Gplv2 from __future__ import unicode_literals from django.shortcuts import render from django.shortcuts import get_object_or_404 from django.template.context_processors import csrf from django.template import Context, RequestContext, loader from django.contrib.auth.decorators import login_required from django.db.models import Q from users.models import User, Ban, Whitelist from machines.models import Machine, Interface from topologie.models import Port, Switch, Room from cotisations.models import Facture from preferences.models import GeneralOption from search.forms import ( SearchForm, SearchFormPlus, CHOICES_USER, CHOICES_AFF, initial_choices ) from re2o.utils import SortTable def get_results(query, request, filters={}): start = filters.get('s', None) end = filters.get('e', None) user_state = filters.get('u', initial_choices(CHOICES_USER)) aff = filters.get('a', initial_choices(CHOICES_AFF)) options, created = GeneralOption.objects.get_or_create() max_result = options.search_display_page user_state_filter = Q() for s in user_state: user_state_filter |= Q(state = s) connexion = [] results = { 'users_list': User.objects.none(), 'machines_list' : Machine.objects.none(), 'factures_list' : Facture.objects.none(), 'bans_list' : Ban.objects.none(), 'whitelists_list': Whitelist.objects.none(), 'rooms_list': Room.objects.none(), 'switch_ports_list': Port.objects.none(), 'switches_list': Switch.objects.none() } users_filter = Q( user__pseudo__icontains = query ) | Q( user__adherent__name__icontains = query ) | Q( user__surname__icontains = query ) if not request.user.has_perms(('cableur',)): users_filter &= Q(user = request.user) # Users if '0' in aff: filter_user_list = Q( adherent__room__name__icontains = query ) | Q( club__room__name__icontains = query ) | Q( pseudo__icontains = query ) | Q( adherent__name__icontains = query ) | Q( surname__icontains = query ) & user_state_filter if not request.user.has_perms(('cableur',)): filter_user_list &= Q(id=request.user.id) results['users_list'] = User.objects.filter(filter_user_list) results['users_list'] = SortTable.sort( results['users_list'], request.GET.get('col'), request.GET.get('order'), SortTable.USERS_INDEX ) # Machines if '1' in aff: filter_machine_list = Q( user__pseudo__icontains = query ) | Q( user__adherent__name__icontains = query ) | Q( user__surname__icontains = query ) | Q( interface__mac_address__icontains = query ) | Q( interface__ipv4__ipv4__icontains = query ) | Q( interface__domain__name__icontains = query ) | Q( interface__domain__related_domain__name__icontains = query ) if not request.user.has_perms(('cableur',)): filter_machine_list &= Q(user__id=request.user.id) results['machines_list'] = Machine.objects.filter(filter_machine_list) results['machines_list'] = SortTable.sort( results['machines_list'], request.GET.get('col'), request.GET.get('order'), SortTable.MACHINES_INDEX ) # Factures if '2' in aff: filter_facture_list = users_filter if start != None: filter_facture_list &= Q(date__gte=start) if end != None: filter_facture_list &= Q(date__lte=end) results['factures_list'] = Facture.objects.filter(filter_facture_list) results['factures_list'] = SortTable.sort( results['factures_list'], request.GET.get('col'), request.GET.get('order'), SortTable.COTISATIONS_INDEX ) # Bans if '3' in aff: date_filter = Q() if start != None: date_filter &= ( Q(date_start__gte=start) & Q(date_end__gte=start) ) | ( Q(date_start__lte=start) & Q(date_end__gte=start) ) | ( Q(date_start__gte=start) & Q(date_end__lte=start) ) if end != None: date_filter &= ( Q(date_start__lte=end) & Q(date_end__lte=end) ) | ( Q(date_start__lte=end) & Q(date_end__gte=end) ) | ( Q(date_start__gte=end) & Q(date_end__lte=end) ) results['bans_list'] = Ban.objects.filter(users_filter & date_filter) results['bans_list'] = SortTable.sort( results['bans_list'], request.GET.get('col'), request.GET.get('order'), SortTable.USERS_INDEX_BAN ) # Whitelists if '4' in aff: date_filter = Q() if start != None: date_filter &= ( Q(date_start__gte=start) & Q(date_end__gte=start) ) | ( Q(date_start__lte=start) & Q(date_end__gte=start) ) | ( Q(date_start__gte=start) & Q(date_end__lte=start) ) if end != None: date_filter &= ( Q(date_start__lte=end) & Q(date_end__lte=end) ) | ( Q(date_start__lte=end) & Q(date_end__gte=end) ) | ( Q(date_start__gte=end) & Q(date_end__lte=end) ) results['whitelists_list'] = Whitelist.objects.filter(users_filter & date_filter) results['whitelists_list'] = SortTable.sort( results['whitelists_list'], request.GET.get('col'), request.GET.get('order'), SortTable.USERS_INDEX_WHITE ) # Rooms if '5' in aff and request.user.has_perms(('cableur',)): filter_rooms_list = Q(details__icontains=query) | Q(name__icontains=query) results['rooms_list'] = Room.objects.filter(filter_rooms_list) results['rooms_list'] = SortTable.sort( results['rooms_list'], request.GET.get('col'), request.GET.get('order'), SortTable.TOPOLOGIE_INDEX_ROOM ) # Switch ports if '6' in aff and request.user.has_perms(('cableur',)): results['switch_ports_list'] = Port.objects.filter(details__icontains=query) results['switch_ports_list'] = SortTable.sort( results['switch_ports_list'], request.GET.get('col'), request.GET.get('order'), SortTable.TOPOLOGIE_INDEX_PORT ) # Switches if '7' in aff and request.user.has_perms(('cableur',)): results['switches_list'] = Switch.objects.filter(details__icontains=query) results['switches_list'] = SortTable.sort( results['switches_list'], request.GET.get('col'), request.GET.get('order'), SortTable.TOPOLOGIE_INDEX ) for r in results.keys(): results[r] = results[r].distinct()[:max_result] results.update({'max_result': max_result}) results.update({'search_term': query}) return results @login_required def search(request): search_form = SearchForm(request.GET or None) if search_form.is_valid(): return render( request, 'search/index.html', get_results( search_form.cleaned_data.get('q', ''), request, search_form.cleaned_data ) ) return render(request, 'search/search.html', {'search_form' : search_form}) @login_required def searchp(request): search_form = SearchFormPlus(request.GET or None) if search_form.is_valid(): return render( request, 'search/index.html', get_results( search_form.cleaned_data.get('q', ''), request, search_form.cleaned_data ) ) return render(request, 'search/search.html', {'search_form' : search_form})