Mercurial > hg > buttercup

annotate buttercup/source.py @ 27:901755a65249

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
start stubbing checkout
author Jeff Hammel <jhammel@mozilla.com>
date Mon, 30 May 2011 21:21:42 -0700
parents 8179c05d7692
children 6089d6ec745a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
18
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
1 import os
20
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
2 try:
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
3 from subprocess import check_call as call
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
4 except:
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
5 from subprocess import call
18
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
6
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
7 class Source(object):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
8 """abstract base class for VCS source"""
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
9 def __init__(self, uri, srcdir=None):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
10 self.uri = uri
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
11 self.srcdir = srcdir or os.getcwd()
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
12
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
13 @classmethod
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
14 def directory_name(cls, uri):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
15 """return relative directory name from self.uri"""
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
16 return uri.rstrip('/').split('/')[-1]
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
17
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
18 def directory(self):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
19 return os.path.join(self.srcdir, self.directory_name(self.uri))
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
20
26
8179c05d7692 note ABC
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
21 def update(self):
8179c05d7692 note ABC
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
22 raise NotImplementedError("`Source` is an abstract base class")
27
901755a65249 start stubbing checkout
Jeff Hammel <jhammel@mozilla.com>
parents: 26
diff changeset
23
26
8179c05d7692 note ABC
Jeff Hammel <jhammel@mozilla.com>
parents: 25
diff changeset
24
18
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
25 class HgSource(Source):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
26 """mercurial source"""
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
27
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
28 def update(self):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
29 """updates a checkout or does one if it does not exist"""
20
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
30 if os.path.exists(self.directory()):
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
31 assert os.path.exists(os.path.join(self.directory(), '.hg'))
22
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
32 call(['hg', 'pull'], cwd=self.directory())
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
33 call(['hg', 'update'], cwd=self.directory())
20
60ddc49acd5f add partial checkout method to hg
Jeff Hammel <jhammel@mozilla.com>
parents: 19
diff changeset
34 else:
21
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
35 if not os.path.exists(self.srcdir):
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
36 os.makedirs(self.srcdir)
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
37 call(['hg', 'clone', self.uri], cwd=self.srcdir)
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
38 # TODO: add a more intelligent .hg/hgrc
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
39
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
40 __call__ = update
1f54294629f9 finish hg updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 20
diff changeset
41
18
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
42
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
43 class GitSource(Source):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
44
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
45 @classmethod
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
46 def directory_name(cls, uri):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
47 ext = '.git'
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
48 if uri.endswith(uri):
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
49 uri = uri[:-len(ext)]
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
50 return Source.directory_name(uri)
151862a0a711 begin restructuring to something more modular and extensible
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
51
22
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
52 def update(self):
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
53 """updates a checkout or does one if it does not exist"""
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
54 if os.path.exists(self.directory()):
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
55 assert os.path.exists(os.path.join(self.directory(), '.git'))
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
56 call(['git', 'pull'], cwd=self.directory())
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
57 else:
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
58 if not os.path.exists(self.srcdir):
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
59 os.makedirs(self.srcdir)
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
60 call(['git', 'clone', self.uri], cwd=self.srcdir)
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
61 # TODO: add a more intelligent .git/config
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
62
1378e96c65ac git updater method
Jeff Hammel <jhammel@mozilla.com>
parents: 21
diff changeset
63 __call__ = update
23
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
64
25
a54543838686 add test for source demunging
Jeff Hammel <jhammel@mozilla.com>
parents: 23
diff changeset
65 def sources(source_dict, **kwargs):
23
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
66 retval = []
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
67 for repo_type, sources in source_dict.items():
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
68 if repo_type == 'hg':
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
69 retval.extend([HgSource(source, **kwargs)
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
70 for source in sources])
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
71 elif repo_type == 'git':
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
72 retval.extend([GitSource(source, **kwargs)
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
73 for source in sources])
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
74 else:
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
75 raise AssertionError("Source type unknown: %s" % repo_type)
425a3c73ac1d add a convenience method for VCS sources
Jeff Hammel <jhammel@mozilla.com>
parents: 22
diff changeset
76 return retval