8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-12-25 00:13:45 +00:00
re2o/machines/forms.py

650 lines
22 KiB
Python
Raw Normal View History

# -*- mode: python; coding: utf-8 -*-
2017-01-15 23:01:18 +00:00
# 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
# Copyright © 2017 Maël Kervella
2017-01-15 23:01:18 +00:00
#
# 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.
2017-10-16 00:41:21 +00:00
"""
Formulaires d'ajout, edition et suppressions de :
- machines
- interfaces
- domain (noms de machine)
- alias (cname)
- service (dhcp, dns..)
- ns (serveur dns)
- mx (serveur mail)
- ports ouverts et profils d'ouverture par interface
"""
2017-01-15 23:01:18 +00:00
from __future__ import unicode_literals
2017-10-16 00:41:21 +00:00
from django.forms import ModelForm, Form
2016-07-07 11:19:03 +00:00
from django import forms
from django.utils.translation import ugettext_lazy as _l
2016-07-06 20:49:16 +00:00
2017-12-30 00:48:23 +00:00
from re2o.field_permissions import FieldPermissionFormMixin
from re2o.mixins import FormRevMixin
2017-12-30 00:48:23 +00:00
2017-10-16 00:41:21 +00:00
from .models import (
Domain,
Machine,
Interface,
IpList,
MachineType,
Extension,
SOA,
2017-10-16 00:41:21 +00:00
Mx,
2017-11-15 14:17:22 +00:00
Txt,
2018-06-25 14:50:45 +00:00
DName,
2017-10-16 00:41:21 +00:00
Ns,
2018-06-23 14:39:03 +00:00
Role,
2017-10-16 00:41:21 +00:00
Service,
Vlan,
2017-11-16 01:33:57 +00:00
Srv,
SshFp,
2017-10-16 00:41:21 +00:00
Nas,
IpType,
OuverturePortList,
Ipv6List,
2017-10-16 00:41:21 +00:00
)
class EditMachineForm(FormRevMixin, FieldPermissionFormMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Formulaire d'édition d'une machine"""
2016-07-06 20:49:16 +00:00
class Meta:
model = Machine
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(EditMachineForm, self).__init__(*args, prefix=prefix, **kwargs)
2016-07-06 20:49:16 +00:00
self.fields['name'].label = 'Nom de la machine'
2017-10-16 00:41:21 +00:00
2016-07-06 20:49:16 +00:00
class NewMachineForm(EditMachineForm):
2017-10-16 00:41:21 +00:00
"""Creation d'une machine, ne renseigne que le nom"""
2016-07-06 20:49:16 +00:00
class Meta(EditMachineForm.Meta):
2016-07-18 17:14:48 +00:00
fields = ['name']
2016-07-06 20:49:16 +00:00
2017-10-16 00:41:21 +00:00
class EditInterfaceForm(FormRevMixin, FieldPermissionFormMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Edition d'une interface. Edition complète"""
2016-07-06 20:49:16 +00:00
class Meta:
model = Interface
fields = ['machine', 'type', 'ipv4', 'mac_address', 'details']
2016-07-06 20:49:16 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
user = kwargs.get('user')
super(EditInterfaceForm, self).__init__(*args, prefix=prefix, **kwargs)
2016-07-06 20:49:16 +00:00
self.fields['mac_address'].label = 'Adresse mac'
2016-07-18 17:14:48 +00:00
self.fields['type'].label = 'Type de machine'
self.fields['type'].empty_label = "Séléctionner un type de machine"
if "ipv4" in self.fields:
2018-04-13 22:48:44 +00:00
self.fields['ipv4'].empty_label = ("Assignation automatique de "
"l'ipv4")
2017-10-16 00:41:21 +00:00
self.fields['ipv4'].queryset = IpList.objects.filter(
interface__isnull=True
)
can_use_all_iptype, _reason = IpType.can_use_all(user)
if not can_use_all_iptype:
self.fields['ipv4'].queryset = IpList.objects.filter(
interface__isnull=True
).filter(ip_type__in=IpType.objects.filter(need_infra=False))
else:
self.fields['ipv4'].queryset = IpList.objects.filter(
interface__isnull=True
)
# Add it's own address
2017-10-16 00:41:21 +00:00
self.fields['ipv4'].queryset |= IpList.objects.filter(
interface=self.instance
)
2017-05-28 17:22:45 +00:00
if "machine" in self.fields:
2017-10-16 00:41:21 +00:00
self.fields['machine'].queryset = Machine.objects.all()\
.select_related('user')
can_use_all_machinetype, _reason = MachineType.can_use_all(user)
if not can_use_all_machinetype:
2017-10-16 00:41:21 +00:00
self.fields['type'].queryset = MachineType.objects.filter(
ip_type__in=IpType.objects.filter(need_infra=False)
)
2016-07-06 20:49:16 +00:00
class AddInterfaceForm(EditInterfaceForm):
"""Ajout d'une interface à une machine. En fonction des droits,
affiche ou non l'ensemble des ip disponibles"""
class Meta(EditInterfaceForm.Meta):
2017-10-16 00:41:21 +00:00
fields = ['type', 'ipv4', 'mac_address', 'details']
class AliasForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout d'un alias (et edition), CNAME, contenant nom et extension"""
class Meta:
2016-12-24 19:04:53 +00:00
model = Domain
2017-10-16 00:41:21 +00:00
fields = ['name', 'extension']
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
2017-12-29 00:42:00 +00:00
user = kwargs.pop('user')
super(AliasForm, self).__init__(*args, prefix=prefix, **kwargs)
2018-04-14 18:19:02 +00:00
can_use_all, _reason = Extension.can_use_all(user)
2017-12-29 00:42:00 +00:00
if not can_use_all:
2017-10-21 21:25:22 +00:00
self.fields['extension'].queryset = Extension.objects.filter(
2018-04-14 18:19:02 +00:00
need_infra=False
2017-10-21 21:25:22 +00:00
)
2017-10-16 00:41:21 +00:00
class DomainForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout et edition d'un enregistrement de nom, relié à interface"""
2017-10-22 12:41:15 +00:00
class Meta:
model = Domain
fields = ['name']
def __init__(self, *args, **kwargs):
if 'user' in kwargs:
user = kwargs.pop('user')
2017-01-09 15:20:13 +00:00
initial = kwargs.get('initial', {})
initial['name'] = user.get_next_domain_name()
2017-10-16 00:41:21 +00:00
kwargs['initial'] = initial
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(DomainForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
class DelAliasForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs objets alias"""
alias = forms.ModelMultipleChoiceField(
queryset=Domain.objects.all(),
label="Alias actuels",
widget=forms.CheckboxSelectMultiple
)
def __init__(self, *args, **kwargs):
interface = kwargs.pop('interface')
super(DelAliasForm, self).__init__(*args, **kwargs)
2017-10-16 00:41:21 +00:00
self.fields['alias'].queryset = Domain.objects.filter(
cname__in=Domain.objects.filter(interface_parent=interface)
)
class MachineTypeForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout et edition d'un machinetype, relié à un iptype"""
2016-07-07 11:19:03 +00:00
class Meta:
model = MachineType
2017-10-16 00:41:21 +00:00
fields = ['type', 'ip_type']
2016-07-07 11:19:03 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(MachineTypeForm, self).__init__(*args, prefix=prefix, **kwargs)
2016-07-07 11:19:03 +00:00
self.fields['type'].label = 'Type de machine à ajouter'
2016-10-22 22:55:58 +00:00
self.fields['ip_type'].label = "Type d'ip relié"
2016-07-07 11:19:03 +00:00
2017-10-16 00:41:21 +00:00
class DelMachineTypeForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs machinetype"""
machinetypes = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=MachineType.objects.none(),
2017-10-16 00:41:21 +00:00
label="Types de machines actuelles",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelMachineTypeForm, self).__init__(*args, **kwargs)
if instances:
self.fields['machinetypes'].queryset = instances
else:
self.fields['machinetypes'].queryset = MachineType.objects.all()
2016-07-07 11:19:03 +00:00
class IpTypeForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Formulaire d'ajout d'un iptype. Pas d'edition de l'ip de start et de
stop après creation"""
2016-10-22 22:55:58 +00:00
class Meta:
model = IpType
2017-10-16 00:41:21 +00:00
fields = ['type', 'extension', 'need_infra', 'domaine_ip_start',
2018-06-25 15:15:48 +00:00
'domaine_ip_stop', 'dnssec_reverse_v4', 'prefix_v6',
'dnssec_reverse_v6', 'vlan', 'ouverture_ports']
2017-10-16 00:41:21 +00:00
2016-10-22 22:55:58 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(IpTypeForm, self).__init__(*args, prefix=prefix, **kwargs)
2016-10-22 22:55:58 +00:00
self.fields['type'].label = 'Type ip à ajouter'
2017-10-16 00:41:21 +00:00
class EditIpTypeForm(IpTypeForm):
2017-10-16 00:41:21 +00:00
"""Edition d'un iptype. Pas d'edition du rangev4 possible, car il faudrait
synchroniser les objets iplist"""
class Meta(IpTypeForm.Meta):
2017-10-16 00:41:21 +00:00
fields = ['extension', 'type', 'need_infra', 'prefix_v6', 'vlan',
2018-06-25 15:15:48 +00:00
'dnssec_reverse_v4', 'dnssec_reverse_v6',
2017-10-16 00:41:21 +00:00
'ouverture_ports']
2016-10-22 22:55:58 +00:00
class DelIpTypeForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs iptype"""
iptypes = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=IpType.objects.none(),
2017-10-16 00:41:21 +00:00
label="Types d'ip actuelles",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelIpTypeForm, self).__init__(*args, **kwargs)
if instances:
self.fields['iptypes'].queryset = instances
else:
self.fields['iptypes'].queryset = IpType.objects.all()
2016-10-22 22:55:58 +00:00
class ExtensionForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Formulaire d'ajout et edition d'une extension"""
2016-07-08 15:54:06 +00:00
class Meta:
model = Extension
fields = '__all__'
2016-07-08 15:54:06 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(ExtensionForm, self).__init__(*args, prefix=prefix, **kwargs)
2016-10-26 12:05:40 +00:00
self.fields['name'].label = 'Extension à ajouter'
self.fields['origin'].label = 'Enregistrement A origin'
self.fields['origin_v6'].label = 'Enregistrement AAAA origin'
self.fields['soa'].label = 'En-tête SOA à utiliser'
2016-07-08 15:54:06 +00:00
2017-10-16 00:41:21 +00:00
class DelExtensionForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'une ou plusieurs extensions"""
extensions = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Extension.objects.none(),
2017-10-16 00:41:21 +00:00
label="Extensions actuelles",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelExtensionForm, self).__init__(*args, **kwargs)
if instances:
self.fields['extensions'].queryset = instances
else:
self.fields['extensions'].queryset = Extension.objects.all()
2016-07-08 15:54:06 +00:00
class Ipv6ListForm(FormRevMixin, FieldPermissionFormMixin, ModelForm):
"""Gestion des ipv6 d'une machine"""
class Meta:
model = Ipv6List
fields = ['ipv6', 'slaac_ip']
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(Ipv6ListForm, self).__init__(*args, prefix=prefix, **kwargs)
class SOAForm(FormRevMixin, ModelForm):
"""Ajout et edition d'un SOA"""
class Meta:
model = SOA
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(SOAForm, self).__init__(*args, prefix=prefix, **kwargs)
class DelSOAForm(FormRevMixin, Form):
"""Suppression d'un ou plusieurs SOA"""
soa = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=SOA.objects.none(),
label="SOA actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelSOAForm, self).__init__(*args, **kwargs)
if instances:
self.fields['soa'].queryset = instances
else:
self.fields['soa'].queryset = SOA.objects.all()
class MxForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout et edition d'un MX"""
class Meta:
model = Mx
fields = ['zone', 'priority', 'name']
2017-10-16 00:41:21 +00:00
2017-01-05 22:48:45 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(MxForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
self.fields['name'].queryset = Domain.objects.exclude(
interface_parent=None
2017-10-18 15:19:51 +00:00
).select_related('extension')
2017-10-16 00:41:21 +00:00
2018-04-13 22:48:44 +00:00
class DelMxForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs MX"""
mx = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Mx.objects.none(),
2017-10-16 00:41:21 +00:00
label="MX actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelMxForm, self).__init__(*args, **kwargs)
if instances:
self.fields['mx'].queryset = instances
else:
self.fields['mx'].queryset = Mx.objects.all()
class NsForm(FormRevMixin, ModelForm):
2017-10-18 15:19:51 +00:00
"""Ajout d'un NS pour une zone
On exclue les CNAME dans les objets domain (interdit par la rfc)
donc on prend uniquemet """
class Meta:
model = Ns
2016-12-26 16:43:41 +00:00
fields = ['zone', 'ns']
2016-12-26 18:45:43 +00:00
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(NsForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
self.fields['ns'].queryset = Domain.objects.exclude(
interface_parent=None
2017-10-18 15:19:51 +00:00
).select_related('extension')
2017-10-16 00:41:21 +00:00
2016-12-26 18:45:43 +00:00
class DelNsForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppresion d'un ou plusieurs NS"""
ns = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Ns.objects.none(),
2017-10-16 00:41:21 +00:00
label="Enregistrements NS actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelNsForm, self).__init__(*args, **kwargs)
if instances:
self.fields['ns'].queryset = instances
else:
self.fields['ns'].queryset = Ns.objects.all()
class TxtForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout d'un txt pour une zone"""
2017-09-05 16:18:41 +00:00
class Meta:
2017-11-15 14:17:22 +00:00
model = Txt
2017-09-05 16:18:41 +00:00
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(TxtForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
class DelTxtForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs TXT"""
txt = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Txt.objects.none(),
2017-10-16 00:41:21 +00:00
label="Enregistrements Txt actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelTxtForm, self).__init__(*args, **kwargs)
if instances:
self.fields['txt'].queryset = instances
else:
self.fields['txt'].queryset = Txt.objects.all()
2018-07-23 21:19:19 +00:00
2018-06-25 14:50:45 +00:00
class DNameForm(FormRevMixin, ModelForm):
2018-07-23 21:19:19 +00:00
"""Add a DNAME entry for a zone"""
2018-06-25 14:50:45 +00:00
class Meta:
model = DName
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(DNameForm, self).__init__(*args, prefix=prefix, **kwargs)
class DelDNameForm(FormRevMixin, Form):
2018-07-23 21:19:19 +00:00
"""Delete a set of DNAME entries"""
2018-06-25 14:50:45 +00:00
dnames = forms.ModelMultipleChoiceField(
queryset=Txt.objects.none(),
2018-07-23 21:19:19 +00:00
label="Existing DNAME entries",
2018-06-25 14:50:45 +00:00
widget=forms.CheckboxSelectMultiple
)
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelDNameForm, self).__init__(*args, **kwargs)
if instances:
self.fields['dnames'].queryset = instances
else:
self.fields['dnames'].queryset = DName.objects.all()
class SrvForm(FormRevMixin, ModelForm):
2017-11-16 01:33:57 +00:00
"""Ajout d'un srv pour une zone"""
class Meta:
model = Srv
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(SrvForm, self).__init__(*args, prefix=prefix, **kwargs)
class DelSrvForm(FormRevMixin, Form):
2017-11-16 01:33:57 +00:00
"""Suppression d'un ou plusieurs Srv"""
srv = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Srv.objects.none(),
2017-11-16 01:33:57 +00:00
label="Enregistrements Srv actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelSrvForm, self).__init__(*args, **kwargs)
if instances:
self.fields['srv'].queryset = instances
else:
self.fields['srv'].queryset = Srv.objects.all()
2017-09-05 16:18:41 +00:00
class NasForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout d'un type de nas (machine d'authentification,
swicths, bornes...)"""
class Meta:
model = Nas
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(NasForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
class DelNasForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs nas"""
nas = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Nas.objects.none(),
2017-10-16 00:41:21 +00:00
label="Enregistrements Nas actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelNasForm, self).__init__(*args, **kwargs)
if instances:
self.fields['nas'].queryset = instances
else:
self.fields['nas'].queryset = Nas.objects.all()
2018-06-23 14:39:03 +00:00
class RoleForm(FormRevMixin, ModelForm):
"""Add and edit role."""
2018-06-23 14:39:03 +00:00
class Meta:
model = Role
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(RoleForm, self).__init__(*args, prefix=prefix, **kwargs)
self.fields['servers'].queryset = (Interface.objects.all()
.select_related(
'domain__extension'
))
class DelRoleForm(FormRevMixin, Form):
"""Deletion of one or several roles."""
2018-06-23 14:39:03 +00:00
role = forms.ModelMultipleChoiceField(
queryset=Role.objects.none(),
label=_l("Current roles"),
2018-06-23 14:39:03 +00:00
widget=forms.CheckboxSelectMultiple
)
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelRoleForm, self).__init__(*args, **kwargs)
if instances:
self.fields['role'].queryset = instances
else:
self.fields['role'].queryset = Role.objects.all()
2018-06-23 14:39:03 +00:00
class ServiceForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout et edition d'une classe de service : dns, dhcp, etc"""
class Meta:
model = Service
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(ServiceForm, self).__init__(*args, prefix=prefix, **kwargs)
2018-04-13 22:48:44 +00:00
self.fields['servers'].queryset = (Interface.objects.all()
.select_related(
'domain__extension'
))
def save(self, commit=True):
2018-04-14 18:19:02 +00:00
# TODO : None of the parents of ServiceForm use the commit
# parameter in .save()
instance = super(ServiceForm, self).save(commit=False)
if commit:
instance.save()
instance.process_link(self.cleaned_data.get('servers'))
return instance
2017-10-16 00:41:21 +00:00
class DelServiceForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs service"""
service = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Service.objects.none(),
2017-10-16 00:41:21 +00:00
label="Services actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelServiceForm, self).__init__(*args, **kwargs)
if instances:
self.fields['service'].queryset = instances
else:
self.fields['service'].queryset = Service.objects.all()
class VlanForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Ajout d'un vlan : id, nom"""
class Meta:
model = Vlan
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(VlanForm, self).__init__(*args, prefix=prefix, **kwargs)
2017-10-16 00:41:21 +00:00
class DelVlanForm(FormRevMixin, Form):
2017-10-16 00:41:21 +00:00
"""Suppression d'un ou plusieurs vlans"""
vlan = forms.ModelMultipleChoiceField(
2017-12-13 17:31:13 +00:00
queryset=Vlan.objects.none(),
2017-10-16 00:41:21 +00:00
label="Vlan actuels",
widget=forms.CheckboxSelectMultiple
)
2017-12-13 17:31:13 +00:00
def __init__(self, *args, **kwargs):
instances = kwargs.pop('instances', None)
super(DelVlanForm, self).__init__(*args, **kwargs)
if instances:
self.fields['vlan'].queryset = instances
else:
self.fields['vlan'].queryset = Vlan.objects.all()
class EditOuverturePortConfigForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Edition de la liste des profils d'ouverture de ports
pour l'interface"""
class Meta:
model = Interface
fields = ['port_lists']
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
2017-10-16 00:41:21 +00:00
super(EditOuverturePortConfigForm, self).__init__(
*args,
prefix=prefix,
**kwargs
)
class EditOuverturePortListForm(FormRevMixin, ModelForm):
2017-10-16 00:41:21 +00:00
"""Edition de la liste des ports et profils d'ouverture
des ports"""
class Meta:
model = OuverturePortList
fields = '__all__'
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
2017-10-16 00:41:21 +00:00
super(EditOuverturePortListForm, self).__init__(
*args,
prefix=prefix,
**kwargs
)
class SshFpForm(FormRevMixin, ModelForm):
"""Edits a SSHFP record."""
class Meta:
model = SshFp
exclude = ('machine',)
def __init__(self, *args, **kwargs):
prefix = kwargs.pop('prefix', self.Meta.model.__name__)
super(SshFpForm, self).__init__(
*args,
prefix=prefix,
**kwargs
)