Git Updater

2011-08-17 17:37:25 +02:00
parent 6735cc9e5e
commit f3c142f531
15 changed files with 1033 additions and 0 deletions
@@ -0,0 +1,76 @@
+# Copyright (c) 2009, Rotem Yaari <vmalloc@gmail.com>
+# 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 organization 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 Rotem Yaari ''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 Rotem Yaari 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 re
+from ref import Ref
+
+class Branch(Ref):
+    def delete(self):
+        raise NotImplementedError()
+    def __repr__(self):
+        return "<branch %s>" % (self.name,)
+class LocalBranch(Branch):
+    def delete(self, force=True):
+        self.repo._executeGitCommandAssertSuccess("git branch -%s %s" % ("D" if force else "d", self.name,))
+    def setRemoteBranch(self, branch):
+        if branch is None:
+            self.repo.config.unsetParameter('branch.%s.remote' % self.name)
+            self.repo.config.unsetParameter('branch.%s.merge' % self.name)
+            return
+        elif not isinstance(branch, RegisteredRemoteBranch):
+            raise ValueError("Remote branch must be a remote branch object (got %r)" % (branch,))
+        self.repo.config.setParameter('branch.%s.remote' % self.name, branch.remote.name)
+        self.repo.config.setParameter('branch.%s.merge' % self.name, 'refs/heads/%s' % branch.name)
+    def getRemoteBranch(self):
+        remote = self.repo.config.getParameter('branch.%s.remote' % self.name)
+        if remote is None:
+            return None
+        remote = self.repo.getRemoteByName(remote)
+        merge = self.repo.config.getParameter('branch.%s.merge' % self.name)
+        merge = re.sub('^refs/heads/', '', merge)
+        remote_branch = remote.getBranchByName(merge)
+        return remote_branch
+        
+class LocalBranchAlias(LocalBranch):
+    def __init__(self, repository, name, dest):
+        super(LocalBranchAlias, self).__init__(repository, name)
+        self.dest = dest
+        
+class RemoteBranch(Branch):
+    pass
+class RegisteredRemoteBranch(RemoteBranch):
+    def __init__(self, repo, remote, name):
+        super(RegisteredRemoteBranch, self).__init__(repo, name)
+        self.remote = remote
+    def getHead(self):
+        return self.repo._getCommitByRefName("%s/%s" % (self.remote.name, self.name))
+    def delete(self):
+        """
+        Deletes the actual branch on the remote repository!
+        """
+        self.repo.push(self.remote, fromBranch="", toBranch=self, force=True)
+    def getNormalizedName(self):
+        return "%s/%s" % (self.remote.name, self.name)
+    def __repr__(self):
+        return "<branch %s on remote %r>" % (self.name, self.remote.name)