class BranchesDialogWrapper(QDialog):
def __init__(self, parent=None):
QDialog.__init__(self)
self.parent = parent
self.ui = Ui_BranchesDialog()
self.ui.setupUi(self)
self.list_local_branches()
self.list_remote_branches()
QObject.connect(self, SIGNAL('accepted()'), self.change_branch)
def list_local_branches(self):
branches = get_local_branches()
for branch in branches:
QListWidgetItem(branch, self.ui.localBranchesListWidget)
def list_remote_branches(self):
branches = get_remote_branches()
for branch in branches:
QListWidgetItem(branch, self.ui.remoteBranchesListWidget)
def error(self):
QMessageBox.critical(self, "Error", "You must choose branch!", QMessageBox.Ok)
def get_default_branch_name(self, name):
name = name.split('/')
return name[1]
def change_lcl_branch(self):
item = self.ui.localBranchesListWidget.currentItem()
if item == None:
self.error()
else:
change_local_branch(item.text())
def change_rmt_branch(self):
item = self.ui.remoteBranchesListWidget.currentItem()
if item == None:
self.error()
else:
name = QInputDialog().getText(self, 'Name', 'Put your branch name:', text=self.get_default_branch_name(item.text()))
if name[1]:
QMessageBox.information(self, 'Change branch', change_remote_branch(item.text(), name[0]), QMessageBox.Ok)
def change_branch(self):
if self.ui.branchesTabWidget.currentIndex() == 0:
self.change_lcl_branch()
else:
self.change_rmt_branch()
def __init__(self, parent=None):
QDialog.__init__(self)
self.parent = parent
self.ui = Ui_BranchesDialog()
self.ui.setupUi(self)
self.list_local_branches()
self.list_remote_branches()
QObject.connect(self, SIGNAL('accepted()'), self.change_branch)
