Package translate :: Package storage :: Package versioncontrol :: Module hg
[hide private]
[frames] | no frames]

Source Code for Module translate.storage.versioncontrol.hg

  1  #!/usr/bin/env python 
  2  # -*- coding: utf-8 -*- 
  3  #  
  4  # Copyright 2004-2008 Zuza Software Foundation 
  5  #  
  6  # This file is part of translate. 
  7  # 
  8  # translate is free software; you can redistribute it and/or modify 
  9  # it under the terms of the GNU General Public License as published by 
 10  # the Free Software Foundation; either version 2 of the License, or 
 11  # (at your option) any later version. 
 12  #  
 13  # translate is distributed in the hope that it will be useful, 
 14  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
 15  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
 16  # GNU General Public License for more details. 
 17  # 
 18  # You should have received a copy of the GNU General Public License 
 19  # along with translate; if not, write to the Free Software 
 20  # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
 21   
 22   
 23  from translate.storage.versioncontrol import run_command 
 24  from translate.storage.versioncontrol import GenericRevisionControlSystem 
 25   
 26   
27 -def is_available():
28 """check if hg is installed""" 29 exitcode, output, error = run_command(["hg", "--version"]) 30 return exitcode == 0
31
32 -def get_version():
33 """return a tuple of (major, minor) for the installed bazaar client""" 34 import re 35 command = ["hg", "--version"] 36 exitcode, output, error = run_command(command) 37 if exitcode == 0: 38 version_line = output.splitlines()[0] 39 version_match = re.search(r"\d+\.\d+", version_line) 40 if version_match: 41 major, minor = version_match.group().split(".") 42 if (major.isdigit() and minor.isdigit()): 43 return (int(major), int(minor)) 44 # if anything broke before, then we return the invalid version number 45 return (0, 0)
46 47
48 -class hg(GenericRevisionControlSystem):
49 """Class to manage items under revision control of mercurial.""" 50 51 RCS_METADIR = ".hg" 52 SCAN_PARENTS = True 53
54 - def update(self, revision=None):
55 """Does a clean update of the given path 56 57 @param revision: ignored for hg 58 """ 59 # revert local changes (avoids conflicts) 60 command = ["hg", "-R", self.root_dir, "revert", 61 "--all", self.location_abs] 62 exitcode, output_revert, error = run_command(command) 63 if exitcode != 0: 64 raise IOError("[Mercurial] error running '%s': %s" % (command, error)) 65 # pull new patches 66 command = ["hg", "-R", self.root_dir, "pull"] 67 exitcode, output_pull, error = run_command(command) 68 if exitcode != 0: 69 raise IOError("[Mercurial] error running '%s': %s" % (command, error)) 70 # update working directory 71 command = ["hg", "-R", self.root_dir, "update"] 72 exitcode, output_update, error = run_command(command) 73 if exitcode != 0: 74 raise IOError("[Mercurial] error running '%s': %s" % (command, error)) 75 return output_revert + output_pull + output_update
76
77 - def commit(self, message=None, author=None):
78 """Commits the file and supplies the given commit message if present""" 79 if message is None: 80 message = "" 81 # commit changes 82 command = ["hg", "-R", self.root_dir, "commit", "-m", message] 83 # add the 'author' argument, if it was given (only supported since v1.0) 84 if author and (get_version() >= (1, 0)): 85 command.extend(["--user", author]) 86 # the location is the last argument 87 command.append(self.location_abs) 88 exitcode, output_commit, error = run_command(command) 89 if exitcode != 0: 90 raise IOError("[Mercurial] Error running '%s': %s" \ 91 % (command, error)) 92 # push changes 93 command = ["hg", "-R", self.root_dir, "push"] 94 exitcode, output_push, error = run_command(command) 95 if exitcode != 0: 96 raise IOError("[Mercurial] Error running '%s': %s" \ 97 % (command, error)) 98 return output_commit + output_push
99
100 - def getcleanfile(self, revision=None):
101 """Get a clean version of a file from the hg repository""" 102 # run hg cat 103 command = ["hg", "-R", self.root_dir, "cat", 104 self.location_abs] 105 exitcode, output, error = run_command(command) 106 if exitcode != 0: 107 raise IOError("[Mercurial] Error running '%s': %s" \ 108 % (command, error)) 109 return output
110