
    >Tf}$                        d dl mZmZ d dlmZmZ d dlmZ d dlm	Z	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lT d d
lT d dlmZmZ d dlmZmZ d dlmZm Z  d dl!m"Z" dZ#d Z$ G d de          Z% G d de          Z&dS )    )GenericForeignKeyGenericRelation)
ArrayFieldIntegerRangeField)ValidationError)MaxValueValidatorMinValueValidator)models)NumericRange)reverse)gettext_lazy)	Interface)*)VLANQuerySetVLANGroupQuerySet)OrganizationalModelPrimaryModel)check_ranges_overlapranges_to_string)VMInterface)VLAN	VLANGroupc                  <    t          t          t          d          gS )Nz[])bounds)r   VLAN_VID_MINVLAN_VID_MAX     6/var/www/html/netbox-4.1.3/netbox/ipam/models/vlans.pydefault_vid_rangesr       s    \<=== r   c                       e Zd ZdZ ej         ed          d          Z ej         ed          d          Z	 ej
        dej         ee          dd	          Z ej        dd
          Z edd          Z e e             ed          e          Z ej        eez
  dz             Z ej                    Z G d d          Zd Z fdZ fdZ d Z!d Z"d Z#e$d             Z% xZ&S )r   z
    A VLAN group is an arbitrary collection of VLANs within which VLAN IDs and names must be unique. Each group must
     define one or more ranges of valid VLAN IDs, and may be assigned a specific scope.
    named   verbose_name
max_lengthslugzcontenttypes.ContentType)	model__inT)to	on_deletelimit_choices_toblanknull)r,   r-   
scope_typescope_id)ct_fieldfk_fieldzVLAN ID ranges)r%   default   )r2   c                       e Zd ZdZ ej        d          fZ ej        dd           ej        dd          fZ e	d	          Z
 e	d
          ZdS )VLANGroup.Meta)r"   pk)r.   r/   )fields)r.   r/   r"   z)%(app_label)s_%(class)s_unique_scope_namer7   r"   )r.   r/   r'   z)%(app_label)s_%(class)s_unique_scope_slugz
VLAN groupzVLAN groupsN)__name__
__module____qualname__orderingr
   IndexindexesUniqueConstraintconstraints_r%   verbose_name_pluralr   r   r   Metar5   E   s        !FL :;;;
 $F#9@   $F#9@  	
 qa..r   rC   c                 0    t          d| j        g          S )Nzipam:vlangroupargsr   r6   selfs    r   get_absolute_urlzVLANGroup.get_absolute_urlW   s    'twi8888r   c                    t                                                       | j        r#| j        st	          t          d                    | j        r#| j        st	          t          d                    | j        r2t          | j                  rt	          dt          d          i          | j        D ]D}|j        |j	        k    r2t	          dt          d          
                    |          i          Ed S )Nz'Cannot set scope_type without scope_id.z'Cannot set scope_id without scope_type.
vid_rangeszRanges cannot overlap.zNMaximum child VID must be greater than or equal to minimum child VID ({value}))value)supercleanr.   r/   r   rA   rL   r   lowerupperformat)rI   	vid_range	__class__s     r   rO   zVLANGroup.cleanZ   s    ? 	P4= 	P!!$M"N"NOOO= 	P 	P!!$M"N"NOOO ? 	O3DODD 	O!<3K1L1L"MNNN 	 	I00% !h# #f9f--'    1	 	r   c                     d| _         | j        D ]"}| xj         |j        |j        z
  dz   z  c_         # t	                      j        |i | d S )Nr   r3   )_total_vlan_idsrL   rQ   rP   rN   save)rI   rF   kwargsrS   rT   s       r   rW   zVLANGroup.saven   sd      	J 	JI  IOio$E$II   d%f%%%%%r   c                 H   t                      }| j        D ]9}|                    d t          |j        |j                  D                       }:|t          t          j                            |           	                    dd                    z  }t          |          S )z?
        Return all available VLANs within this group.
        c                     h | ]}|S r   r   ).0vids     r   	<setcomp>z/VLANGroup.get_available_vids.<locals>.<setcomp>{   s%     5 5 55 5 5r   groupr\   T)flat)setrL   unionrangerP   rQ   r   objectsfiltervalues_listsorted)rI   available_vlans
