a I_@spdZddlZddlmZmZGdddeZGdddeZGdd d eZGd d d eZd d Z ddZ dS)zManipulating sites.N) FLAG_MOD_ADDLdbErrorc@s eZdZdZddZddZdS) SiteExceptionzBase element for Sites errorscCs ||_dS)N)value)selfrr-/usr/lib/python3/dist-packages/samba/sites.py__init__szSiteException.__init__cCsd|jj|jfS)Nz%s: %s) __class____name__r)rrrr__str__szSiteException.__str__N)r __module__ __qualname____doc__r r rrrrrsrc@seZdZdZdS)SiteNotFoundExceptionz>Raised when the site is not found and it's expected to exists.Nr r rrrrrrr"src@seZdZdZdS)SiteAlreadyExistsExceptionzBRaised when the site is not found and it's expected not to exists.Nrrrrrr&src@seZdZdZdS)SiteServerNotEmptyExceptionz0Raised when the site still has servers attached.Nrrrrrr*srcCs|j|tjd|d}t|dkr.td|t}t|d|t|f|_t dt d|d<| |t}t|dt|j|_t d t d|d<| |t}t|d t|j|_t d t d|d<| |d S) a Create a site :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param siteName: Name of the site to create :return: True upon success :raise SiteAlreadyExists: if the site to be created already exists. z(&(objectclass=Site)(cn=%s))baseZscopeZ expressionrz&A site with the name %s already existszCn=%s,CN=Sites,%sZsiteZ objectclasszCn=NTDS Site Settings,%sZnTDSSiteSettingsz Cn=Servers,%sZserversContainerT) searchldbZ SCOPE_SUBTREElenrZMessageDnstrZdnZMessageElementradd)samdbconfigDnsiteNameretmZm2Zm3rrr create_site.s$      r!c Cs t|d}||dkr"td|ddkr8td|dd|t|d}||z.|j|tjd d }t|d krt d |WnDt y}z,|j \}}|tj krt d |WYd }~n d }~00|j|tj dd }t|dkrtd|||dgdS)aV Delete a site :param samdb: A samdb connection :param configDn: The DN of the configuration partition :param siteName: Name of the site to delete :return: True upon success :raise SiteNotFoundException: if the site to be deleted do not exists. :raise SiteServerNotEmpty: if the site has still servers in it. zCN=SitesFzdnsites.add_base() failedzCN=Xzdnsites.add_child() failedrZCNz CN=ServerszobjectClass=siterzSite %s does not existNz(objectclass=server)z9Site %s still has servers in it, move them before removalz tree_delete:0T)rrZadd_baserZ add_childZ set_componentrZ SCOPE_BASErrrargsZERR_NO_SUCH_OBJECTZSCOPE_ONELEVELrdelete) rrrZdnsiteZ dnserversreenumZestrrrr delete_siteSs2       "   r') rrrr Exceptionrrrrr!r'rrrrs %