@ -15,351 +15,351 @@ import ipaddress
class WhoisObjectFormMixin ( object ) : class WhoisObjectFormMixin ( object ) :
def __init__ ( self , user , * args , * * kwargs ) : def __init__ ( self , user , * args , * * kwargs ) :
super ( WhoisObjectFormMixin , self ) . __init__ ( * args , * * kwargs ) super ( WhoisObjectFormMixin , self ) . __init__ ( * args , * * kwargs )
self . _user = user self . _user = user
instance = getattr ( self , ' instance ' , None ) instance = getattr ( self , ' instance ' , None )
if instance and instance . pk : if instance and instance . pk :
self . _create = False self . _create = False
#self.fields['handle'].disabled = True #self.fields['handle'].disabled = True
else : else :
self . _create = True self . _create = True
if ' handle ' in self . fields : if ' handle ' in self . fields :
self . fields [ ' handle ' ] . help_text = " Handle for this object in uppercase with a suffix of - %s " % instance . handleSuffix self . fields [ ' handle ' ] . help_text = " Handle for this object in uppercase with a suffix of - %s " % instance . handleSuffix
# only show users contacts and already present contacts # only show users contacts and already present contacts
mnts = self . _user . maintainer_set . all ( ) mnts = self . _user . maintainer_set . all ( )
if ' admin_c ' in self . fields : if ' admin_c ' in self . fields :
self . fields [ ' admin_c ' ] . queryset = Contact . getMntQueryset ( mnts , self . instance , " admin_c " ) self . fields [ ' admin_c ' ] . queryset = Contact . getMntQueryset ( mnts , self . instance , " admin_c " )
def clean_handle ( self ) : def clean_handle ( self ) :
HandleValidatorWithSuffix ( self . instance . handleSuffix ) ( self . cleaned_data [ ' handle ' ] ) HandleValidatorWithSuffix ( self . instance . handleSuffix ) ( self . cleaned_data [ ' handle ' ] )
return self . cleaned_data [ ' handle ' ] return self . cleaned_data [ ' handle ' ]
def clean ( self ) : def clean ( self ) :
cleaned_data = super ( WhoisObjectFormMixin , self ) . clean ( ) cleaned_data = super ( WhoisObjectFormMixin , self ) . clean ( )
if cleaned_data . get ( " handle " ) == " AUTO " and not self . errors : if cleaned_data . get ( " handle " ) == " AUTO " and not self . errors :
name = cleaned_data . get ( " name " ) name = cleaned_data . get ( " name " )
if name is None : if name is None :
name = self . _user . username name = self . _user . username
cleaned_data [ ' handle ' ] = self . _meta . model . genGenericHandle ( name ) cleaned_data [ ' handle ' ] = self . _meta . model . genGenericHandle ( name )
# XXX: Find a better position to update last_changed # XXX: Find a better position to update last_changed
self . instance . last_modified = timezone . now ( ) self . instance . last_modified = timezone . now ( )
return cleaned_data return cleaned_data
class MntFormMixin ( object ) : class MntFormMixin ( object ) :
protectedFields = [ ] protectedFields = [ ]
def __init__ ( self , lower = False , * args , * * kwargs ) : def __init__ ( self , lower = False , * args , * * kwargs ) :
super ( MntFormMixin , self ) . __init__ ( * args , * * kwargs ) super ( MntFormMixin , self ) . __init__ ( * args , * * kwargs )
self . _editLower = lower self . _editLower = lower
if self . _editLower : if self . _editLower :
for key in self . protectedFields : for key in self . protectedFields :
self . fields [ key ] . disabled = True self . fields [ key ] . disabled = True
instance = getattr ( self , " instance " , None ) instance = getattr ( self , " instance " , None )
if not hasattr ( self , " _create " ) : if not hasattr ( self , " _create " ) :
self . _create = not ( instance and instance . pk ) self . _create = not ( instance and instance . pk )
mntQs = orderQueryset ( Maintainer . objects . all ( ) , Q ( auth = self . _user ) , Q ( pk__in = instance . mnt_by . all ( ) ) if not self . _create else None ) mntQs = orderQueryset ( Maintainer . objects . all ( ) , Q ( auth = self . _user ) , Q ( pk__in = instance . mnt_by . all ( ) ) if not self . _create else None )
self . fields [ " mnt_by " ] . queryset = mntQs self . fields [ " mnt_by " ] . queryset = mntQs
if " mnt_lower " in self . fields : if " mnt_lower " in self . fields :
self . fields [ " mnt_lower " ] . queryset = mntQs self . fields [ " mnt_lower " ] . queryset = mntQs
def clean ( self ) : def clean ( self ) :
cleaned_data = super ( MntFormMixin , self ) . clean ( ) cleaned_data = super ( MntFormMixin , self ) . clean ( )
if not self . errors : if not self . errors :
if self . _create : if self . _create :
# at least one own mnt on creation # at least one own mnt on creation
mnts = self . _user . maintainer_set . all ( ) mnts = self . _user . maintainer_set . all ( )
for mnt in cleaned_data [ ' mnt_by ' ] : for mnt in cleaned_data [ ' mnt_by ' ] :
if mnt in mnts : if mnt in mnts :
break break
else : else :
raise forms . ValidationError ( " On object creation at least one maintainer needs to be under your control " ) raise forms . ValidationError ( " On object creation at least one maintainer needs to be under your control " )
if cleaned_data [ ' mnt_by ' ] . count ( ) > 5 or " mnt_lower " in self . fields and cleaned_data [ ' mnt_lower ' ] . count ( ) > 5 : if cleaned_data [ ' mnt_by ' ] . count ( ) > 5 or " mnt_lower " in self . fields and cleaned_data [ ' mnt_lower ' ] . count ( ) > 5 :
raise forms . ValidationError ( " Currently only 5 MNTs per object allowed. If you need more ask @ irc/rrequest and state your use case " ) raise forms . ValidationError ( " Currently only 5 MNTs per object allowed. If you need more ask @ irc/rrequest and state your use case " )
return cleaned_data return cleaned_data
class MntForm ( WhoisObjectFormMixin , forms . ModelForm ) : class MntForm ( WhoisObjectFormMixin , forms . ModelForm ) :
class Meta : class Meta :
model = Maintainer model = Maintainer
#fields = ['handle', 'description', 'admin_c', 'auth'] #fields = ['handle', 'description', 'admin_c', 'auth']
fields = [ ' handle ' , ' description ' , ' admin_c ' ] fields = [ ' handle ' , ' description ' , ' admin_c ' ]
widgets = { ' auth ' : MultiTextInput ( ) } widgets = { ' auth ' : MultiTextInput ( ) }
help_texts = { help_texts = {
' auth ' : ' Enter names of users which can edit this object (space separated; ' ' auth ' : ' Enter names of users which can edit this object (space separated; '
' and yes, validation is somewhat broken (values disappear on error - just reload)) ' ' and yes, validation is somewhat broken (values disappear on error - just reload)) '
} }
def __init__ ( self , * args , * * kwargs ) : def __init__ ( self , * args , * * kwargs ) :
super ( MntForm , self ) . __init__ ( * args , * * kwargs ) super ( MntForm , self ) . __init__ ( * args , * * kwargs )
print ( args , kwargs ) print ( args , kwargs )
#if self._create: #if self._create:
# self.fields['auth'].text("noot") # self.fields['auth'].text("noot")
class MntInitialForm ( MntForm ) : class MntInitialForm ( MntForm ) :
class Meta : class Meta :
model = Maintainer model = Maintainer
fields = [ ' handle ' , ' description ' ] fields = [ ' handle ' , ' description ' ]
class ContactForm ( WhoisObjectFormMixin , forms . ModelForm ) : class ContactForm ( WhoisObjectFormMixin , forms . ModelForm ) :
class Meta : class Meta :
model = Contact model = Contact
fields = [ ' handle ' , ' name ' , ' mnt_by ' ] fields = [ ' handle ' , ' name ' , ' mnt_by ' ]
def __init__ ( self , * args , * * kwargs ) : def __init__ ( self , * args , * * kwargs ) :
super ( ContactForm , self ) . __init__ ( * args , * * kwargs ) super ( ContactForm , self ) . __init__ ( * args , * * kwargs )
if " mnt_by " in self . fields : if " mnt_by " in self . fields :
self . fields [ ' mnt_by ' ] . queryset = Maintainer . objects . filter ( auth = self . _user ) . distinct ( ) self . fields [ ' mnt_by ' ] . queryset = Maintainer . objects . filter ( auth = self . _user ) . distinct ( )
class ContactInitialForm ( ContactForm ) : class ContactInitialForm ( ContactForm ) :
class Meta : class Meta :
model = Contact model = Contact
fields = [ ' handle ' , ' name ' ] fields = [ ' handle ' , ' name ' ]
class InetNumForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) : class InetNumForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) :
prefix = forms . CharField ( ) prefix = forms . CharField ( )
protectedFields = [ ' protocol ' , ' parent_range ' , ' mnt_by ' , ' prefix ' ] protectedFields = [ ' protocol ' , ' parent_range ' , ' mnt_by ' , ' prefix ' ]
class Meta : class Meta :
model = InetNum model = InetNum
fields = [ ' handle ' , ' protocol ' , ' parent_range ' , ' prefix ' , ' name ' , ' description ' , ' origin_as ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ] fields = [ ' handle ' , ' protocol ' , ' parent_range ' , ' prefix ' , ' name ' , ' description ' , ' origin_as ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ]
def __init__ ( self , * args , * * kwargs ) : def __init__ ( self , * args , * * kwargs ) :
super ( InetNumForm , self ) . __init__ ( * args , * * kwargs ) super ( InetNumForm , self ) . __init__ ( * args , * * kwargs )
if self . _editLower : if self . _editLower :
for key in self . protectedFields : for key in self . protectedFields :
self . fields [ key ] . disabled = True self . fields [ key ] . disabled = True
mnts = self . _user . maintainer_set . all ( ) mnts = self . _user . maintainer_set . all ( )
#self.fields['parent_range'].queryset = InetNum.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts)) #self.fields['parent_range'].queryset = InetNum.objects.filter(Q(mnt_by__in=mnts) | Q(mnt_lower__in=mnts))
#if not self._create: #if not self._create:
# self.fields['parent_range'].queryset |= InetNum.objects.filter(pk=self.instance.pk) # self.fields['parent_range'].queryset |= InetNum.objects.filter(pk=self.instance.pk)
#self.fields['parent_range'].queryset = self.fields['parent_range'].queryset.distinct() #self.fields['parent_range'].queryset = self.fields['parent_range'].queryset.distinct()
self . fields [ ' parent_range ' ] . queryset = InetNum . getMntQueryset ( mnts , self . instance , " parent_range " ) self . fields [ ' parent_range ' ] . queryset = InetNum . getMntQueryset ( mnts , self . instance , " parent_range " )
self . fields [ ' origin_as ' ] . queryset = ASNumber . getMntQueryset ( mnts , self . instance , " origin_as " ) self . fields [ ' origin_as ' ] . queryset = ASNumber . getMntQueryset ( mnts , self . instance , " origin_as " )
def clean_prefix ( self ) : def clean_prefix ( self ) :
# make sure this is a subnet we're getting # make sure this is a subnet we're getting
net = self . cleaned_data [ ' prefix ' ] . lower ( ) net = self . cleaned_data [ ' prefix ' ] . lower ( )
IP46CIDRValidator ( net ) IP46CIDRValidator ( net )
try : try :
net = ipaddress . ip_network ( net ) net = ipaddress . ip_network ( net )
except ValueError as e : except ValueError as e :
raise forms . ValidationError ( str ( e ) ) raise forms . ValidationError ( str ( e ) )
return net return net
def clean_parent_range ( self ) : def clean_parent_range ( self ) :
parent_range = self . cleaned_data . get ( ' parent_range ' , None ) parent_range = self . cleaned_data . get ( ' parent_range ' , None )
# allow parent range to be unset for already present objects # allow parent range to be unset for already present objects
if not parent_range and ( self . _create or not self . _create and self . instance . parent_range ) : if not parent_range and ( self . _create or not self . _create and self . instance . parent_range ) :
raise forms . ValidationError ( " Parent range must be set " ) raise forms . ValidationError ( " Parent range must be set " )
if not self . _create and parent_range : if not self . _create and parent_range :
# make sure we don't have circular dependencies # make sure we don't have circular dependencies
obj = parent_range obj = parent_range
while obj . parent_range : while obj . parent_range :
if obj . pk == self . instance . pk : if obj . pk == self . instance . pk :
raise forms . ValidationError ( " No circular dependencies allowed " ) raise forms . ValidationError ( " No circular dependencies allowed " )
obj = obj . parent_range obj = obj . parent_range
if parent_range . origin_as . count ( ) > 0 : if parent_range . origin_as . count ( ) > 0 :
raise forms . ValidationError ( " Parent range has origin as set " ) raise forms . ValidationError ( " Parent range has origin as set " )
return parent_range return parent_range
def clean ( self ) : def clean ( self ) :
cleaned_data = super ( InetNumForm , self ) . clean ( ) cleaned_data = super ( InetNumForm , self ) . clean ( )
if not self . _editLower : if not self . _editLower :
if not self . errors : if not self . errors :
if not self . _create and self . cleaned_data [ ' origin_as ' ] : if not self . _create and self . cleaned_data [ ' origin_as ' ] :
if self . instance . inetnum_set . count ( ) > 0 : if self . instance . inetnum_set . count ( ) > 0 :
ranges = " , " . join ( map ( str , self . instance . inetnum_set . all ( ) ) ) ranges = " , " . join ( map ( str , self . instance . inetnum_set . all ( ) ) )
raise forms . ValidationError ( " You cannot set an origin as if there are already existing subranges ( %s ) " % ( ranges ) ) raise forms . ValidationError ( " You cannot set an origin as if there are already existing subranges ( %s ) " % ( ranges ) )
prefix = cleaned_data [ ' prefix ' ] prefix = cleaned_data [ ' prefix ' ]
parent = cleaned_data [ ' parent_range ' ] parent = cleaned_data [ ' parent_range ' ]
if parent : if parent :
parentNet = parent . getNetwork ( ) parentNet = parent . getNetwork ( )
if cleaned_data [ ' protocol ' ] != parent . protocol : if cleaned_data [ ' protocol ' ] != parent . protocol :
raise forms . ValidationError ( " Protocol type for prefix must be same as parent network " ) raise forms . ValidationError ( " Protocol type for prefix must be same as parent network " )
# check if in parent block # check if in parent block
if prefix . network_address not in parentNet or prefix . prefixlen < parentNet . prefixlen : if prefix . network_address not in parentNet or prefix . prefixlen < parentNet . prefixlen :
raise forms . ValidationError ( " Prefix must be inside parent network range " ) raise forms . ValidationError ( " Prefix must be inside parent network range " )
# check if parent block has net that overlaps with us # check if parent block has net that overlaps with us
for otherNet in parent . inetnum_set . all ( ) : for otherNet in parent . inetnum_set . all ( ) :
if self . instance and self . instance . pk == otherNet . pk : if self . instance and self . instance . pk == otherNet . pk :
continue continue
if otherNet . getNetwork ( ) . overlaps ( prefix ) : if otherNet . getNetwork ( ) . overlaps ( prefix ) :
raise forms . ValidationError ( " The given prefix overlaps with network %s " % otherNet . handle ) raise forms . ValidationError ( " The given prefix overlaps with network %s " % otherNet . handle )
# check if subnets to this subnet are (still) in current network # check if subnets to this subnet are (still) in current network
if not self . _create : if not self . _create :
for subnet in self . instance . inetnum_set . all ( ) : for subnet in self . instance . inetnum_set . all ( ) :
if subnet . getNetwork ( ) . network_address not in self . instance . getNetwork ( ) : if subnet . getNetwork ( ) . network_address not in self . instance . getNetwork ( ) :
raise forms . ValidationError ( " Subnet %s with %s is not in block anymore " % ( subnet , subnet . getNetwork ( ) ) ) raise forms . ValidationError ( " Subnet %s with %s is not in block anymore " % ( subnet , subnet . getNetwork ( ) ) )
self . instance . address = str ( prefix . network_address ) self . instance . address = str ( prefix . network_address )
self . instance . netmask = prefix . prefixlen self . instance . netmask = prefix . prefixlen
return cleaned_data return cleaned_data
class ASBlockForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) : class ASBlockForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) :
protectedFields = [ ' parent_block ' , ' asBegin ' , ' asEnd ' , ' mnt_by ' ] protectedFields = [ ' parent_block ' , ' asBegin ' , ' asEnd ' , ' mnt_by ' ]
# FIXME: Filter blocks # FIXME: Filter blocks
class Meta : class Meta :
model = ASBlock model = ASBlock
fields = [ ' handle ' , ' parent_block ' , ' asBegin ' , ' asEnd ' , ' name ' , ' description ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ] fields = [ ' handle ' , ' parent_block ' , ' asBegin ' , ' asEnd ' , ' name ' , ' description ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ]
def __init__ ( self , * args , * * kwargs ) : def __init__ ( self , * args , * * kwargs ) :
super ( ASBlockForm , self ) . __init__ ( * args , * * kwargs ) super ( ASBlockForm , self ) . __init__ ( * args , * * kwargs )
if not self . instance or self . instance and self . instance and self . instance . parent_block : if not self . instance or self . instance and self . instance and self . instance . parent_block :
self . fields [ " parent_block " ] . required = True self . fields [ " parent_block " ] . required = True
if self . instance and self . instance . pk : if self . instance and self . instance . pk :
self . fields [ " parent_block " ] . disabled = True self . fields [ " parent_block " ] . disabled = True
mnts = self . _user . maintainer_set . all ( ) mnts = self . _user . maintainer_set . all ( )
self . fields [ ' parent_block ' ] . queryset = ASBlock . getMntQueryset ( mnts , self . instance , " parent_block " ) self . fields [ ' parent_block ' ] . queryset = ASBlock . getMntQueryset ( mnts , self . instance , " parent_block " )
def clean_parent_block ( self ) : def clean_parent_block ( self ) :
parent_block = self . cleaned_data . get ( ' parent_block ' , None ) parent_block = self . cleaned_data . get ( ' parent_block ' , None )
# allow parent range to be unset for already present objects # allow parent range to be unset for already present objects
if not parent_block and ( self . _create or not self . _create and self . instance . parent_block ) : if not parent_block and ( self . _create or not self . _create and self . instance . parent_block ) :
raise forms . ValidationError ( " Parent block must be set " ) raise forms . ValidationError ( " Parent block must be set " )
if not self . _create and parent_block : if not self . _create and parent_block :
# make sure we don't have circular dependencies # make sure we don't have circular dependencies
obj = parent_block obj = parent_block
while obj . parent_block : while obj . parent_block :
if obj . pk == self . instance . pk : if obj . pk == self . instance . pk :
raise forms . ValidationError ( " No circular dependencies allowed " ) raise forms . ValidationError ( " No circular dependencies allowed " )
obj = obj . parent_block obj = obj . parent_block
return parent_block return parent_block
def clean ( self ) : def clean ( self ) :
cleaned_data = super ( ASBlockForm , self ) . clean ( ) cleaned_data = super ( ASBlockForm , self ) . clean ( )
if not self . errors : if not self . errors :
asBegin = cleaned_data [ ' asBegin ' ] asBegin = cleaned_data [ ' asBegin ' ]
asEnd = cleaned_data [ ' asEnd ' ] asEnd = cleaned_data [ ' asEnd ' ]
parent = cleaned_data [ ' parent_block ' ] parent = cleaned_data [ ' parent_block ' ]
# check if somebody is already using this block # check if somebody is already using this block
# check if in range # check if in range
if asBegin > asEnd : if asBegin > asEnd :
raise forms . ValidationError ( " AS beginning must be smaller or equal to AS end " ) raise forms . ValidationError ( " AS beginning must be smaller or equal to AS end " )
if parent : if parent :
if parent . asnumber_set . count ( ) > 0 : if parent . asnumber_set . count ( ) > 0 :
raise forms . ValidationError ( " The parent AS block is already references by following AS number objects: %s " % ( " , " . join ( map ( lambda _x : _x . handle , parent . asnumber_set . all ( ) ) ) , ) ) raise forms . ValidationError ( " The parent AS block is already references by following AS number objects: %s " % ( " , " . join ( map ( lambda _x : _x . handle , parent . asnumber_set . all ( ) ) ) , ) )
# check if same range # check if same range
if not ( asBegin > = parent . asBegin and asEnd < = parent . asEnd ) : if not ( asBegin > = parent . asBegin and asEnd < = parent . asEnd ) :
raise forms . ValidationError ( " AS beginning and end must be inside the range of the parent AS block " ) raise forms . ValidationError ( " AS beginning and end must be inside the range of the parent AS block " )
if parent . asBegin == asBegin and parent . asEnd == asEnd : if parent . asBegin == asBegin and parent . asEnd == asEnd :
raise forms . ValidationError ( " The range of this block cannot be the same range AS the parent AS block " ) raise forms . ValidationError ( " The range of this block cannot be the same range AS the parent AS block " )
# check for overlap with other asblocks # check for overlap with other asblocks
for block in parent . asblock_set . all ( ) : for block in parent . asblock_set . all ( ) :
if self . instance and self . instance . pk and block . pk == self . instance . pk : if self . instance and self . instance . pk and block . pk == self . instance . pk :
continue continue
if block . asBegin < = asBegin < = block . asEnd or block . asBegin < = asEnd < = block . asEnd or \ if block . asBegin < = asBegin < = block . asEnd or block . asBegin < = asEnd < = block . asEnd or \
asBegin < = block . asBegin < = asEnd or asBegin < = block . asEnd < = asEnd : asBegin < = block . asBegin < = asEnd or asBegin < = block . asEnd < = asEnd :
raise forms . ValidationError ( " Block overlaps with block %s " % block . handle ) raise forms . ValidationError ( " Block overlaps with block %s " % block . handle )
if not self . _create : if not self . _create :
# check if subblocks are still in range # check if subblocks are still in range
for subblock in self . instance . asblock_set . all ( ) : for subblock in self . instance . asblock_set . all ( ) :
if not ( asBegin < = subblock . asBegin < = asEnd and asBegin < = subblock . asEnd < = asEnd ) : if not ( asBegin < = subblock . asBegin < = asEnd and asBegin < = subblock . asEnd < = asEnd ) :
raise forms . ValidationError ( " Subblock %s ( %s - %s ) is not contained in this block anymore " % ( subblock , subblock . asBegin , subblock . asEnd ) ) raise forms . ValidationError ( " Subblock %s ( %s - %s ) is not contained in this block anymore " % ( subblock , subblock . asBegin , subblock . asEnd ) )
# check if asnumbers are still in range # check if asnumbers are still in range
for asnumber in self . instance . asnumber_set . all ( ) : for asnumber in self . instance . asnumber_set . all ( ) :
if not ( asBegin < = asnumber . number < = asEnd ) : if not ( asBegin < = asnumber . number < = asEnd ) :
raise forms . ValidationError ( " AS %s ( %s ) is not contained in this block anymore " % ( asnumber , asnumber . number ) ) raise forms . ValidationError ( " AS %s ( %s ) is not contained in this block anymore " % ( asnumber , asnumber . number ) )
return cleaned_data return cleaned_data
class ASNumberForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) : class ASNumberForm ( MntFormMixin , WhoisObjectFormMixin , forms . ModelForm ) :
protectedFields = [ ' asblock ' , ' number ' , ' mnt_by ' ] protectedFields = [ ' asblock ' , ' number ' , ' mnt_by ' ]
class Meta : class Meta :
model = ASNumber model = ASNumber
fields = [ ' handle ' , ' asblock ' , ' number ' , ' volatile ' , ' name ' , ' description ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ] fields = [ ' handle ' , ' asblock ' , ' number ' , ' volatile ' , ' name ' , ' description ' , ' mnt_by ' , ' mnt_lower ' , ' admin_c ' ]
def __init__ ( self , * args , * * kwargs ) : def __init__ ( self , * args , * * kwargs ) :
super ( ASNumberForm , self ) . __init__ ( * args , * * kwargs ) super ( ASNumberForm , self ) . __init__ ( * args , * * kwargs )
if not ( self . instance and self . instance . pk ) : if not ( self . instance and self . instance . pk ) :
self . fields [ " asblock " ] . required = True self . fields [ " asblock " ] . required = True
else : else :
self . fields [ " asblock " ] . disabled = True self . fields [ " asblock " ] . disabled = True
mnts = self . _user . maintainer_set . all ( ) mnts = self . _user . maintainer_set . all ( )
self . fields [ ' asblock ' ] . queryset = ASBlock . getMntQueryset ( mnts , self . instance , " asblock " ) self . fields [ ' asblock ' ] . queryset = ASBlock . getMntQueryset ( mnts , self . instance , " asblock " )
def clean ( self ) : def clean ( self ) :
cleaned_data = super ( ASNumberForm , self ) . clean ( ) cleaned_data = super ( ASNumberForm , self ) . clean ( )
if not self . errors : if not self . errors :
number = cleaned_data [ ' number ' ] number = cleaned_data [ ' number ' ]
block = cleaned_data [ ' asblock ' ] block = cleaned_data [ ' asblock ' ]
# belongs to asblock? # belongs to asblock?
if number < block . asBegin or number > block . asEnd : if number < block . asBegin or number > block . asEnd :
raise forms . ValidationError ( " AS number is not inside AS block " ) raise forms . ValidationError ( " AS number is not inside AS block " )
# does an entry already exist? # does an entry already exist?
try : try :
otherAS = ASNumber . objects . get ( number = number ) otherAS = ASNumber . objects . get ( number = number )
if not ( self . instance and self . instance . pk and self . instance . pk == otherAS . pk ) : if not ( self . instance and self . instance . pk and self . instance . pk == otherAS . pk ) :
raise forms . ValidationError ( " This AS number is already represented by %s " % otherAS . handle ) raise forms . ValidationError ( " This AS number is already represented by %s " % otherAS . handle )
except ASNumber . DoesNotExist : except ASNumber . DoesNotExist :
pass pass
# has already other asblock? # has already other asblock?
if block . asblock_set . count ( ) > 0 : if block . asblock_set . count ( ) > 0 :
raise forms . ValidationError ( " The given AS block is already references by following sub AS blocks: %s " % ( " , " . join ( map ( lambda _x : _x . handle , block . asblock_set . all ( ) ) ) , ) ) raise forms . ValidationError ( " The given AS block is already references by following sub AS blocks: %s " % ( " , " . join ( map ( lambda _x : _x . handle , block . asblock_set . all ( ) ) ) , ) )
