add distribution script and launchd pkg postinstall

code/pkgtemplate/Scripts_distribution/launchctl.py

@@ -0,0 +1,25 @@
+#!/usr/bin/python
+"""
+Postinstall script to load munki's launchdaemons.
+"""
+
+
+def getconsoleuser():
+    '''Uses Apple's SystemConfiguration framework to get the current
+    console user'''
+    from SystemConfiguration import SCDynamicStoreCopyConsoleUser
+    cfuser = SCDynamicStoreCopyConsoleUser(None, None, None)
+    return cfuser[0]
+
+
+def main():
+    # This returns the conditions on whether or not a restart is required
+    # for the launchd pkg.
+    consoleuser = getconsoleuser()
+    if consoleuser is None or consoleuser == u"loginwindow" or consoleuser == u"_mbsetupuser":
+        exit(0)
+    else:
+        exit(1)
+
+if __name__ == '__main__':
+    main()

code/pkgtemplate/Scripts_launchd/postinstall

@@ -0,0 +1,41 @@
+#!/usr/bin/python
+"""
+Postinstall script to load munki's launchdaemons.
+"""
+import os
+import subprocess
+
+
+def getconsoleuser():
+    '''Uses Apple's SystemConfiguration framework to get the current
+    console user'''
+    from SystemConfiguration import SCDynamicStoreCopyConsoleUser
+    cfuser = SCDynamicStoreCopyConsoleUser(None, None, None)
+    return cfuser[0]
+
+
+def launchctld(identifier):
+    path = os.path.join('/Library/LaunchDaemons/', identifier, '.plist')
+    try:
+        cmd = ['/bin/launchctl', 'load', path]
+        proc = subprocess.Popen(cmd, stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE)
+        output, err = proc.communicate()
+        return output
+    except error:
+        pass
+
+
+def main():
+    # Only load the launch daemons if there isn't a console user or at the
+    # loginwindow. launchctl.py will take care of the restart via the pkg
+    # distribution.xml
+    consoleuser = getconsoleuser()
+    if consoleuser is None or consoleuser == u"loginwindow" or consoleuser == u"_mbsetupuser":
+        launchctld('com.googlecode.munki.managedsoftwareupdate-check')
+        launchctld('com.googlecode.munki.managedsoftwareupdate-install')
+        launchctld('com.googlecode.munki.managedsoftwareupdate-check')
+        launchctld('com.googlecode.munki.logouthelper')
+
+if __name__ == '__main__':
+    main()