Mercurial > hg > fetch
annotate fetch.py @ 16:c77d29a10e08
get it working
| author | Jeff Hammel <jhammel@mozilla.com> |
|---|---|
| date | Wed, 09 Nov 2011 16:33:29 -0800 |
| parents | bc7d6763357e |
| children | e2af4bc5159c |
| rev | line source |
|---|---|
| 0 | 1 #!/usr/bin/env python |
| 2 | |
| 3 """ | |
| 4 fetch stuff from the interwebs | |
| 5 """ | |
| 6 | |
| 7 import os | |
| 8 import sys | |
| 9 import optparse | |
| 10 | |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
11 __all__ = ['Fetcher', 'Fetch', 'main'] |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
12 |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
13 def which(executable, path=os.environ['PATH']): |
| 15 | 14 """python equivalent of which; should really be in the stdlib""" |
| 15 dirs = path.split(os.pathsep) | |
| 16 for dir in dirs: | |
| 17 if os.path.isfile(os.path.join(dir, executable)): | |
| 18 return os.path.join(dir, executable) | |
| 7 | 19 |
| 0 | 20 class Fetcher(object): |
| 15 | 21 """abstract base class for resource fetchers""" |
| 0 | 22 |
| 15 | 23 @classmethod |
| 24 def match(cls, _type): | |
| 25 return _type == cls.type | |
| 0 | 26 |
| 15 | 27 def __init__(self, url): |
| 28 self.url = url | |
| 0 | 29 |
| 15 | 30 def __call__(self, dest): |
| 31 raise NotImplementedError | |
| 0 | 32 |
| 7 | 33 ### standard dispatchers - always available |
| 0 | 34 |
| 7 | 35 import tarfile |
| 0 | 36 import urllib2 |
| 7 | 37 from StringIO import StringIO |
| 0 | 38 |
| 5 | 39 class FileFetcher(Fetcher): |
| 15 | 40 """fetch a single file""" |
| 0 | 41 |
| 15 | 42 type = 'file' |
| 0 | 43 |
| 15 | 44 @classmethod |
| 45 def download(cls, url): | |
| 46 return urllib2.urlopen(url).read() | |
| 0 | 47 |
| 15 | 48 def __call__(self, dest): |
| 49 if os.path.isdir(dest): | |
| 50 filename = self.url.rsplit('/', 1)[-1] | |
| 51 dest = os.path.join(dest, filename) | |
| 52 f = file(dest, 'w') | |
| 53 f.write(self.download(self.url)) | |
| 54 f.close() | |
| 0 | 55 |
|
6
86f6f99e421b
add types for unimplemented dispatchers
Jeff Hammel <jhammel@mozilla.com>
parents:
5
diff
changeset
|
56 |
| 5 | 57 class TarballFetcher(FileFetcher): |
| 15 | 58 """fetch and extract a tarball""" |
| 0 | 59 |
| 15 | 60 type = 'tar' |
| 0 | 61 |
| 15 | 62 def __call__(self, dest): |
| 63 assert os.path.isdir(dest) | |
| 64 buffer = StringIO() | |
| 65 buffer.write(self.download(self.url)) | |
| 66 buffer.seek(0) | |
| 67 tf = tarfile.open(mode='r', fileobj=buffer) | |
| 68 tf.extract(dest) | |
| 7 | 69 |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
70 fetchers = [FileFetcher, TarballFetcher] |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
71 |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
72 ### VCS fetchers using executable |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
73 |
|
11
726c3d288733
* add convenience import in __init__
Jeff Hammel <jhammel@mozilla.com>
parents:
10
diff
changeset
|
74 import subprocess |
|
726c3d288733
* add convenience import in __init__
Jeff Hammel <jhammel@mozilla.com>
parents:
10
diff
changeset
|
75 |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
76 if which('hg'): |
|
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
77 |
| 15 | 78 class HgFetcher(Fetcher): |
| 79 """checkout a mercurial repository""" | |
| 80 type = 'hg' | |
| 0 | 81 |
| 15 | 82 def __call__(self, dest): |
| 83 if os.path.exits(dest): | |
| 84 assert os.path.isdir(dest) and os.path.exists(os.path.join(dest, '.hg')) | |
| 85 pass # TODO | |
|
11
726c3d288733
* add convenience import in __init__
Jeff Hammel <jhammel@mozilla.com>
parents:
10
diff
changeset
|
86 |
| 15 | 87 fetchers.append(HgFetcher) |
|
6
86f6f99e421b
add types for unimplemented dispatchers
Jeff Hammel <jhammel@mozilla.com>
parents:
5
diff
changeset
|
88 |
| 15 | 89 if which('git'): |
| 90 class GitFetcher(Fetcher): | |
| 91 """checkout a git repository""" | |
| 92 type = 'git' | |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
93 |
| 16 | 94 __all__ += [i.__name__ for i in fetchers] |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
95 |
| 0 | 96 class Fetch(object): |
| 97 | |
| 15 | 98 def __init__(self, fetchers, relative_to=None, strict=True): |
| 99 self.fetchers = fetchers | |
| 100 self.relative_to = relative_to | |
| 101 self.strict = strict | |
| 0 | 102 |
| 15 | 103 def fetcher(self, _type): |
| 104 """find the fetcher for the appropriate type""" | |
| 105 for fetcher in fetchers: | |
| 106 if fetcher.match(_type): | |
| 107 return fetcher | |
| 0 | 108 |
| 15 | 109 def __call__(self, url, destination, type, **options): |
| 110 fetcher = self.fetcher(type) | |
| 111 assert fetcher is not None, "No fetcher found for type '%s'" % type | |
| 112 fetcher = fetcher(url, **options) | |
| 113 fetcher(destination) | |
| 2 | 114 |
| 15 | 115 def fetch(self, *items): |
| 2 | 116 |
| 15 | 117 if self.strict: |
| 118 # ensure all the required fetchers are available | |
| 119 types = set([i['type'] for i in items]) | |
| 120 assert not [i for i in types | |
| 121 if [True for fetcher in fetchers if fetcher.match(i)]] | |
| 4 | 122 |
| 15 | 123 for item in items: |
| 4 | 124 |
| 15 | 125 # fix up relative paths |
| 126 dest = item['dest'] | |
| 127 if not os.path.isabs(dest): | |
| 128 relative_to = self.relative_to or os.path.dirname(os.path.abspath(item['manifest'])) | |
| 129 dest = os.path.join(relative_to, dest) | |
| 4 | 130 |
| 15 | 131 # fetch the items |
| 132 self(item['url'], destination=dest, type=item['type'], **item['options']) | |
| 0 | 133 |
| 134 format_string = "[URL] [destination] [type] <options>" | |
| 135 def read_manifests(*manifests): | |
| 15 | 136 """ |
| 137 read some manifests and return the items | |
| 138 | |
| 139 Format: | |
| 140 %s | |
| 141 """ % format_string | |
| 0 | 142 |
| 15 | 143 # sanity check |
| 144 assert not [i for i in manifests if not os.path.exists(i)] | |
| 0 | 145 |
| 15 | 146 retval = [] |
| 0 | 147 |
| 15 | 148 for manifest in manifests: |
| 149 for line in file(i).readlines(): | |
| 150 line = line.strip() | |
| 151 if line.startswith('#') or not line: | |
| 152 continue | |
| 153 line = line.split() | |
| 154 if len(line) not in (3,4): | |
| 155 raise Exception("Format should be: %s; line %s" % (format_string, line)) | |
| 156 options = {} | |
| 157 if len(line) == 4: | |
| 158 option_string = line.pop().rstrip(',') | |
| 159 try: | |
| 160 options = dict([[j.strip() for j in i.split('=', 1)] | |
| 161 for i in option_string.split(',')]) | |
| 162 except: | |
| 163 raise Exception("Options format should be: key=value,key2=value2,...; got %s" % option_string) | |
| 0 | 164 |
| 15 | 165 url, dest, _type = line |
| 166 retval.append(dict(url=url, dest=dest, type=_type, options=options, manifest=manifest)) | |
| 167 return retval | |
| 0 | 168 |
| 2 | 169 def main(args=sys.argv[1:]): |
| 0 | 170 |
| 15 | 171 # parse command line options |
| 172 usage = '%prog [options] manifest [manifest] [...]' | |
| 0 | 173 |
| 15 | 174 class PlainDescriptionFormatter(optparse.IndentedHelpFormatter): |
| 175 def format_description(self, description): | |
| 176 if description: | |
| 177 return description + '\n' | |
| 178 else: | |
| 179 return '' | |
| 0 | 180 |
| 15 | 181 parser = optparse.OptionParser(usage=usage, description=__doc__, formatter=PlainDescriptionFormatter()) |
| 182 parser.add_option('-o', '--output', | |
| 183 help="output relative to this location vs. the manifest location") | |
| 184 parser.add_option('-d', '--dest', | |
| 185 action='append', | |
| 186 help="output only these destinations") | |
| 187 parser.add_option('-s', '--strict', | |
| 188 action='store_true', default=False, | |
| 189 help="fail on error") | |
| 190 parser.add_option('--list-fetchers', dest='list_fetchers', | |
| 191 action='store_true', default=False, | |
| 192 help='list available fetchers and exit') | |
| 193 options, args = parser.parse_args(args) | |
| 0 | 194 |
| 15 | 195 if options.list_fetchers: |
| 16 | 196 |
| 15 | 197 for name in sorted(fetchers.keys()): |
| 198 print name | |
| 199 parser.exit() | |
|
8
cf00d46b1bfb
pretend like we have a pluggable system to start debugging it
Jeff Hammel <jhammel@mozilla.com>
parents:
7
diff
changeset
|
200 |
| 15 | 201 if not args: |
| 202 parser.print_help() | |
| 203 parser.exit() | |
| 0 | 204 |
| 15 | 205 items = read_manifests(*args) |
| 16 | 206 fetch = Fetch(fetchers, strict=options.strict) |
| 0 | 207 |
| 15 | 208 # download the files |
| 209 fetch.fetch(*items) | |
| 0 | 210 |
| 211 if __name__ == '__main__': | |
| 15 | 212 main() |
| 0 | 213 |