#!/usr/bin/env python3
import json
from commands.base import BaseCommand
from gitlab import get_mrs_for_branch, get_comments
class Comments(BaseCommand):
width = 5
@staticmethod
def sort_comments(comments):
result = {}
for comment in comments:
author = comment["author"]["username"], comment["author"]["name"]
if author not in result:
result[author] = list()
result[author].append(comment)
return result
def print_comments(comments):
comments = Comments.sort_comments(comments)
for author in comments:
login, name = author
print(f"***** {name} ({login}) *****")
for comment in comments[author]:
if comment["system"]:
continue
if comment["type"] == "DiffNote":
body = comment["body"].replace(
"\n", "\n" + " " * (Comments.width + 2)
)
print(
f"""{f'L{comment["position"]["new_line"]}':>{Comments.width}}: {body}"""
else:
print(f"""[{comment["created_at"]}]\n{comment["body"]}""")
print()
def exec(self, login: str, submission: str):
mrs = get_mrs_for_branch(self.branch)
if not mrs:
return
iid = mrs[0]["iid"]
comments = get_comments(iid)
print(f"# {self.branch}")
self.print_comments(comments)