a I_0!@sddlZddlmZmZmZddlmZGdddeZGdddeZGd d d eZ Gd d d eZ Gd ddeZ ddZ ddZ ddZddZdS)N) FLAG_MOD_ADDFLAG_MOD_REPLACELdbError)SiteNotFoundExceptionc@seZdZdZdS)SubnetExceptionzBase element for Subnet errorsN__name__ __module__ __qualname____doc__r r //usr/lib/python3/dist-packages/samba/subnets.pyrsrc@seZdZdZdS)SubnetNotFoundz$The subnet requested does not exist.Nrr r r rr!src@seZdZdZdS)SubnetAlreadyExistsz&The subnet being added already exists.Nrr r r rr&src@seZdZdZdS) SubnetInvalidzThe subnet CIDR is invalid.Nrr r r rr+src@seZdZdZdS) SiteNotFoundz2The site to be used for the subnet does not exist.Nrr r r rr0src CsP|j|tjdt|d}t|dkr4td||dj}t|tsTt d|t |d}| |dkrvt d | d dkrt d |dd |zDt}||_td td|d<tt|td|d<||WnltjyJ}zP|j\} } | tjkrt d|| fn| tjkr4td|nWYd}~n d}~00dS)ajCreate a subnet and associate it with a site. :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param subnet_name: name of the subnet to create (a CIDR range) :return: None :raise SubnetAlreadyExists: if the subnet to be created already exists. :raise SiteNotFound: if the site does not exist. z(&(objectclass=Site)(cn=%s))baseZscopeZ expressionrz&A site with the name %s does not existrz'%s is not a valid subnet (not a string)CN=Subnets,CN=SitesFdnsubnet.add_base() failedCN=Xdnsubnet.add_child() failedCNZsubnetZ objectclass siteObject%s is not a valid subnet: %s(A subnet with the CIDR %s already existsN)searchldbZ SCOPE_SUBTREEZ binary_encodelenrdn isinstancestrrDnadd_baser add_child set_componentMessageMessageElementraddrargsERR_INVALID_DN_SYNTAXERR_ENTRY_ALREADY_EXISTSr) samdbconfigDn subnet_name site_nameretZdn_sitednsubnetmeenumestrr r r create_subnet5sJ          r7c Cst|d}||dkr"td|ddkr8td|dd|z.|j|tjdd }t|d krrt d |WnDt y}z,|j \}}|tj krt d |WYd }~n d }~00| |d S) aDelete a subnet. :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param subnet_name: Name of the subnet to delete :return: None :raise SubnetNotFound: if the subnet to be deleted does not exist. rFrrrrrobjectClass=subnetrrSubnet %s does not existN)rr#r$rr%r&r SCOPE_BASErrrr*ERR_NO_SUCH_OBJECTdelete)r-r.r/r2r1Ze1r5r6r r r delete_subnetis"     "r=c Cst|d}||dkr"td|ddkr8td|dd|t|t|}|dd|z|||Wn|ty}zd|j \}}|tj krt d|n6|tj krt d |n|tjkrtd ||fnWYd }~n d }~00d S) avRename a subnet. :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param subnet_name: Name of the subnet to rename :param new_name: New name for the subnet :return: None :raise SubnetNotFound: if the subnet to be renamed does not exist. :raise SubnetExists: if the subnet to be created already exists. rFrrrrrr9rrN)rr#r$rr%r&r"renamerr*r;rr,rr+r) r-r.r/new_namer2Z newdnsubnetZe2r5r6r r r rename_subnets.      r@c Cst|d}||dkr"td|ddkr8td|dd|z.|j|tjdd }t|d krrt d |WnDt y}z,|j \}}|tj krt d |WYd }~n d }~00t|d } | |dkrtd| ddkrtd| dd|t|d} | | z0|j| tjdd }t|d krDt d|WnHt y} z.| j \}}|tj krzt d|WYd } ~ n d } ~ 00t|dj} t} || _t| td| d<|| d S)aAssign a subnet to a site. This dissociates the subnet from its previous site. :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param subnet_name: Name of the subnet :param site_name: Name of the site :return: None :raise SubnetNotFound: if the subnet does not exist. :raise SiteNotFound: if the site does not exist. rFrrrrrr8rrr9NzCN=Siteszdnsites.add_base() failedzdnsites.add_child() failedz CN=ServerszobjectClass=sitezSite %s does not existr)rr#r$rr%r&rr:rrrr*r;rr"r r'r(rZmodify)r-r.r/r0r2r1Ze3r5r6ZdnsiteZ dnserversZe4ZsiteDnr3r r rset_subnet_sitesR     "      "rA)rrrrZsitesr Exceptionrrrrrr7r=r@rAr r r rs 4$