1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
#!/usr/bin/env python
"""
Create desktop shortcuts for Darwin / MacOS
"""
import os
import sys
import shutil
from pathlib import Path
from collections import namedtuple
from .utils import get_pyexe, get_homedir
def get_startmenu():
"get start menu location"
return ''
def get_desktop():
"get desktop location"
return Path(get_homedir(), 'Desktop').resolve().as_posix()
def get_folders():
"""get user-specific folders
Returns:
-------
Named tuple with fields 'home', 'desktop', 'startmenu'
Example:
-------
>>> from pyshortcuts import get_folders
>>> folders = get_folders()
>>> print("Home, Desktop, StartMenu ",
... folders.home, folders.desktop, folders.startmenu)
"""
UserFolders = namedtuple("UserFolders", ("home", "desktop", "startmenu"))
return UserFolders(get_homedir(), get_desktop(), get_startmenu())
def make_shortcut(script, name=None, description=None, icon=None, working_dir=None,
folder=None, terminal=True, desktop=True,
startmenu=False, executable=None, noexe=False):
"""create shortcut
Arguments:
---------
script (str) path to script, may include command-line arguments
name (str, None) name to display for shortcut [name of script]
description (str, None) longer description of script [`name`]
icon (str, None) path to icon file [python icon]
working_dir (str, None) directory where to run the script in
folder (str, None) subfolder of Desktop for shortcut [None] (See Note 1)
terminal (bool) whether to run in a Terminal [True]
desktop (bool) whether to add shortcut to Desktop [True]
startmenu (bool) whether to add shortcut to Start Menu [False] (See Note 2)
executable (str, None) name of executable to use [this Python] (see Note 3)
noexe (bool) whether to use no executable (script is entire command) [False]
Notes:
------
1. `folder` will place shortcut in a subfolder of Desktop and/or Start Menu
2. Start Menu does not exist for Darwin / MacOSX
3. executable defaults to the Python executable used to make shortcut.
"""
if not desktop:
return None
userfolders = get_folders()
if working_dir is None:
working_dir = ''
from .shortcut import shortcut
scut = shortcut(script, userfolders, name=name, description=description,
working_dir=working_dir, folder=folder, icon=icon)
if noexe:
full_script =scut.script
executable = ''
else:
full_script =scut.full_script
if executable is None:
executable = get_pyexe()
executable = Path(executable).resolve().as_posix()
if Path(scut.full_script).resolve() == Path(executable).resolve():
executable = ''
if not Path(scut.desktop_dir).exists():
os.makedirs(scut.desktop_dir)
osascript = f'{full_script} {scut.arguments}'
osascript = osascript.replace(' ', '\\ ')
dest = Path(scut.desktop_dir, scut.target).resolve().as_posix()
if Path(dest).exists():
shutil.rmtree(dest)
os.mkdir(dest)
os.mkdir(Path(dest, 'Contents'))
os.mkdir(Path(dest, 'Contents', 'MacOS'))
os.mkdir(Path(dest, 'Contents', 'Resources'))
opts = {'name': scut.name,
'desc': scut.description,
'script': full_script,
'workdir': scut.working_dir,
'args': scut.arguments,
'prefix': Path(sys.prefix).resolve().as_posix(),
'exe': executable,
'osascript': osascript}
info = """<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN"
"http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
<key>CFBundleGetInfoString</key> <string>{desc:s}</string>
<key>CFBundleName</key> <string>{name:s}</string>
<key>CFBundleExecutable</key> <string>{name:s}</string>
<key>CFBundleIconFile</key> <string>{name:s}</string>
<key>CFBundlePackageType</key> <string>APPL</string>
</dict>
</plist>
"""
text = ['#!/bin/bash',
"export EXE={exe:s}",
"export SCRIPT={script:s}",
"export ARGS='{args:s}'"]
if scut.working_dir not in (None, ''):
text.append("cd {workdir:s}")
if terminal:
text.append(r"""osascript -e 'tell application "Terminal"
do script "'${{EXE}}\ {osascript:s}'"
end tell
'
""")
else:
text.append("$EXE $SCRIPT $ARGS")
text.append('\n')
text = '\n'.join(text)
with open(Path(dest, 'Contents', 'Info.plist'), 'w') as fout:
fout.write(info.format(**opts))
ascript_name = Path(dest, 'Contents', 'MacOS', scut.name).as_posix()
with open(ascript_name, 'w') as fout:
fout.write(text.format(**opts))
os.chmod(ascript_name, 493) # = octal 755 / rwxr-xr-x
icon_dest = Path(dest, 'Contents', 'Resources', scut.name + '.icns').as_posix()
shutil.copy(scut.icon, icon_dest)
return scut
|
