aboutsummaryrefslogtreecommitdiffstats
path: root/functions.py
diff options
context:
space:
mode:
authorFilipp Lepalaan <filipp@mac.com>2016-09-23 17:39:08 +0300
committerFilipp Lepalaan <filipp@mac.com>2016-09-23 17:39:08 +0300
commit712f5bb479b7183a49feb1ee6cf279d1dbd39f6b (patch)
treea571c468d90f6937c6840e40efc75e4ca58a616b /functions.py
parent932e24d6187daea8b63baec6a19d0319f9270980 (diff)
downloadmachammer-712f5bb479b7183a49feb1ee6cf279d1dbd39f6b.tar.gz
machammer-712f5bb479b7183a49feb1ee6cf279d1dbd39f6b.tar.bz2
machammer-712f5bb479b7183a49feb1ee6cf279d1dbd39f6b.zip
Return None as path to EULA-image volumes
Diffstat (limited to 'functions.py')
-rw-r--r--functions.py219
1 files changed, 0 insertions, 219 deletions
diff --git a/functions.py b/functions.py
deleted file mode 100644
index bf176ba..0000000
--- a/functions.py
+++ /dev/null
@@ -1,219 +0,0 @@
-# -*- coding: utf-8 -*-
-
-import os
-import plistlib
-import subprocess
-import sys
-import tempfile
-
-from system_profiler import SystemProfile
-
-
-SERVICEDIR = '/Library/Services'
-
-
-def call(**args):
- """Shorthand for subprocess.call.
-
- > call('ls', '/Users')
- """
- subprocess.call(args)
-
-
-def rmdir(path):
- """Shortcut for deleting a directory."""
- call('rm', '-r', path)
-
-
-def display_notification(msg, title='', subtitle=''):
- """Display notification with optional title and subtitle.
-
- display_notification('Mymessage')
- """
- msg = msg.replace('"', '\\"')
- title = title.replace('"', '\\"')
- subtitle = subtitle.replace('"', '\\"')
- osascript('display notification "{0}" with title "{1}" subtitle "{2}"'.format(msg, title, subtitle))
-
-
-def ditto(src, dst):
- subprocess.call(['/usr/bin/ditto', src, dst])
-
-
-def rsync(src, dst, flags='auE'):
- subprocess.call(['/usr/bin/rsync', '-' + flags, src, dst])
-
-
-def dscl(domain='.', *args):
- subprocess.call(['/usr/bin/dscl', domain] + args)
-
-
-def exec_jar(path, user):
- javapath = '/Library/Internet Plug-Ins/JavaAppletPlugin.plugin/Contents/Home/bin/java'
- if not os.path.exists(javapath):
- raise ValueError('Looks like your machine does not have Java installed')
-
- subprocess.call(['/bin/launchctl', 'asuser', user, javapath, '-jar', path, '-silent'])
-
-
-def osascript(s):
- subprocess.call(['/usr/bin/osascript', '-e', s])
-
-
-def tell_app(app, s):
- osascript('tell application "%s" to %s' % (app, s))
-
-
-def quit_app(app):
- tell_app(app, 'quit')
-
-
-def copy_app(path):
- """Copy path to /Applications folder."""
- rsync(path.rstrip('/'), '/Applications/')
-
-
-def add_login_item(path, name=None, hidden=True):
- """Add login item to the current user."""
- if not name:
- name = os.path.basename(path)
-
- hidden = 'true' if hidden else 'false'
- tell_app('System Events',
- 'make login item at end with properties {path:"%s", hidden:%s, name:"%s"}' % (path, hidden, name))
-
-
-def remove_login_item(**kwargs):
- """Remove login item from the current user."""
- if not (kwargs.get('name') or kwargs.get('path')):
- raise ValueError('Need either path or name')
-
- if kwargs.get('name') and kwargs.get('path'):
- raise ValueError('Please specify only path OR name')
-
- k, v = kwargs.items()[0]
- tell_app('System Events', 'delete every login item whose %s is "%s"' % (k, v))
-
-
-def create_user(username, realname, password):
- """Create a user account."""
- os.system("""dscl . create /Users/{0}
- dscl . create /Users/{0} RealName "{1}"
- dscl . passwd /Users/{0} {2}
- dscl . create /Users/{0} UniqueID 501
- dscl . create /Users/{0} PrimaryGroupID 80
- dscl . create /Users/{0} UserShell /bin/bash
- dscl . create /Users/{0} NFSHomeDirectory /Users/{0}
- cp -R /System/Library/User\ Template/English.lproj /Users/{0}
- chown -R {0}:staff /Users/{0}""".format(username, realname, password))
-
-
-def hide_user(username, hide_home=True):
- """Hide a user account."""
- path = '/Users/%s' % username
- subprocess.call(['dscl', '.', 'create', path, 'IsHidden', '1'])
-
- if hide_home:
- subprocess.call(['chflags', 'hidden', path])
-
-
-def delete_user(username, delete_home=True):
- """Delete a user account."""
- path = '/Users/' + username
- dscl = subprocess.check_output(['dscl', '-plist', '.', 'read', path])
- userinfo = plistlib.readPlistFromString(dscl)
-
- subprocess.call(['dscl', '.', 'delete', path])
-
- if delete_home:
- homedir = userinfo['dsAttrTypeStandard:NFSHomeDirectory'][0]
- os.rmdir(homedir)
-
-
-def make_admin(username):
- """Give admin rights to username."""
- subprocess.call(['dscl', '.', '-append', '/Groups/admin', 'users', username])
-
-
-def enable_ard(username):
- """Enable Apple Remote Desktop for username."""
- subprocess.call(['/System/Library/CoreServices/RemoteManagement/ARDAgent.app/Contents/Resources/kickstart',
- '-activate', '-configure',
- '-access', '-on',
- '-users', username,
- '-privs', '-all',
- '-restart', '-agent'])
-
-
-def is_laptop():
- profile = SystemProfile('Hardware')
- return 'Book' in profile.machine_model
-
-
-def is_desktop():
- return not is_laptop()
-
-
-def mount_image(dmg):
- """Mount disk image and return path to mountpoint."""
- r = subprocess.check_output(['/usr/bin/hdiutil', 'mount', '-plist', '-nobrowse', dmg])
- plist = plistlib.readPlistFromString(r)
- for p in [p.get('mount-point') for p in plist.get('system-entities')]:
- if p and os.path.exists(p):
- return p
-
- raise ValueError('Failed to mount %s' % dmg)
-
-
-def mount_and_install(dmg, pkg):
- """Mountsthe DMG and installs the PKG."""
- p = mount_image(dmg)
- install_pkg(os.path.join(p, pkg))
-
-
-def install_profile(path):
- """Install a configuration profile."""
- subprocess.call(['/usr/bin/profiles', '-I', '-F', path])
-
-
-def install_pkg(pkg, target='/'):
- """Install a package."""
- subprocess.call(['/usr/sbin/installer', '-pkg', pkg, '-target', target])
-
-
-def mount_afp(url, username, password, mountpoint=None):
- """Mount AFP share."""
- if mountpoint is None:
- mountpoint = tempfile.mkdtemp()
- subprocess.call(['/sbin/mount_afp', 'afp://%s:%s@%s' % (username, password, url), mountpoint])
- return mountpoint
-
-
-def umount(path):
- """Unmount path."""
- subprocess.call(['/sbin/umount', path])
-
-
-def install_su(restart=True):
- """Install all available Apple software Updates, restart if update requires it."""
- su_results = subprocess.check_output(['/usr/sbin/softwareupdate', '-ia'])
- if restart and ('restart' in su_results):
- tell_app('Finder', 'restart')
- sys.exit(0)
-
-
-def disable_wifi(port='en1'):
- subprocess.call(['/usr/sbin/networksetup', '-setairportpower', port, 'off'])
- subprocess.call(['/usr/sbin/networksetup', '-setnetworkserviceenabled', 'Wi-Fi', 'off'])
-
-
-def log(msg):
- print('*** %s...' % msg)
-
-
-def install_service(src):
- """Copy .service to systemwide Services folder."""
- if not os.path.exists(SERVICEDIR):
- os.mkdir(SERVICEDIR)
-
- ditto(src, SERVICEDIR)