DFS_shares_for_NoMAD.py

#!/usr/bin/python
# Version 0.1
# This was (re)written quick and dirty. And by dirty I mean embarrasing.
import subprocess
import tempfile
import os
import re

dfsroot="//example.com/dfs"
dfsrootserveraddress='dfs01.example.com'
domain='example.com'




myshares = []

tempmountpoint = tempfile.mkdtemp(".nomaddfs")
dfsroot = dfsroot.replace('/'+domain+'/','/'+dfsrootserveraddress+'/')
subprocess.check_output(["mount","-a","-o","nobrowse,rdonly","-t","smbfs",dfsroot,tempmountpoint])
dfsanchors = os.listdir(tempmountpoint)
#print dfsanchors
for dfsanchor in dfsanchors:
	anchorinfo = subprocess.check_output(["smbutil","dfs",("%s/%s"%(dfsroot,dfsanchor))])
	for line in anchorinfo.split('\n'):
		#print line
		anchorurl = re.search('Network Address: (.+)',anchorinfo)
		if anchorurl:
			myshares.append([dfsanchor,"smb:/%s"%anchorurl.group(1),("smb:%s/%s"%(dfsroot,dfsanchor))])
			break
#print subprocess.check_output("df")
subprocess.call(["umount",tempmountpoint])
os.rmdir(tempmountpoint)
pliststrings = []
pliststrings.append('<?xml version="1.0" encoding="UTF-8"?>\n<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">\n<plist version="1.0">')
pliststrings.append( """<dict>
     <key>HomeMount</key>
     <dict>
          <key>Groups</key>
          <array>
               <string>All</string>
          </array>
          <key>Mount</key>
          <false/>
          <key>Options</key>
          <array/>
     </dict>""")
pliststrings.append( '<key>Shares</key><array>' )
for myshare in myshares:
	#print "<dict>\n\t<key>%s</key>\n\t<string>%s</string>\n\t<key>%s</key>\n\t<string>%s</string>\n\t<key>%s</key>\n\t<string>%s</string>\n</dict>"%("Name",myshare[0],"ShareURL",myshare[1],"MountURL",myshare[2])
	pliststrings.append( """<dict>
               <key>AutoMount</key>
               <false/>
               <key>ConnectedOnly</key>
               <true/>
               <key>Groups</key>
               <array/>
               <key>LocalMount</key>
               <string></string>
               <key>Name</key>
               <string>%s</string>
               <key>Options</key>
               <array/>
               <key>URL</key>
               <string>%s</string>
          </dict>"""%(myshare[0],myshare[2]))
pliststrings.append( '</array>' )
pliststrings.append( '<key>Version</key><string>1</string></dict>' )
pliststrings.append( '</plist>' )

with open(os.path.expanduser('~/Library/Preferences/menu.nomad.shares.plist'),'w') as outfile:
	outfile.write("\n".join(pliststrings))
print "\n".join(pliststrings)