File indexing completed on 2024-04-14 05:34:31

 """
 Module containing classes for listing merge requests
 """
 
 # SPDX-FileCopyrightText: 2020 Jonah Brüchert <jbb@kaidan.im>
 #
 # SPDX-License-Identifier: GPL-2.0-or-later
 import argparse
 from typing import List
 
 from gitlab.v4.objects import ProjectMergeRequest
 
 from lab.repositoryconnection import RepositoryConnection
 from lab.utils import TextFormatting
 from lab.table import Table
 
 
 def parser(
     subparsers: argparse._SubParsersAction,  # pylint: disable=protected-access
 ) -> argparse.ArgumentParser:
     """
     Subparser for merge request list command
     :param subparsers: subparsers object from global parser
     :return: merge request list subparser
     """
     lister_parser: argparse.ArgumentParser = subparsers.add_parser(
         "mrs", aliases=["list"], help="List open merge requests"
     )
     lister_parser.add_argument(
         "--project",
         help="Show merge requests of the current project, not of the user",
         action="store_true",
     )
     lister_parser.add_argument(
         "--opened",
         help="Show opened merge requests",
         action="store_true",
     )
     lister_parser.add_argument(
         "--merged",
         help="Show merged merge requests",
         action="store_true",
     )
     lister_parser.add_argument(
         "--closed",
         help="Show closed merge requests",
         action="store_true",
     )
     lister_parser.add_argument(
         "--url",
         help="Show web url of merge requests (default false)",
         action="store_true",
     )
     return lister_parser
 
 
 def run(args: argparse.Namespace) -> None:
     """
     run merge request list command
     :param args: parsed arguments
     """
     lister = MergeRequestList(args.project, args.merged, args.opened, args.closed, args.url)
     lister.print_formatted_list()
 
 
 class MergeRequestList(RepositoryConnection):
     """
     Lists all merge requests of the current repository
     """
 
     for_project: bool = False
     merged: bool = True
     opened: bool = True
     closed: bool = True
     show_url: bool = True
 
     def __init__(  # pylint: disable=too-many-arguments
         self, for_project: bool, merged: bool, opened: bool, closed: bool, show_url: bool
     ) -> None:
         RepositoryConnection.__init__(self)
         self.for_project = for_project
         self.show_url = show_url
 
         if not merged and not opened and not closed:
             return
 
         self.merged = merged
         self.opened = opened
         self.closed = closed
 
     def print_formatted_list(self) -> None:  # pylint: disable=too-many-branches
         """
         prints the list of merge requests to the terminal formatted as a table
         """
         merge_requests: List[ProjectMergeRequest] = []
 
         if self.for_project:
             base = self._remote_project
         else:
             base = self._connection
 
         if self.merged and self.opened and self.closed:
             merge_requests = base.mergerequests.list()
         else:
             if self.merged:
                 merge_requests += base.mergerequests.list(state="merged")
             if self.opened:
                 merge_requests += base.mergerequests.list(state="opened")
             if self.closed:
                 merge_requests += base.mergerequests.list(state="closed")
 
         table = Table()
 
         for merge_request in merge_requests:
             row: List[str] = []
 
             # Show merge request reference or url according to command line options
             if self.show_url:
                 row.append(merge_request.web_url)
             else:
                 row.append(
                     TextFormatting.BOLD + merge_request.references["full"] + TextFormatting.END
                 )
 
             row.append(merge_request.title)
 
             if merge_request.state == "merged":
                 row.append(TextFormatting.GREEN + merge_request.state + TextFormatting.END)
             elif merge_request.state == "opened":
                 row.append(merge_request.state)
             elif merge_request.state == "closed":
                 row.append(TextFormatting.RED + merge_request.state + TextFormatting.END)
 
             table.add_row(row)
 
         table.print()