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
159
160
161
162
163
|
#!/bin/env python
print "This determines the site-packages directory of python and installs extended_threading there."
import sys
import os
import time
import py_compile
verbose = False;
src = "."
pslen = len(os.sep)
def install(dest, src="."):
"""
A generic install function. it will parse through the directory in which it is placed, and try to install all the files into the
chosen destination directory (dest). It will skip over install.py. The source directory is by default ".", though it can be
changed. Do not use a file separator with src.
"""
path = src
contents = os.listdir(path)
if(dest[-pslen:] != os.sep):
dest += os.sep
tpath = path
if(path[0:pslen+1] == ("." + os.sep)):
tpath = path[pslen+1:]
elif(path == "."):
tpath = ""
#ensure the destination directory exists
if(not os.path.exists(dest + tpath) and tpath != ""):
if(verbose):
print "Creating: " + dest+tpath
os.mkdir(dest+tpath)
for c in contents:
if(c == "install.py" or c == "samples" or
c[-4:] == ".txt" or c[-4:] == ".TXT"):
continue
install_sub(dest, path + os.sep, c)
#def install()
def install_sub(dest, path, c):
"""
Take file [c] in [path], and install it to [dest]/[path]. If [c] is a directory, create it and install the contents of [c].
"""
tpath = path
if(path[0:pslen+1] == ("." + os.sep)):
tpath = path[pslen+1:]
if(os.path.isdir(path + c)):
#ensure the destination directory exists
if(not os.path.exists(dest + tpath + c)):
if(verbose):
print "Creating: " + dest+tpath
os.mkdir(dest+tpath+c)
#now install each file in it:
path += c + os.sep
contents = os.listdir(path)
for co in contents:
install_sub(dest, path, co)
#if(os.path.isdir(path)):
else:
if(verbose):
print " Copying: " + path + c + " --> " + dest + tpath + c
sf = open(path+c, "rb")
df = open(dest+tpath+c, "wb")
#buffer input file
sf.seek(0, 2)
flen = sf.tell()
sf.seek(0)
total = 0
if(verbose):
sys.stdout.write(" % 9d of % 9d bytes" % (0, flen))
while(total < flen):
buff = sf.read()
if(buff == ""):
break
df.write(buff)
total += len(buff)
if(verbose):
sys.stdout.write("\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b")
sys.stdout.write("% 9d of % 9d bytes" % (total, flen))
time.sleep(0.25)
print ""
sf.close()
df.close()
#compile if it is python
if(c[-3:] == ".py"):
if(verbose):
print " Compiling..."
py_compile.compile(dest+tpath+c)
#def install_sub(dest, path, c):
for v in sys.argv[1:]:
if(v == "-v" or v == "--verbose"):
verbose = not verbose
elif(v[:9] == "--source="):
src = v[9:]
else:
print "Unknown argument, " + v
#for v in sys.argv[1:]:
#find the site-packages directories
possibles = []
for p in sys.path:
if(p.find("site-packages") > 0):
possibles.append(p)
if(len(possibles) == 0):
print "No install path found."
sys.exit(1)
#get the directory if there are multiple possibilities:
while(len(possibles) > 1):
i = 0
print ""
print "Choose an install path"
while(i < len(possibles)):
print str(i+1) + ". " + possibles[i]
i += 1
print "Install to: ",
l = sys.stdin.readline()
if(l == ""):
sys.exit(0)
l = l.strip()
try:
l = int(l)
if(l >= 1 and l <= len(possibles)):
possibles = [possibles[l-1]]
except Exception, e:
pass
#while(len(possibles) > 1):
ans = ""
while(not (ans == "y" or ans == "yes" or
ans == "n" or ans == "no")):
print ""
print "Installing to: " + possibles[0] + " (yes/no): ",
ans = sys.stdin.readline()
if(ans == ""):
sys.exit(0)
ans = ans.strip().lower()
#while(not (ans == "y" or ans == "yes" or
# ans == "n" or ans == "no")):
if(ans == "y" or ans == "yes"):
#install
print ""
path = possibles[0]
install(path)
#if(ans == "y" or ans == "yes"):
|
