import active_directory
import menu
import win32com.client
import os

# name
# 2/12/11

print "1. Create password"
print "2. AD script 1 - print user list"
print "3. AD script 2 - print computers"
print "4. WMI Script 2: print user acc info"
print "5. WMI Script 2"
print "6. Print specific info about os"

selection = raw_input("Please enter selection number ") #input which selection want

if selection == "1": # create password
    username = raw_input("Please enter username ")
    password = raw_input("Please enter password ")
    print "Your username is ",username
    print "Your password is ",password

    print "Now writing to file"

    html_file=open("logfile.htm","w") #open html file for writing data to
    html_file.write ("Username: ")
    html_file.write (username)
    html_file.write ("\n")
    html_file.write ("Password: ")
    html_file.write (password)   #put data into html format
    html_file.close()               #close logfile
    os.startfile('logfile.htm')     #display  logfile in browser


elif selection == "2": # ad script1 print user list
    for xUser in active_directory.search(objectclass = "User"):
        print xUser.cn

elif selection == "3":# ad script2 print computers
    for Computer in active_directory.search(objectclass = "Computer"):
        print Computer.cn

elif selection == "4": # wmi script 2: print user acc info
    #locate WMI service &connect to namespace where objects stored
    strComputer = "."
    objWMIService = win32com.client.Dispatch("WbemScripting.SWbemLocator")
    objSWbemServices = objWMIService.ConnectServer(strComputer,"root\cimv2")

    #execute query & print results
    colItems = objSWbemServices.ExecQuery("Select * from Win32_UserAccount")
    for objItem in colItems:

        print "Caption: ", objItem.Caption
        print "Domain: domainofzeus", objItem.Domain

#elif selection == "5": # wmi script 2
#    print "5"

#else:
#    selection == 6 #print specific info about os
#    print "no"