Mercurial > hgsubversion
diff tests/fixtures/rsvn.py @ 581:90efea2c19df
Add rsvn.py to test tools
author | Patrick Mezard <pmezard@gmail.com> |
---|---|
date | Tue, 02 Mar 2010 17:06:06 +0100 |
parents | |
children | e9af7eba88db |
line wrap: on
line diff
new file mode 100755 --- /dev/null +++ b/tests/fixtures/rsvn.py @@ -0,0 +1,260 @@ +#!/usr/bin/env python2 + +# LICENSE +# +# Copyright (c) 2004, Francois Beausoleil +# All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions +# are met: +# +# * Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in +# the documentation and/or other materials provided with the +# distribution. +# * Neither the name of the Francois Beausoleil nor the names of its +# contributors may be used to endorse or promote products derived +# from this software without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +import getopt +import sys +import os +import re +import traceback +from svn import core, repos, fs + +VERSION='$Id: rsvn.py 20 2005-09-23 15:08:08Z fbos $' +RCOPY_RE = re.compile('^\s*rcopy\s+(.+)\s+(.+)$') +RMOVE_RE = re.compile('^\s*rmove\s+(.+)\s+(.+)$') +RMKDIR_RE = re.compile('^\s*rmkdir\s+(.+)$') +RDELETE_RE = re.compile('^\s*rdelete\s+(.+)$') +COMMENT_RE = re.compile('(?:^\s*#)|(?:^\s*$)') + +def usage(error=None): + if error: + print 'Error: %s\n' % error + print 'USAGE: %s --message=MESSAGE repos_path [--username=USERNAME]' % ( + os.path.basename(sys.argv[0])) + print '' + print ' --help, -h print this usage message and exit with success' + print ' --version print the version number' + print ' --username=USERNAME Username to execute the commands under' + print ' --message=LOG_MSG Log message to execute the commit with' + print '' + print 'Reads STDIN and parses the following commands, to execute them on the server, ' + print 'all within the same transaction:' + print '' + print ' rcopy SRC DEST Copy the HEAD revision of a file or folder' + print ' rmove SRC DEST Copy + delete the HEAD revision of a file or folder' + print ' rdelete TARGET Deletes something from the repository' + print ' rmkdir TARGET Creates a new folder (must create parents first)' + print ' # Initiates a comment' + print '' +# 12345678901234567890123456789012345678901234567890123456789012345678901234567890 + + +class Transaction: + """Represents a transaction in a Subversion repository + + Transactions are long-lived objects which exist in the repository, + and are used to build an intermediate representation of a new + revision. Once the transaction is committed, the repository + bumps the revision number, and links the new transaction in the + Subversion filesystem.""" + + def __init__(self, repository, rev, username, message, pool, logger=None): + if logger: + self.logger = logger + else: + self.logger = sys.stdout + self.pool = pool + self.rev = rev + + self.fsptr = repos.svn_repos_fs(repository) + self.rev_root = fs.revision_root(self.fsptr, self.rev, + self.pool) + self.txnp = repos.svn_repos_fs_begin_txn_for_commit( + repository, self.rev, username, message, self.pool) + self.txn_root = fs.txn_root(self.txnp, self.pool) + self.log('Base revision %d\n' % rev) + + def commit(self): + values = fs.commit_txn(self.txnp, self.pool) + return values[1] + + def rollback(self): + fs.abort_txn(self.txnp, self.pool) + + def copy(self, src, dest, subpool): + self.log('A + %s\n' % dest) + fs.copy(self.rev_root, src, self.txn_root, dest, subpool) + + def delete(self, entry, subpool): + self.log('D %s\n' % entry) + fs.delete(self.txn_root, entry, subpool) + + def mkdir(self, entry, subpool): + self.log('A %s\n' % entry) + fs.make_dir(self.txn_root, entry, subpool) + + def move(self, src, dest, subpool): + self.copy(src, dest, subpool) + self.delete(src, subpool) + + def log(self, msg): + self.logger.write(msg) + + +class Repository: + """Represents a Subversion repository, and allows common operations + on it.""" + + def __init__(self, repos_path, pool, logger=None): + if logger: + self.logger = logger + else: + self.logger = sys.stdout + self.pool = pool + assert self.pool + + self.repo = repos.svn_repos_open(repos_path, self.pool) + self.fsptr = repos.svn_repos_fs(self.repo) + + def get_youngest(self): + """Returns the youngest revision in the repository.""" + return fs.youngest_rev(self.fsptr, self.pool) + + def begin(self, username, log_msg): + """Initiate a new Transaction""" + return Transaction(self.repo, self.get_youngest(), username, + log_msg, self.pool, self.logger) + + def close(self): + """Close the repository, aborting any uncommitted transactions""" + core.svn_pool_destroy(self.pool) + core.apr_terminate() + + def subpool(self): + """Instantiates a new pool from the master pool""" + return core.svn_pool_create(self.pool) + + def delete_pool(self, pool): + """Deletes the passed-in pool. Returns None, to assign to pool in + caller.""" + core.svn_pool_destroy(pool) + return None + +def rsvn(pool): + log_msg = None + + try: + opts, args = getopt.getopt(sys.argv[1:], 'vh', + ["help", "username=", "message=", "version"]) + except getopt.GetoptError, e: + sys.stderr.write(str(e) + '\n\n') + usage() + sys.exit(1) + + for opt, value in opts: + if opt == '--version': + print '%s version %s' % (os.path.basename(sys.argv[0]), VERSION) + sys.exit(0) + elif opt == '--help' or opt == '-h': + usage() + sys.exit(0) + elif opt == '--username': + username = value + elif opt == '--message': + log_msg = value + + if log_msg == None: + usage('Missing --message argument') + sys.exit(1) + + if len(args) != 1: + usage('Missing repository path argument') + sys.exit(1) + + repos_path = args[0] + print 'Accessing repository at [%s]' % repos_path + + repository = Repository(repos_path, pool) + sub = repository.subpool() + + try: + txn = repository.begin(username, log_msg) + + # Read commands from STDIN + lineno = 0 + for line in sys.stdin: + lineno += 1 + + core.svn_pool_clear(sub) + try: + if COMMENT_RE.search(line): + continue + + match = RCOPY_RE.search(line) + if match: + src = match.group(1) + dest = match.group(2) + txn.copy(src, dest, sub) + continue + + match = RMOVE_RE.search(line) + if match: + src = match.group(1) + dest = match.group(2) + txn.move(src, dest, sub) + continue + + match = RMKDIR_RE.search(line) + if match: + entry = match.group(1) + txn.mkdir(entry, sub) + continue + + match = RDELETE_RE.search(line) + if match: + entry = match.group(1) + txn.delete(entry, sub) + continue + + raise NameError, ('Unknown command [%s] on line %d' % + (line, lineno)) + + except: + sys.stderr.write(('Exception occured while processing line %d:\n' % + lineno)) + etype, value, tb = sys.exc_info() + traceback.print_exception(etype, value, tb, None, sys.stderr) + sys.stderr.write('\n') + txn.rollback() + sys.exit(1) + + new_rev = txn.commit() + print '\nCommitted revision %d.' % new_rev + + finally: + print '\nRepository closed.' + +def main(): + core.run_app(rsvn) + +if __name__ == '__main__': + main()