#!/usr/bin/evn python

import os
import sys
import string
import tempfile

from time import sleep

_hostname = None
_username = None
_password = None
_ldapDataFile = '/tmp/kabc_ldap_record.txt'
NumberOfThreads = 5

if _hostname is None or _username is None or _password is None or _ldapDataFile is None:
    sys.stderr.write("Please first set the credentials properly.\n")
    sys.exit(1)

_ldapPreCommand = 'ldapsearch -LLL -z 0 "(Sn='
_ldapPostCommand = '*)" -h ' + _hostname + ' -x -D "'  + _username + '" -b Cn=users,DC=hq,DC=domain,DC=com' + " -w " + _password

#char = 's'
#ldapCommand = _ldapPreCommand + char + _ldapPostCommand
#x = os.system(ldapCommand)
#if x == 1024:
#    print "I'm in the exception"
#print x
#sys.exit(1)

ldap_attributes = ['dn:', 'cn:', 'sn:', 'l:', 'st:', 'title:', 'description:', 'postalCode:', 'telephoneNumber:', 'facsimileTelephoneNumber:',
                   'givenName:', 'mail:', 'homePhone:', 'mobile:', 'pager:']

try:
    writeFile = open(_ldapDataFile, 'w')
except IOError:
    sys.stderr.write("Couldn't open file %s to write.\n" % (writeFile) )
    sys.exit(1)
    
(temp_file_fd, temp_file_name) = tempfile.mkstemp()
os.environ['__kabc_ldap'] = temp_file_name
print "Writing data to %s\n" % (temp_file_name)

def RecordFetchher(char):
        sleep(5) #Let's not thrash the exchange server ;-)
        ldapCommand = _ldapPreCommand + char + _ldapPostCommand
        if os.system(ldapCommand + "> $__kabc_ldap") != 0:
            sys.stderr.write("Couldn't execute the command %s\n" % (ldapCommand) )
            sys.exit(1)
        #temp_file_name = "/tmp/tmpnhYrps"
        readFile = open(temp_file_name, 'r')
        for record in readFile.readlines():
            if record.startswith(' '): # Remove the junk
                pass
            record = string.rstrip(record, "\n")
            for attrib in ldap_attributes:
                if record.startswith(attrib):
                    if ldap_attributes[0] == attrib: #This attribute is common/mandatory in all records
                        writeFile.write("\n")
                    writeFile.write(record)
                    writeFile.write("\n")
                    break
            writeFile.flush()
        readFile.close()
        writeFile.write("\n")
    
for char in string.lowercase:
    # I know this is ugly. Too many cycles
    # But ldapsearch or exchange is restricting, the query max result limit is 1000
    for subChar in string.lowercase:
        RecordFetchher(char+subChar)

print "Data writted to file %s\n" % (temp_file_name)
writeFile.close()