wip

Author: carlosmn

pygit2/blame.py

@@ -0,0 +1,162 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright 2010-2014 The pygit2 contributors
+#
+# This file is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2,
+# as published by the Free Software Foundation.
+#
+# In addition to the permissions in the GNU General Public License,
+# the authors give you unlimited permission to link the compiled
+# version of this file into combinations with other programs,
+# and to distribute those combinations without any restriction
+# coming from the use of this file.  (The General Public License
+# restrictions do apply in other respects; for example, they cover
+# modification of the file, and distribution when not linked into
+# a combined executable.)
+#
+# This file is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; see the file COPYING.  If not, write to
+# the Free Software Foundation, 51 Franklin Street, Fifth Floor,
+# Boston, MA 02110-1301, USA.
+
+# Import from the future
+from __future__ import absolute_import, unicode_literals
+
+# Import from pygit2
+from .errors import check_error
+from .ffi import ffi, C
+from .utils import to_bytes, is_string, to_str
+from _pygit2 import Signature, Oid
+
+def wrap_signature(csig):
+    if not csig:
+        return None
+
+    return Signature(ffi.string(csig.name).decode('utf-8'),
+                     ffi.string(csig.email).decode('utf-8'),
+                     csig.when.time, csig.when.offset, 'utf-8')
+
+class BlameHunk(object):
+
+    @classmethod
+    def _from_c(cls, blame, ptr):
+        hunk = cls.__new__(cls)
+        hunk._blame = blame
+        hunk._hunk = ptr
+        return hunk
+
+    @property
+    def lines_in_hunk(self):
+        """Number of lines"""
+        return self._hunk.lines_in_hunk
+
+    @property
+    def boundary(self):
+        """Tracked to a boundary commit"""
+        
+        # Casting directly to bool via cffi does not seem to work
+        return int(ffi.cast('int', self._hunk.boundary)) != 0
+
+    @property
+    def final_start_line_number(self):
+        """Final start line number"""
+        return self._hunk.final_start_line_number
+
+    @property
+    def final_committer(self):
+        """Final committer"""
+        return wrap_signature(self._hunk.final_signature)
+
+    @property
+    def final_commit_id(self):
+        return Oid(raw=bytes(ffi.buffer(ffi.addressof(self._hunk, 'final_commit_id'))[:]))
+
+    @property
+    def orig_start_line_number(self):
+        """Origin start line number"""
+        return self._hunk.orig_start_line_number
+
+    @property
+    def orig_committer(self):
+        """Original committer"""
+        return wrap_signature(self._hunk.orig_signature)
+
+    @property
+    def orig_commit_id(self):
+        return Oid(raw=bytes(ffi.buffer(ffi.addressof(self._hunk, 'orig_commit_id'))[:]))
+
+    @property
+    def orig_path(self):
+        """Original path"""
+        path = self._hunk.orig_path
+        if not path:
+            return None
+
+        return ffi.string(path).decode()
+
+
+class Blame(object):
+                  
+    @classmethod
+    def _from_c(cls, repo, ptr):
+        blame = cls.__new__(cls)
+        blame._repo = repo
+        blame._blame = ptr
+        return blame
+
+    def __del__(self):
+        C.git_blame_free(self._blame)
+
+    def __len__(self):
+        return C.git_blame_get_hunk_count(self._blame)
+
+    def __getitem__(self, index):
+        chunk = C.git_blame_get_hunk_byindex(self._blame, index)
+        if not chunk:
+            raise IndexError
+
+        return BlameHunk._from_c(self, chunk)
+
+    def for_line(self, line_no):
+        """for_line(line_no) -> BlameHunk
+
+        Returns the blame hunk data for a given line given its number
+        in the current Blame.
+
+        Arguments:
+
+        line_no
+            Line number, starts at 1.
+        """
+        if line_no < 0:
+            raise IndexError
+
+        chunk = C.git_blame_get_hunk_byline(self._blame, line_no)
+        if not chunk:
+            raise IndexError
+
+        return BlameHunk._from_c(self, chunk)
+        
+class BlameIterator(object):
+    def __init__(self, blame):
+        self._count = len(blame)
+        self._index = 0
+        self._blame = blame
+
+    def __next__(self):
+        if self._index >= self._count:
+            raise StopIteration
+
+        hunk = self._blame[self._blame]
+        self._index += 1
+
+        return hunk
+
+    def next(self):
+        return self.__next__()