rainbowCoder/diff.py at master · ManojRayudu/rainbowCoder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import re
from git import *

class PostDiff(object):
  git_path = ".git"

  def __init__(self):
    self.repo = Repo(self.git_path)
    self.origin = self.repo.remotes.origin

  def _cur_local_hexsha(self):
    hcommit = self.repo.head.commit
    return hcommit.hexsha

  def _remote_idiff(self):
    hcommit = self.repo.head.commit
    self.origin.pull()
    local_hexsha = self._cur_local_hexsha()
    return hcommit.diff(local_hexsha)


  def diff_last_time(self, md_file):
    re_s = "^Date:(.*)$"
    patt = re.compile(re_s)

    log = self.repo.git.log(md_file)
    date = re.search(re_s, log, re.M).group(0)
    return patt.match(date).groups()[0]


  def _diff_(self, idiff, change_type):
    ret = []
    for v in idiff.iter_change_type(change_type):
      file_name = None
      if v.a_blob:
        file_name = v.a_blob.path
      elif v.b_blob:
        file_name = v.b_blob.path
      else:
        assert(False)

      ret.append({
          "file_name": file_name,
          "change_type": change_type
        })
    return ret

  def diff_list(self):
    ret = []
    idiff = self._remote_idiff()

    ret.extend(self._diff_(idiff, 'A')) # insert add file
    ret.extend(self._diff_(idiff, 'M')) # insert modify file
    return ret