from javax.naming import *
from java.util import *
from javax.naming.directory import *
class ADS:
    '''
    The parameters required for making the connections.
    the user name should be a user with previliges to log into the
    LDAP machine.
    '''
    ads_server="<adsserver>"
    ads_user="<valid_username>"
    ads_password="<valid_password>"
    ads_base_dns="DC=something,DC=com"
    '''
        The constructor: This intialises the ads object 
    '''
    def __init__(self):
        self.url="ldap://%s/%s" % (self.ads_server,self.ads_base_dns)
        env=Hashtable()
        env.put(Context.INITIAL_CONTEXT_FACTORY, "com.sun.jndi.ldap.LdapCtxFactory")
        env.put(Context.PROVIDER_URL, self.url)
        env.put(Context.SECURITY_AUTHENTICATION, "simple")
        env.put(Context.SECURITY_PRINCIPAL, self.ads_user)
        env.put(Context.SECURITY_CREDENTIALS, self.ads_password)
        ctx =InitialDirContext(env)
        self.ctx=ctx
    '''
        The string method is overriden to print the url used to connect to the ads server.
    '''
    def __str__(self):
        return self.url
    '''
    Method for adding an attribute with a specific value to ADS
    '''
    def modify_attribute(self,name,value,username=None):
        myAttrs = BasicAttributes(1)
        oc = BasicAttribute(name)
        oc.add(value)
        myAttrs.put(oc)
        if username!=None:
            self.ctx.modifyAttributes("cn=%s,ou=portal,ou=admins" % username, DirContext.REPLACE_ATTRIBUTE, myAttrs)
        else:
            results=self.search_user("*")
            for result in results:
                attrs=result.getAttributes()
                try:
                    self.ctx.modifyAttributes("cn=%s,ou=portal,ou=admins" % attrs.get("name").get(), DirContext.REPLACE_ATTRIBUTE, myAttrs)
                except:
                    pass
    '''
    Method for searching a user by it's name for wild card search use *
    '''
    def search_user(self,username):
        return self.search(username,"user")
    '''
    Method for searching a group by it's name for wild card search user *
    '''
    def search_group(self,groupname):
        return self.search(groupname,"group")
    '''
    The main search class
    '''
    def search(self,criteria,objectclass):
        srch =SearchControls()
        srch.setSearchScope(SearchControls.SUBTREE_SCOPE)
        results = self.ctx.search("", "(&(CN=%s) (objectClass=%s))" % (criteria,objectclass), srch)
        return results
        
'''
The class for handling the output results of the ads search
'''
class Output:
    '''
    Method for printing all the attributes present in the search result
    of a user or group.
    '''
    def print_attributes(self,results):
        for result in results:
            attributes=result.getAttributes()
            
            for atr in attributes.getIDs():
                print str(atr)
    '''
    Method for printing the values of results in ADS.
    '''
    def print(self,results,*attributes):
        for result in results:
            ads_attributes=result.getAttributes()
            print self.attribute_string(ads_attributes,attributes)
    '''
    Method for writing the results of the the output to a file
    '''
    def log(self,results,*attributes):
        filename="output.ads"
        fd=open(filename,"w")
        for result in results:
            ads_attributes=result.getAttributes()        
            fd.write("%s\n" % self.attribute_string(ads_attributes,attributes))
        fd.flush()
        fd.close()
    '''
    Generic method for displaying the ADS attribute
    '''
    def attribute_string(self,ads_attributes,attributes):
        result=str(ads_attributes.get("name"))
        for attribute in attributes:
            result="%s,%s" % (result,ads_attributes.get(attribute))
        return result