# drivespace.py - M.Keranen (mksql@yahoo.com) [10/05/2004]
# -------------------------------------------------------------
# Reads a file containing a list of share names,
# and prints a report of the current free space on the volumes. 
# -------------------------------------------------------------

import getpass,os,string, sys
import win32file, win32net, win32wnet, win32netcon

cfgfile=sys.path[0]+'/drivespace2.cfg'
cfglines = open(cfgfile,'r').readlines()
cfglines.sort()

# Compile a list of domains in the config file
domains = []
for line in cfglines:
	target = string.split(string.strip(line),'\\')
	if len(target)>2 and target[0] not in domains: domains.append(target[0])	

# Create a dict of domain\user_id:password
domid ={}
#domid['(local)']=getpass.getuser(),getpass.getpass()
for domain in domains:
	string.strip(domain)
	uid = string.strip(raw_input('UserID for %s: ' % domain))
	if uid != '':
		pw = string.strip(getpass.getpass())
		domid[domain] = domain+'\\'+uid,pw
	else: domid[domain] = '',''
	print

lastserv = ""
xit = " "
while xit == " ":
	print "\n%s \t %s \t %s\t%s" % ("Server\share","%Free","Avail MB","Total MB")
	print   "%s \t %s \t %s\t%s" % ("------------","-----","--------","--------")
	for line in cfglines:
		if line[0]<>'#':
			line = string.strip(line)
			target = string.split(line,'\\')
	
			if len(target)==2:
				server = '\\\\'+target[0]
				share = '\\\\'+line
				uid = getpass.getuser()
				pw = None
			elif len(target)==3:
				domain = target[0]
				server = '\\\\'+target[1]
				share = '\\\\'+target[1]+'\\'+target[2]
				uid = domid[domain][0]
				pw = domid[domain][1]

			if server != lastserv and lastserv != "":
				try: win32wnet.WNetCancelConnection2(lastserv, 0, 0)
				except: warn = "!"
			if server != lastserv and uid != '':
				try: win32wnet.WNetAddConnection2(win32netcon.RESOURCETYPE_DISK, None, server, None, uid, pw, 0)
				except: print "%s\t  (WNetAddConnection2 failed [%s])" % (server,uid)
				lastserv = server

			fs,ts,fp = 0,0,0

			if uid != '':
				try: space = win32file.GetDiskFreeSpaceEx(share)
				except:
					warn = " "
					fp = "(GetDiskFreeSpaceEx failed)"
					fs,ts = '',''
				else:
					fs = int(space[0]/1048576)
					ts = int(space[1]/1048576)
					fp = int((float(space[0])/float(space[1]))*100)
					if fp<16: warn = "!"
					else: warn = " "

				print "%s \t %s%s \t %s\t\t%s" % (share[2:],warn,fp,fs,ts)
	
	try: win32wnet.WNetCancelConnection2(lastserv, 0, 0)
	except: warn = "!"
	xit = raw_input('\nPress Enter to exit (Space, Enter to rerun)...')

"""
Example of drivespace.cfg file:
-------------------------------
local1\c$
domain2\remote2\c$
"""