1
# Copyright (C) 2005-2009, 2011-2013 Aaron Bentley <aaron@aaronbentley.com>
2
# Copyright (C) 2007 John Arbash Meinel
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
from contextlib import contextmanager
20
from bzrlib import urlutils
21
from bzrlib.errors import (
26
from bzrlib.bzrdir import BzrDir
27
from bzrlib.transport import get_transport
31
def read_locked(lockable):
32
"""Read-lock a tree, branch or repository in this context."""
40
def short_committer(committer):
41
new_committer = re.sub('<.*>', '', committer).strip(' ')
42
if len(new_committer) < 2:
48
"""Screen-scrape Apache listings"""
49
apache_dir = '<img border="0" src="/icons/folder.gif" alt="[dir]">'\
52
t._remote_path = lambda x: t.base
57
expr = re.compile('<a[^>]*href="([^>]*)\/"[^>]*>', flags=re.I)
59
match = expr.search(line)
63
if url.startswith('http://') or url.startswith('/') or '../' in url:
71
def is_inside(branch):
72
return bool(branch.base.startswith(t.base))
74
if t.base.startswith('http://'):
77
branch = bzrdir.open_branch()
82
except NotBranchError:
84
return [b for b in BzrDir.find_bzrdirs(t, list_current=apache_ls,
85
evaluate=evaluate) if b is not None]
86
elif not t.listable():
87
raise BzrCommandError("Can't list this type of location.")
88
return [b for b in BzrDir.find_branches(t) if is_inside(b)]
91
def evaluate_branch_tree(bzrdir):
93
tree, branch = bzrdir._get_tree_branch()
94
except NotBranchError:
97
return True, (branch, tree)
100
def iter_branch_tree(t, lister=None):
101
return (x for x in BzrDir.find_bzrdirs(t, evaluate=evaluate_branch_tree,
102
list_current=lister) if x is not None)
105
def open_from_url(location):
106
location = urlutils.normalize_url(location)
107
dirname, basename = urlutils.split(location)
108
if location.endswith('/') and not basename.endswith('/'):
110
return get_transport(dirname).get(basename)
9
dirname = tempfile.mkdtemp("temp-branch")
10
return bzrlib.Branch(dirname, init=True)
13
shutil.rmtree(br.base)
15
def is_clean(cur_branch):
17
Return true if no files are modifed or unknown
18
>>> br = temp_branch()
21
>>> fooname = os.path.join(br.base, "foo")
22
>>> file(fooname, "wb").write("bar")
25
>>> bzrlib.add.smart_add([fooname])
28
>>> br.commit("added file")
33
old_tree = cur_branch.basis_tree()
34
new_tree = cur_branch.working_tree()
35
for path, file_class, kind, file_id in new_tree.list_files():
38
delta = bzrlib.compare_trees(old_tree, new_tree, want_unchanged=False)
39
if len(delta.added) > 0 or len(delta.removed) > 0 or \
40
len(delta.modified) > 0:
44
def set_pull_data(br, location, rev_id):
45
pull_file = file (br.controlfilename("x-pull-data"), "wb")
46
pull_file.write("%s\n%s\n" % (location, rev_id))
48
def get_pull_data(br):
50
>>> br = temp_branch()
53
>>> set_pull_data(br, 'http://somewhere', '888-777')
55
('http://somewhere', '888-777')
58
filename = br.controlfilename("x-pull-data")
59
if not os.path.exists(filename):
61
pull_file = file (filename, "rb")
62
location, rev_id = [f.rstrip('\n') for f in pull_file]
63
return location, rev_id
65
def set_push_data(br, location):
66
push_file = file (br.controlfilename("x-push-data"), "wb")
67
push_file.write("%s\n" % location)
69
def get_push_data(br):
71
>>> br = temp_branch()
72
>>> get_push_data(br) is None
74
>>> set_push_data(br, 'http://somewhere')
79
filename = br.controlfilename("x-push-data")
80
if not os.path.exists(filename):
82
push_file = file (filename, "rb")
83
(location,) = [f.rstrip('\n') for f in push_file]
87
>>> shell_escape('hello')
90
def shell_escape(arg):
91
return "".join(['\\'+c for c in arg])
93
def safe_system(args):
95
>>> real_system = os.system
96
>>> os.system = sys.stdout.write
97
>>> safe_system(['a', 'b', 'cd'])
99
>>> os.system = real_system
101
arg_str = " ".join([shell_escape(a) for a in args])
102
return os.system(arg_str)
104
def rsync(source, target, ssh=False, exclude_globs=()):
106
>>> real_system = os.system
107
>>> os.system = sys.stdout.write
109
\\r\\s\\y\\n\\c \\-\\a\\v \\-\\-\\d\\e\\l\\e\\t\\e \\a \\b
110
>>> rsync("a", "b", exclude_globs=("*.py",))
111
\\r\\s\\y\\n\\c \\-\\a\\v \\-\\-\\d\\e\\l\\e\\t\\e\
112
\\-\\-\\e\\x\\c\\l\\u\\d\\e \\*\\.\\p\\y \\a \\b
113
>>> os.system = real_system
115
cmd = ["rsync", "-av", "--delete"]
117
cmd.extend(('-e', 'ssh'))
118
for exclude in exclude_globs:
119
cmd.extend(('--exclude', exclude))
120
cmd.extend((source, target))
123
exclusions = ('x-push-data', 'x-pull-data')
126
def pull(cur_branch, location=None, overwrite=False):
127
pull_location, pull_revision = get_pull_data(cur_branch)
128
if pull_location is not None:
129
if not overwrite and cur_branch.last_patch() != pull_revision:
130
print "Aborting: This branch has had commits, so pull would lose data."
132
if location is not None:
133
pull_location = location
134
if not pull_location.endswith('/'):
137
if pull_location is None:
138
print "No pull location saved. Please specify one on the command line."
141
if not is_clean(cur_branch):
142
print "Error: This tree has uncommitted changes or unknown (?) files."
145
print "Synchronizing with %s" % pull_location
146
rsync (pull_location, cur_branch.base+'/', exclude_globs=exclusions)
148
set_pull_data(cur_branch, pull_location, cur_branch.last_patch())
151
def push(cur_branch, location=None):
152
push_location = get_push_data(cur_branch)
153
if location is not None:
154
if not location.endswith('/'):
156
push_location = location
158
if push_location is None:
159
print "No push location saved. Please specify one on the command line."
162
if not is_clean(cur_branch):
163
print "Error: This tree has uncommitted changes or unknown (?) files."
166
print "Pushing to %s" % push_location
167
rsync(cur_branch.base+'/', push_location, ssh=True,
168
exclude_globs=exclusions)
170
set_push_data(cur_branch, push_location)
added
removed
scriptlib.py
