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