vlan_ranges      r   get_available_vidszVLANGroup.get_available_vidsu   s     %%/ 	 	J-33 5 5$Z%5z7GHH5 5 5  OO 	3t|222>>JJ5W[J\\]]]o&&&r   c                 B    |                                  }|r|d         S dS )zK
        Return the first available VLAN ID (1-4094) in the group.
        r   N)rj   )rI   available_vidss     r   get_next_available_vidz VLANGroup.get_next_available_vid   s-     0022 	%!!$$tr   c                 h    t           j                            |                               d          S )z5
        Return all VLANs within this group.
        r^   r\   )r   rd   re   order_byrH   s    r   get_child_vlanszVLANGroup.get_child_vlans   s+     |"""..77>>>r   c                 *    t          | j                  S N)r   rL   rH   s    r   vid_ranges_listzVLANGroup.vid_ranges_list   s    000r   )'r9   r:   r;   __doc__r
   	CharFieldrA   r"   	SlugFieldr'   
ForeignKeyCASCADEQVLANGROUP_SCOPE_TYPESr.   PositiveBigIntegerFieldr/   r   scoper   r   r    rL   r   r   rV   r   
as_managerrd   rC   rJ   rO   rW   rj   rm   rp   propertyrs   __classcell__rT   s   @r   r   r      s         6QvYY  D 6QvYY  D #"%.%:;;;  J .v-  H   E Q'(("  J
 5f4|+a/  O +*,,G/ / / / / / / /$9 9 9    (& & & & &' ' '  ? ? ? 1 1 X1 1 1 1 1r   r   c            
           e Zd ZdZ ej        dej        ddd ed                    Z ej        dej        ddd ed                    Z	 ej
         ed	           ee           ee          f ed
                    Z ej         ed          d          Z ej        dej        ddd          Z ej         ed          deej         ed                    Z ej        dej        ddd ed                    Z edddd          Z ej                    Zg dZ G d d          Zd Zd  Z  fd!Z!d" Z"d# Z#d$ Z$e%d%             Z& xZ'S )&r   a  
    A VLAN is a distinct layer two forwarding domain identified by a 12-bit integer (1-4094). Each VLAN must be assigned
    to a Site, however VLAN IDs need not be unique within a Site. A VLAN may optionally be assigned to a VLANGroup,
    within which all VLAN IDs and names but be unique.

    Like Prefixes, each VLAN is assigned an operational status and optionally a user-defined Role. A VLAN can have zero
    or more Prefixes assigned to it.
    z	dcim.SitevlansTz9The specific site to which this VLAN is assigned (if any))r)   r*   related_namer,   r-   	help_textzipam.VLANGroupzVLAN group (optional)zVLAN IDzNumeric VLAN ID (1-4094))r%   
validatorsr   r"   @   r$   ztenancy.Tenant)r)   r*   r   r,   r-   status2   zOperational status of this VLAN)r%   r&   choicesr2   r   z	ipam.Rolez!The primary function of this VLANzvpn.L2VPNTerminationassigned_object_typeassigned_object_idvlan)r)   content_type_fieldobject_id_fieldrelated_query_name)siter_   tenantr   roledescriptionc                       e Zd ZdZ ej        dd           ej        dd          fZ ed          Z ed          Z	d	S )
	VLAN.Meta)r   r_   r\   r6   )r_   r\   z(%(app_label)s_%(class)s_unique_group_vidr8   )r_   r"   z)%(app_label)s_%(class)s_unique_group_namer   VLANsN)
r9   r:   r;   r<   r
   r?   r@   rA   r%   rB   r   r   r   rC   r      su        1#F#'?   $F#(@  	
 qyyajjr   rC   c                 &    | j          d| j         dS )Nz ())r"   r\   rH   s    r   __str__zVLAN.__str__   s    )**tx****r   c                 0    t          d| j        g          S )Nz	ipam:vlanrE   rG   rH   s    r   rJ   zVLAN.get_absolute_url   s    {$'3333r   c                     t                                                        j        rb j        r[ j        j         j        k    rFt          t          d                               j         j        j         j                             j        rtt           fd j        j	        D                       sQt          dt          d                              t           j        j	                   j                  i          d S d S )NzVVLAN is assigned to group {group} (scope: {scope}); cannot also assign to site {site}.)r_   r|   r   c                 "    g | ]}j         |v S r   )r\   )r[   rrI   s     r   
<listcomp>zVLAN.clean.<locals>.<listcomp>   s    EEE!AEEEr   r\   z9VID must be in ranges {ranges} for VLANs in group {group})rangesr_   )rN   rO   r_   r   r|   r   rA   rR   anyrL   r   )rI   rT   s   `r   rO   z
VLAN.clean   s    : 	$) 	
(8DI(E(E!l &tz1A	&RR   : 	EEEEtz/DEEEFF %1S f$4TZ5J$K$KSWS]f^^'   	 	 r   c                 J    t           j                            | j                  S rr   )VLANStatusChoicescolorsgetr   rH   s    r   get_status_colorzVLAN.get_status_color  s     '++DK888r   c                     t           j                            t          | j                  t          | j                  z                                            S N)untagged_vlan_id)tagged_vlans)r   rd   re   ry   r6   distinctrH   s    r   get_interfaceszVLAN.get_interfaces
  sM     ''tw'''47###$
 
 (**	r   c                     t           j                            t          | j                  t          | j                  z                                            S r   )r   rd   re   ry   r6   r   rH   s    r   get_vminterfaceszVLAN.get_vminterfaces  sM    "))tw'''47###$
 
 (**	r   c                 4    | j                                         S rr   )l2vpn_terminationsfirstrH   s    r   l2vpn_terminationzVLAN.l2vpn_termination  s    &,,...r   )(r9   r:   r;   rt   r
   rw   PROTECTrA   r   r_   PositiveSmallIntegerFieldr	   r   r   r   r\   ru   r"   r   r   STATUS_ACTIVEr   SET_NULLr   r   r   r   r}   rd   clone_fieldsrC   r   rJ   rO   r   r   r   r~   r   r   r   s   @r   r   r      s         6.!OPP  D F.!+,,  E +&
*Qy\\l++l++
 !.//  C 6QvYY  D V.  F VQx[[!!/!566  F 6/!788  D )!1,!	   &l%''G  L) ) ) ) ) ) ) )+ + +4 4 4    (9 9 9     / / X/ / / / /r   r   N)'"django.contrib.contenttypes.fieldsr   r   django.contrib.postgres.fieldsr   r   django.core.exceptionsr   django.core.validatorsr   r	   	django.dbr
   )django.db.backends.postgresql.psycopg_anyr   django.urlsr   django.utils.translationr   rA   dcim.modelsr   ipam.choicesipam.constantsipam.querysetsr   r   netbox.modelsr   r   utilities.datar   r   virtualization.modelsr   __all__r    r   r   r   r   r   <module>r      s   Q Q Q Q Q Q Q Q H H H H H H H H 2 2 2 2 2 2 G G G G G G G G       B B B B B B       6 6 6 6 6 6 ! ! ! ! ! !         : : : : : : : : ; ; ; ; ; ; ; ; A A A A A A A A - - - - - -  u1 u1 u1 u1 u1# u1 u1 u1pD/ D/ D/ D/ D/< D/ D/ D/ D/ D/r   