git-branch-deleter

#!/usr/bin/env python3

# MIT License
#
# Copyright (c) 2019 Jussi Lind <jussi.lind@iki.fi>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
#

import argparse
import signal
import subprocess
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
from gi.repository import GLib


class Application:

    class MainWindow(Gtk.Window):

        def __init__(self, app):
            Gtk.Window.__init__(self, title="Git Branch Deleter")

            self._app = app
            self._toggleColumnIndex = 2
            self.set_default_size(800, 600)
            self.set_border_width(5)

            sw = Gtk.ScrolledWindow()
            sw.set_border_width(10)
            sw.set_hexpand(True)
            sw.set_vexpand(True)

            self.applyButton = Gtk.Button.new_with_mnemonic("_Apply")
            self.applyButton.connect("clicked", self.onApplyClicked)
            self.applyButton.set_sensitive(False)

            self.listStore = Gtk.ListStore(str, str, bool)
            self.updateListStore()

            treeView = Gtk.TreeView(model=self.listStore)

            branchNameRendererText = Gtk.CellRendererText()
            branchNameText = Gtk.TreeViewColumn("Local Branch", branchNameRendererText, text=0)
            treeView.append_column(branchNameText)

            branchUseRendererText = Gtk.CellRendererText()
            branchUseText = Gtk.TreeViewColumn("Last Time Used", branchUseRendererText, text=1)
            treeView.append_column(branchUseText)

            rendererToggle = Gtk.CellRendererToggle()
            rendererToggle.connect("toggled", self.onCellToggled)
            columnToggle = Gtk.TreeViewColumn("Delete", rendererToggle, active=self._toggleColumnIndex)
            treeView.append_column(columnToggle)

            grid = Gtk.Grid()
            sw.add(treeView)
            grid.attach(sw, 0, 0, 12, 6)
            grid.attach(self.applyButton, 0, 13, 12, 1)
            self.add(grid)

        def atLeastOneChecked(self):
            for toggle in self.listStore:
                if toggle[self._toggleColumnIndex]:
                    return True
            return False

        def onCellToggled(self, widget, path):
            self.listStore[path][self._toggleColumnIndex] = not self.listStore[path][self._toggleColumnIndex]
            self.updateApplyButton()

        def onApplyClicked(self, button):
            dialog = Gtk.MessageDialog(parent=self, flags=0, message_type=Gtk.MessageType.QUESTION, buttons=Gtk.ButtonsType.YES_NO, text="Confirm Deletion")
            dialog.format_secondary_text("Delete selected branches?")
            response = dialog.run()
            if response == Gtk.ResponseType.YES:
                self.deleteSelectedBranches()
            dialog.destroy()

        def getSelectedBranches(self):
            branches = []
            for toggle in self.listStore:
                if toggle[self._toggleColumnIndex]:
                    branches.append(toggle[0])
            return branches

        def deleteSelectedBranches(self):
            for branch in self.getSelectedBranches():
                self._app.deleteBranch(branch)
            self.updateListStore()

        def updateListStore(self):
            self.listStore.clear()
            for branch in self._app.getBranches():
                self.listStore.append([branch, self._app.getBranchUse(branch), False])
            self.updateApplyButton()

        def updateApplyButton(self):
            self.applyButton.set_sensitive(self.atLeastOneChecked())

    def __init__(self, useD=False):
        self._useD = useD
        self._mainWindow = Application.MainWindow(self)
        self._mainWindow.set_position(Gtk.WindowPosition.CENTER)
        self._mainWindow.connect("destroy", Gtk.main_quit)

    def deleteBranch(self, branch):
        try:
            subprocess.run(["git", "branch", "-D" if self._useD else "-d", branch], check=True)
        except Exception as e:
            dialog = Gtk.MessageDialog(parent=self._mainWindow, flags=0, message_type=Gtk.MessageType.ERROR, buttons=Gtk.ButtonsType.OK, text="Error")
            dialog.format_secondary_text("Failed to delete branch '{0}': {1}".format(branch, e))
            dialog.run()
            dialog.destroy()

    def getBranches(self):
        try:
            def filterDeletableBranches(branch):
                return "*" not in branch and not branch.isspace() and len(branch) > 0

            output = subprocess.check_output(["git", "branch", "--list", "--no-color"], universal_newlines=True)
            branches = output.split('\n')
            branches = [x.strip() for x in branches]
            return filter(filterDeletableBranches, branches)
        except Exception:
            return []

    def getBranchUse(self, branch):
        try:
            output = subprocess.check_output(["git", "log", branch, "--pretty=format:\"%cr\""], universal_newlines=True)
            return output.split('\n')[0].strip('"')
        except Exception:
            return ""

    def run(self):
        self._mainWindow.show_all()
        Gtk.main()


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-D', action='store_true', help='Delete branches with -D instead of -d')
    args = parser.parse_args()
    app = Application(args.D)
    GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, signal.SIGINT, Gtk.main_quit)
    app.run()


if __name__ == '__main__':
    main()