ugit/views.py

   1 import os
   2 import time
   3 from PyQt4 import QtCore
   4 from PyQt4.QtGui import QDialog
   5 from PyQt4.QtGui import QMainWindow
   6 from PyQt4.QtGui import QCheckBox
   7 from PyQt4.QtGui import QSplitter
   8 from maingui import Ui_maingui
   9 from outputgui import Ui_outputgui
  10 from optionsgui import Ui_optionsgui
  11 from branchgui import Ui_branchgui
  12 from commitgui import Ui_commitgui
  13 from createbranchgui import Ui_createbranchgui
  14 from pushgui import Ui_pushgui
  15 from syntax import DiffSyntaxHighlighter
  16 from syntax import LogSyntaxHighlighter
  17 import qtutils
  18
  19 class View(Ui_maingui, QMainWindow):
  20         '''The main ugit interface.'''
  21         def __init__(self, parent=None):
  22                 QMainWindow.__init__(self, parent)
  23                 Ui_maingui.__init__(self)
  24                 self.setupUi(self)
  25                 self.staged.setAlternatingRowColors(True)
  26                 self.unstaged.setAlternatingRowColors(True)
  27                 self.set_display = self.display_text.setText
  28                 self.set_info = self.displayLabel.setText
  29                 self.action_undo = self.commitmsg.undo
  30                 self.action_redo = self.commitmsg.redo
  31                 self.action_paste = self.commitmsg.paste
  32                 self.action_select_all = self.commitmsg.selectAll
  33
  34                 # Handle automatically setting the horizontal/vertical orientation
  35                 self.splitter.resizeEvent = self.splitter_resize_event
  36
  37                 # Qt does not support noun/verbs
  38                 self.commit_button.setText(qtutils.tr('Commit@@verb'))
  39                 self.commit_menu.setTitle(qtutils.tr('Commit@@verb'))
  40                 # Default to creating a new commit(i.e. not an amend commit)
  41                 self.new_commit_radio.setChecked(True)
  42                 self.toolbar_show_log = self.toolbar.addAction(
  43                                 qtutils.get_qicon('git.png'),
  44                                 'Show/Hide Log Window')
  45                 self.toolbar_show_log.setEnabled(True)
  46
  47                 # Setup the default dock layout
  48                 self.tabifyDockWidget(self.diff_dock, self.editor_dock)
  49
  50                 dock_area = QtCore.Qt.TopDockWidgetArea
  51                 self.addDockWidget(dock_area, self.status_dock)
  52
  53                 toolbar_area = QtCore.Qt.BottomToolBarArea
  54                 self.addToolBar(toolbar_area, self.toolbar)
  55
  56                 # Diff/patch syntax highlighter
  57                 DiffSyntaxHighlighter(self.display_text.document())
  58
  59         def splitter_resize_event(self, event):
  60                 width = self.splitter.width()
  61                 height = self.splitter.height()
  62                 if width > height:
  63                         self.splitter.setOrientation(QtCore.Qt.Horizontal)
  64                 else:
  65                         self.splitter.setOrientation(QtCore.Qt.Vertical)
  66                 QSplitter.resizeEvent(self.splitter, event)
  67
  68         def action_cut(self):
  69                 self.action_copy()
  70                 self.action_delete()
  71         def action_copy(self):
  72                 cursor = self.commitmsg.textCursor()
  73                 selection = cursor.selection().toPlainText()
  74                 qtutils.set_clipboard(selection)
  75         def action_delete(self):
  76                 self.commitmsg.textCursor().removeSelectedText()
  77         def reset_display(self):
  78                 self.set_display('')
  79                 self.set_info('')
  80         def copy_display(self):
  81                 cursor = self.display_text.textCursor()
  82                 selection = cursor.selection().toPlainText()
  83                 qtutils.set_clipboard(selection)
  84         def diff_selection(self):
  85                 cursor = self.display_text.textCursor()
  86                 offset = cursor.position()
  87                 selection = cursor.selection().toPlainText()
  88                 num_selected_lines = selection.count('\n')
  89                 return offset, selection
  90
  91 class OutputGUI(Ui_outputgui, QDialog):
  92         '''A simple dialog to display command output.'''
  93         def __init__(self, parent=None, output=None):
  94                 QDialog.__init__(self, parent)
  95                 Ui_outputgui.__init__(self)
  96                 self.setupUi(self)
  97                 self.setWindowTitle(self.tr('Git Command Output'))
  98                 # Syntax highlight the log window
  99                 LogSyntaxHighlighter(self.output_text.document())
 100                 if output: self.set_output(output)
 101         def clear(self):
 102                 self.output_text.clear()
 103         def set_output(self, output):
 104                 self.output_text.setText(output)
 105         def log(self, output):
 106                 if not output: return
 107                 cursor = self.output_text.textCursor()
 108                 cursor.movePosition(cursor.End)
 109                 text = self.output_text
 110                 cursor.insertText(time.asctime() + '\n')
 111                 for line in unicode(output).splitlines():
 112                         cursor.insertText(line + '\n')
 113                 cursor.insertText('\n')
 114                 cursor.movePosition(cursor.End)
 115                 text.setTextCursor(cursor)
 116
 117 class BranchGUI(Ui_branchgui, QDialog):
 118         '''A dialog for choosing branches.'''
 119         def __init__(self, parent=None, branches=None):
 120                 QDialog.__init__(self, parent)
 121                 Ui_branchgui.__init__(self)
 122                 self.setupUi(self)
 123                 self.reset()
 124                 if branches: self.add(branches)
 125         def reset(self):
 126                 self.branches = []
 127                 self.branch_combo.clear()
 128         def add(self, branches):
 129                 for branch in branches:
 130                         self.branches.append(branch)
 131                         self.branch_combo.addItem(branch)
 132         def get_selected(self):
 133                 self.show()
 134                 if self.exec_() == QDialog.Accepted:
 135                         return self.branches[self.branch_combo.currentIndex()]
 136                 else:
 137                         return None
 138
 139 class CommitGUI(Ui_commitgui, QDialog):
 140         def __init__(self, parent=None, title=None):
 141                 QDialog.__init__(self, parent)
 142                 Ui_commitgui.__init__(self)
 143                 self.setupUi(self)
 144                 if title: self.setWindowTitle(title)
 145                 # Make the list widget slighty larger
 146                 self.splitter.setSizes([ 50, 200 ])
 147                 DiffSyntaxHighlighter(self.commit_text.document(),
 148                                 whitespace=False)
 149
 150 class CreateBranchGUI(Ui_createbranchgui, QDialog):
 151         def __init__(self, parent=None):
 152                 QDialog.__init__(self, parent)
 153                 Ui_createbranchgui.__init__(self)
 154                 self.setupUi(self)
 155
 156 class PushGUI(Ui_pushgui, QDialog):
 157         def __init__(self, parent=None):
 158                 QDialog.__init__(self, parent)
 159                 Ui_pushgui.__init__(self)
 160                 self.setupUi(self)
 161
 162 class OptionsGUI(Ui_optionsgui, QDialog):
 163         def __init__(self, parent=None):
 164                 QDialog.__init__(self, parent)
 165                 Ui_optionsgui.__init__(self)
 166                 self.setupUi(self)