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)
174
result = doctest.testmod()
177
print "All tests passed"
179
print "No tests to run"
180
if __name__ == "__main__":
added
removed
bzrtools.py
