def files(self):
diffstat = PatchOps.get_diffstat(self.body())
f = []
for line in diffstat.splitlines():
m = re.search("#? (\S+) \| ", line)
if m:
f.append(m.group(1))
if not f:
return None
return f
def add_diffstat(self):
for line in self.message.get_payload().splitlines():
if re.search("[0-9]+ files? changed, [0-9]+ insertion", line):
return
diffstat = PatchOps.get_diffstat(self.body())
text = ""
switched = False
need_sep = True
body = ""
for line in self.header().splitlines():
if re.match("^---$", line) and not switched:
need_sep = False
if need_sep:
diffstat = "---\n" + diffstat
else:
diffstat = "\n" + diffstat
diffstat += "\n"
header = self.header().rstrip() + "\n"
self.message.set_payload(header + diffstat + self.body())
