Mercurial > hg > PaInt
annotate paint/package.py @ 37:36e70712fc9a
things are just awful
| author | Jeff Hammel <jhammel@mozilla.com> |
|---|---|
| date | Fri, 30 Mar 2012 15:06:17 -0700 |
| parents | f59da9e6be37 |
| children | c9d8abb2026f |
| rev | line source |
|---|---|
| 4 | 1 """ |
| 2 package model for python PAckage INTrospection | |
| 3 """ | |
| 4 | |
| 36 | 5 # TODO: use pkginfo.sdist more |
| 6 | |
| 7 | 7 import os |
| 28 | 8 import pip |
|
29
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
9 import pypi |
|
9
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
10 import shutil |
| 19 | 11 import subprocess |
| 12 import sys | |
| 7 | 13 import tarfile |
| 5 | 14 import tempfile |
| 7 | 15 import urllib2 |
|
21
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
16 import urlparse |
| 7 | 17 import utils |
| 5 | 18 |
| 19 | 19 try: |
| 20 from subprocess import check_call as call | |
| 21 except ImportError: | |
| 22 from subporcess import call | |
| 23 | |
| 10 | 24 __all__ = ['Package'] |
| 25 | |
| 4 | 26 class Package(object): |
| 33 | 27 """ |
| 28 class for python package introspection. | |
| 29 constructor takes the package 'src' | |
| 30 """ | |
| 4 | 31 |
| 32 def __init__(self, src): | |
| 33 self.src = src | |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
34 |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
35 # ephemeral data |
| 5 | 36 self._tmppath = None |
| 28 | 37 self._egg_info_path = None |
| 33 | 38 self._build_path = None |
| 5 | 39 |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
40 # TODO: list of temporary files/directories to be deleted |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
41 |
| 34 | 42 def _path(self): |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
43 """filesystem path to package directory""" |
| 7 | 44 |
| 45 # return cached copy if it exists | |
| 46 if self._tmppath: | |
| 47 return self._tmppath | |
| 48 | |
| 49 # fetch from the web if a URL | |
| 50 tmpfile = None | |
| 51 src = self.src | |
| 5 | 52 if utils.isURL(self.src): |
| 8 | 53 tmpfile = src = self.fetch() |
| 7 | 54 |
| 55 # unpack if an archive | |
| 28 | 56 if self._is_archive(src): |
| 8 | 57 try: |
| 37 | 58 self._unpack(src) |
| 8 | 59 finally: |
| 60 if tmpfile: | |
| 61 os.remove(tmpfile) | |
| 7 | 62 return self._tmppath |
| 63 | |
| 5 | 64 return self.src |
| 65 | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
66 def fetch(self, filename=None): |
| 8 | 67 """fetch from remote source to a temporary file""" |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
68 if filename is None: |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
69 fd, filename = tempfile.mkstemp() |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
70 os.close(fd) |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
71 fp = file(filename, 'w') |
|
15
8c8b7482772f
fix a few failures but still failing
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
72 resource = urllib2.urlopen(self.src) |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
73 fp.write(resource.read()) |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
74 fp.close() |
| 7 | 75 return filename |
| 76 | |
| 37 | 77 def _unpack(self, archive): |
| 7 | 78 """unpack the archive to a temporary destination""" |
| 79 # TODO: should handle zipfile additionally at least | |
| 80 # Ideally, this would be pluggable, etc | |
| 8 | 81 assert tarfile.is_tarfile(archive), "%s is not an archive" % self.src |
|
9
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
82 tf = tarfile.TarFile.open(archive) |
|
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
83 self._tmppath = tempfile.mkdtemp() |
| 19 | 84 members = tf.getmembers() |
| 85 | |
| 86 # cut off the top level directory | |
| 37 | 87 members = [i for i in members if os.path.sep in i.name] |
| 19 | 88 tld = set() |
| 89 for member in members: | |
| 90 directory, member.name = member.name.split(os.path.sep, 1) | |
| 91 tld.add(directory) | |
| 92 assert len(tld) == 1 | |
| 93 | |
| 94 # extract | |
| 95 for member in members: | |
| 96 tf.extract(member, path=self._tmppath) | |
| 97 tf.close() | |
| 7 | 98 |
| 28 | 99 def _is_archive(self, path): |
| 7 | 100 """returns if the filesystem path is an archive""" |
| 101 # TODO: should handle zipfile additionally at least | |
| 102 # Ideally, this would be pluggable, etc | |
| 103 return tarfile.is_tarfile(path) | |
|
9
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
104 |
| 28 | 105 def _cleanup(self): |
|
9
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
106 if self._tmppath: |
|
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
107 shutil.rmtree(self._tmppath) |
|
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
108 self._tmppath = None |
|
562230cc2e86
now the package class should at least extracts things. it doesnt actually do anything useful
Jeff Hammel <jhammel@mozilla.com>
parents:
8
diff
changeset
|
109 |
|
18
4c4d1e194bf7
leave in debugging code for now
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
110 # __del__ = cleanup |
| 11 | 111 |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
112 ### python-package-specific functionality |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
113 |
| 28 | 114 def _egg_info(self): |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
115 """build the egg_info directory""" |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
116 |
| 28 | 117 if self._egg_info_path: |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
118 # return cached copy |
| 28 | 119 return self._egg_info_path |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
120 |
| 34 | 121 directory = self._path() |
| 16 | 122 setup_py = os.path.join(directory, 'setup.py') |
| 123 if not os.path.exists(setup_py): | |
| 17 | 124 raise AssertionError("%s does not exist" % setup_py) |
| 11 | 125 |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
126 # setup the egg info |
| 19 | 127 call([sys.executable, 'setup.py', 'egg_info'], cwd=directory, stdout=subprocess.PIPE) |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
128 |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
129 # get the .egg-info directory |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
130 egg_info = [i for i in os.listdir(directory) |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
131 if i.endswith('.egg-info')] |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
132 assert len(egg_info) == 1, 'Expected one .egg-info directory in %s, got: %s' % (directory, egg_info) |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
133 egg_info = os.path.join(directory, egg_info[0]) |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
134 assert os.path.isdir(egg_info), "%s is not a directory" % egg_info |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
135 |
|
14
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
136 # cache it |
| 28 | 137 self._egg_info_path = egg_info |
| 138 return self._egg_info_path | |
|
14
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
139 |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
140 def info(self): |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
141 """return info dictionary for package""" |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
142 # could use pkginfo |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
143 |
| 28 | 144 egg_info = self._egg_info() |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
145 |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
146 # read the package information |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
147 pkg_info = os.path.join(egg_info, 'PKG-INFO') |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
148 info_dict = {} |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
149 for line in file(pkg_info).readlines(): |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
150 if not line or line[0].isspace(): |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
151 continue # XXX neglects description |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
152 assert ':' in line |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
153 key, value = [i.strip() for i in line.split(':', 1)] |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
154 info_dict[key] = value |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
155 |
|
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
156 # return the information |
|
20
11bc01a089e7
return only the dictionary
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
157 return info_dict |
|
13
0dd1f8f83be2
moving towards some sort of organization
Jeff Hammel <jhammel@mozilla.com>
parents:
12
diff
changeset
|
158 |
| 11 | 159 def dependencies(self): |
|
14
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
160 """return the dependencies""" |
| 32 | 161 # TODO: should probably have a more detailed dict: |
| 162 # {'mozinfo': {'version': '>= 0.2', | |
| 163 # 'url': 'http://something.com/'}} | |
|
21
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
164 # get the egg_info directory |
| 28 | 165 egg_info = self._egg_info() |
|
14
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
166 |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
167 # read the dependencies |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
168 requires = os.path.join(egg_info, 'requires.txt') |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
169 if os.path.exists(requires): |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
170 dependencies = [i.strip() for i in file(requires).readlines() if i.strip()] |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
171 else: |
|
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
172 dependencies = [] |
|
21
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
173 dependencies = dict([(i, None) for i in dependencies]) |
|
14
6d27c2136163
a decent structure; need to reuse more
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
174 |
|
21
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
175 # read the dependency links |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
176 dependency_links = os.path.join(egg_info, 'dependency_links.txt') |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
177 if os.path.exists(dependency_links): |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
178 links = [i.strip() for i in file(dependency_links).readlines() if i.strip()] |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
179 for link in links: |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
180 # XXX pretty ghetto |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
181 assert '#egg=' in link |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
182 url, dep = link.split('#egg=', 1) |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
183 if dep in dependencies: |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
184 dependencies[dep] = link |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
185 |
|
4df3e715817d
now have passing tests again
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
186 return dependencies |
| 24 | 187 |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
188 def extension(self): |
| 35 | 189 """filename extension of the package""" |
| 190 | |
| 191 package = self.package() | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
192 |
| 35 | 193 # determine the extension (XXX hacky) |
| 194 extensions = ('.tar.gz', '.zip', '.tar.bz2') | |
| 195 for ext in extensions: | |
| 36 | 196 if package.endswith(ext): |
| 35 | 197 return ext |
| 198 | |
| 199 raise Exception("Extension %s not found: %s" % (extensions, package)) | |
| 200 | |
| 201 def package(self, destination=None): | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
202 """ |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
203 repackage the package to ensure its actually in the right form |
| 34 | 204 and return the path to the destination |
| 205 - destination: if given, path to put the build in [TODO] | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
206 """ |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
207 |
| 33 | 208 if self._build_path: |
| 34 | 209 if destination: |
| 210 shutil.copy(self._build_path, destination) | |
| 211 return os.path.abspath(destination) | |
| 212 | |
| 213 # return cached copy | |
| 33 | 214 return self._build_path |
| 215 | |
| 34 | 216 path = self._path() |
| 217 dist = os.path.join(path, 'dist') | |
| 218 if os.path.exists(dist): | |
| 219 shutil.rmtree(dist) | |
| 220 | |
| 221 call([sys.executable, 'setup.py', 'sdist'], cwd=path, stdout=subprocess.PIPE) | |
| 222 | |
| 223 assert os.path.exists(dist) | |
| 224 contents = os.listdir(dist) | |
| 225 assert len(contents) == 1 | |
| 226 | |
| 227 self._build_path = os.path.join(dist, contents[0]) | |
| 228 | |
| 229 # destination | |
| 230 # use an evil recursive trick | |
| 231 if destination: | |
| 36 | 232 return self.package(destination=destination) |
| 34 | 233 |
| 234 return self._build_path | |
| 235 | |
| 24 | 236 def download(self, directory): |
| 237 """download a package and all its dependencies using pip""" | |
| 28 | 238 if not os.path.exists(directory): |
| 239 os.makedirs(directory) | |
| 240 assert os.path.isdir(directory) | |
| 241 pip.main(['install', '--download', directory, self.src]) | |
| 242 | |
| 243 def pypi(self, directory): | |
| 244 """ | |
| 245 download packages for a pypi directory structure | |
| 246 http://k0s.org/portfolio/pypi.html | |
| 247 """ | |
|
29
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
248 if not os.path.exists(directory): |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
249 os.makedirs(directory) |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
250 assert os.path.isdir(directory) |
| 28 | 251 tempdir = tempfile.mkdtemp() |
| 252 try: | |
|
29
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
253 self.download(tempdir) |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
254 for package in os.listdir(tempdir): |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
255 |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
256 # full path |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
257 src = os.path.join(tempdir, package) |
|
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
258 |
| 35 | 259 # make a package of the thing |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
260 package = Package(src) |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
261 |
| 36 | 262 # get destination dirname, filename |
| 263 dirname, filename = package.pypi_path() | |
|
29
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
264 |
| 36 | 265 # make the directory if it doesn't exist |
| 266 subdir = os.path.join(directory, dirname) | |
| 267 if not os.path.exists(subdir): | |
| 268 os.makedirs(subdir) | |
| 269 assert os.path.isdir(subdir) | |
|
29
59524b6d25c0
implementing making pypi directory strcuture from a package
Jeff Hammel <jhammel@mozilla.com>
parents:
28
diff
changeset
|
270 |
| 36 | 271 # move the file |
| 272 self.package(destination=os.path.join(subdir, filename)) | |
| 28 | 273 finally: |
| 274 shutil.rmtree(tempdir) | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
275 |
| 36 | 276 def pypi_path(self): |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
277 """ |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
278 returns subpath 2-tuple appropriate for pypi path structure: |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
279 http://k0s.org/portfolio/pypi.html |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
280 """ |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
281 info = self.info() |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
282 |
| 36 | 283 # determine the extension |
| 284 extension = self.extension() | |
|
31
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
285 |
|
5fb1844db0b2
back to the drawring broad
Jeff Hammel <jhammel@mozilla.com>
parents:
30
diff
changeset
|
286 # get the filename destination |
| 36 | 287 name = info['Name'] |
| 288 version = info['Version'] | |
| 289 filename = '%s-%s%s' % (name, version, extension) | |
| 290 return name, filename |