getspace_osx.py

#!/usr/bin/env python

import os
import subprocess

# OSX NOTIFICATION FUNCTION
def notify(title, subtitle, text, kind):

	if kind == 'success':
		sound = 'Submarine'
	else:
		sound = 'Basso'

	os.system("""
		osascript -e 'display notification "{}" with title "{}" subtitle "{}" sound name "{}"'
		""".format(text, title, subtitle, sound))

# OSX SCREEN SIZE FUNCTION
def getscreensize():

	# import AppKit
	# # wrong because in a multi screen setup, it would return the size of the last screen in the array
	# for screen in AppKit.NSScreen.screens(): 
	# 	vw = screen.frame().size.width
	# 	vh = screen.frame().size.height
	# return (vw, vh)

	# system_profiler SPDisplaysDataType | grep Resolution | grep -oE '[0-9]+' | grep -Eo '[0-9]+$'
	systemprofile = subprocess.Popen(["system_profiler", "SPDisplaysDataType"], stdout=subprocess.PIPE)
	stdout = systemprofile.communicate()
	stdout = stdout[0].splitlines()
	for line in stdout:
		if "Resolution" in line:
			res = [int(s) for s in line.split() if s.isdigit()]
			vw = res[0]
			vh = res[1]
	return (vw, vh)

# OSX FONT DEFINITION VARIABLES
def getfontvars(height, explanation):
	
	from PIL import ImageFont
	from math import floor
	import textwrap

	# 'FUTURA+AVENIR' STYLE
	# SET SIZING VARS
	fsizehead = int(floor(height / 36))
	fsizetext = int(floor(height / 65))
	wrapped = textwrap.fill(explanation, 100)
	# SET FONT VARS
	headfont = ImageFont.truetype("/Library/Fonts/Futura.ttc",fsizehead,index=2)
	textfont = ImageFont.truetype("/System/Library/Fonts/Avenir.ttc",fsizetext)

	# 'MENLO' STYLE
	# SET SIZING VARS
	# fsizehead = int(floor(height / 30))
	# fsizetext = int(floor(height / 70))
	# wrapped = textwrap.fill(explanation, 105)
	# SET FONT VARS
	# headfont = ImageFont.truetype("/System/Library/Fonts/Menlo.ttc",fsizehead,index=1)
	# textfont = ImageFont.truetype("/System/Library/Fonts/Menlo.ttc",fsizetext)

	# 'GILL SANS' STYLE
	# SET SIZING VARS
	# fsizehead = int(floor(height / 35))
	# fsizetext = int(floor(height / 55))
	# wrapped = textwrap.fill(explanation, 110)
	# SET FONT VARS
	# headfont = ImageFont.truetype("/Library/Fonts/GillSans.ttc",fsizehead,index=1)
	# textfont = ImageFont.truetype("/Library/Fonts/GillSans.ttc",fsizetext,index=7)

	# 'HELVETICA NEUE' STYLE
	# SET SIZING VARS
	# fsizehead = int(floor(height / 30))
	# fsizetext = int(floor(height / 57))
	# wrapped = textwrap.fill(explanation, 100)
	# SET FONT VARS
	# headfont = ImageFont.truetype("/System/Library/Fonts/HelveticaNeue.dfont",fsizehead,index=6)
	# textfont = ImageFont.truetype("/System/Library/Fonts/HelveticaNeue.dfont",fsizetext,index=1)

	# 'SAN FRANCISCO' STYLE
	# SET SIZING VARS
	# fsizehead = int(floor(height / 35))
	# fsizetext = int(floor(height / 65))
	# wrapped = textwrap.fill(explanation, 105)
	# SET FONT VARS
	# headfont = ImageFont.truetype("/System/Library/Fonts/SFCompactText-Heavy.otf",fsizehead)
	# textfont = ImageFont.truetype("/System/Library/Fonts/SFCompactText-Light.otf",fsizetext)

	return (fsizehead, fsizetext, wrapped, headfont, textfont)


# OSX SET BACKGROUND
def dosetbackground(path):

	# GET OSX VERSION NUMBER
	import platform
	version = platform.mac_ver()

	# SET COMMAND DEPENDING ON OSX VERSION
	if version[0] >= 10.9:
		# AFTER MAVERICKS: SQLITE SOLUTION
		# seems to be working though it uses killall Dock and all desktops have to be created from the first
		# add an  WHERE ROWID='1' ??
		setbackgroundcmd = """
		sqlite3 ~/Library/Application\ Support/Dock/desktoppicture.db "UPDATE data SET value='{}'" && killall Dock
		""".format(path)

	else:
		# BEFORE MAVERICKS: APPLESCRIPT SOLUTION
		setbackgroundcmd = """
		osascript -e 'tell application "System Events"
		set desktopCount to count of desktops
		repeat with desktopNumber from 1 to desktopCount
			tell desktop desktopNumber
				set picture to "{}"
			end tell
		end repeat
		end tell'
		""".format(path)

		# setbackgroundcmd = """
		# osascript -e 'tell application "System Events" to set picture of every desktop to ("{}" as POSIX file as alias)'
		# """.format(path)

	os.popen(setbackgroundcmd)


# ADD FUNCTION FOR CHANGING ADMIN LOGIN PNG
# dosetbackground(savein)
	# loginimage = '/Library/Caches/com.apple.desktop.admin.png'
	
	# try:
	# 	os.remove(loginimage) # OSX REGENERATES FILE AUTOMATICALLY / or not
	# except OSError:
	# 	pass

	# admin = img.filter(ImageFilter.GaussianBlur(radius=15))
	# admin.save(loginimage)