setup.py: switch to python3

setup.py

@@ -1,29 +1,30 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python3
 
 import errno
 import os
 import sys
 
-if os.path.dirname(__file__) != '.':
-	print 'This script must be run from the dotfiles directory.'
+dirname = os.path.dirname(__file__)
+if dirname[len(os.path.commonprefix([os.path.abspath(__file__), dirname])):] != '.':
+	print('This script must be run from the dotfiles directory.')
 	sys.exit(1)
 script = os.path.basename(__file__)
 
 def link_file(target, name):
 	try:
 		os.symlink(target, name)
-		print '\033[36;1m' + name + '\033[0m ->', target
+		print('\033[36;1m' + name + '\033[0m ->', target)
 	except OSError as e:
 		if e.errno != errno.EEXIST:
 			raise
 		try:
 			if os.readlink(name) == target:
-				print name, 'is already linked'
+				print(name, 'is already linked')
 			else:
-				print name, '\033[31;1mexists\033[0m (symlink)'
+				print(name, '\033[31;1mexists\033[0m (symlink)')
 		except OSError as e:
 			if e.errno == errno.EINVAL:
-				print name, '\033[31;1mexists\033[0m'
+				print(name, '\033[31;1mexists\033[0m')
 			else:
 				raise
 
@@ -42,10 +43,10 @@ for f in os.listdir('.'):
 			try:
 				dirname = '../.' + dirpath
 				os.mkdir(dirname)
-				print 'Created \033[34;1m' + dirname + '\033[0m'
+				print('Created \033[34;1m' + dirname + '\033[0m')
 			except OSError as e:
 				if e.errno == errno.EEXIST:
-					print 'No need to create', dirname
+					print('No need to create', dirname)
 				else:
 					raise
 			prefix = '../' * (dirpath.count('/') + 1)
@@ -57,7 +58,7 @@ link_file('../dotfiles/vim', '../.config/nvim')
 
 vim_swap = '../.vim_swap'
 if os.path.exists(vim_swap):
-	print 'No need to create', vim_swap
+	print('No need to create', vim_swap)
 else:
-	os.mkdir(vim_swap, 0700)
-	print 'Created \033[34;1m' + vim_swap + '\033[0m'
+	os.mkdir(vim_swap, 0o700)
+	print('Created \033[34;1m' + vim_swap + '\033[0m')