Mercurial > hg > buttercup
comparison buttercup/source.py @ 18:151862a0a711
begin restructuring to something more modular and extensible
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Mon, 30 May 2011 13:40:13 -0700 |
parents | |
children | d3b1bf9d8235 |
comparison
equal
deleted
inserted
replaced
17:8b4c21b853e9 | 18:151862a0a711 |
---|---|
1 import os | |
2 import subprocess | |
3 | |
4 class Source(object): | |
5 """abstract base class for VCS source""" | |
6 def __init__(self, uri, srcdir=None): | |
7 self.uri = uri | |
8 self.srcdir = srcdir or os.getcwd() | |
9 | |
10 @classmethod | |
11 def directory_name(cls, uri): | |
12 """return relative directory name from self.uri""" | |
13 return uri.rstrip('/').split('/')[-1] | |
14 | |
15 def directory(self): | |
16 return os.path.join(self.srcdir, self.directory_name(self.uri)) | |
17 | |
18 class HgSource(Source): | |
19 """mercurial source""" | |
20 | |
21 def update(self): | |
22 """updates a checkout or does one if it does not exist""" | |
23 | |
24 class GitSource(Source): | |
25 | |
26 @classmethod | |
27 def directory_name(cls, uri): | |
28 ext = '.git' | |
29 if uri.endswith(uri): | |
30 uri = uri[:-len(ext)] | |
31 return Source.directory_name(uri) | |
32 | |
33 if __name__ == '__main__': | |
34 source = HgSource('http://k0s.org/hg/pyloader') | |
35 print source.directory() | |
36 source = GitSource('git://github.com/mozilla/toolbox.git') | |
37 print source.directory() |