mirror of
https://gitlab.federez.net/re2o/re2o
synced 2024-06-02 07:52:23 +00:00
309 lines
10 KiB
Python
309 lines
10 KiB
Python
# 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 = users_filter
|
|
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(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 = users_filter
|
|
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(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
|
|
) | Q(
|
|
port__details=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',)):
|
|
filter_switch_ports_list = Q(
|
|
details__icontains=query
|
|
) | Q(
|
|
switch__switch_interface__domain__name__icontains=query
|
|
) | Q(
|
|
room__name__icontains=query
|
|
) | Q(
|
|
machine_interface__domain__name__icontains=query
|
|
) | Q(
|
|
related__switch__switch_interface__domain__name__icontains=query
|
|
) | Q(
|
|
radius__icontains=query
|
|
) | Q(
|
|
vlan_force__name__icontains=query
|
|
)
|
|
results['switch_ports_list'] = Port.objects.filter(filter_switch_ports_list)
|
|
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',)):
|
|
filter_switches = Q(
|
|
details__icontains=query
|
|
) | Q(
|
|
switch_interface__domain__name__icontains=query
|
|
) | Q(
|
|
switch_interface__ipv4__ipv4__icontains=query
|
|
) | Q(
|
|
location__icontains=query
|
|
) | Q(
|
|
stack__name__icontains=query
|
|
) | Q(
|
|
model__reference__icontains=query
|
|
) | Q(
|
|
model__constructor__name__icontains=query
|
|
)
|
|
results['switches_list'] = Switch.objects.filter(filter_switches)
|
|
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})
|