FilesExpand file tree

 master

/

ipython_win_post_install.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

147 lines (119 loc) · 4.87 KB

 master

/

ipython_win_post_install.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

147 lines (119 loc) · 4.87 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

#!python

"""Distutils post installation script for Windows.

http://docs.python.org/2/distutils/builtdist.html#the-postinstallation-script

"""

from __future__ import print_function

import os

import sys

import shutil

try:

import setuptools

have_setuptools = True

except ImportError:

have_setuptools = False

pjoin = os.path.join

# suffix for start menu folder names

pyver = "(Py%i.%i %i bit)" % (sys.version_info[0], sys.version_info[1],

(32, 64)[sys.maxsize > 2**32])

def mkshortcut(target, description, linkdir, arguments="", iconpath='',

workdir="%HOMEDRIVE%%HOMEPATH%", iconindex=0):

"""Make a shortcut if it doesn't exist and register its creation."""

filename = pjoin(linkdir, description + '.lnk')

description = "%s %s" % (description, pyver)

create_shortcut(target, description, filename, arguments, workdir,

iconpath, iconindex)

file_created(filename)

def arguments(scriptsdir, script, scriptargs=''):

"""Return command line arguments to be passed to the python executable."""

cmdbase = suffix(pjoin(scriptsdir, script))

if have_setuptools:

cmdbase += '-script.py'

return '"%s" %s' % (cmdbase, scriptargs)

def suffix(s):

"""Add '3' suffix to programs for Python 3."""

if sys.version_info[0] == 3:

s = s + '3'

return s

def install():

"""Routine to be run by the win32 installer with the -install switch."""

# Get some system constants

python = pjoin(sys.prefix, 'python.exe')

pythonw = pjoin(sys.prefix, 'pythonw.exe')

if not have_setuptools:

# This currently doesn't work without setuptools,

# so don't bother making broken links

print("Setuptools is required to"

" create Start Menu items.", file=sys.stderr)

print("Re-run this installer after installing"

" Setuptools to get Start Menu items.", file=sys.stderr)

return

# Lookup path to common startmenu ...

ip_start_menu = pjoin(get_special_folder_path('CSIDL_COMMON_PROGRAMS'),

'IPython %s' % pyver)

# Create IPython entry ...

if not os.path.isdir(ip_start_menu):

os.mkdir(ip_start_menu)

directory_created(ip_start_menu)

# Create .py and .bat files to make things available from

# the Windows command line. Thanks to the Twisted project

# for this logic!

programs = [

'ipython',

'iptest',

'ipcontroller',

'ipengine',

'ipcluster',

]

programs = [suffix(p) for p in programs]

scripts = pjoin(sys.prefix, 'scripts')

if not have_setuptools:

# only create .bat files if we don't have setuptools

for program in programs:

raw = pjoin(scripts, program)

bat = raw + '.bat'

py = raw + '.py'

# Create .py versions of the scripts

shutil.copy(raw, py)

# Create .bat files for each of the scripts

bat_file = file(bat, 'w')

bat_file.write("@%s %s %%*" % (python, py))

bat_file.close()

# Create Start Menu shortcuts

iconpath = pjoin(scripts, 'ipython.ico')

mkshortcut(python, 'IPython', ip_start_menu,

arguments(scripts, 'ipython'), iconpath)

mkshortcut(python, 'IPython (pylab mode)', ip_start_menu,

arguments(scripts, 'ipython', '--pylab'), iconpath)

mkshortcut(python, 'IPython Controller', ip_start_menu,

arguments(scripts, 'ipcontroller'), iconpath)

mkshortcut(python, 'IPython Engine', ip_start_menu,

arguments(scripts, 'ipengine'), iconpath)

mkshortcut(pythonw, 'IPython Qt Console', ip_start_menu,

arguments(scripts, 'ipython', 'qtconsole'), iconpath)

iconpath = pjoin(scripts, 'ipython_nb.ico')

mkshortcut(python, 'IPython Notebook', ip_start_menu,

arguments(scripts, 'ipython', 'notebook'), iconpath)

mkshortcut(pythonw, 'IPython Documentation', ip_start_menu,

'-m webbrowser -t "http://ipython.org/documentation.html',

iconpath='url.dll')

# Disable pysh Start item until the profile restores functionality

# Most of this code is in IPython/deathrow, and needs to be updated

# to 0.11 APIs

#mkshortcut(python, 'IPython%s (command prompt mode)', ip_start_menu,

# arguments(scripts, 'ipython', 'profile=pysh --init'))

def remove():

"""Routine to be run by the win32 installer with the -remove switch."""

pass

# main()

if len(sys.argv) > 1:

if sys.argv[1] == '-install':

try:

install()

except OSError:

print("Failed to create Start Menu items, try running the"

" installer as administrator.", file=sys.stderr)

elif sys.argv[1] == '-remove':

remove()

else:

print("Script was called with option %s" % sys.argv[1],

file=sys.stderr)