]> Tony Duckles's Git Repositories (git.nynim.org) - svn2svn.git/blob - svn2svn.py
* Fix changed_path sorting in parse_svn_log_xml()
[svn2svn.git] / svn2svn.py
1 #!/usr/bin/env python
2 """
3 svn2svn.py
4
5 Replicate (replay) changesets from one SVN repository to another:
6 * Maintains full logical history (e.g. uses "svn copy" for renames).
7 * Maintains original commit messages.
8 * Cannot maintain original commit date, but appends original commit date
9 for each commit message: "Date: %d".
10 * Optionally maintain source author info. (Only supported if accessing
11 target SVN repo via file://)
12 * Optionally run an external shell script before each replayed commit
13 to give the ability to dynamically exclude or modify files as part
14 of the replay.
15
16 License: GPLv2, the same as hgsvn.
17 Author: Tony Duckles (https://github.com/tonyduckles/svn2svn)
18 (This is a forked and modified verison of http://code.google.com/p/svn2svn/)
19 """
20
21 import os
22 import sys
23 import time
24 import locale
25 import shutil
26 import select
27 import calendar
28 import traceback
29
30 from optparse import OptionParser
31 from subprocess import Popen, PIPE
32 from datetime import datetime
33 from operator import itemgetter
34
35 try:
36 from xml.etree import cElementTree as ET
37 except ImportError:
38 try:
39 from xml.etree import ElementTree as ET
40 except ImportError:
41 try:
42 import cElementTree as ET
43 except ImportError:
44 from elementtree import ElementTree as ET
45
46 svn_log_args = ['log', '--xml']
47 svn_info_args = ['info', '--xml']
48 svn_checkout_args = ['checkout', '-q']
49 svn_status_args = ['status', '--xml', '-v', '--ignore-externals']
50
51 # Setup debug options
52 debug = False
53 debug_runsvn_timing = False # Display how long each "svn" OS command took to run?
54 # Setup verbosity options
55 runsvn_showcmd = False # Display every "svn" OS command we run?
56 runsvn_showout = False # Display the stdout results from every "svn" OS command we run?
57 svnlog_verbose = True # Display each action + changed-path as we walk the history?
58
59 # define exception class
60 class ExternalCommandFailed(RuntimeError):
61 """
62 An external command failed.
63 """
64
65 def display_error(message, raise_exception = True):
66 """
67 Display error message, then terminate.
68 """
69 print "Error:", message
70 print
71 if raise_exception:
72 raise ExternalCommandFailed
73 else:
74 sys.exit(1)
75
76 # Windows compatibility code by Bill Baxter
77 if os.name == "nt":
78 def find_program(name):
79 """
80 Find the name of the program for Popen.
81 Windows is finnicky about having the complete file name. Popen
82 won't search the %PATH% for you automatically.
83 (Adapted from ctypes.find_library)
84 """
85 # See MSDN for the REAL search order.
86 base, ext = os.path.splitext(name)
87 if ext:
88 exts = [ext]
89 else:
90 exts = ['.bat', '.exe']
91 for directory in os.environ['PATH'].split(os.pathsep):
92 for e in exts:
93 fname = os.path.join(directory, base + e)
94 if os.path.exists(fname):
95 return fname
96 return None
97 else:
98 def find_program(name):
99 """
100 Find the name of the program for Popen.
101 On Unix, popen isn't picky about having absolute paths.
102 """
103 return name
104
105 def shell_quote(s):
106 if os.name == "nt":
107 q = '"'
108 else:
109 q = "'"
110 return q + s.replace('\\', '\\\\').replace("'", "'\"'\"'") + q
111
112 locale_encoding = locale.getpreferredencoding()
113
114 def run_svn(args, fail_if_stderr=False, encoding="utf-8"):
115 """
116 Run svn cmd in PIPE
117 exit if svn cmd failed
118 """
119 def _transform_arg(a):
120 if isinstance(a, unicode):
121 a = a.encode(encoding or locale_encoding)
122 elif not isinstance(a, str):
123 a = str(a)
124 return a
125 t_args = map(_transform_arg, args)
126
127 cmd = find_program("svn")
128 cmd_string = str(" ".join(map(shell_quote, [cmd] + t_args)))
129 if runsvn_showcmd:
130 print "$", "("+os.getcwd()+")", cmd_string
131 if debug_runsvn_timing:
132 time1 = time.time()
133 pipe = Popen([cmd] + t_args, executable=cmd, stdout=PIPE, stderr=PIPE)
134 out, err = pipe.communicate()
135 if debug_runsvn_timing:
136 time2 = time.time()
137 print "(" + str(round(time2-time1,4)) + " elapsed)"
138 if out and runsvn_showout:
139 print out
140 if pipe.returncode != 0 or (fail_if_stderr and err.strip()):
141 display_error("External program failed (return code %d): %s\n%s"
142 % (pipe.returncode, cmd_string, err))
143 return out
144
145 def svn_date_to_timestamp(svn_date):
146 """
147 Parse an SVN date as read from the XML output and
148 return the corresponding timestamp.
149 """
150 # Strip microseconds and timezone (always UTC, hopefully)
151 # XXX there are various ISO datetime parsing routines out there,
152 # cf. http://seehuhn.de/comp/pdate
153 date = svn_date.split('.', 2)[0]
154 time_tuple = time.strptime(date, "%Y-%m-%dT%H:%M:%S")
155 return calendar.timegm(time_tuple)
156
157 def parse_svn_info_xml(xml_string):
158 """
159 Parse the XML output from an "svn info" command and extract
160 useful information as a dict.
161 """
162 d = {}
163 tree = ET.fromstring(xml_string)
164 entry = tree.find('.//entry')
165 if entry:
166 d['url'] = entry.find('url').text
167 d['revision'] = int(entry.get('revision'))
168 d['repos_url'] = tree.find('.//repository/root').text
169 d['last_changed_rev'] = int(tree.find('.//commit').get('revision'))
170 d['kind'] = entry.get('kind')
171 return d
172
173 def parse_svn_log_xml(xml_string):
174 """
175 Parse the XML output from an "svn log" command and extract
176 useful information as a list of dicts (one per log changeset).
177 """
178 l = []
179 tree = ET.fromstring(xml_string)
180 for entry in tree.findall('logentry'):
181 d = {}
182 d['revision'] = int(entry.get('revision'))
183 # Some revisions don't have authors, most notably
184 # the first revision in a repository.
185 author = entry.find('author')
186 d['author'] = author is not None and author.text or None
187 d['date'] = svn_date_to_timestamp(entry.find('date').text)
188 # Some revisions may have empty commit message
189 message = entry.find('msg')
190 message = message is not None and message.text is not None \
191 and message.text.strip() or ""
192 # Replace DOS return '\r\n' and MacOS return '\r' with unix return '\n'
193 d['message'] = message.replace('\r\n', '\n').replace('\n\r', '\n'). \
194 replace('\r', '\n')
195 paths = []
196 for path in entry.findall('.//path'):
197 copyfrom_rev = path.get('copyfrom-rev')
198 if copyfrom_rev:
199 copyfrom_rev = int(copyfrom_rev)
200 paths.append({
201 'path': path.text,
202 'kind': path.get('kind'),
203 'action': path.get('action'),
204 'copyfrom_path': path.get('copyfrom-path'),
205 'copyfrom_revision': copyfrom_rev,
206 })
207 # Need to sort paths (i.e. into hierarchical order), so that process_svn_log_entry()
208 # can process actions in depth-first order.
209 d['changed_paths'] = sorted(paths, key=itemgetter('path'))
210 l.append(d)
211 return l
212
213 def parse_svn_status_xml(xml_string, base_dir=None):
214 """
215 Parse the XML output from an "svn status" command and extract
216 useful info as a list of dicts (one per status entry).
217 """
218 l = []
219 tree = ET.fromstring(xml_string)
220 for entry in tree.findall('.//entry'):
221 d = {}
222 path = entry.get('path')
223 if base_dir is not None:
224 assert path.startswith(base_dir)
225 path = path[len(base_dir):].lstrip('/\\')
226 d['path'] = path
227 wc_status = entry.find('wc-status')
228 if wc_status.get('item') == 'external':
229 d['type'] = 'external'
230 elif wc_status.get('revision') is not None:
231 d['type'] = 'normal'
232 else:
233 d['type'] = 'unversioned'
234 l.append(d)
235 return l
236
237 def get_svn_info(svn_url_or_wc, rev_number=None):
238 """
239 Get SVN information for the given URL or working copy,
240 with an optionally specified revision number.
241 Returns a dict as created by parse_svn_info_xml().
242 """
243 if rev_number is not None:
244 args = [svn_url_or_wc + "@" + str(rev_number)]
245 else:
246 args = [svn_url_or_wc]
247 xml_string = run_svn(svn_info_args + args, fail_if_stderr=True)
248 return parse_svn_info_xml(xml_string)
249
250 def svn_checkout(svn_url, checkout_dir, rev_number=None):
251 """
252 Checkout the given URL at an optional revision number.
253 """
254 args = []
255 if rev_number is not None:
256 args += ['-r', rev_number]
257 args += [svn_url, checkout_dir]
258 return run_svn(svn_checkout_args + args)
259
260 def run_svn_log(svn_url_or_wc, rev_start, rev_end, limit, stop_on_copy=False, get_changed_paths=True):
261 """
262 Fetch up to 'limit' SVN log entries between the given revisions.
263 """
264 if stop_on_copy:
265 args = ['--stop-on-copy']
266 else:
267 args = []
268 url = str(svn_url_or_wc)
269 if rev_start != 'HEAD' and rev_end != 'HEAD':
270 args += ['-r', '%s:%s' % (rev_start, rev_end)]
271 if not "@" in svn_url_or_wc:
272 url += "@" + str(max(rev_start, rev_end))
273 if get_changed_paths:
274 args += ['-v']
275 args += ['--limit', str(limit), url]
276 xml_string = run_svn(svn_log_args + args)
277 return parse_svn_log_xml(xml_string)
278
279 def get_svn_status(svn_wc, flags=None):
280 """
281 Get SVN status information about the given working copy.
282 """
283 # Ensure proper stripping by canonicalizing the path
284 svn_wc = os.path.abspath(svn_wc)
285 args = []
286 if flags:
287 args += [flags]
288 args += [svn_wc]
289 xml_string = run_svn(svn_status_args + args)
290 return parse_svn_status_xml(xml_string, svn_wc)
291
292 def get_one_svn_log_entry(svn_url, rev_start, rev_end, stop_on_copy=False, get_changed_paths=True):
293 """
294 Get the first SVN log entry in the requested revision range.
295 """
296 entries = run_svn_log(svn_url, rev_start, rev_end, 1, stop_on_copy, get_changed_paths)
297 if not entries:
298 display_error("No SVN log for %s between revisions %s and %s" %
299 (svn_url, rev_start, rev_end))
300
301 return entries[0]
302
303 def get_first_svn_log_entry(svn_url, rev_start, rev_end, get_changed_paths=True):
304 """
305 Get the first log entry after/at the given revision number in an SVN branch.
306 By default the revision number is set to 0, which will give you the log
307 entry corresponding to the branch creaction.
308
309 NOTE: to know whether the branch creation corresponds to an SVN import or
310 a copy from another branch, inspect elements of the 'changed_paths' entry
311 in the returned dictionary.
312 """
313 return get_one_svn_log_entry(svn_url, rev_start, rev_end, stop_on_copy=True, get_changed_paths=True)
314
315 def get_last_svn_log_entry(svn_url, rev_start, rev_end, get_changed_paths=True):
316 """
317 Get the last log entry before/at the given revision number in an SVN branch.
318 By default the revision number is set to HEAD, which will give you the log
319 entry corresponding to the latest commit in branch.
320 """
321 return get_one_svn_log_entry(svn_url, rev_end, rev_start, stop_on_copy=True, get_changed_paths=True)
322
323
324 log_duration_threshold = 10.0
325 log_min_chunk_length = 10
326
327 def iter_svn_log_entries(svn_url, first_rev, last_rev):
328 """
329 Iterate over SVN log entries between first_rev and last_rev.
330
331 This function features chunked log fetching so that it isn't too nasty
332 to the SVN server if many entries are requested.
333 """
334 cur_rev = first_rev
335 chunk_length = log_min_chunk_length
336 chunk_interval_factor = 1.0
337 while last_rev == "HEAD" or cur_rev <= last_rev:
338 start_t = time.time()
339 stop_rev = min(last_rev, cur_rev + int(chunk_length * chunk_interval_factor))
340 entries = run_svn_log(svn_url, cur_rev, stop_rev, chunk_length)
341 duration = time.time() - start_t
342 if not entries:
343 if stop_rev == last_rev:
344 break
345 cur_rev = stop_rev + 1
346 chunk_interval_factor *= 2.0
347 continue
348 for e in entries:
349 yield e
350 cur_rev = e['revision'] + 1
351 # Adapt chunk length based on measured request duration
352 if duration < log_duration_threshold:
353 chunk_length = int(chunk_length * 2.0)
354 elif duration > log_duration_threshold * 2:
355 chunk_length = max(log_min_chunk_length, int(chunk_length / 2.0))
356
357 def commit_from_svn_log_entry(entry, files=None, keep_author=False):
358 """
359 Given an SVN log entry and an optional sequence of files, do an svn commit.
360 """
361 # TODO: Run optional external shell hook here, for doing pre-commit filtering
362 # This will use the local timezone for displaying commit times
363 timestamp = int(entry['date'])
364 svn_date = str(datetime.fromtimestamp(timestamp))
365 # Uncomment this one one if you prefer UTC commit times
366 #svn_date = "%d 0" % timestamp
367 if keep_author:
368 options = ["ci", "--force-log", "-m", entry['message'] + "\nDate: " + svn_date, "--username", entry['author']]
369 else:
370 options = ["ci", "--force-log", "-m", entry['message'] + "\nDate: " + svn_date + "\nAuthor: " + entry['author']]
371 if files:
372 options += list(files)
373 print "(Committing source rev #"+str(entry['revision'])+"...)"
374 run_svn(options)
375
376 def in_svn(p):
377 """
378 Check if a given file/folder is being tracked by Subversion.
379 Prior to SVN 1.6, we could "cheat" and look for the existence of ".svn" directories.
380 With SVN 1.7 and beyond, WC-NG means only a single top-level ".svn" at the root of the working-copy.
381 Use "svn status" to check the status of the file/folder.
382 """
383 # TODO: Is there a better way to do this?
384 entries = get_svn_status(p)
385 if not entries:
386 return False
387 d = entries[0]
388 return (d['type'] == 'normal')
389
390 def find_svn_ancestors(source_repos_url, source_base, source_offset, copyfrom_path, copyfrom_rev):
391 """
392 Given a copy-from path (copyfrom_path), walk the SVN history backwards to inspect
393 the ancestory of that path. Build a collection of copyfrom_path+revision pairs
394 for each of the branch-copies since the initial branch-creation. If we find a
395 copyfrom_path which source_base is a substring match of (e.g. we crawled back to
396 the initial branch-copy from trunk), then return the collection of ancestor paths.
397 Otherwise, copyfrom_path has no ancestory compared to source_base.
398
399 This is useful when comparing "trunk" vs. "branch" paths, to handle cases where a
400 file/folder was renamed in a branch and then that branch was merged back to trunk.
401
402 PARAMETERS:
403 * source_repos_url = Full URL to root of repository, e.g. 'file:///path/to/repos'
404 * source_base = e.g. '/trunk'
405 * source_offset = e.g. 'projectA/file1.txt'
406 * copyfrom_path = e.g. '/branches/bug123/projectA/file1.txt'
407 """
408
409 done = False
410 working_path = copyfrom_path
411 working_base = copyfrom_path[:-len(source_offset)].rstrip('/')
412 working_offset = source_offset.strip('/')
413 working_rev = copyfrom_rev
414 ancestors = [{'path': [working_base, working_offset], 'revision': working_rev}]
415 while not done:
416 # Get the first "svn log" entry for this path (relative to @rev)
417 #working_path = working_base + "/" + working_offset
418 if debug:
419 print ">> find_svn_ancestors: " + source_repos_url + working_path + "@" + str(working_rev) + \
420 " (" + working_base + " " + working_offset + ")"
421 log_entry = get_first_svn_log_entry(source_repos_url + working_path + "@" + str(working_rev), 1, str(working_rev), True)
422 if not log_entry:
423 done = True
424 # Find the action for our working_path in this revision
425 for d in log_entry['changed_paths']:
426 path = d['path']
427 if not path in working_path:
428 continue
429 # Check action-type for this file
430 action = d['action']
431 if action not in 'MARD':
432 display_error("In SVN rev. %d: action '%s' not supported. \
433 Please report a bug!" % (log_entry['revision'], action))
434 if debug:
435 debug_desc = ": " + action + " " + path
436 if d['copyfrom_path']:
437 debug_desc += " (from " + d['copyfrom_path'] + "@" + str(d['copyfrom_revision']) + ")"
438 print debug_desc
439
440 if action == 'R':
441 # If file/folder was replaced, it has no ancestor
442 return []
443 if action == 'D':
444 # If file/folder was deleted, it has no ancestor
445 return []
446 if action == 'A':
447 # If file/folder was added but not a copy, it has no ancestor
448 if not d['copyfrom_path']:
449 return []
450 # Else, file/folder was added and is a copy, so check ancestors
451 path_old = d['copyfrom_path']
452 working_path = working_path.replace(path, path_old)
453 if working_base in working_path:
454 # If the new and old working_path share the same working_base, just need to update working_offset.
455 working_offset = working_path[len(working_base)+1:]
456 else:
457 # Else, assume that working_base has changed but working_offset is the same, e.g. a re-branch.
458 # TODO: Is this a safe assumption?!
459 working_base = working_path[:-len(working_offset)].rstrip('/')
460 working_rev = d['copyfrom_revision']
461 if debug:
462 print ">> find_svn_ancestors: copy-from: " + working_base + " " + working_offset + "@" + str(working_rev)
463 ancestors.append({'path': [working_base, working_offset], 'revision': working_rev})
464 # If we found a copy-from case which matches our source_base, we're done
465 if (path_old == source_base) or (path_old.startswith(source_base + "/")):
466 return ancestors
467 # Else, follow the copy and keep on searching
468 break
469 return None
470
471 def replay_svn_ancestors(ancestors, source_repos_url, source_url, target_url):
472 """
473 Given an array of ancestor info (find_svn_ancestors), replay the history
474 to correctly track renames ("svn copy/move") across branch-merges.
475
476 For example, consider a sequence of events like this:
477 1. svn copy /trunk /branches/fix1
478 2. (Make some changes on /branches/fix1)
479 3. svn copy /branches/fix1/Proj1 /branches/fix1/Proj2 " Rename folder
480 4. svn copy /branches/fix1/Proj2/file1.txt /branches/fix1/Proj2/file2.txt " Rename file inside renamed folder
481 5. svn co /trunk && svn merge /branches/fix1
482 After the merge and commit, "svn log -v" with show a delete of /trunk/Proj1
483 and and add of /trunk/Proj2 comp-from /branches/fix1/Proj2. If we were just
484 to do a straight "svn export+add" based on the /branches/fix1/Proj2 folder,
485 we'd lose the logical history that Proj2/file2.txt is really a descendant
486 of Proj1/file1.txt.
487
488 'source_repos_url' is the full URL to the root of the source repository.
489 'ancestors' is the array returned by find_svn_ancestors() with the final
490 destination info appended to it by process_svn_log_entry().
491 'dest_path'
492 """
493 # Ignore ancestors[0], which is the original (pre-branch-copy) trunk path
494 # Ignore ancestors[1], which is the original branch-creation commit
495 # Ignore ancestors[n], which is the final commit back to trunk
496 for idx in range(1, len(ancestors)-1):
497 ancestor = ancestors[idx]
498 source_base = ancestor['path'][0]
499 source_offset = ancestor['path'][1]
500 source_path = source_base + "/" + source_offset
501 source_rev = ancestor['revision']
502 source_rev_next = ancestors[idx+1]['revision']
503 # Do a "svn log" on the _parent_ directory of source_path, since trying to get log info
504 # for the "old path" on the revision where the copy/move happened will fail.
505 if "/" in source_path:
506 p_source_path = source_path[:source_path.rindex('/')]
507 else:
508 p_source_path = ""
509 if debug:
510 print ">> replay_svn_ancestors: ["+str(idx)+"]" + source_path+"@"+str(source_rev) + " ["+p_source_path+"@"+str(source_rev)+":"+str(source_rev_next-1)+"]"
511 it_log_entries = iter_svn_log_entries(source_repos_url+p_source_path, source_rev, source_rev_next-1)
512 for log_entry in it_log_entries:
513 #print ">> replay_svn_ancestors: log_entry: (" + source_repos_url+source_base + ")"
514 #print log_entry
515 # TODO: Hit a problem case with a rename-situation where the "remove" was committed ahead of the "add (copy)".
516 # Do we maybe need to buffer all the remove's until the end of the entire replay session?
517 # Or can we maybe work around this by passing an explicit rev # into "svn copy"?
518 process_svn_log_entry(log_entry, source_repos_url, source_repos_url+source_base, target_url)
519
520 def process_svn_log_entry(log_entry, source_repos_url, source_url, target_url):
521 """
522 Process SVN changes from the given log entry.
523 Returns array of all the paths in the working-copy that were changed,
524 i.e. the paths which need to be "svn commit".
525
526 'log_entry' is the array structure built by parse_svn_log_xml().
527 'source_repos_url' is the full URL to the root of the source repository.
528 'source_url' is the full URL to the source path in the source repository.
529 'target_url' is the full URL to the target path in the target repository.
530 """
531 # Get the relative offset of source_url based on source_repos_url, e.g. u'/branches/bug123'
532 source_base = source_url[len(source_repos_url):]
533 if debug:
534 print ">> process_svn_log_entry: " + source_url + " (" + source_base + ")"
535
536 svn_rev = log_entry['revision']
537 # Get current target revision, for "svn copy" support
538 dup_info = get_svn_info(target_url)
539 dup_rev = dup_info['revision']
540
541 removed_paths = []
542 modified_paths = []
543 unrelated_paths = []
544 commit_paths = []
545
546 for d in log_entry['changed_paths']:
547 # Get the full path for this changed_path
548 # e.g. u'/branches/bug123/projectA/file1.txt'
549 path = d['path']
550 if not path.startswith(source_base + "/"):
551 # Ignore changed files that are not part of this subdir
552 if path != source_base:
553 print ">> process_svn_log_entry: Unrelated path: " + path + " (" + source_base + ")"
554 unrelated_paths.append(path)
555 continue
556 # Calculate the offset (based on source_base) for this changed_path
557 # e.g. u'projectA/file1.txt'
558 # (path = source_base + "/" + path_offset)
559 path_offset = path[len(source_base):].strip("/")
560 # Get the action for this path
561 action = d['action']
562 if action not in 'MARD':
563 display_error("In SVN rev. %d: action '%s' not supported. \
564 Please report a bug!" % (svn_rev, action))
565
566 # Try to be efficient and keep track of an explicit list of paths in the
567 # working copy that changed. If we commit from the root of the working copy,
568 # then SVN needs to crawl the entire working copy looking for pending changes.
569 # But, if we gather too many paths to commit, then we wipe commit_paths below
570 # and end-up doing a commit at the root of the working-copy.
571 if len (commit_paths) < 100:
572 commit_paths.append(path_offset)
573
574 # Special-handling for replace's
575 is_replace = False
576 if action == 'R':
577 if svnlog_verbose:
578 msg = " " + action + " " + d['path']
579 if d['copyfrom_path']:
580 msg += " (from " + d['copyfrom_path'] + "@" + str(d['copyfrom_revision']) + ")"
581 print msg
582 # If file was "replaced" (deleted then re-added, all in same revision),
583 # then we need to run the "svn rm" first, then change action='A'. This
584 # lets the normal code below handle re-"svn add"'ing the files. This
585 # should replicate the "replace".
586 run_svn(["up", path_offset])
587 run_svn(["remove", "--force", path_offset])
588 action = 'A'
589 is_replace = True
590
591 # Handle all the various action-types
592 # (Handle "add" first, for "svn copy/move" support)
593 if action == 'A':
594 if svnlog_verbose:
595 msg = " " + action + " " + d['path']
596 if d['copyfrom_path']:
597 msg += " (from " + d['copyfrom_path'] + "@" + str(d['copyfrom_revision']) + ")"
598 print msg
599 # Determine where to export from
600 copyfrom_rev = svn_rev
601 copyfrom_path = path
602 svn_copy = False
603 # Handle cases where this "add" was a copy from another URL in the source repos
604 if d['copyfrom_revision']:
605 copyfrom_rev = d['copyfrom_revision']
606 copyfrom_path = d['copyfrom_path']
607 if debug:
608 print ">> process_svn_log_entry: copy-to: " + source_base + " " + path_offset
609 if source_base in copyfrom_path:
610 # If the copy-from path is inside the current working-copy, no need to check ancestry.
611 ancestors = []
612 copyfrom_path = copyfrom_path[len(source_base):].strip("/")
613 if debug:
614 print ">> process_svn_log_entry: Found copy: " + copyfrom_path+"@"+str(copyfrom_rev)
615 svn_copy = True
616 else:
617 ancestors = find_svn_ancestors(source_repos_url, source_base, path_offset,
618 copyfrom_path, copyfrom_rev)
619 if ancestors:
620 # Reverse the list, so that we loop in chronological order
621 ancestors.reverse()
622 # Append the current revision
623 ancestors.append({'path': [source_base, path_offset], 'revision': svn_rev})
624 # ancestors[0] is the original (pre-branch-copy) trunk path.
625 # ancestors[1] is the first commit on the new branch.
626 copyfrom_rev = ancestors[0]['revision']
627 copyfrom_base = ancestors[0]['path'][0]
628 copyfrom_offset = ancestors[0]['path'][1]
629 copyfrom_path = copyfrom_base + copyfrom_offset
630 if debug:
631 print ">> process_svn_log_entry: FOUND PARENT:"
632 for idx in range(0,len(ancestors)):
633 ancestor = ancestors[idx]
634 print " ["+str(idx)+"] " + ancestor['path'][0]+" "+ancestor['path'][1]+"@"+str(ancestor['revision'])
635 #print ">> process_svn_log_entry: copyfrom_path (before): " + copyfrom_path + " source_base: " + source_base + " p: " + p
636 copyfrom_path = copyfrom_path[len(source_base):].strip("/")
637 #print ">> process_svn_log_entry: copyfrom_path (after): " + copyfrom_path
638 svn_copy = True
639 # If this add was a copy-from, do a smart replay of the ancestors' history.
640 # Else just copy/export the files from the source repo and "svn add" them.
641 if svn_copy:
642 if debug:
643 print ">> process_svn_log_entry: svn_copy: copy-from: " + copyfrom_path+"@"+str(copyfrom_rev) + " source_base: "+source_base + " len(ancestors): " + str(len(ancestors))
644 # If we don't have any ancestors, then this is just a straight "svn copy" in the current working-copy.
645 if not ancestors:
646 # ...but not if the target is already tracked, because this might run several times for the same path.
647 # TODO: Is there a better way to avoid recusion bugs? Maybe a collection of processed paths?
648 if not in_svn(path_offset):
649 run_svn(["copy", "-r", dup_rev, target_url+"/"+copyfrom_path+"@"+str(dup_rev), path_offset])
650 else:
651 if d['kind'] == 'dir':
652 # Replay any actions which happened to this folder from the ancestor path(s).
653 replay_svn_ancestors(ancestors, source_repos_url, source_url, target_url)
654 else:
655 # Just do a straight "svn copy" for files. There isn't any kind of "dependent"
656 # history we might need to replay like for folders.
657 run_svn(["copy", "-r", dup_rev, target_url+"/"+copyfrom_path+"@"+str(dup_rev), path_offset])
658 else:
659 # Create (parent) directory if needed
660 if d['kind'] == 'dir':
661 p_path = path_offset
662 else:
663 p_path = os.path.dirname(path_offset).strip() or '.'
664 if not os.path.exists(p_path):
665 os.makedirs(p_path)
666 # Export the entire added tree.
667 run_svn(["export", "--force", "-r", str(copyfrom_rev),
668 source_repos_url + copyfrom_path + "@" + str(copyfrom_rev), path_offset])
669 # TODO: The "no in_svn" condition here is wrong for replace cases.
670 # Added the in_svn condition here originally since "svn export" is recursive
671 # but "svn log" will have an entry for each indiv file, hence we run into a
672 # cannot-re-add-file-which-is-already-added issue.
673 if (not in_svn(path_offset)) or (is_replace):
674 run_svn(["add", "--parents", path_offset])
675 # TODO: Need to copy SVN properties from source repos
676
677 elif action == 'D':
678 # Queue "svn remove" commands, to allow the action == 'A' handling the opportunity
679 # to do smart "svn copy" handling on copy/move/renames.
680 removed_paths.append(path_offset)
681
682 elif action == 'R':
683 # TODO
684 display_error("Internal Error: Handling for action='R' not implemented yet.")
685
686 elif action == 'M':
687 modified_paths.append(path_offset)
688
689 else:
690 display_error("Internal Error: pull_svn_rev: Unhandled 'action' value: '" + action + "'")
691
692 if removed_paths:
693 for r in removed_paths:
694 if svnlog_verbose:
695 print " D " + r
696 # TODO: Is the "svn up" here needed?
697 run_svn(["up", r])
698 run_svn(["remove", "--force", r])
699
700 if modified_paths:
701 for m in modified_paths:
702 if svnlog_verbose:
703 print " M " + m
704 # TODO: Is the "svn up" here needed?
705 run_svn(["up", m])
706 m_url = source_url + "/" + m
707 out = run_svn(["merge", "-c", str(svn_rev), "--non-recursive",
708 "--non-interactive", "--accept=theirs-full",
709 m_url+"@"+str(svn_rev), m])
710
711 if unrelated_paths:
712 print "Unrelated paths: (vs. '" + source_base + "')"
713 print "*", unrelated_paths
714
715 return commit_paths
716
717 def pull_svn_rev(log_entry, source_repos_url, source_url, target_url, keep_author=False):
718 """
719 Pull SVN changes from the given log entry.
720 Returns the new SVN revision.
721 If an exception occurs, it will rollback to revision 'svn_rev - 1'.
722 """
723 ## Get the relative offset of source_url based on source_repos_url, e.g. u'/branches/bug123'
724 #source_base = source_url[len(source_repos_url):]
725
726 svn_rev = log_entry['revision']
727 print "\n(Starting source rev #"+str(svn_rev)+":)"
728 print "r"+str(log_entry['revision']) + " | " + \
729 log_entry['author'] + " | " + \
730 str(datetime.fromtimestamp(int(log_entry['date'])).isoformat(' '))
731 print log_entry['message']
732 print "------------------------------------------------------------------------"
733 commit_paths = process_svn_log_entry(log_entry, source_repos_url, source_url, target_url)
734
735 # If we had too many individual paths to commit, wipe the list and just commit at
736 # the root of the working copy.
737 if len (commit_paths) > 99:
738 commit_paths = []
739
740 # TODO: Use SVN properties to track source URL + rev in the target repo?
741 # This would provide a more reliable resume-support
742 try:
743 commit_from_svn_log_entry(log_entry, commit_paths, keep_author=keep_author)
744 except ExternalCommandFailed:
745 # try to ignore the Properties conflicts on files and dirs
746 # use the copy from original_wc
747 # TODO: Need to re-work this?
748 #has_Conflict = False
749 #for d in log_entry['changed_paths']:
750 # p = d['path']
751 # p = p[len(source_base):].strip("/")
752 # if os.path.isfile(p):
753 # if os.path.isfile(p + ".prej"):
754 # has_Conflict = True
755 # shutil.copy(original_wc + os.sep + p, p)
756 # p2=os.sep + p.replace('_', '__').replace('/', '_') \
757 # + ".prej-" + str(svn_rev)
758 # shutil.move(p + ".prej", os.path.dirname(original_wc) + p2)
759 # w="\n### Properties conflicts ignored:"
760 # print "%s %s, in revision: %s\n" % (w, p, svn_rev)
761 # elif os.path.isdir(p):
762 # if os.path.isfile(p + os.sep + "dir_conflicts.prej"):
763 # has_Conflict = True
764 # p2=os.sep + p.replace('_', '__').replace('/', '_') \
765 # + "_dir__conflicts.prej-" + str(svn_rev)
766 # shutil.move(p + os.sep + "dir_conflicts.prej",
767 # os.path.dirname(original_wc) + p2)
768 # w="\n### Properties conflicts ignored:"
769 # print "%s %s, in revision: %s\n" % (w, p, svn_rev)
770 # out = run_svn(["propget", "svn:ignore",
771 # original_wc + os.sep + p])
772 # if out:
773 # run_svn(["propset", "svn:ignore", out.strip(), p])
774 # out = run_svn(["propget", "svn:externel",
775 # original_wc + os.sep + p])
776 # if out:
777 # run_svn(["propset", "svn:external", out.strip(), p])
778 ## try again
779 #if has_Conflict:
780 # commit_from_svn_log_entry(log_entry, commit_paths, keep_author=keep_author)
781 #else:
782 raise ExternalCommandFailed
783 print "(Finished source rev #"+str(svn_rev)+")"
784
785
786 def main():
787 usage = "Usage: %prog [-a] [-c] [-r SVN rev] <Source SVN URL> <Target SVN URL>"
788 parser = OptionParser(usage)
789 parser.add_option("-a", "--keep-author", action="store_true",
790 dest="keep_author", help="Keep revision Author or not")
791 parser.add_option("-c", "--continue-from-break", action="store_true",
792 dest="cont_from_break",
793 help="Continue from previous break")
794 parser.add_option("-r", "--svn-rev", type="int", dest="svn_rev",
795 help="SVN revision to checkout from")
796 (options, args) = parser.parse_args()
797 if len(args) != 2:
798 display_error("incorrect number of arguments\n\nTry: svn2svn.py --help",
799 False)
800
801 source_url = args.pop(0).rstrip("/")
802 target_url = args.pop(0).rstrip("/")
803 if options.keep_author:
804 keep_author = True
805 else:
806 keep_author = False
807
808 # Find the greatest_rev in the source repo
809 svn_info = get_svn_info(source_url)
810 greatest_rev = svn_info['revision']
811
812 dup_wc = "_dup_wc"
813
814 # if old working copy does not exist, disable continue mode
815 # TODO: Better continue support. Maybe include source repo's rev # in target commit info?
816 if not os.path.exists(dup_wc):
817 options.cont_from_break = False
818
819 if not options.cont_from_break:
820 # Warn if Target SVN URL existed
821 cmd = find_program("svn")
822 pipe = Popen([cmd] + ["list"] + [target_url], executable=cmd,
823 stdout=PIPE, stderr=PIPE)
824 out, err = pipe.communicate()
825 if pipe.returncode == 0:
826 print "Target SVN URL: %s existed!" % target_url
827 if out:
828 print out
829 print "Press 'Enter' to Continue, 'Ctrl + C' to Cancel..."
830 print "(Timeout in 5 seconds)"
831 rfds, wfds, efds = select.select([sys.stdin], [], [], 5)
832
833 # Get log entry for the SVN revision we will check out
834 if options.svn_rev:
835 # If specify a rev, get log entry just before or at rev
836 svn_start_log = get_last_svn_log_entry(source_url, 1, options.svn_rev, False)
837 else:
838 # Otherwise, get log entry of branch creation
839 # TODO: This call is *very* expensive on a repo with lots of revisions.
840 # Even though the call is passing --limit 1, it seems like that limit-filter
841 # is happening after SVN has fetched the full log history.
842 svn_start_log = get_first_svn_log_entry(source_url, 1, greatest_rev, False)
843
844 # This is the revision we will start from for source_url
845 svn_rev = svn_start_log['revision']
846
847 # Check out a working copy of target_url
848 dup_wc = os.path.abspath(dup_wc)
849 if os.path.exists(dup_wc):
850 shutil.rmtree(dup_wc)
851 svn_checkout(target_url, dup_wc)
852 os.chdir(dup_wc)
853
854 # For the initial commit to the target URL, export all the contents from
855 # the source URL at the start-revision.
856 paths = run_svn(["list", "-r", str(svn_rev), source_url+"@"+str(svn_rev)])
857 paths = paths.strip("\n").split("\n")
858 for path in paths:
859 if not path:
860 # Skip null lines
861 break
862 # Directories have a trailing slash in the "svn list" output
863 if path[-1] == "/":
864 path=path.rstrip('/')
865 if not os.path.exists(path):
866 os.makedirs(path)
867 run_svn(["export", "--force", "-r" , str(svn_rev), source_url+"/"+path+"@"+str(svn_rev), path])
868 run_svn(["add", path])
869 commit_from_svn_log_entry(svn_start_log, [], keep_author)
870 else:
871 dup_wc = os.path.abspath(dup_wc)
872 os.chdir(dup_wc)
873 # TODO: Need better resume support. For the time being, expect caller explictly passes in resume revision.
874 svn_rev = options.svn_rev
875 if svn_rev < 1:
876 display_error("Invalid arguments\n\nNeed to pass result rev # (-r) when using continue-mode (-c)", False)
877
878
879 # Get SVN info
880 svn_info = get_svn_info(source_url)
881 # Get the base URL for the source repos, e.g. u'svn://svn.example.com/svn/repo'
882 source_repos_url = svn_info['repos_url']
883
884 # Load SVN log starting from svn_rev + 1
885 it_log_entries = iter_svn_log_entries(source_url, svn_rev + 1, greatest_rev)
886
887 try:
888 for log_entry in it_log_entries:
889 # Replay this revision from source_url into target_url
890 pull_svn_rev(log_entry, source_repos_url, source_url, target_url, keep_author)
891 # Update our target working-copy, to ensure everything says it's at the new HEAD revision
892 run_svn(["up", dup_wc])
893
894 except KeyboardInterrupt:
895 print "\nStopped by user."
896 run_svn(["cleanup"])
897 run_svn(["revert", "--recursive", "."])
898 except:
899 print "\nCommand failed with following error:\n"
900 traceback.print_exc()
901 run_svn(["cleanup"])
902 run_svn(["revert", "--recursive", "."])
903 finally:
904 run_svn(["up"])
905 print "\nFinished!"
906
907
908 if __name__ == "__main__":
909 main()
910
911 # vim:sts=4:sw=4: