def setup_menubar(self):
menu_bar = QMenuBar()
file_menu = menu_bar.addMenu("&File")
file_menu.addAction(self.new_project_action)
file_menu.addAction(self.show_settings_action)
file_menu.addSeparator()
file_menu.addAction(self.quit_action)
view_menu = menu_bar.addMenu("&View")
dockWidgets = self.findChildren(QDockWidget)
for widget in dockWidgets:
view_menu.addAction(widget.toggleViewAction())
debug_menu = menu_bar.addMenu("&Debug")
debug_menu.addAction(self.start_listening_action)
debug_menu.addAction(self.stop_listening_action)
debug_menu.addSeparator()
debug_menu.addAction(self.stop_debug_action)
debug_menu.addAction(self.detach_debug_action)
debug_menu.addSeparator()
debug_menu.addAction(self.run_debug_action)
debug_menu.addAction(self.step_over_action)
debug_menu.addAction(self.step_into_action)
debug_menu.addAction(self.step_out_action)
self.setMenuBar(menu_bar)
def __init__(self, parent=None):
"""
Initialize the menubar.
:param parent: the parent widget
"""
QMenuBar.__init__(self, parent)
self.initUI()
def __init__(self, parent, actionManager):
QMenuBar.__init__(self, parent)
self._manager = actionManager
for action in self._manager.allActions():
self._onActionInserted(action)
self._manager.actionInserted.connect(self._onActionInserted)
self._manager.actionRemoved.connect(self._onActionRemoved)
def initUI(self):
QToolTip.setFont(QFont('SansSerif', 10))
self.setToolTip('This is a <b>QWidget</b> widget.')
#quitBtn = QPushButton('Quit', self)
#quitBtn.clicked.connect(self.quitBtnEvent())
#quitBtn.clicked.connect(QCoreApplication.instance().quit)
#quitBtn.setToolTip('This is a <b>QPushButton</b> widget.')
#quitBtn.resize(quitBtn.sizeHint())
exitAction = QAction(QIcon('application-exit-4.png'), '&Exit', self)
exitAction.setShortcut('Alt+F4')
exitAction.setStatusTip('Exit Application')
exitAction.triggered.connect(qApp.quit)
menuBar = QMenuBar()
fileMenu = menuBar.addMenu('&File')
fileMenu.addAction(exitAction)
fileMenu.resize(fileMenu.sizeHint())
toolBar = QToolBar(self)
toolBar.addAction(exitAction)
#toolBar.resize(toolBar.sizeHint())
toolBar.setFixedHeight(60)
hozLine = QFrame()
hozLine.setFrameStyle(QFrame.HLine)
#hozLine.setSizePolicy(QSizePolicy.Minimum,QSizePolicy.Expanding)
statusBar = QStatusBar(self)
statusBar.showMessage('Ready')
grid = QGridLayout()
lbl_1 = QLabel('1,1')
lbl_2 = QLabel('1,2')
lbl_3 = QLabel('2,1')
lbl_4 = QLabel('2,2')
grid.addWidget(lbl_1, 1, 1)
grid.addWidget(lbl_2, 1, 2)
grid.addWidget(lbl_3, 2, 1)
grid.addWidget(lbl_4, 2, 2)
vbox = QVBoxLayout()
vbox.addWidget(menuBar)
vbox.addWidget(hozLine)
vbox.addWidget(toolBar)
# vbox.addWidget(hozLine)
vbox.addLayout(grid)
vbox.addStretch(1)
vbox.addWidget(statusBar)
self.setLayout(vbox)
self.setGeometry(300, 300, 500, 500)
self.setWindowTitle('Photos')
self.setWindowIcon(QIcon('camera-photo-5.png'))
self.center()
self.show()
def __init__(self, app):
QMainWindow.__init__(self, None)
self.l.debug('Initializing MainWindow ...')
self.setWindowTitle('MynPad')
app.setWindowIcon(QIcon(':/icons/mynpad.png'))
if os.name == 'nt':
# On Windows, make sure to use a unique Application User Model Id, otherwise
# Windows shows the default python icon in the taskbar
# see http://stackoverflow.com/questions/1551605/how-to-set-applications-taskbar-icon-in-windows-7
myappid = 'afester.mynpad'
import ctypes; ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(myappid)
self.theApplication = app
app.aboutToQuit.connect(self.saveState)
# read the local configuration file
iniPath = 'mynpad.ini'
if not os.path.exists(iniPath):
iniPath = os.path.join(expanduser("~"), iniPath)
self.settings = Settings(iniPath)
self.settings.load()
# Set up the menu bar
menuBar = QMenuBar(self)
exitAction = QAction("Exit", self, triggered=self.theApplication.exit)
fileMenu = menuBar.addMenu("&File")
fileMenu.addAction(exitAction)
aboutAction = QAction("About ...", self, triggered = self.handleAbout)
helpMenu = menuBar.addMenu("&Help")
helpMenu.addAction(aboutAction)
self.setMenuBar(menuBar)
# Setup the status bar
self.statusBar = QStatusBar()
self.statusBar.showMessage("Ready.")
self.setStatusBar(self.statusBar)
self.mainWidget = CentralWidget(self, self.settings)
self.mainWidget.updateWindowTitle.connect(self.updateWindowTitle)
self.setCentralWidget(self.mainWidget)
# Reset main window size and position
pos = self.settings.getMainWindowPos()
self.move(pos.x(), pos.y())
size = self.settings.getMainWindowSize()
self.resize(size)
# initialize the browser tree (add the top nodes and expand the saved path)
self.mainWidget.browserWidget.initialize()
def menubar():
"""Return a newly created parent-less menu bar that's used when there is no main window."""
m = QMenuBar()
m.addMenu(menu_file(m))
m.addMenu(menu_edit(m))
m.addMenu(menu_window(m))
m.addMenu(menu_sessions(m))
m.addMenu(menu_help(m))
return m
def createMenus(self):
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 1029, 22))
self.menubar.setDefaultUp(False)
self.menubar.setNativeMenuBar(False)
self.menubar.setObjectName("menubar")
self.menuFile = QMenu(self.menubar)
self.setMenuBar(self.menubar)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.newAct)
self.fileMenu.addAction(self.openAct)
self.fileMenu.addAction(self.closeAct)
self.fileMenu.addAction(self.saveAct)
self.fileMenu.addAction(self.saveAsAct)
self.fileMenu.addAction(self.printAct)
self.fileMenu.addSeparator();
self.fileMenu.addAction(self.exitAct)
self.editMenu = self.menuBar().addMenu("&Edit")
self.editMenu.addAction(self.cutAct)
self.editMenu.addAction(self.copyAct)
self.editMenu.addAction(self.pasteAct)
self.menuBar().addSeparator()
self.helpMenu = self.menuBar().addMenu("&Help")
self.helpMenu.addAction(self.aboutAct)
def createMenuBar(self):
self.menuBar = QMenuBar()
self.fileMenu = QMenu("&File", self)
self.exitAction = self.fileMenu.addAction("E&xit")
self.menuBar.addMenu(self.fileMenu)
self.exitAction.triggered.connect(self.accept)
def __init__(self):
super(Main_Window, self).__init__()
self.menubar = QMenuBar(self)
self.statusbar = QStatusBar(self)
self.centralwidget = QWidget(self)
self.image_view = Image_View(self.centralwidget)
self.horizontal_slider = QSlider(self.centralwidget)
self.setup_ui()
def __init__(self, parent=None, title="Source"):
super(ViewSourceDialog, self).__init__()
self.setParent(parent)
self.menuBar = QMenuBar(self)
self.menuBar.setStyleSheet("""QMenuBar { background: transparent; border: 0; }
QMenuBar::item { background: transparent;
color: palette(window-text); }
QMenuBar::item:pressed { background: palette(highlight);
color: palette(highlighted-text); }""")
self.setMenuBar(self.menuBar)
self.text = ""
self.findFlag = None
self.fileMenu = QMenu(tr("&File"), self.menuBar)
self.menuBar.addMenu(self.fileMenu)
self.saveAsAction = QAction(tr("&Save As..."), self)
self.saveAsAction.setShortcut("Ctrl+S")
self.saveAsAction.triggered.connect(self.saveAs)
self.addAction(self.saveAsAction)
self.fileMenu.addAction(self.saveAsAction)
self.viewMenu = QMenu(tr("&View"), self.menuBar)
self.menuBar.addMenu(self.viewMenu)
self.findAction = QAction(tr("&Find..."), self)
self.findAction.setShortcut("Ctrl+F")
self.findAction.triggered.connect(self.find)
self.addAction(self.findAction)
self.viewMenu.addAction(self.findAction)
self.findNextAction = QAction(tr("Find Ne&xt"), self)
self.findNextAction.setShortcut("Ctrl+G")
self.findNextAction.triggered.connect(self.findNext)
self.addAction(self.findNextAction)
self.viewMenu.addAction(self.findNextAction)
self.findPreviousAction = QAction(tr("Find Pre&vious"), self)
self.findPreviousAction.setShortcut("Ctrl+Shift+G")
self.findPreviousAction.triggered.connect(self.findPrevious)
self.addAction(self.findPreviousAction)
self.viewMenu.addAction(self.findPreviousAction)
self.sourceView = SourceView(self)
self.sourceView.setReadOnly(True)
self.sourceView.setFontFamily("monospace")
self.setCentralWidget(self.sourceView)
closeWindowAction = QAction(self)
closeWindowAction.setShortcut("Ctrl+W")
closeWindowAction.triggered.connect(self.close)
self.addAction(closeWindowAction)
self.setWindowTitle(title)
self.resize(640, 480)
def createMenu(self):
self.menuBar = QMenuBar()
self.fileMenu = self.menuBar.addMenu("&File")
self.helpMenu = self.menuBar.addMenu("&Help")
self.aNewProject = self.fileMenu.addAction("Start new project")
self.aOpen = self.fileMenu.addAction("Open")
self.aSave = self.fileMenu.addAction("Save as...")
self.aExport = self.fileMenu.addAction("Export as...")
self.fileMenu.addSeparator()
self.aExitApp = self.fileMenu.addAction("Exit")
self.setMenuBar(self.menuBar)
def createMenu(self):
self.menuBar = QMenuBar()
self.fileMenu = QMenu("&Fichier", self)
self.exitAction = self.fileMenu.addAction("E&xit")
apropos = QMenu("&Infos", self)
apro = QAction("À &propos", self)
apro.triggered.connect(self._createApropos)
apropos.addAction(apro)
self.menuBar.addMenu(self.fileMenu)
self.menuBar.addMenu(apropos)
class Main_Window(QMainWindow):
def __init__(self):
super(Main_Window, self).__init__()
self.menubar = QMenuBar(self)
self.statusbar = QStatusBar(self)
self.centralwidget = QWidget(self)
self.image_view = Image_View(self.centralwidget)
self.horizontal_slider = QSlider(self.centralwidget)
self.setup_ui()
def setup_ui(self):
# 1. self info
self.setObjectName("Demo_1")
self.resize(800, 600)
# 2.1 sub widgets
self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 22))
self.menubar.setObjectName("menubar")
self.setMenuBar(self.menubar)
# 2.2
self.statusbar.setObjectName("statusbar")
self.setStatusBar(self.statusbar)
# 2.3
self.centralwidget.setObjectName("centralwidget")
self.setCentralWidget(self.centralwidget)
self.image_view.setGeometry(QtCore.QRect(144, 10, 512, 512))
self.image_view.setObjectName("image_view")
self.horizontal_slider.setGeometry(QtCore.QRect(144, 530, 512, 22))
self.horizontal_slider.setOrientation(QtCore.Qt.Horizontal)
self.horizontal_slider.setObjectName("horizontal_slider")
# 3 signals and slots
self.horizontal_slider.valueChanged['int'].connect(self.image_view.repaint)
def createMenu(self):
self.menuBar = QMenuBar()
self.fileMenu = QMenu("&File", self)
self.exitAction = self.fileMenu.addAction("E&xit")
self.exitAction.setShortcut('Ctrl+Q')
self.exitAction.setStatusTip('Exit application')
#self.exitAction.triggered.connect(qApp.quit)
self.menuBar.addMenu(self.fileMenu)
self.exitAction.triggered.connect(self.accept)
def __init__(self):
super().__init__()
# GroupBox Initialization
self.mainlayout = QGridLayout() # basicinfo layout
self.names = []
self.menuBar = QMenuBar()
# PushButton Initialization
self.BeginTestButton = QPushButton('开始测试')
self.CancelTestButton = QPushButton('取消测试')
self.PauseTestButton = QPushButton('暂停测试')
self.SaveBasicinfoButton = QPushButton('保存基本信息')
self.initUI()
def setupUI(self):
menu_bar = QMenuBar()
file = menu_bar.addMenu("&File")
help = menu_bar.addMenu("&Help")
build = menu_bar.addMenu("&Build")
exit = menu_bar.addMenu("&Exit")
tabs = QTabWidget()
centerwidget = QWidget()
vshadertab = QWidget()
fshadertab = QWidget()
vshadertab_layout = QVBoxLayout(vshadertab)
fshadertab_layout = QVBoxLayout(fshadertab)
fshadereditor = GLSLEditor()
vshadereditor = GLSLEditor()
fshadertab_layout.addWidget(fshadereditor)
vshadertab_layout.addWidget(vshadereditor)
tabs.addTab(vshadertab, "Vertex Shader")
tabs.addTab(fshadertab, "Fragment Shader")
vbox = QVBoxLayout()
vbox.addWidget(menu_bar)
vbox.addWidget(tabs)
centerwidget.setLayout(vbox)
self.setCentralWidget(centerwidget)
pass
def __init__(self, paths=[]):
QMainWindow.__init__( self )
self.setWindowTitle( 'Subtle' )
self.setAcceptDrops( True )
self._menu = QMenuBar( self )
fileMenu = self._menu.addMenu( '&File' )
fileMenu.addAction( QAction('&Open', fileMenu) )
close = QAction('&Quit', fileMenu)
close.setShortcut( QKeySequence(QKeySequence.Quit) )
close.triggered.connect( self._close_action )
fileMenu.addAction( close )
self.setMenuBar( self._menu )
self._tree = SubtleTreeWidget(self, paths_to_items_list(self, paths))
self.setCentralWidget( self._tree )
def __init__(self, cellular_automaton):
super().__init__()
self.speed = DEFAULT_SIMULATION_SPEED
self.cell_size = CELL_SIZE
self.cellular_automaton = cellular_automaton
self.num_of_cells_x = self.cellular_automaton.size_x
self.num_of_cells_y = self.cellular_automaton.size_y
self.run = False
self.setWindowTitle('Cellular automaton')
self.setToolTip('Select cell')
self.start_btn = QPushButton('Start', self)
self.start_btn.setToolTip('Start simulation')
self.clear_btn = QPushButton('Clear', self)
self.clear_btn.setToolTip('Clear board')
self.start_btn.clicked.connect(self.toggle)
self.clear_btn.clicked.connect(self.clean)
self.timer = QTimer(self)
self.timer.timeout.connect(self.paint_update)
self.slider = QSlider(Qt.Horizontal, self)
self.slider.setMinimum(MIN_SIMULATION_SPEED)
self.slider.setMaximum(MAX_SIMULATION_SPEED)
self.slider.setValue(DEFAULT_SIMULATION_SPEED)
self.slider.setToolTip('Speed of simulation')
self.slider.valueChanged.connect(self.set_value)
btn_layout = QHBoxLayout()
btn_layout.addWidget(self.start_btn)
btn_layout.addWidget(self.clear_btn)
layout = QVBoxLayout()
layout.setAlignment(Qt.AlignBottom)
layout.addLayout(btn_layout, Qt.AlignCenter)
layout.addWidget(self.slider, Qt.AlignCenter)
self.setLayout(layout)
self.menu_bar = QMenuBar(self)
self.add_menu_bar()
self.margin_left = MARGIN
self.margin_top = MARGIN
self.resize(WINDOW_X_SIZE, WINDOW_Y_SIZE)
self.setMinimumSize(WINDOW_X_SIZE, WINDOW_Y_SIZE)
self.resizeEvent = self.on_resize
def createMenus(self):
self.menuBar = QMenuBar(self)
self.fileMenu = QMenu("&File", self)
self.fileMenu.addAction(self.exitAct)
self.viewMenu = QMenu("&View", self)
self.viewMenu.addAction(self.zoomInAct)
self.viewMenu.addAction(self.zoomOutAct)
self.viewMenu.addAction(self.normalSizeAct)
self.viewMenu.addSeparator()
self.viewMenu.addAction(self.fitToWindowAct)
self.viewMenu.addSeparator()
self.viewMenu.addAction(self.nextPageAct)
self.viewMenu.addAction(self.previousPageAct)
self.menuBar.addMenu(self.fileMenu)
self.menuBar.addMenu(self.viewMenu)
def _setupMenu(self):
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 42, 22))
self.menuFile = QMenu(self.menubar)
self.menuFile.setTitle(tr("File"))
self.menuView = QMenu(self.menubar)
self.menuView.setTitle(tr("View"))
self.menuHelp = QMenu(self.menubar)
self.menuHelp.setTitle(tr("Help"))
self.menuLoadRecent = QMenu(self.menuFile)
self.menuLoadRecent.setTitle(tr("Load Recent Results"))
self.setMenuBar(self.menubar)
self.menuFile.addAction(self.actionLoadResults)
self.menuFile.addAction(self.menuLoadRecent.menuAction())
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionClearPictureCache)
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionQuit)
self.menuView.addAction(self.app.actionPreferences)
self.menuView.addAction(self.actionShowResultsWindow)
self.menuView.addAction(self.app.actionIgnoreList)
self.menuHelp.addAction(self.app.actionShowHelp)
self.menuHelp.addAction(self.app.actionOpenDebugLog)
self.menuHelp.addAction(self.app.actionAbout)
self.menubar.addAction(self.menuFile.menuAction())
self.menubar.addAction(self.menuView.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
# Recent folders menu
self.menuRecentFolders = QMenu()
self.menuRecentFolders.addAction(self.actionAddFolder)
self.menuRecentFolders.addSeparator()
# Recent results menu
self.menuRecentResults = QMenu()
self.menuRecentResults.addAction(self.actionLoadResults)
self.menuRecentResults.addSeparator()
def __init__(self, path, parent=None):
super(DragWidget, self).__init__(parent)
self.setMinimumSize(400, 200)
self.setAcceptDrops(True)
self.menubar = QMenuBar()
fileMenu = self.menubar.addMenu('File')
editMenu = self.menubar.addMenu('Edit')
exitAction = QAction('Open', self)
prefAction = QAction('Pref', self)
fileMenu.addAction(exitAction)
editMenu.addAction(prefAction)
self.path = path
self.icons = []
# self.clipicon = None
for name in os.listdir(path):
icon_widget = IconWidget(self, name=name, path=self.path)
icon_widget.new_window.connect(self.windowclass_signal.emit)
icon_widget.clipboard.connect(self.on_clipboard)
self.icons.append(icon_widget)
# self.icons[-1].move(DragWidget.spacerX, DragWidget.spacerY)
self.icons[-1].setAttribute(Qt.WA_DeleteOnClose)
self.clean_up()
class SubtleMainWindow(QMainWindow):
def __init__(self, paths=[]):
QMainWindow.__init__( self )
self.setWindowTitle( 'Subtle' )
self.setAcceptDrops( True )
self._menu = QMenuBar( self )
fileMenu = self._menu.addMenu( '&File' )
fileMenu.addAction( QAction('&Open', fileMenu) )
close = QAction('&Quit', fileMenu)
close.setShortcut( QKeySequence(QKeySequence.Quit) )
close.triggered.connect( self._close_action )
fileMenu.addAction( close )
self.setMenuBar( self._menu )
self._tree = SubtleTreeWidget(self, paths_to_items_list(self, paths))
self.setCentralWidget( self._tree )
@QtCore.pyqtSlot()
def _close_action(self):
self.close()
def dragEnterEvent(self, event):
event.acceptProposedAction()
def dropEvent(self, event):
items = paths_to_items_list(self._tree, event.mimeData().urls())
self._tree.addTopLevelItems( items )
self.update()
event.acceptProposedAction()
def update(self):
items = [ self._tree.itemAt(i, 0)
for i in range(0, self._tree.topLevelItemCount()) ]
for item in items:
print( 'item {}'.format(item.data(0, SubtleTreeWidget.ITEM_ROLE)) )
def __init__(self):
super().__init__(flags=QtCore.Qt.Window)
self.setWindowIcon(QIcon(QPixmap(":/icon/codeblock_icon.svg")))
scale_x = QDesktopWidget().screenGeometry().width()/1920
scale_y = QDesktopWidget().screenGeometry().height()/1080
self.resize(1280*scale_x, 720*scale_y)
self.centralwidget = QWidget()
self.centralwidget.setObjectName("centralwidget")
self.horizontalLayout = QHBoxLayout(self.centralwidget)
self.horizontalLayout.setObjectName("horizontalLayout")
self.splitter = QSplitter(self.centralwidget)
self.tabWidget = QTabWidget(self.centralwidget)
sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(4)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(self.tabWidget.sizePolicy().hasHeightForWidth())
self.tabWidget.setSizePolicy(sizePolicy)
self.tabWidget.setTabShape(QTabWidget.Rounded)
self.tabWidget.setMovable(True)
self.tabWidget.setObjectName("tabWidget")
self.tab = QWidget()
self.tab.setObjectName("tab")
self.verticalLayout2 = QVBoxLayout(self.tab)
self.classView = QTreeWidget(self.tab)
self.classView.header().setVisible(False)
self.verticalLayout2.addWidget(self.classView)
self.tabWidget.addTab(self.tab, "")
self.splitter.addWidget(self.tabWidget)
self.tabWidget_2 = QTabWidget(self.centralwidget)
self.tab_3 = QWidget()
self.tab_3.setObjectName("tab_3")
self.tabWidget_2.addTab(self.tab_3, "")
self.scrollAreaWidgetContents = QWidget()
self.scrollAreaWidgetContents.setGeometry(QtCore.QRect(0, 0, 184, 148))
self.scrollAreaWidgetContents.setObjectName("scrollAreaWidgetContents")
sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(7)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(self.tabWidget_2.sizePolicy().hasHeightForWidth())
self.tabWidget_2.setSizePolicy(sizePolicy)
self.tabWidget_2.setTabsClosable(False)
self.tabWidget_2.setMovable(True)
self.tabWidget_2.setObjectName("tabWidget_2")
self.splitter.addWidget(self.tabWidget_2)
self.horizontalLayout.addWidget(self.splitter)
self.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar()
self.menubar.setGeometry(QtCore.QRect(0, 0, 1280, 23))
self.menubar.setObjectName("menubar")
self.menuFile = QMenu(self.menubar)
self.menuHelp = QMenu(self.menubar)
self.actionOpen = QAction(self)
self.actionOpen.setText("Open")
self.menuFile.addAction(self.actionOpen)
self.menuFile.setTitle("File")
self.menubar.addAction(self.actionOpen)
self.actionAbout = QAction(self)
self.actionAbout.setText("About Codeblock Visual Studio")
self.actionTutorial = QAction(self)
self.actionTutorial.setText("Rerun Tutorial")
self.menuHelp.addAction(self.actionAbout)
self.menuHelp.addAction(self.actionTutorial)
self.menuHelp.setTitle("Help")
self.menubar.addAction(self.menuHelp.menuAction())
self.setMenuBar(self.menubar)
self.statusbar = QStatusBar()
self.statusbar.setObjectName("statusbar")
self.setStatusBar(self.statusbar)
self.frameLayout = QHBoxLayout(self.tab_3)
self.scroll = QScrollArea()
self.scrollContents = QWidget()
self.scrollContents.setGeometry(QtCore.QRect(0, 0, 5000, 50000))
self.scrollLayout = QHBoxLayout(self.scrollContents)
self.codeArea = QFrame(self.scrollContents)
self.codeArea.setMinimumSize(QtCore.QSize(200, 2000))
self.scrollLayout.addWidget(self.codeArea)
self.frameLayout.addWidget(self.scroll)
self.scroll.setWidget(self.scrollContents)
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab), "Class View")
self.tabWidget_2.setTabText(self.tabWidget_2.indexOf(self.tab_3), "Code Area")
class FrontPanel(QDialog):
"""This is FiledSensor test program front panel class"""
def __init__(self):
super().__init__()
# GroupBox Initialization
self.mainlayout = QGridLayout() # basicinfo layout
self.names = []
self.menuBar = QMenuBar()
# PushButton Initialization
self.BeginTestButton = QPushButton('开始测试')
self.CancelTestButton = QPushButton('取消测试')
self.PauseTestButton = QPushButton('暂停测试')
self.SaveBasicinfoButton = QPushButton('保存基本信息')
self.initUI()
def initUI(self):
"""initialise the UI"""
self.BasicinfoList()
self.ButtonList()
self.CreateMenu()
self.StretchSet()
self.mainlayout.addWidget(self.SaveBasicinfoButton, 12, 0, 1, 2)
self.mainlayout.setMenuBar(self.menuBar)
self.setLayout(self.mainlayout)
self.setGeometry(300, 100, 900, 700) # 设置窗口位置大小
self.setWindowTitle('FieldSensor')
self.show()
def CreateMenu(self):
"""create the menu"""
file_menu = self.menuBar.addMenu('&File')
open_action = QAction('&打开', self)
exit_action = QAction('&退出', self)
file_menu.addAction(open_action)
file_menu.addAction(exit_action)
exit_action.triggered.connect(self.accept)
def ButtonList(self):
names = [self.BeginTestButton, self.CancelTestButton,
self.PauseTestButton]
for i in range(3):
self.mainlayout.addWidget(names[i], i + 4, 4)
def BasicinfoList(self):
"""basicinfo layout set"""
certificate_num_label = QLabel('证书编号')
certificate_serial_label = QLabel('证书序列号')
manufacturer_label = QLabel('生产厂家')
device_name_label = QLabel('设备名称')
custom_address_label = QLabel('客户地址')
custom_name_label = QLabel('客户名称')
calibration_address_label = QLabel('校准地点')
temperature_label = QLabel('温 度')
humidity_label = QLabel('湿 度')
tester_label = QLabel('测试人员')
date_label = QLabel('测试日期')
# LineEdit Initialization
certificatenumber = QLineEdit()
certificateserial = QLineEdit()
manufacturer = QLineEdit()
devicename = QLineEdit()
customaddress = QLineEdit()
customname = QLineEdit()
calibrationaddress = QLineEdit()
temperature = QLineEdit()
humidity = QLineEdit()
tester = QLineEdit()
date = QLineEdit()
self.names = [certificate_num_label, certificatenumber,
certificate_serial_label, certificateserial,
manufacturer_label, manufacturer,
device_name_label, devicename,
custom_address_label, customaddress,
custom_name_label, customname,
calibration_address_label, calibrationaddress,
temperature_label, temperature,
humidity_label, humidity,
tester_label, tester,
date_label, date]
for i in range(22):
if i % 2 == 0:
self.mainlayout.addWidget(self.names[i], i / 2, 0)
else:
self.mainlayout.addWidget(self.names[i], (i - 1) / 2, 1)
def StretchSet(self):
# set column stretch
self.mainlayout.setColumnStretch(0, 0.5)
self.mainlayout.setColumnStretch(1, 2)
self.mainlayout.setColumnStretch(2, 1)
self.mainlayout.setColumnStretch(3, 1)
class DirectoriesDialog(QMainWindow):
def __init__(self, app, **kwargs):
super().__init__(None, **kwargs)
self.app = app
self.lastAddedFolder = platform.INITIAL_FOLDER_IN_DIALOGS
self.recentFolders = Recent(self.app, 'recentFolders')
self._setupUi()
self._updateScanTypeList()
self.directoriesModel = DirectoriesModel(self.app.model.directory_tree, view=self.treeView)
self.directoriesDelegate = DirectoriesDelegate()
self.treeView.setItemDelegate(self.directoriesDelegate)
self._setupColumns()
self.app.recentResults.addMenu(self.menuLoadRecent)
self.app.recentResults.addMenu(self.menuRecentResults)
self.recentFolders.addMenu(self.menuRecentFolders)
self._updateAddButton()
self._updateRemoveButton()
self._updateLoadResultsButton()
self._updateActionsState()
self._setupBindings()
def _setupBindings(self):
self.appModeRadioBox.itemSelected.connect(self.appModeButtonSelected)
self.showPreferencesButton.clicked.connect(self.app.actionPreferences.trigger)
self.scanButton.clicked.connect(self.scanButtonClicked)
self.loadResultsButton.clicked.connect(self.actionLoadResults.trigger)
self.addFolderButton.clicked.connect(self.actionAddFolder.trigger)
self.removeFolderButton.clicked.connect(self.removeFolderButtonClicked)
self.treeView.selectionModel().selectionChanged.connect(self.selectionChanged)
self.app.recentResults.itemsChanged.connect(self._updateLoadResultsButton)
self.recentFolders.itemsChanged.connect(self._updateAddButton)
self.recentFolders.mustOpenItem.connect(self.app.model.add_directory)
self.directoriesModel.foldersAdded.connect(self.directoriesModelAddedFolders)
self.app.willSavePrefs.connect(self.appWillSavePrefs)
def _setupActions(self):
# (name, shortcut, icon, desc, func)
ACTIONS = [
('actionLoadResults', 'Ctrl+L', '', tr("Load Results..."), self.loadResultsTriggered),
('actionShowResultsWindow', '', '', tr("Results Window"), self.app.showResultsWindow),
('actionAddFolder', '', '', tr("Add Folder..."), self.addFolderTriggered),
]
createActions(ACTIONS, self)
def _setupMenu(self):
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 42, 22))
self.menuFile = QMenu(self.menubar)
self.menuFile.setTitle(tr("File"))
self.menuView = QMenu(self.menubar)
self.menuView.setTitle(tr("View"))
self.menuHelp = QMenu(self.menubar)
self.menuHelp.setTitle(tr("Help"))
self.menuLoadRecent = QMenu(self.menuFile)
self.menuLoadRecent.setTitle(tr("Load Recent Results"))
self.setMenuBar(self.menubar)
self.menuFile.addAction(self.actionLoadResults)
self.menuFile.addAction(self.menuLoadRecent.menuAction())
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionClearPictureCache)
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionQuit)
self.menuView.addAction(self.app.actionPreferences)
self.menuView.addAction(self.actionShowResultsWindow)
self.menuView.addAction(self.app.actionIgnoreList)
self.menuHelp.addAction(self.app.actionShowHelp)
self.menuHelp.addAction(self.app.actionOpenDebugLog)
self.menuHelp.addAction(self.app.actionAbout)
self.menubar.addAction(self.menuFile.menuAction())
self.menubar.addAction(self.menuView.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
# Recent folders menu
self.menuRecentFolders = QMenu()
self.menuRecentFolders.addAction(self.actionAddFolder)
self.menuRecentFolders.addSeparator()
# Recent results menu
self.menuRecentResults = QMenu()
self.menuRecentResults.addAction(self.actionLoadResults)
self.menuRecentResults.addSeparator()
def _setupUi(self):
self.setWindowTitle(self.app.NAME)
self.resize(420, 338)
self.centralwidget = QWidget(self)
self.verticalLayout = QVBoxLayout(self.centralwidget)
hl = QHBoxLayout()
label = QLabel(tr("Application Mode:"), self)
label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
hl.addWidget(label)
self.appModeRadioBox = RadioBox(
self,
items=[tr("Standard"), tr("Music"), tr("Picture")],
spread=False
)
hl.addWidget(self.appModeRadioBox)
self.verticalLayout.addLayout(hl)
hl = QHBoxLayout()
hl.setAlignment(Qt.AlignLeft)
label = QLabel(tr("Scan Type:"), self)
label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
hl.addWidget(label)
self.scanTypeComboBox = QComboBox(self)
self.scanTypeComboBox.setSizePolicy(QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed))
self.scanTypeComboBox.setMaximumWidth(400)
hl.addWidget(self.scanTypeComboBox)
self.showPreferencesButton = QPushButton(tr("More Options"), self.centralwidget)
self.showPreferencesButton.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
hl.addWidget(self.showPreferencesButton)
self.verticalLayout.addLayout(hl)
self.promptLabel = QLabel(tr("Select folders to scan and press \"Scan\"."), self.centralwidget)
self.verticalLayout.addWidget(self.promptLabel)
self.treeView = QTreeView(self.centralwidget)
self.treeView.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.treeView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.treeView.setAcceptDrops(True)
triggers = QAbstractItemView.DoubleClicked | QAbstractItemView.EditKeyPressed\
| QAbstractItemView.SelectedClicked
self.treeView.setEditTriggers(triggers)
self.treeView.setDragDropOverwriteMode(True)
self.treeView.setDragDropMode(QAbstractItemView.DropOnly)
self.treeView.setUniformRowHeights(True)
self.verticalLayout.addWidget(self.treeView)
self.horizontalLayout = QHBoxLayout()
self.removeFolderButton = QPushButton(self.centralwidget)
self.removeFolderButton.setIcon(QIcon(QPixmap(":/minus")))
self.removeFolderButton.setShortcut("Del")
self.horizontalLayout.addWidget(self.removeFolderButton)
self.addFolderButton = QPushButton(self.centralwidget)
self.addFolderButton.setIcon(QIcon(QPixmap(":/plus")))
self.horizontalLayout.addWidget(self.addFolderButton)
spacerItem1 = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem1)
self.loadResultsButton = QPushButton(self.centralwidget)
self.loadResultsButton.setText(tr("Load Results"))
self.horizontalLayout.addWidget(self.loadResultsButton)
self.scanButton = QPushButton(self.centralwidget)
self.scanButton.setText(tr("Scan"))
self.scanButton.setDefault(True)
self.horizontalLayout.addWidget(self.scanButton)
self.verticalLayout.addLayout(self.horizontalLayout)
self.setCentralWidget(self.centralwidget)
self._setupActions()
self._setupMenu()
if self.app.prefs.directoriesWindowRect is not None:
self.setGeometry(self.app.prefs.directoriesWindowRect)
else:
moveToScreenCenter(self)
def _setupColumns(self):
header = self.treeView.header()
header.setStretchLastSection(False)
header.setSectionResizeMode(0, QHeaderView.Stretch)
header.setSectionResizeMode(1, QHeaderView.Fixed)
header.resizeSection(1, 100)
def _updateActionsState(self):
self.actionShowResultsWindow.setEnabled(self.app.resultWindow is not None)
def _updateAddButton(self):
if self.recentFolders.isEmpty():
self.addFolderButton.setMenu(None)
else:
self.addFolderButton.setMenu(self.menuRecentFolders)
def _updateRemoveButton(self):
indexes = self.treeView.selectedIndexes()
if not indexes:
self.removeFolderButton.setEnabled(False)
return
self.removeFolderButton.setEnabled(True)
def _updateLoadResultsButton(self):
if self.app.recentResults.isEmpty():
self.loadResultsButton.setMenu(None)
else:
self.loadResultsButton.setMenu(self.menuRecentResults)
def _updateScanTypeList(self):
try:
self.scanTypeComboBox.currentIndexChanged[int].disconnect(self.scanTypeChanged)
except TypeError:
# Not connected, ignore
pass
self.scanTypeComboBox.clear()
scan_options = self.app.model.SCANNER_CLASS.get_scan_options()
for scan_option in scan_options:
self.scanTypeComboBox.addItem(scan_option.label)
SCAN_TYPE_ORDER = [so.scan_type for so in scan_options]
selected_scan_type = self.app.prefs.get_scan_type(self.app.model.app_mode)
scan_type_index = SCAN_TYPE_ORDER.index(selected_scan_type)
self.scanTypeComboBox.setCurrentIndex(scan_type_index)
self.scanTypeComboBox.currentIndexChanged[int].connect(self.scanTypeChanged)
self.app._update_options()
#--- QWidget overrides
def closeEvent(self, event):
event.accept()
if self.app.model.results.is_modified:
title = tr("Unsaved results")
msg = tr("You have unsaved results, do you really want to quit?")
if not self.app.confirm(title, msg):
event.ignore()
if event.isAccepted():
QApplication.quit()
#--- Events
def addFolderTriggered(self):
title = tr("Select a folder to add to the scanning list")
flags = QFileDialog.ShowDirsOnly
dirpath = str(QFileDialog.getExistingDirectory(self, title, self.lastAddedFolder, flags))
if not dirpath:
return
self.lastAddedFolder = dirpath
self.app.model.add_directory(dirpath)
self.recentFolders.insertItem(dirpath)
def appModeButtonSelected(self, index):
if index == 2:
mode = AppMode.Picture
elif index == 1:
mode = AppMode.Music
else:
mode = AppMode.Standard
self.app.model.app_mode = mode
self._updateScanTypeList()
def appWillSavePrefs(self):
self.app.prefs.directoriesWindowRect = self.geometry()
def directoriesModelAddedFolders(self, folders):
for folder in folders:
self.recentFolders.insertItem(folder)
def loadResultsTriggered(self):
title = tr("Select a results file to load")
files = ';;'.join([tr("dupeGuru Results (*.dupeguru)"), tr("All Files (*.*)")])
destination = QFileDialog.getOpenFileName(self, title, '', files)
if destination:
self.app.model.load_from(destination)
self.app.recentResults.insertItem(destination)
def removeFolderButtonClicked(self):
self.directoriesModel.model.remove_selected()
def scanButtonClicked(self):
if self.app.model.results.is_modified:
title = tr("Start a new scan")
msg = tr("You have unsaved results, do you really want to continue?")
if not self.app.confirm(title, msg):
return
self.app.model.start_scanning()
def scanTypeChanged(self, index):
scan_options = self.app.model.SCANNER_CLASS.get_scan_options()
self.app.prefs.set_scan_type(self.app.model.app_mode, scan_options[index].scan_type)
self.app._update_options()
def selectionChanged(self, selected, deselected):
self._updateRemoveButton()
def _setupUi(self): # has to take place *before* base elements creation
self.setWindowTitle("moneyGuru")
self.resize(700, 580)
self.centralwidget = QWidget(self)
self.verticalLayout = QVBoxLayout(self.centralwidget)
self.verticalLayout.setSpacing(0)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.topBar = QWidget(self.centralwidget)
self.horizontalLayout_2 = QHBoxLayout(self.topBar)
self.horizontalLayout_2.setContentsMargins(2, 0, 2, 0)
spacerItem = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout_2.addItem(spacerItem)
self.dateRangeSelectorView = DateRangeSelectorView(self.topBar)
self.dateRangeSelectorView.setMinimumSize(QSize(220, 0))
self.horizontalLayout_2.addWidget(self.dateRangeSelectorView)
spacerItem1 = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout_2.addItem(spacerItem1)
self.searchLineEdit = SearchEdit(self.topBar)
self.searchLineEdit.setMaximumSize(QSize(240, 16777215))
self.horizontalLayout_2.addWidget(self.searchLineEdit)
self.verticalLayout.addWidget(self.topBar)
self.tabBar = TabBarPlus(self.centralwidget)
self.tabBar.setMinimumSize(QSize(0, 20))
self.verticalLayout.addWidget(self.tabBar)
self.mainView = QStackedWidget(self.centralwidget)
self.verticalLayout.addWidget(self.mainView)
# Bottom buttons & status label
self.bottomBar = QWidget(self.centralwidget)
self.horizontalLayout = QHBoxLayout(self.bottomBar)
self.horizontalLayout.setContentsMargins(2, 2, 2, 2)
self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
self.newItemButton = QPushButton(self.bottomBar)
buttonSizePolicy = QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
buttonSizePolicy.setHorizontalStretch(0)
buttonSizePolicy.setVerticalStretch(0)
buttonSizePolicy.setHeightForWidth(self.newItemButton.sizePolicy().hasHeightForWidth())
self.newItemButton.setSizePolicy(buttonSizePolicy)
self.newItemButton.setIcon(QIcon(QPixmap(':/plus_8')))
self.horizontalLayout.addWidget(self.newItemButton)
self.deleteItemButton = QPushButton(self.bottomBar)
self.deleteItemButton.setSizePolicy(buttonSizePolicy)
self.deleteItemButton.setIcon(QIcon(QPixmap(':/minus_8')))
self.horizontalLayout.addWidget(self.deleteItemButton)
self.editItemButton = QPushButton(self.bottomBar)
self.editItemButton.setSizePolicy(buttonSizePolicy)
self.editItemButton.setIcon(QIcon(QPixmap(':/info_gray_12')))
self.horizontalLayout.addWidget(self.editItemButton)
self.horizontalLayout.addItem(horizontalSpacer(size=20))
self.graphVisibilityButton = QPushButton()
self.graphVisibilityButton.setSizePolicy(buttonSizePolicy)
self.graphVisibilityButton.setIcon(QIcon(QPixmap(':/graph_visibility_on_16')))
self.horizontalLayout.addWidget(self.graphVisibilityButton)
self.piechartVisibilityButton = QPushButton()
self.piechartVisibilityButton.setSizePolicy(buttonSizePolicy)
self.piechartVisibilityButton.setIcon(QIcon(QPixmap(':/piechart_visibility_on_16')))
self.horizontalLayout.addWidget(self.piechartVisibilityButton)
self.columnsVisibilityButton = QPushButton()
self.columnsVisibilityButton.setSizePolicy(buttonSizePolicy)
self.columnsVisibilityButton.setIcon(QIcon(QPixmap(':/columns_16')))
self.horizontalLayout.addWidget(self.columnsVisibilityButton)
self.statusLabel = QLabel(tr("Status"))
self.statusLabel.setAlignment(Qt.AlignCenter)
self.horizontalLayout.addWidget(self.statusLabel)
self.verticalLayout.addWidget(self.bottomBar)
self.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 700, 20))
self.menuFile = QMenu(tr("File"))
self.menuOpenRecent = QMenu(tr("Open Recent"))
self.menuView = QMenu(tr("View"))
self.menuDateRange = QMenu(tr("Date Range"))
self.menuEdit = QMenu(tr("Edit"))
self.menuHelp = QMenu(tr("Help"))
self.setMenuBar(self.menubar)
self.actionOpenDocument = QAction(tr("Open..."), self)
self.actionOpenDocument.setShortcut("Ctrl+O")
self.actionShowNetWorth = QAction(tr("Net Worth"), self)
self.actionShowNetWorth.setShortcut("Ctrl+1")
self.actionShowNetWorth.setIcon(QIcon(QPixmap(':/balance_sheet_48')))
self.actionShowProfitLoss = QAction(escapeamp(tr("Profit & Loss")), self)
self.actionShowProfitLoss.setShortcut("Ctrl+2")
self.actionShowProfitLoss.setIcon(QIcon(QPixmap(':/income_statement_48')))
self.actionShowTransactions = QAction(tr("Transactions"), self)
self.actionShowTransactions.setShortcut("Ctrl+3")
self.actionShowTransactions.setIcon(QIcon(QPixmap(':/transaction_table_48')))
self.actionShowSelectedAccount = QAction(tr("Show Account"), self)
self.actionShowSelectedAccount.setShortcut("Ctrl+]")
self.actionNewItem = QAction(tr("New Item"), self)
self.actionNewItem.setShortcut("Ctrl+N")
self.actionDeleteItem = QAction(tr("Remove Selected"), self)
self.actionEditItem = QAction(tr("Show Info"), self)
self.actionEditItem.setShortcut("Ctrl+I")
self.actionToggleGraph = QAction(tr("Toggle Graph"), self)
self.actionToggleGraph.setShortcut("Ctrl+Alt+G")
self.actionTogglePieChart = QAction(tr("Toggle Pie Chart"), self)
self.actionTogglePieChart.setShortcut("Ctrl+Alt+P")
self.actionMoveUp = QAction(tr("Move Up"), self)
self.actionMoveUp.setShortcut("Ctrl++")
self.actionMoveDown = QAction(tr("Move Down"), self)
self.actionMoveDown.setShortcut("Ctrl+-")
self.actionNavigateBack = QAction(tr("Go Back"), self)
self.actionNavigateBack.setShortcut("Ctrl+[")
self.actionNewAccountGroup = QAction(tr("New Account Group"), self)
self.actionNewAccountGroup.setShortcut("Ctrl+Shift+N")
self.actionShowNextView = QAction(tr("Next View"), self)
self.actionShowNextView.setShortcut("Ctrl+Shift+]")
self.actionShowPreviousView = QAction(tr("Previous View"), self)
self.actionShowPreviousView.setShortcut("Ctrl+Shift+[")
self.actionNewDocument = QAction(tr("New Document"), self)
self.actionImport = QAction(tr("Import..."), self)
self.actionImport.setShortcut("Ctrl+Alt+I")
self.actionExport = QAction(tr("Export..."), self)
self.actionExport.setShortcut("Ctrl+Alt+E")
self.actionSave = QAction(tr("Save"), self)
self.actionSave.setShortcut("Ctrl+S")
self.actionSaveAs = QAction(tr("Save As..."), self)
self.actionSaveAs.setShortcut("Ctrl+Shift+S")
self.actionAbout = QAction(tr("About moneyGuru"), self)
self.actionToggleReconciliationMode = QAction(tr("Toggle Reconciliation Mode"), self)
self.actionToggleReconciliationMode.setShortcut("Ctrl+Shift+R")
self.actionToggleAccountExclusion = QAction(tr("Toggle Exclusion Status of Account"), self)
self.actionToggleAccountExclusion.setShortcut("Ctrl+Shift+X")
self.actionShowSchedules = QAction(tr("Schedules"), self)
self.actionShowSchedules.setShortcut("Ctrl+4")
self.actionShowSchedules.setIcon(QIcon(QPixmap(':/schedules_48')))
self.actionReconcileSelected = QAction(tr("Reconcile Selection"), self)
self.actionReconcileSelected.setShortcut("Ctrl+R")
self.actionMakeScheduleFromSelected = QAction(tr("Make Schedule from Selected"), self)
self.actionMakeScheduleFromSelected.setShortcut("Ctrl+M")
self.actionShowPreferences = QAction(tr("Preferences..."), self)
self.actionPrint = QAction(tr("Print..."), self)
self.actionPrint.setShortcut("Ctrl+P")
self.actionQuit = QAction(tr("Quit moneyGuru"), self)
self.actionQuit.setShortcut("Ctrl+Q")
self.actionUndo = QAction(tr("Undo"), self)
self.actionUndo.setShortcut("Ctrl+Z")
self.actionRedo = QAction(tr("Redo"), self)
self.actionRedo.setShortcut("Ctrl+Y")
self.actionShowHelp = QAction(tr("moneyGuru Help"), self)
self.actionShowHelp.setShortcut("F1")
self.actionDuplicateTransaction = QAction(tr("Duplicate Transaction"), self)
self.actionDuplicateTransaction.setShortcut("Ctrl+D")
self.actionJumpToAccount = QAction(tr("Jump to Account..."), self)
self.actionJumpToAccount.setShortcut("Ctrl+Shift+A")
self.actionNewTab = QAction(tr("New Tab"), self)
self.actionNewTab.setShortcut("Ctrl+T")
self.actionCloseTab = QAction(tr("Close Tab"), self)
self.actionCloseTab.setShortcut("Ctrl+W")
self.menuFile.addAction(self.actionNewDocument)
self.menuFile.addAction(self.actionNewTab)
self.menuFile.addAction(self.actionOpenDocument)
self.menuFile.addAction(self.menuOpenRecent.menuAction())
self.menuFile.addAction(self.actionImport)
self.menuFile.addSeparator()
self.menuFile.addAction(self.actionCloseTab)
self.menuFile.addAction(self.actionSave)
self.menuFile.addAction(self.actionSaveAs)
self.menuFile.addAction(self.actionExport)
self.menuFile.addAction(self.actionPrint)
self.menuFile.addAction(self.actionQuit)
self.menuView.addAction(self.actionShowNetWorth)
self.menuView.addAction(self.actionShowProfitLoss)
self.menuView.addAction(self.actionShowTransactions)
self.menuView.addAction(self.actionShowSchedules)
self.menuView.addAction(self.actionShowPreviousView)
self.menuView.addAction(self.actionShowNextView)
self.menuView.addAction(self.menuDateRange.menuAction())
self.menuView.addAction(self.actionShowPreferences)
self.menuView.addAction(self.actionToggleGraph)
self.menuView.addAction(self.actionTogglePieChart)
self.menuEdit.addAction(self.actionNewItem)
self.menuEdit.addAction(self.actionNewAccountGroup)
self.menuEdit.addAction(self.actionDeleteItem)
self.menuEdit.addAction(self.actionEditItem)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionMoveUp)
self.menuEdit.addAction(self.actionMoveDown)
self.menuEdit.addAction(self.actionDuplicateTransaction)
self.menuEdit.addAction(self.actionMakeScheduleFromSelected)
self.menuEdit.addAction(self.actionReconcileSelected)
self.menuEdit.addAction(self.actionToggleReconciliationMode)
self.menuEdit.addAction(self.actionToggleAccountExclusion)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionShowSelectedAccount)
self.menuEdit.addAction(self.actionNavigateBack)
self.menuEdit.addAction(self.actionJumpToAccount)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionUndo)
self.menuEdit.addAction(self.actionRedo)
self.menuHelp.addAction(self.actionShowHelp)
self.menuHelp.addAction(self.actionAbout)
mainmenus = [self.menuFile, self.menuEdit, self.menuView, self.menuHelp]
for menu in mainmenus:
self.menubar.addAction(menu.menuAction())
setAccelKeys(menu)
setAccelKeys(self.menubar)
self.tabBar.setMovable(True)
self.tabBar.setTabsClosable(True)
self.tabBar.setExpanding(False)
seq = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Right)
self._shortcutNextTab = QShortcut(seq, self)
seq = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Left)
self._shortcutPrevTab = QShortcut(seq, self)
class MainWindow(QMainWindow):
def __init__(self, app):
QMainWindow.__init__(self, None)
self.documentPath = None
self.doc = DocumentModel(app=app.model)
self.app = app
self._setupUi()
# Create base elements
self.model = MainWindowModel(document=self.doc)
self.model2view = {}
self.alookup = Lookup(self, model=self.model.account_lookup)
self.clookup = Lookup(self, model=self.model.completion_lookup)
self.drsel = DateRangeSelector(mainwindow=self, view=self.dateRangeSelectorView)
self.sfield = SearchField(model=self.model.search_field, view=self.searchLineEdit)
self.recentDocuments = Recent(self.app, 'recentDocuments')
self.recentDocuments.addMenu(self.menuOpenRecent)
self.doc.view = self
self.model.view = self
self._updateUndoActions()
self._bindSignals()
def _setupUi(self): # has to take place *before* base elements creation
self.setWindowTitle("moneyGuru")
self.resize(700, 580)
self.centralwidget = QWidget(self)
self.verticalLayout = QVBoxLayout(self.centralwidget)
self.verticalLayout.setSpacing(0)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.topBar = QWidget(self.centralwidget)
self.horizontalLayout_2 = QHBoxLayout(self.topBar)
self.horizontalLayout_2.setContentsMargins(2, 0, 2, 0)
spacerItem = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout_2.addItem(spacerItem)
self.dateRangeSelectorView = DateRangeSelectorView(self.topBar)
self.dateRangeSelectorView.setMinimumSize(QSize(220, 0))
self.horizontalLayout_2.addWidget(self.dateRangeSelectorView)
spacerItem1 = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout_2.addItem(spacerItem1)
self.searchLineEdit = SearchEdit(self.topBar)
self.searchLineEdit.setMaximumSize(QSize(240, 16777215))
self.horizontalLayout_2.addWidget(self.searchLineEdit)
self.verticalLayout.addWidget(self.topBar)
self.tabBar = TabBarPlus(self.centralwidget)
self.tabBar.setMinimumSize(QSize(0, 20))
self.verticalLayout.addWidget(self.tabBar)
self.mainView = QStackedWidget(self.centralwidget)
self.verticalLayout.addWidget(self.mainView)
# Bottom buttons & status label
self.bottomBar = QWidget(self.centralwidget)
self.horizontalLayout = QHBoxLayout(self.bottomBar)
self.horizontalLayout.setContentsMargins(2, 2, 2, 2)
self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
self.newItemButton = QPushButton(self.bottomBar)
buttonSizePolicy = QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
buttonSizePolicy.setHorizontalStretch(0)
buttonSizePolicy.setVerticalStretch(0)
buttonSizePolicy.setHeightForWidth(self.newItemButton.sizePolicy().hasHeightForWidth())
self.newItemButton.setSizePolicy(buttonSizePolicy)
self.newItemButton.setIcon(QIcon(QPixmap(':/plus_8')))
self.horizontalLayout.addWidget(self.newItemButton)
self.deleteItemButton = QPushButton(self.bottomBar)
self.deleteItemButton.setSizePolicy(buttonSizePolicy)
self.deleteItemButton.setIcon(QIcon(QPixmap(':/minus_8')))
self.horizontalLayout.addWidget(self.deleteItemButton)
self.editItemButton = QPushButton(self.bottomBar)
self.editItemButton.setSizePolicy(buttonSizePolicy)
self.editItemButton.setIcon(QIcon(QPixmap(':/info_gray_12')))
self.horizontalLayout.addWidget(self.editItemButton)
self.horizontalLayout.addItem(horizontalSpacer(size=20))
self.graphVisibilityButton = QPushButton()
self.graphVisibilityButton.setSizePolicy(buttonSizePolicy)
self.graphVisibilityButton.setIcon(QIcon(QPixmap(':/graph_visibility_on_16')))
self.horizontalLayout.addWidget(self.graphVisibilityButton)
self.piechartVisibilityButton = QPushButton()
self.piechartVisibilityButton.setSizePolicy(buttonSizePolicy)
self.piechartVisibilityButton.setIcon(QIcon(QPixmap(':/piechart_visibility_on_16')))
self.horizontalLayout.addWidget(self.piechartVisibilityButton)
self.columnsVisibilityButton = QPushButton()
self.columnsVisibilityButton.setSizePolicy(buttonSizePolicy)
self.columnsVisibilityButton.setIcon(QIcon(QPixmap(':/columns_16')))
self.horizontalLayout.addWidget(self.columnsVisibilityButton)
self.statusLabel = QLabel(tr("Status"))
self.statusLabel.setAlignment(Qt.AlignCenter)
self.horizontalLayout.addWidget(self.statusLabel)
self.verticalLayout.addWidget(self.bottomBar)
self.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 700, 20))
self.menuFile = QMenu(tr("File"))
self.menuOpenRecent = QMenu(tr("Open Recent"))
self.menuView = QMenu(tr("View"))
self.menuDateRange = QMenu(tr("Date Range"))
self.menuEdit = QMenu(tr("Edit"))
self.menuHelp = QMenu(tr("Help"))
self.setMenuBar(self.menubar)
self.actionOpenDocument = QAction(tr("Open..."), self)
self.actionOpenDocument.setShortcut("Ctrl+O")
self.actionShowNetWorth = QAction(tr("Net Worth"), self)
self.actionShowNetWorth.setShortcut("Ctrl+1")
self.actionShowNetWorth.setIcon(QIcon(QPixmap(':/balance_sheet_48')))
self.actionShowProfitLoss = QAction(escapeamp(tr("Profit & Loss")), self)
self.actionShowProfitLoss.setShortcut("Ctrl+2")
self.actionShowProfitLoss.setIcon(QIcon(QPixmap(':/income_statement_48')))
self.actionShowTransactions = QAction(tr("Transactions"), self)
self.actionShowTransactions.setShortcut("Ctrl+3")
self.actionShowTransactions.setIcon(QIcon(QPixmap(':/transaction_table_48')))
self.actionShowSelectedAccount = QAction(tr("Show Account"), self)
self.actionShowSelectedAccount.setShortcut("Ctrl+]")
self.actionNewItem = QAction(tr("New Item"), self)
self.actionNewItem.setShortcut("Ctrl+N")
self.actionDeleteItem = QAction(tr("Remove Selected"), self)
self.actionEditItem = QAction(tr("Show Info"), self)
self.actionEditItem.setShortcut("Ctrl+I")
self.actionToggleGraph = QAction(tr("Toggle Graph"), self)
self.actionToggleGraph.setShortcut("Ctrl+Alt+G")
self.actionTogglePieChart = QAction(tr("Toggle Pie Chart"), self)
self.actionTogglePieChart.setShortcut("Ctrl+Alt+P")
self.actionMoveUp = QAction(tr("Move Up"), self)
self.actionMoveUp.setShortcut("Ctrl++")
self.actionMoveDown = QAction(tr("Move Down"), self)
self.actionMoveDown.setShortcut("Ctrl+-")
self.actionNavigateBack = QAction(tr("Go Back"), self)
self.actionNavigateBack.setShortcut("Ctrl+[")
self.actionNewAccountGroup = QAction(tr("New Account Group"), self)
self.actionNewAccountGroup.setShortcut("Ctrl+Shift+N")
self.actionShowNextView = QAction(tr("Next View"), self)
self.actionShowNextView.setShortcut("Ctrl+Shift+]")
self.actionShowPreviousView = QAction(tr("Previous View"), self)
self.actionShowPreviousView.setShortcut("Ctrl+Shift+[")
self.actionNewDocument = QAction(tr("New Document"), self)
self.actionImport = QAction(tr("Import..."), self)
self.actionImport.setShortcut("Ctrl+Alt+I")
self.actionExport = QAction(tr("Export..."), self)
self.actionExport.setShortcut("Ctrl+Alt+E")
self.actionSave = QAction(tr("Save"), self)
self.actionSave.setShortcut("Ctrl+S")
self.actionSaveAs = QAction(tr("Save As..."), self)
self.actionSaveAs.setShortcut("Ctrl+Shift+S")
self.actionAbout = QAction(tr("About moneyGuru"), self)
self.actionToggleReconciliationMode = QAction(tr("Toggle Reconciliation Mode"), self)
self.actionToggleReconciliationMode.setShortcut("Ctrl+Shift+R")
self.actionToggleAccountExclusion = QAction(tr("Toggle Exclusion Status of Account"), self)
self.actionToggleAccountExclusion.setShortcut("Ctrl+Shift+X")
self.actionShowSchedules = QAction(tr("Schedules"), self)
self.actionShowSchedules.setShortcut("Ctrl+4")
self.actionShowSchedules.setIcon(QIcon(QPixmap(':/schedules_48')))
self.actionReconcileSelected = QAction(tr("Reconcile Selection"), self)
self.actionReconcileSelected.setShortcut("Ctrl+R")
self.actionMakeScheduleFromSelected = QAction(tr("Make Schedule from Selected"), self)
self.actionMakeScheduleFromSelected.setShortcut("Ctrl+M")
self.actionShowPreferences = QAction(tr("Preferences..."), self)
self.actionPrint = QAction(tr("Print..."), self)
self.actionPrint.setShortcut("Ctrl+P")
self.actionQuit = QAction(tr("Quit moneyGuru"), self)
self.actionQuit.setShortcut("Ctrl+Q")
self.actionUndo = QAction(tr("Undo"), self)
self.actionUndo.setShortcut("Ctrl+Z")
self.actionRedo = QAction(tr("Redo"), self)
self.actionRedo.setShortcut("Ctrl+Y")
self.actionShowHelp = QAction(tr("moneyGuru Help"), self)
self.actionShowHelp.setShortcut("F1")
self.actionDuplicateTransaction = QAction(tr("Duplicate Transaction"), self)
self.actionDuplicateTransaction.setShortcut("Ctrl+D")
self.actionJumpToAccount = QAction(tr("Jump to Account..."), self)
self.actionJumpToAccount.setShortcut("Ctrl+Shift+A")
self.actionNewTab = QAction(tr("New Tab"), self)
self.actionNewTab.setShortcut("Ctrl+T")
self.actionCloseTab = QAction(tr("Close Tab"), self)
self.actionCloseTab.setShortcut("Ctrl+W")
self.menuFile.addAction(self.actionNewDocument)
self.menuFile.addAction(self.actionNewTab)
self.menuFile.addAction(self.actionOpenDocument)
self.menuFile.addAction(self.menuOpenRecent.menuAction())
self.menuFile.addAction(self.actionImport)
self.menuFile.addSeparator()
self.menuFile.addAction(self.actionCloseTab)
self.menuFile.addAction(self.actionSave)
self.menuFile.addAction(self.actionSaveAs)
self.menuFile.addAction(self.actionExport)
self.menuFile.addAction(self.actionPrint)
self.menuFile.addAction(self.actionQuit)
self.menuView.addAction(self.actionShowNetWorth)
self.menuView.addAction(self.actionShowProfitLoss)
self.menuView.addAction(self.actionShowTransactions)
self.menuView.addAction(self.actionShowSchedules)
self.menuView.addAction(self.actionShowPreviousView)
self.menuView.addAction(self.actionShowNextView)
self.menuView.addAction(self.menuDateRange.menuAction())
self.menuView.addAction(self.actionShowPreferences)
self.menuView.addAction(self.actionToggleGraph)
self.menuView.addAction(self.actionTogglePieChart)
self.menuEdit.addAction(self.actionNewItem)
self.menuEdit.addAction(self.actionNewAccountGroup)
self.menuEdit.addAction(self.actionDeleteItem)
self.menuEdit.addAction(self.actionEditItem)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionMoveUp)
self.menuEdit.addAction(self.actionMoveDown)
self.menuEdit.addAction(self.actionDuplicateTransaction)
self.menuEdit.addAction(self.actionMakeScheduleFromSelected)
self.menuEdit.addAction(self.actionReconcileSelected)
self.menuEdit.addAction(self.actionToggleReconciliationMode)
self.menuEdit.addAction(self.actionToggleAccountExclusion)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionShowSelectedAccount)
self.menuEdit.addAction(self.actionNavigateBack)
self.menuEdit.addAction(self.actionJumpToAccount)
self.menuEdit.addSeparator()
self.menuEdit.addAction(self.actionUndo)
self.menuEdit.addAction(self.actionRedo)
self.menuHelp.addAction(self.actionShowHelp)
self.menuHelp.addAction(self.actionAbout)
mainmenus = [self.menuFile, self.menuEdit, self.menuView, self.menuHelp]
for menu in mainmenus:
self.menubar.addAction(menu.menuAction())
setAccelKeys(menu)
setAccelKeys(self.menubar)
self.tabBar.setMovable(True)
self.tabBar.setTabsClosable(True)
self.tabBar.setExpanding(False)
seq = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Right)
self._shortcutNextTab = QShortcut(seq, self)
seq = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Left)
self._shortcutPrevTab = QShortcut(seq, self)
def _bindSignals(self):
self.newItemButton.clicked.connect(self.actionNewItem.trigger)
self.deleteItemButton.clicked.connect(self.actionDeleteItem.trigger)
self.editItemButton.clicked.connect(self.actionEditItem.trigger)
self.graphVisibilityButton.clicked.connect(self.actionToggleGraph.trigger)
self.piechartVisibilityButton.clicked.connect(self.actionTogglePieChart.trigger)
self.columnsVisibilityButton.clicked.connect(self.columnsVisibilityButtonClicked)
self.recentDocuments.mustOpenItem.connect(self.open)
self.tabBar.currentChanged.connect(self.currentTabChanged)
self.tabBar.tabCloseRequested.connect(self.tabCloseRequested)
self.tabBar.tabMoved.connect(self.tabMoved)
self.tabBar.plusClicked.connect(self.model.new_tab)
# Views
self.actionShowNetWorth.triggered.connect(self.showNetWorthTriggered)
self.actionShowProfitLoss.triggered.connect(self.showProfitLossTriggered)
self.actionShowTransactions.triggered.connect(self.showTransactionsTriggered)
self.actionShowSchedules.triggered.connect(self.showSchedulesTriggered)
self.actionShowPreviousView.triggered.connect(self.showPreviousViewTriggered)
self.actionShowNextView.triggered.connect(self.showNextViewTriggered)
self.actionShowPreferences.triggered.connect(self.app.showPreferences)
self.actionToggleGraph.triggered.connect(self.toggleGraphTriggered)
self.actionTogglePieChart.triggered.connect(self.togglePieChartTriggered)
# Document Edition
self.actionNewItem.triggered.connect(self.newItemTriggered)
self.actionNewAccountGroup.triggered.connect(self.newAccountGroupTriggered)
self.actionDeleteItem.triggered.connect(self.deleteItemTriggered)
self.actionEditItem.triggered.connect(self.editItemTriggered)
self.actionMoveUp.triggered.connect(self.moveUpTriggered)
self.actionMoveDown.triggered.connect(self.moveDownTriggered)
self.actionDuplicateTransaction.triggered.connect(self.model.duplicate_item)
self.actionUndo.triggered.connect(self.model.undo)
self.actionRedo.triggered.connect(self.model.redo)
# Open / Save / Import / Export / New
self.actionNewDocument.triggered.connect(self.new)
self.actionOpenDocument.triggered.connect(self.openDocument)
self.actionImport.triggered.connect(self.importDocument)
self.actionSave.triggered.connect(self.save)
self.actionSaveAs.triggered.connect(self.saveAs)
self.actionExport.triggered.connect(self.model.export)
# Misc
self.actionNewTab.triggered.connect(self.model.new_tab)
self.actionCloseTab.triggered.connect(self.closeTabTriggered)
self.actionShowSelectedAccount.triggered.connect(self.model.show_account)
self.actionNavigateBack.triggered.connect(self.navigateBackTriggered)
self.actionJumpToAccount.triggered.connect(self.jumpToAccountTriggered)
self.actionMakeScheduleFromSelected.triggered.connect(self.makeScheduleFromSelectedTriggered)
self.actionReconcileSelected.triggered.connect(self.reconcileSelectedTriggered)
self.actionToggleReconciliationMode.triggered.connect(self.toggleReconciliationModeTriggered)
self.actionToggleAccountExclusion.triggered.connect(self.toggleAccountExclusionTriggered)
self.actionPrint.triggered.connect(self._print)
self.actionShowHelp.triggered.connect(self.app.showHelp)
self.actionAbout.triggered.connect(self.aboutTriggered)
self.actionQuit.triggered.connect(self.close)
# Extra Shortcuts
self._shortcutNextTab.activated.connect(self.showNextViewTriggered)
self._shortcutPrevTab.activated.connect(self.showPreviousViewTriggered)
# --- QWidget overrides
def closeEvent(self, event):
if self.confirmDestructiveAction():
event.accept()
else:
event.ignore()
# --- Private
def _print(self):
dialog = QPrintDialog(self)
if dialog.exec_() != QPrintDialog.Accepted:
return
printer = dialog.printer()
currentView = self.mainView.currentWidget()
viewPrinter = ViewPrinter(printer, currentView)
currentView.fitViewsForPrint(viewPrinter)
viewPrinter.render()
def _getViewforPane(self, pane_type, pane_view):
if pane_view in self.model2view:
view = self.model2view[pane_view]
else:
view = PANETYPE2VIEWCLASS[pane_type](model=pane_view, mainwindow=self)
self.model2view[pane_view] = view
self.mainView.addWidget(view)
view.restoreSubviewsSize()
return view
def _setTabIndex(self, index):
if not self.tabBar.count():
return
self.tabBar.setCurrentIndex(index)
self._updateActionsState()
pane_type = self.model.pane_type(index)
pane_view = self.model.pane_view(index)
view = self._getViewforPane(pane_type, pane_view)
self.mainView.setCurrentWidget(view)
view.setFocus()
def _activeView(self):
paneIndex = self.model.current_pane_index
return self.model.pane_view(paneIndex)
def _updateActionsState(self):
# Updates enable/disable checked/unchecked state of all actions. These state can change
# under various conditions: main view change, date range type change and when reconciliation
# mode is toggled
# Determine what actions are enabled
view = self._activeView()
viewType = view.VIEW_TYPE
isSheet = viewType in {PaneType.NetWorth, PaneType.Profit}
isTransactionOrEntryTable = viewType in {PaneType.Transaction, PaneType.Account}
canToggleReconciliation = viewType == PaneType.Account and view.can_toggle_reconciliation_mode
newItemLabel = {
PaneType.NetWorth: tr("New Account"),
PaneType.Profit: tr("New Account"),
PaneType.Transaction: tr("New Transaction"),
PaneType.Account: tr("New Transaction"),
PaneType.Schedule: tr("New Schedule"),
PaneType.GeneralLedger: tr("New Transaction"),
}.get(viewType, tr("New Item")) # XXX make "New Item" disabled
self.actionNewItem.setText(newItemLabel)
self.actionNewAccountGroup.setEnabled(isSheet)
self.actionMoveDown.setEnabled(isTransactionOrEntryTable)
self.actionMoveUp.setEnabled(isTransactionOrEntryTable)
self.actionDuplicateTransaction.setEnabled(isTransactionOrEntryTable)
self.actionMakeScheduleFromSelected.setEnabled(isTransactionOrEntryTable)
self.actionReconcileSelected.setEnabled(viewType == PaneType.Account and view.reconciliation_mode)
self.actionShowNextView.setEnabled(self.model.current_pane_index < self.model.pane_count-1)
self.actionShowPreviousView.setEnabled(self.model.current_pane_index > 0)
self.actionShowSelectedAccount.setEnabled(isSheet or isTransactionOrEntryTable)
self.actionNavigateBack.setEnabled(viewType == PaneType.Account)
self.actionToggleReconciliationMode.setEnabled(canToggleReconciliation)
self.actionToggleAccountExclusion.setEnabled(isSheet)
def _updateUndoActions(self):
if self.doc.can_undo():
self.actionUndo.setEnabled(True)
self.actionUndo.setText(tr("Undo {0}").format(self.doc.undo_description()))
else:
self.actionUndo.setEnabled(False)
self.actionUndo.setText(tr("Undo"))
if self.doc.can_redo():
self.actionRedo.setEnabled(True)
self.actionRedo.setText(tr("Redo {0}").format(self.doc.redo_description()))
else:
self.actionRedo.setEnabled(False)
self.actionRedo.setText(tr("Redo"))
# --- Actions
# Document open/save/close
def confirmDestructiveAction(self):
# Asks whether the user wants to continue before continuing with an action that will replace
# the current document. Will save the document as needed. Returns True if the action can
# continue.
if not self.model.document.is_dirty():
return True
title = tr("Unsaved Document")
msg = tr("Do you want to save your changes before continuing?")
buttons = QMessageBox.Save | QMessageBox.Cancel | QMessageBox.Discard
result = QMessageBox.question(self.app.mainWindow, title, msg, buttons)
if result == QMessageBox.Save:
self.save()
if self.model.document.is_dirty(): # "save as" was cancelled
return False
else:
return True
elif result == QMessageBox.Cancel:
return False
elif result == QMessageBox.Discard:
return True
def new(self):
if not self.confirmDestructiveAction():
return
self.model.close()
self.documentPath = None
self.model.clear()
self.documentPathChanged()
def open(self, docpath, initial=False):
# initial flag is true when open() is called at the document's initialization. When that's
# the case, we need to create a new document when we fail opening this one.
if not self.confirmDestructiveAction():
return
self.model.close()
try:
self.model.load_from_xml(docpath)
self.documentPath = docpath
except FileFormatError as e:
QMessageBox.warning(self.app.mainWindow, tr("Cannot load file"), str(e))
if initial:
self.new()
self.documentPathChanged()
self.recentDocuments.insertItem(docpath)
def openDocument(self):
title = tr("Select a document to load")
filters = tr("moneyGuru Documents (*.moneyguru)")
docpath, filetype = QFileDialog.getOpenFileName(self.app.mainWindow, title, '', filters)
if docpath:
self.open(docpath)
def save(self):
if self.documentPath is not None:
self.model.save_to_xml(self.documentPath)
else:
self.saveAs()
def saveAs(self):
title = tr("Save As")
filters = tr("moneyGuru Documents (*.moneyguru)")
docpath = QFileDialog.getSaveFileName(self.app.mainWindow, title, '', filters)[0]
if docpath:
if not docpath.endswith('.moneyguru'):
docpath += '.moneyguru'
self.model.save_to_xml(docpath)
self.documentPath = docpath
self.documentPathChanged()
self.recentDocuments.insertItem(docpath)
# Views
def showNetWorthTriggered(self):
self.model.select_pane_of_type(PaneType.NetWorth)
def showProfitLossTriggered(self):
self.model.select_pane_of_type(PaneType.Profit)
def showTransactionsTriggered(self):
self.model.select_pane_of_type(PaneType.Transaction)
def showSchedulesTriggered(self):
self.model.select_pane_of_type(PaneType.Schedule)
def showPreviousViewTriggered(self):
self.model.select_previous_view()
def showNextViewTriggered(self):
self.model.select_next_view()
# Document Edition
def newItemTriggered(self):
self.model.new_item()
def newAccountGroupTriggered(self):
self.model.new_group()
def deleteItemTriggered(self):
self.model.delete_item()
def editItemTriggered(self):
self.model.edit_item()
def moveUpTriggered(self):
self.model.move_up()
def moveDownTriggered(self):
self.model.move_down()
# Misc
def closeTabTriggered(self):
self.model.close_pane(self.model.current_pane_index)
def navigateBackTriggered(self):
self.model.navigate_back()
def jumpToAccountTriggered(self):
self.model.jump_to_account()
def makeScheduleFromSelectedTriggered(self):
self.model.make_schedule_from_selected()
def reconcileSelectedTriggered(self):
self._activeView().etable.toggle_reconciled()
def toggleReconciliationModeTriggered(self):
self._activeView().toggle_reconciliation_mode()
self._updateActionsState()
def toggleAccountExclusionTriggered(self):
viewType = self.model.pane_type(self.model.current_pane_index)
if viewType in {PaneType.NetWorth, PaneType.Profit}:
self._activeView().sheet.toggle_excluded()
def toggleGraphTriggered(self):
self.model.toggle_area_visibility(PaneArea.BottomGraph)
def togglePieChartTriggered(self):
self.model.toggle_area_visibility(PaneArea.RightChart)
def columnsVisibilityButtonClicked(self):
items = self.model.column_menu_items()
if not items:
return
menu = QMenu()
for i, (display, marked) in enumerate(items):
action = menu.addAction(display)
action.setCheckable(True)
action.setChecked(marked)
action.setData(i)
action.triggered.connect(self.columnsMenuItemWasClicked)
self._columnMenuHolder = menu # we need to hold a reference to it while it popups
button = self.columnsVisibilityButton
menu.popup(button.parentWidget().mapToGlobal(button.geometry().topLeft()))
def columnsMenuItemWasClicked(self):
action = self.sender()
if action is not None:
index = action.data()
self.model.toggle_column_menu_item(index)
def aboutTriggered(self):
self.app.showAboutBox()
def importDocument(self):
title = tr("Select a document to import")
filters = tr("Supported files (*.moneyguru *.ofx *.qfx *.qif *.csv *.txt);;All files (*)")
docpath, filetype = QFileDialog.getOpenFileName(self.app.mainWindow, title, '', filters)
# There's a strange glitch under GNOME where, right after the dialog is gone, the main
# window isn't the active window, but it will become active if we give it enough time. If we
# start showing the import window before that happens, we'll end up with an import window
# under the main window, which is bad. Therefore, we process events until this happens. We
# do this in a big forloop instead of a while to avoid a possible infinite loop.
for i in range(10000):
if self.app.mainWindow.isActiveWindow():
break
QApplication.processEvents()
if docpath:
try:
self.model.parse_file_for_import(docpath)
except FileFormatError as e:
QMessageBox.warning(self.app.mainWindow, tr("Cannot import file"), str(e))
# --- Other Signals
def currentTabChanged(self, index):
self.model.current_pane_index = index
self._setTabIndex(index)
def documentPathChanged(self):
if self.documentPath:
title = "moneyGuru ({})".format(self.documentPath)
else:
title = "moneyGuru"
self.setWindowTitle(title)
def tabCloseRequested(self, index):
self.model.close_pane(index)
def tabMoved(self, fromIndex, toIndex):
# We don't refresh panes because tabMoved is apparently now called *during* drag operations.
# If we start a full pane refresh during a drag operation, we segfault.
self.model.move_pane(fromIndex, toIndex, refresh_panes=False)
# --- document model --> view
def query_for_schedule_scope(self):
if QApplication.keyboardModifiers() & Qt.ShiftModifier:
return ScheduleScope.Global
if not self.app.model.show_schedule_scope_dialog:
return ScheduleScope.Local
dialog = ScheduleScopeDialog(self)
return dialog.queryForScope()
# --- model --> view
def change_current_pane(self):
self._setTabIndex(self.model.current_pane_index)
def get_panel_view(self, model):
if isinstance(model, CustomDateRangePanelModel):
return CustomDateRangePanel(model, self)
elif isinstance(model, CSVOptionsModel):
return CSVOptionsWindow(model, self)
elif isinstance(model, ImportWindowModel):
return ImportWindow(model, self, self.app.prefs)
else:
return ExportPanel(model, self)
def refresh_panes(self):
while self.tabBar.count() < self.model.pane_count:
self.tabBar.addTab('')
for i in range(self.model.pane_count):
pane_label = self.model.pane_label(i)
pane_label = escapeamp(pane_label)
self.tabBar.setTabText(i, pane_label)
pane_type = self.model.pane_type(i)
pane_view = self.model.pane_view(i)
# Ensure that the view's "view" has been created and bound
self._getViewforPane(pane_type, pane_view)
iconname = PANETYPE2ICON.get(pane_type)
icon = QIcon(QPixmap(':/{0}'.format(iconname))) if iconname else QIcon()
self.tabBar.setTabIcon(i, icon)
# It's important that we proceed with tab removal *after* we've completed tab initialization.
# We're walking on eggshells here. refresh_panes() can be called in multiple situations, one
# of them is during the opening of a document. When that happens when another document was
# previously opened, all views' model are uninitalized and don't have their "view" attribute
# set yet. If we proceed with the setCurrentIndex() call below before _getViewforPane()
# could be called above, we get a crash.
if self.tabBar.currentIndex() >= self.model.pane_count:
# Normally, we don't touch the tabBar index here and wait for change_current_pane,
# but when we remove tabs, it's possible that currentTabChanged end up being called and
# then the tab selection is bugged. I tried disconnecting/reconnecting the signal, but
# this is buggy. So when a selected tab is about to be removed and is out of bounds,
# we change the selection to the last index in the model. We don't use
# self.model.current_pane_index because in some cases, it's -1 and prevents this crash
# preventer from preventing its crash.
self.tabBar.setCurrentIndex(self.model.pane_count - 1)
while self.tabBar.count() > self.model.pane_count:
self.tabBar.removeTab(self.tabBar.count()-1)
self.tabBar.setTabsClosable(self.model.pane_count > 1)
def refresh_status_line(self):
self.statusLabel.setText(self.model.status_line)
def refresh_undo_actions(self):
self._updateUndoActions()
def restore_window_frame(self, frame):
self.setGeometry(*frame)
def save_window_frame(self):
r = self.geometry()
return (r.x(), r.y(), r.width(), r.height())
def show_message(self, msg):
title = tr("Warning")
QMessageBox.warning(self, title, msg)
def update_area_visibility(self):
hidden = self.model.hidden_areas
graphimg = ':/graph_visibility_{}_16'.format('off' if PaneArea.BottomGraph in hidden else 'on')
pieimg = ':/piechart_visibility_{}_16'.format('off' if PaneArea.RightChart in hidden else 'on')
self.graphVisibilityButton.setIcon(QIcon(QPixmap(graphimg)))
self.piechartVisibilityButton.setIcon(QIcon(QPixmap(pieimg)))
def view_closed(self, index):
self.tabBar.removeTab(index)
self.tabBar.setTabsClosable(self.model.pane_count > 1)
class BackgroundPlotter(QtInteractor):
ICON_TIME_STEP = 5.0
def __init__(self,
show=True,
app=None,
shape=(1, 1),
window_size=None,
off_screen=None,
**kwargs):
if not has_pyqt:
raise AssertionError('Requires PyQt5')
self.active = True
self.saved_camera_positions = []
self.counters = []
if window_size is None:
window_size = rcParams['window_size']
# Remove notebook argument in case user passed it
kwargs.pop('notebook', None)
# ipython magic
if scooby.in_ipython(): # pragma: no cover
from IPython import get_ipython
ipython = get_ipython()
ipython.magic('gui qt')
from IPython.external.qt_for_kernel import QtGui
QtGui.QApplication.instance()
else:
ipython = None
# run within python
if app is None:
from PyQt5.QtWidgets import QApplication
app = QApplication.instance()
if not app: # pragma: no cover
app = QApplication([''])
self.app = app
self.app_window = MainWindow()
self.app_window.setWindowTitle(kwargs.get('title', rcParams['title']))
self.frame = QFrame()
self.frame.setFrameStyle(QFrame.NoFrame)
QtInteractor.__init__(self,
parent=self.frame,
shape=shape,
off_screen=off_screen,
**kwargs)
self.app_window.signal_close.connect(self.quit)
# build main menu
self.main_menu = QMenuBar(parent=self.app_window)
self.app_window.signal_close.connect(self.main_menu.clear)
file_menu = self.main_menu.addMenu('File')
file_menu.addAction('Take Screenshot', self._qt_screenshot)
file_menu.addAction('Export as VTKjs', self._qt_export_vtkjs)
file_menu.addSeparator()
file_menu.addAction('Exit', self.app_window.close)
view_menu = self.main_menu.addMenu('View')
view_menu.addAction('Toggle Eye Dome Lighting', self._toggle_edl)
view_menu.addAction('Scale Axes', self.scale_axes_dialog)
view_menu.addAction('Clear All', self.clear)
tool_menu = self.main_menu.addMenu('Tools')
tool_menu.addAction('Enable Cell Picking (through)',
self.enable_cell_picking)
tool_menu.addAction('Enable Cell Picking (visible)',
lambda: self.enable_cell_picking(through=False))
cam_menu = view_menu.addMenu('Camera')
cam_menu.addAction('Toggle Parallel Projection',
self._toggle_parallel_projection)
cam_menu.addAction('Reset Camera', self.reset_camera)
cam_menu.addAction('Isometric View', self.view_isometric)
cam_menu.addAction('View XY Plane', self.view_xy)
cam_menu.addAction('View XZ Plane', self.view_xz)
cam_menu.addAction('View YZ Plane', self.view_yz)
cam_menu.addSeparator()
cam_menu.addAction('Save Current Camera Position',
self.save_camera_position)
cam_menu.addAction('Clear Saved Positions',
self.clear_camera_positions)
view_menu.addSeparator()
# Orientation marker
orien_menu = view_menu.addMenu('Orientation Marker')
orien_menu.addAction('Show', self.show_axes)
orien_menu.addAction('Hide', self.hide_axes)
# Bounds axes
axes_menu = view_menu.addMenu('Bounds Axes')
axes_menu.addAction('Add Bounds Axes (front)', self.show_bounds)
axes_menu.addAction('Add Bounds Grid (back)', self.show_grid)
axes_menu.addAction('Add Bounding Box', self.add_bounding_box)
axes_menu.addSeparator()
axes_menu.addAction('Remove Bounding Box', self.remove_bounding_box)
axes_menu.addAction('Remove Bounds', self.remove_bounds_axes)
# A final separator to seperate OS options
view_menu.addSeparator()
self.saved_camera_menu = self.main_menu.addMenu('Camera Positions')
vlayout = QVBoxLayout()
vlayout.addWidget(self)
self.frame.setLayout(vlayout)
self.app_window.setCentralWidget(self.frame)
if off_screen is None:
off_screen = pyvista.OFF_SCREEN
if show and not off_screen: # pragma: no cover
self.app_window.show()
self.show()
self._spawn_background_rendering()
self.window_size = window_size
self._last_update_time = time.time(
) - BackgroundPlotter.ICON_TIME_STEP / 2
self._last_window_size = self.window_size
self._last_camera_pos = self.camera_position
def scale_axes_dialog(self, show=True):
""" Open scale axes dialog """
return ScaleAxesDialog(self.app_window, self, show=show)
def clear_camera_positions(self):
""" clears all camera positions """
for action in self.saved_camera_menu.actions():
self.saved_camera_menu.removeAction(action)
def save_camera_position(self):
""" Saves camera position to saved camera menu for recall """
self.saved_camera_positions.append(self.camera_position)
ncam = len(self.saved_camera_positions)
camera_position = self.camera_position[:] # py2.7 copy compatibility
def load_camera_position():
self.camera_position = camera_position
self.saved_camera_menu.addAction('Camera Position %2d' % ncam,
load_camera_position)
def _spawn_background_rendering(self, rate=5.0):
"""
Spawns a thread that updates the render window.
Sometimes directly modifiying object data doesn't trigger
Modified() and upstream objects won't be updated. This
ensures the render window stays updated without consuming too
many resources.
"""
twait = (rate**-1) * 1000.0
self.render_timer = QTimer(parent=self.app_window)
self.render_timer.timeout.connect(self._render)
self.app_window.signal_close.connect(self.render_timer.stop)
self.render_timer.start(twait)
def quit(self):
QtInteractor.quit(self)
def close(self):
self.app_window.close()
def add_actor(self,
actor,
reset_camera=None,
name=None,
loc=None,
culling=False,
pickable=True):
actor, prop = super(BackgroundPlotter,
self).add_actor(actor,
reset_camera=reset_camera,
name=name,
loc=loc,
culling=culling,
pickable=pickable)
self.update_app_icon()
return actor, prop
def update_app_icon(self):
"""
Update the app icon if the user is not trying to resize the window.
"""
if os.name == 'nt' or not hasattr(
self, '_last_window_size'): # pragma: no cover
# DO NOT EVEN ATTEMPT TO UPDATE ICON ON WINDOWS
return
cur_time = time.time()
if self._last_window_size != self.window_size: # pragma: no cover
# Window size hasn't remained constant since last render.
# This means the user is resizing it so ignore update.
pass
elif ((cur_time - self._last_update_time >
BackgroundPlotter.ICON_TIME_STEP)
and self._last_camera_pos != self.camera_position):
# its been a while since last update OR
# the camera position has changed and its been at leat one second
# Update app icon as preview of the window
img = pad_image(self.image)
qimage = QtGui.QImage(img.copy(), img.shape[1], img.shape[0],
QtGui.QImage.Format_RGB888)
icon = QtGui.QIcon(QtGui.QPixmap.fromImage(qimage))
self.app.setWindowIcon(icon)
# Update trackers
self._last_update_time = cur_time
self._last_camera_pos = self.camera_position
# Update trackers
self._last_window_size = self.window_size
def _qt_screenshot(self, show=True):
return FileDialog(self.app_window,
filefilter=['Image File (*.png)', 'JPEG (*.jpeg)'],
show=show,
directory=os.getcwd(),
callback=self.screenshot)
def _qt_export_vtkjs(self, show=True):
"""
Spawn an save file dialog to export a vtkjs file.
"""
return FileDialog(self.app_window,
filefilter=['VTK JS File(*.vtkjs)'],
show=show,
directory=os.getcwd(),
callback=self.export_vtkjs)
def _toggle_edl(self):
if hasattr(self.renderer, 'edl_pass'):
return self.renderer.disable_eye_dome_lighting()
return self.renderer.enable_eye_dome_lighting()
def _toggle_parallel_projection(self):
if self.camera.GetParallelProjection():
return self.disable_parallel_projection()
return self.enable_parallel_projection()
@pyqtSlot()
def _render(self):
super(BackgroundPlotter, self)._render()
self.update_app_icon()
self.ren_win.Render() # force rendering
return
@property
def window_size(self):
""" returns render window size """
the_size = self.app_window.baseSize()
return the_size.width(), the_size.height()
@window_size.setter
def window_size(self, window_size):
""" set the render window size """
BasePlotter.window_size.fset(self, window_size)
self.app_window.setBaseSize(*window_size)
self.app_window.resize(*window_size)
def __del__(self): # pragma: no cover
self.app_window.close()
def add_callback(self, func, interval=1000, count=None):
"""Add a function that can update the scene in the background
Parameters
----------
func : callable
Function to be called with no arguments.
interval : int
Time interval between calls to `func` in milliseconds.
count : int, optional
Number of times `func` will be called. If None,
`func` will be called until the main window is closed.
"""
timer = QTimer(parent=self.app_window)
timer.timeout.connect(func)
timer.start(interval)
self.app_window.signal_close.connect(timer.stop)
if count is not None:
counter = Counter(count)
counter.signal_finished.connect(timer.stop)
timer.timeout.connect(counter.decrease)
self.counters.append(counter)
class ResultWindow(QMainWindow):
def __init__(self, parent, app, **kwargs):
super().__init__(parent, **kwargs)
self.app = app
self._setupUi()
self.resultsModel = app.RESULT_MODEL_CLASS(self.app, self.resultsView)
self.stats = StatsLabel(app.model.stats_label, self.statusLabel)
self._update_column_actions_status()
self.menuColumns.triggered.connect(self.columnToggled)
self.resultsView.doubleClicked.connect(self.resultsDoubleClicked)
self.resultsView.spacePressed.connect(self.resultsSpacePressed)
self.detailsButton.clicked.connect(self.actionDetails.triggered)
self.dupesOnlyCheckBox.stateChanged.connect(self.powerMarkerTriggered)
self.deltaValuesCheckBox.stateChanged.connect(self.deltaTriggered)
self.searchEdit.searchChanged.connect(self.searchChanged)
self.app.willSavePrefs.connect(self.appWillSavePrefs)
def _setupActions(self):
# (name, shortcut, icon, desc, func)
ACTIONS = [
('actionDetails', 'Ctrl+I', '', tr("Details"), self.detailsTriggered),
('actionActions', '', '', tr("Actions"), self.actionsTriggered),
('actionPowerMarker', 'Ctrl+1', '', tr("Show Dupes Only"), self.powerMarkerTriggered),
('actionDelta', 'Ctrl+2', '', tr("Show Delta Values"), self.deltaTriggered),
('actionDeleteMarked', 'Ctrl+D', '', tr("Send Marked to Recycle Bin..."), self.deleteTriggered),
('actionMoveMarked', 'Ctrl+M', '', tr("Move Marked to..."), self.moveTriggered),
('actionCopyMarked', 'Ctrl+Shift+M', '', tr("Copy Marked to..."), self.copyTriggered),
('actionRemoveMarked', 'Ctrl+R', '', tr("Remove Marked from Results"), self.removeMarkedTriggered),
('actionReprioritize', '', '', tr("Re-Prioritize Results..."), self.reprioritizeTriggered),
('actionRemoveSelected', 'Ctrl+Del', '', tr("Remove Selected from Results"), self.removeSelectedTriggered),
('actionIgnoreSelected', 'Ctrl+Shift+Del', '', tr("Add Selected to Ignore List"), self.addToIgnoreListTriggered),
('actionMakeSelectedReference', 'Ctrl+Space', '', tr("Make Selected into Reference"), self.app.model.make_selected_reference),
('actionOpenSelected', 'Ctrl+O', '', tr("Open Selected with Default Application"), self.openTriggered),
('actionRevealSelected', 'Ctrl+Shift+O', '', tr("Open Containing Folder of Selected"), self.revealTriggered),
('actionRenameSelected', 'F2', '', tr("Rename Selected"), self.renameTriggered),
('actionMarkAll', 'Ctrl+A', '', tr("Mark All"), self.markAllTriggered),
('actionMarkNone', 'Ctrl+Shift+A', '', tr("Mark None"), self.markNoneTriggered),
('actionInvertMarking', 'Ctrl+Alt+A', '', tr("Invert Marking"), self.markInvertTriggered),
('actionMarkSelected', '', '', tr("Mark Selected"), self.markSelectedTriggered),
('actionExportToHTML', '', '', tr("Export To HTML"), self.app.model.export_to_xhtml),
('actionExportToCSV', '', '', tr("Export To CSV"), self.app.model.export_to_csv),
('actionSaveResults', 'Ctrl+S', '', tr("Save Results..."), self.saveResultsTriggered),
('actionInvokeCustomCommand', 'Ctrl+Alt+I', '', tr("Invoke Custom Command"), self.app.invokeCustomCommand),
]
createActions(ACTIONS, self)
self.actionDelta.setCheckable(True)
self.actionPowerMarker.setCheckable(True)
def _setupMenu(self):
self.menubar = QMenuBar()
self.menubar.setGeometry(QRect(0, 0, 630, 22))
self.menuFile = QMenu(self.menubar)
self.menuFile.setTitle(tr("File"))
self.menuMark = QMenu(self.menubar)
self.menuMark.setTitle(tr("Mark"))
self.menuActions = QMenu(self.menubar)
self.menuActions.setTitle(tr("Actions"))
self.menuColumns = QMenu(self.menubar)
self.menuColumns.setTitle(tr("Columns"))
self.menuView = QMenu(self.menubar)
self.menuView.setTitle(tr("View"))
self.menuHelp = QMenu(self.menubar)
self.menuHelp.setTitle(tr("Help"))
self.setMenuBar(self.menubar)
self.menuActions.addAction(self.actionDeleteMarked)
self.menuActions.addAction(self.actionMoveMarked)
self.menuActions.addAction(self.actionCopyMarked)
self.menuActions.addAction(self.actionRemoveMarked)
self.menuActions.addAction(self.actionReprioritize)
self.menuActions.addSeparator()
self.menuActions.addAction(self.actionRemoveSelected)
self.menuActions.addAction(self.actionIgnoreSelected)
self.menuActions.addAction(self.actionMakeSelectedReference)
self.menuActions.addSeparator()
self.menuActions.addAction(self.actionOpenSelected)
self.menuActions.addAction(self.actionRevealSelected)
self.menuActions.addAction(self.actionInvokeCustomCommand)
self.menuActions.addAction(self.actionRenameSelected)
self.menuMark.addAction(self.actionMarkAll)
self.menuMark.addAction(self.actionMarkNone)
self.menuMark.addAction(self.actionInvertMarking)
self.menuMark.addAction(self.actionMarkSelected)
self.menuView.addAction(self.actionPowerMarker)
self.menuView.addAction(self.actionDelta)
self.menuView.addSeparator()
self.menuView.addAction(self.actionDetails)
self.menuView.addAction(self.app.actionIgnoreList)
self.menuView.addAction(self.app.actionPreferences)
self.menuHelp.addAction(self.app.actionShowHelp)
self.menuHelp.addAction(self.app.actionOpenDebugLog)
self.menuHelp.addAction(self.app.actionAbout)
self.menuFile.addAction(self.actionSaveResults)
self.menuFile.addAction(self.actionExportToHTML)
self.menuFile.addAction(self.actionExportToCSV)
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionQuit)
self.menubar.addAction(self.menuFile.menuAction())
self.menubar.addAction(self.menuMark.menuAction())
self.menubar.addAction(self.menuActions.menuAction())
self.menubar.addAction(self.menuColumns.menuAction())
self.menubar.addAction(self.menuView.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
# Columns menu
menu = self.menuColumns
self._column_actions = []
for index, (display, visible) in enumerate(self.app.model.result_table.columns.menu_items()):
action = menu.addAction(display)
action.setCheckable(True)
action.setChecked(visible)
action.item_index = index
self._column_actions.append(action)
menu.addSeparator()
action = menu.addAction(tr("Reset to Defaults"))
action.item_index = -1
# Action menu
actionMenu = QMenu(tr("Actions"), self.menubar)
actionMenu.addAction(self.actionDeleteMarked)
actionMenu.addAction(self.actionMoveMarked)
actionMenu.addAction(self.actionCopyMarked)
actionMenu.addAction(self.actionRemoveMarked)
actionMenu.addSeparator()
actionMenu.addAction(self.actionRemoveSelected)
actionMenu.addAction(self.actionIgnoreSelected)
actionMenu.addAction(self.actionMakeSelectedReference)
actionMenu.addSeparator()
actionMenu.addAction(self.actionOpenSelected)
actionMenu.addAction(self.actionRevealSelected)
actionMenu.addAction(self.actionInvokeCustomCommand)
actionMenu.addAction(self.actionRenameSelected)
self.actionActions.setMenu(actionMenu)
self.actionsButton.setMenu(self.actionActions.menu())
def _setupUi(self):
self.setWindowTitle(tr("{} Results").format(self.app.NAME))
self.resize(630, 514)
self.centralwidget = QWidget(self)
self.verticalLayout = QVBoxLayout(self.centralwidget)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.verticalLayout.setSpacing(0)
self.actionsButton = QPushButton(tr("Actions"))
self.detailsButton = QPushButton(tr("Details"))
self.dupesOnlyCheckBox = QCheckBox(tr("Dupes Only"))
self.deltaValuesCheckBox = QCheckBox(tr("Delta Values"))
self.searchEdit = SearchEdit()
self.searchEdit.setMaximumWidth(300)
self.horizontalLayout = horizontalWrap([self.actionsButton, self.detailsButton,
self.dupesOnlyCheckBox, self.deltaValuesCheckBox, None, self.searchEdit, 8])
self.horizontalLayout.setSpacing(8)
self.verticalLayout.addLayout(self.horizontalLayout)
self.resultsView = ResultsView(self.centralwidget)
self.resultsView.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.resultsView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.resultsView.setSortingEnabled(True)
self.resultsView.verticalHeader().setVisible(False)
h = self.resultsView.horizontalHeader()
h.setHighlightSections(False)
h.setSectionsMovable(True)
h.setStretchLastSection(False)
h.setDefaultAlignment(Qt.AlignLeft)
self.verticalLayout.addWidget(self.resultsView)
self.setCentralWidget(self.centralwidget)
self._setupActions()
self._setupMenu()
self.statusbar = QStatusBar(self)
self.statusbar.setSizeGripEnabled(True)
self.setStatusBar(self.statusbar)
self.statusLabel = QLabel(self)
self.statusbar.addPermanentWidget(self.statusLabel, 1)
if self.app.prefs.resultWindowIsMaximized:
self.setWindowState(self.windowState() | Qt.WindowMaximized)
else:
if self.app.prefs.resultWindowRect is not None:
self.setGeometry(self.app.prefs.resultWindowRect)
else:
moveToScreenCenter(self)
#--- Private
def _update_column_actions_status(self):
# Update menu checked state
menu_items = self.app.model.result_table.columns.menu_items()
for action, (display, visible) in zip(self._column_actions, menu_items):
action.setChecked(visible)
#--- Actions
def actionsTriggered(self):
self.actionsButton.showMenu()
def addToIgnoreListTriggered(self):
self.app.model.add_selected_to_ignore_list()
def copyTriggered(self):
self.app.model.copy_or_move_marked(True)
def deleteTriggered(self):
self.app.model.delete_marked()
def deltaTriggered(self, state=None):
# The sender can be either the action or the checkbox, but both have a isChecked() method.
self.resultsModel.delta_values = self.sender().isChecked()
self.actionDelta.setChecked(self.resultsModel.delta_values)
self.deltaValuesCheckBox.setChecked(self.resultsModel.delta_values)
def detailsTriggered(self):
self.app.show_details()
def markAllTriggered(self):
self.app.model.mark_all()
def markInvertTriggered(self):
self.app.model.mark_invert()
def markNoneTriggered(self):
self.app.model.mark_none()
def markSelectedTriggered(self):
self.app.model.toggle_selected_mark_state()
def moveTriggered(self):
self.app.model.copy_or_move_marked(False)
def openTriggered(self):
self.app.model.open_selected()
def powerMarkerTriggered(self, state=None):
# see deltaTriggered
self.resultsModel.power_marker = self.sender().isChecked()
self.actionPowerMarker.setChecked(self.resultsModel.power_marker)
self.dupesOnlyCheckBox.setChecked(self.resultsModel.power_marker)
def preferencesTriggered(self):
self.app.show_preferences()
def removeMarkedTriggered(self):
self.app.model.remove_marked()
def removeSelectedTriggered(self):
self.app.model.remove_selected()
def renameTriggered(self):
index = self.resultsView.selectionModel().currentIndex()
# Our index is the current row, with column set to 0. Our filename column is 1 and that's
# what we want.
index = index.sibling(index.row(), 1)
self.resultsView.edit(index)
def reprioritizeTriggered(self):
dlg = PrioritizeDialog(self, self.app)
result = dlg.exec()
if result == QDialog.Accepted:
dlg.model.perform_reprioritization()
def revealTriggered(self):
self.app.model.reveal_selected()
def saveResultsTriggered(self):
title = tr("Select a file to save your results to")
files = tr("dupeGuru Results (*.dupeguru)")
destination, chosen_filter = QFileDialog.getSaveFileName(self, title, '', files)
if destination:
if not destination.endswith('.dupeguru'):
destination = '{}.dupeguru'.format(destination)
self.app.model.save_as(destination)
self.app.recentResults.insertItem(destination)
#--- Events
def appWillSavePrefs(self):
prefs = self.app.prefs
prefs.resultWindowIsMaximized = self.isMaximized()
prefs.resultWindowRect = self.geometry()
def columnToggled(self, action):
index = action.item_index
if index == -1:
self.app.model.result_table.columns.reset_to_defaults()
self._update_column_actions_status()
else:
visible = self.app.model.result_table.columns.toggle_menu_item(index)
action.setChecked(visible)
def contextMenuEvent(self, event):
self.actionActions.menu().exec_(event.globalPos())
def resultsDoubleClicked(self, modelIndex):
self.app.model.open_selected()
def resultsSpacePressed(self):
self.app.model.toggle_selected_mark_state()
def searchChanged(self):
self.app.model.apply_filter(self.searchEdit.text())
class MainWindow(QWidget):
def __init__(self, x, y, width, height, title, icon=None):
# Init for window
super().__init__()
self.setGeometry(x, y, width, height)
self.setWindowTitle(title)
if icon is not None:
self.setWindowIcon(QIcon(icon))
# Declare textboxes and menubar
self.menubar = None
self.api_box = None
self.stock_box = None
self.resp_data = None
self.current_grabber = api_wrapper.TimeSeries()
# Initialize the window elements
self.init_menubar()
self.init_textboxes()
self.home()
def init_menubar(self):
# Create menubar buttons
self.menubar = QMenuBar(self)
file_menu = self.menubar.addMenu('&File')
extract_action = QAction('&Quit', self)
extract_action.setShortcut('Ctrl+W')
extract_action.setStatusTip('Quit the app')
extract_action.triggered.connect(QCoreApplication.instance().quit)
file_menu.addAction(extract_action)
about_menu = self.menubar.addMenu('&Help')
extract_action = QAction('&About', self)
about_menu.addAction(extract_action)
def init_textboxes(self):
button_layout = QFormLayout()
self.api_box = QLineEdit(self)
self.api_box.move(20, self.menubar.geometry().height() + 20)
self.api_box.resize(self.api_box.sizeHint())
self.stock_box = QLineEdit(self)
self.stock_box.move(
20,
self.menubar.geometry().height() + 40 +
self.api_box.geometry().height())
self.stock_box.resize(self.stock_box.sizeHint())
btn = QPushButton('Fetch data', self)
btn.clicked.connect(self.clicked)
button_layout.addRow(" ", None)
button_layout.addRow("API Key", self.api_box)
button_layout.addRow("Stock Ticker", self.stock_box)
button_layout.addRow(btn)
self.setLayout(button_layout)
def home(self):
btn = QPushButton('Quit', self)
btn.clicked.connect(QCoreApplication.instance().quit)
btn.resize(btn.sizeHint())
btn.move(self.geometry().width() - btn.geometry().width(),
self.geometry().height() - btn.geometry().height())
self.show()
@pyqtSlot()
def clicked(self):
""" activates when the fetch data button is clicked.
todo:
* elegant way to exclude certain columns in the result df
"""
self.current_grabber.set_api_key(self.api_box.text())
try:
self.resp_data = self.current_grabber.get_daily(
self.stock_box.text())
# Raise an exception if the received data is an error
if isinstance(self.resp_data, dict):
for key, val in self.resp_data.items():
if 'Error' or 'Information' in key:
raise api_wrapper.ResponseError(val)
# Plot the data if it exists
self.resp_data = self.resp_data.T
exclude = ['5. volume']
self.resp_data.ix[:,
self.resp_data.columns.difference(exclude)].plot(
)
plt.show()
except api_wrapper.MissingApiKey:
button_reply = QMessageBox.warning(self, 'Error',
'No API Key was provided',
QMessageBox.Ok, QMessageBox.Ok)
self.resp_data = None # reset the response data
except api_wrapper.ResponseError as e:
button_reply = QMessageBox.warning(self, 'Error', '{}'.format(e),
QMessageBox.Ok, QMessageBox.Ok)
self.resp_data = None # reset the response data
class EditorMainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setupUi(self)
self.retranslateUi(self)
self.level = None
path = get_default_path()
if path is None:
self.default_path = ""
else:
self.default_path = path
self.dragging = False
self.last_x = None
self.last_y = None
self.dragged_time = None
self.deleting_item = False # Hack for preventing focusing on next item after deleting the previous one
self.moving = False
self.resetting = False
self.entity_list_widget.currentItemChanged.connect(self.action_listwidget_change_selection)
self.button_zoom_in.pressed.connect(self.zoom_in)
self.button_zoom_out.pressed.connect(self.zoom_out)
self.button_remove_entity.pressed.connect(self.remove_position)
self.button_move_entity.pressed.connect(self.move_entity)
self.button_clone_entity.pressed.connect(self.action_clone_entity)
self.button_show_passengers.pressed.connect(self.action_passenger_window)
self.button_edit_xml.pressed.connect(self.action_open_xml_editor)
self.button_edit_base_xml.pressed.connect(self.action_open_basexml_editor)
self.lineedit_angle.editingFinished.connect(self.action_lineedit_changeangle)
self.bw_map_screen.mouse_clicked.connect(self.get_position)
self.bw_map_screen.entity_clicked.connect(self.entity_position)
self.bw_map_screen.mouse_dragged.connect(self.mouse_move)
self.bw_map_screen.mouse_released.connect(self.mouse_release)
self.bw_map_screen.mouse_wheel.connect(self.mouse_wheel_scroll_zoom)
status = self.statusbar
self.bw_map_screen.setMouseTracking(True)
self.passenger_window = BWPassengerWindow()
self.passenger_window.passengerlist.currentItemChanged.connect(self.passengerwindow_action_choose_entity)
self.xmlobject_textbox = BWEntityXMLEditor()
self.xmlobject_textbox.button_xml_savetext.pressed.connect(self.xmleditor_action_save_object_xml)
self.xmlobject_textbox.triggered.connect(self.action_open_xml_editor_unlimited)
self.basexmlobject_textbox = BWEntityXMLEditor(windowtype="XML Base Object")
self.basexmlobject_textbox.button_xml_savetext.pressed.connect(self.xmleditor_action_save_base_object_xml)
self.basexmlobject_textbox.triggered.connect(self.action_open_xml_editor_unlimited)
self.types_visible = {}
self.terrain_image = None
status.showMessage("Ready")
self.xml_windows = {}
print("We are now ready!")
def reset(self):
self.resetting = True
self.statusbar.clearMessage()
self.dragged_time = None
self.moving = False
self.dragging = False
self.last_x = None
self.last_y = None
self.dragged_time = None
self.moving = False
self.entity_list_widget.clearSelection()
self.entity_list_widget.clear()
self.bw_map_screen.reset()
self.clear_visibility_toggles()
for window in (self.passenger_window, self.xmlobject_textbox, self.basexmlobject_textbox):
window.close()
window.reset()
for id in self.xml_windows:
self.destroy_xml_editor(id)
self.resetting = False
print("reset done")
def destroy_xml_editor(self, id):
pass
@catch_exception
def open_xml_editor(self, objectid, offsetx=0, offsety=0):
selected = objectid
if self.level is not None and selected in self.level.obj_map:
delete = []
for objid, window in self.xml_windows.items():
if not window.isVisible() and objid != selected:
window.destroy()
delete.append(objid)
for objid in delete:
del self.xml_windows[objid]
if selected == self.basexmlobject_textbox.entity or selected == self.xmlobject_textbox.entity:
pass # No need to make a new window
elif selected in self.xml_windows and self.xml_windows[selected].isVisible():
self.xml_windows[selected].activateWindow()
self.xml_windows[selected].update()
else:
xml_window = BWEntityXMLEditor()
def xmleditor_save_object_unlimited():
self.statusbar.showMessage("Saving object changes...")
try:
xmlnode = xml_window.get_content()
#assert self.bw_map_screen.current_entity == self.basexmlobject_textbox.entity
assert xml_window.entity == xmlnode.get("id") # Disallow changing the id of the base object
self.level.remove_object(xmlnode.get("id"))
self.level.add_object(xmlnode)
self.statusbar.showMessage("Saved base object {0} as {1}".format(
xml_window.entity, self.level.obj_map[xmlnode.get("id")].name))
except:
self.statusbar.showMessage("Saving object failed")
traceback.print_exc()
xml_window.button_xml_savetext.pressed.connect(xmleditor_save_object_unlimited)
xml_window.triggered.connect(self.action_open_xml_editor_unlimited)
obj = self.level.obj_map[selected]
xml_window.set_title(obj.name)
xml_window.set_content(obj._xml_node)
#xml_window.move(QPoint(xml_editor_owner.pos().x()+20, xml_editor_owner.pos().y()+20))
xml_window.move(QPoint(offsetx, offsety))
xml_window.show()
xml_window.update()
self.xml_windows[selected] = xml_window
@catch_exception
def action_open_xml_editor_unlimited(self, xml_editor_owner):
selected = xml_editor_owner.textbox_xml.textCursor().selectedText()
self.open_xml_editor(selected,
offsetx=xml_editor_owner.pos().x()+20,
offsety=xml_editor_owner.pos().y()+20)
@catch_exception
def action_open_basexml_editor(self):
"""
if not self.basexmlobject_textbox.isVisible():
self.basexmlobject_textbox.destroy()
self.basexmlobject_textbox = BWEntityXMLEditor(windowtype="XML Base Object")
self.basexmlobject_textbox.button_xml_savetext.pressed.connect(self.xmleditor_action_save_base_object_xml)
self.basexmlobject_textbox.triggered.connect(self.action_open_xml_editor_unlimited)
self.basexmlobject_textbox.show()
self.basexmlobject_textbox.activateWindow()"""
if self.level is not None and self.bw_map_screen.current_entity is not None:
obj = self.level.obj_map[self.bw_map_screen.current_entity]
if not obj.has_attr("mBase"):
pass
else:
baseobj = self.level.obj_map[obj.get_attr_value("mBase")]
#self.basexmlobject_textbox.set_title(baseobj.id)
self.open_xml_editor(baseobj.id)
def xmleditor_action_save_base_object_xml(self):
self.statusbar.showMessage("Saving base object changes...")
try:
xmlnode = self.basexmlobject_textbox.get_content()
#assert self.bw_map_screen.current_entity == self.basexmlobject_textbox.entity
assert self.basexmlobject_textbox.entity == xmlnode.get("id") # Disallow changing the id of the base object
self.level.remove_object(xmlnode.get("id"))
self.level.add_object(xmlnode)
self.statusbar.showMessage("Saved base object {0} as {1}".format(
self.basexmlobject_textbox.entity, self.level.obj_map[xmlnode.get("id")].name))
except:
self.statusbar.showMessage("Saving base object failed")
traceback.print_exc()
def action_open_xml_editor(self):
"""
if not self.xmlobject_textbox.isVisible():
self.xmlobject_textbox.destroy()
self.xmlobject_textbox = BWEntityXMLEditor()
self.xmlobject_textbox.button_xml_savetext.pressed.connect(self.xmleditor_action_save_object_xml)
self.xmlobject_textbox.triggered.connect(self.action_open_xml_editor_unlimited)
self.xmlobject_textbox.show()
self.xmlobject_textbox.activateWindow()"""
if self.level is not None and self.bw_map_screen.current_entity is not None:
entityobj = self.level.obj_map[self.bw_map_screen.current_entity]
self.open_xml_editor(objectid=entityobj.id)
update_mapscreen(self.bw_map_screen, self.level.obj_map[entityobj.id])
self.bw_map_screen.update()
"""self.xmlobject_textbox.set_title(entityobj.id)
self.xmlobject_textbox.set_content(entityobj._xml_node)
self.xmlobject_textbox.update()"""
def xmleditor_action_save_object_xml(self):
self.statusbar.showMessage("Saving object changes...")
try:
xmlnode = self.xmlobject_textbox.get_content()
#assert self.bw_map_screen.current_entity == self.xmlobject_textbox.entity
assert self.xmlobject_textbox.entity == xmlnode.get("id") or xmlnode.get("id") not in self.level.obj_map
if self.passenger_window.isVisible():
self.passenger_window.close()
if self.xmlobject_textbox.entity != xmlnode.get("id"):
#obj = self.level.obj_map[xmlnode.get("id")]
self.level.remove_object(self.xmlobject_textbox.entity)
print("adding", xmlnode.get("id"), xmlnode.get("id") in self.level.obj_map )
self.level.add_object(xmlnode)
pos = self.get_entity_item_pos(self.xmlobject_textbox.entity)
item = self.entity_list_widget.takeItem(pos)
self.entity_list_widget.removeItemWidget(item)
self.add_item_sorted(xmlnode.get("id"))
self.bw_map_screen.rename_entity(self.xmlobject_textbox.entity, xmlnode.get("id"))
assert xmlnode.get("id") in self.level.obj_map
self.xmlobject_textbox.entity = xmlnode.get("id")
self.xmlobject_textbox.set_title(xmlnode.get("id"))
else:
self.level.remove_object(xmlnode.get("id"))
self.level.add_object(xmlnode)
update_mapscreen(self.bw_map_screen, self.level.obj_map[xmlnode.get("id")])
self.statusbar.showMessage("Saved object {0} as {1}".format(
self.xmlobject_textbox.entity, self.level.obj_map[xmlnode.get("id")].name))
self.bw_map_screen.update()
except:
self.statusbar.showMessage("Saving object failed")
traceback.print_exc()
def action_clone_entity(self):
entities = []
if self.bw_map_screen.current_entity is not None:
entities.append(self.bw_map_screen.current_entity)
elif len(self.bw_map_screen.selected_entities) > 0:
entities.extend(self.bw_map_screen.selected_entities.keys())
if len(entities) > 0:
dont_clone = {}
for entity in entities:
obj = self.level.obj_map[entity]
if obj.has_attr("mPassenger"):
passengers = obj.get_attr_elements("mPassenger")
for passenger in passengers:
if passenger != "0":
dont_clone[passenger] = True
select = []
for entity in entities:
if entity in dont_clone:
continue
obj = self.level.obj_map[entity]
xml_node = deepcopy(obj._xml_node)
try:
cloned_id = self.level.generate_unique_id(entity)
xml_node.set("id", cloned_id)
self.level.add_object(xml_node)
bw_x, bw_y, angle = object_get_position(self.level, cloned_id)
x, y = bw_coords_to_image_coords(bw_x, bw_y)
self.add_item_sorted(cloned_id)
self.bw_map_screen.add_entity(x, y, cloned_id, obj.type)
clonedobj = self.level.obj_map[cloned_id]
select.append(cloned_id)
update_mapscreen(self.bw_map_screen, clonedobj)
if clonedobj.has_attr("mPassenger"):
orig_x = bw_x
orig_y = bw_y
passengers = clonedobj.get_attr_elements("mPassenger")
passengers_added = []
for i, passenger in enumerate(passengers):
if passenger != "0":
obj = self.level.obj_map[passenger]
xml_node = deepcopy(obj._xml_node)
clonedpassenger_id = self.level.generate_unique_id(passenger)
xml_node.set("id", clonedpassenger_id)
#print("orig passenger: {0}, new passenger: {1}, alreadyexists: {2}".format(
# passenger, clonedpassenger_id, clonedpassenger_id in self.level.obj_map
#))
#print(type(passenger), type(clonedpassenger_id))
self.level.add_object(xml_node)
#x, y = object_get_position(self.level, newid)
x = orig_x + (i+1)*8
y = orig_y + (i+1)*8
#print(orig_x, orig_y, x, y)
object_set_position(self.level, clonedpassenger_id, x, y)
x, y = bw_coords_to_image_coords(x, y)
self.add_item_sorted(clonedpassenger_id)
self.bw_map_screen.add_entity(x, y, clonedpassenger_id, obj.type)
update_mapscreen(self.bw_map_screen, self.level.obj_map[clonedpassenger_id])
passengers_added.append(passenger)
clonedobj.set_attr_value("mPassenger", clonedpassenger_id, i)
select.append(clonedpassenger_id)
#print("passengers added:", passengers_added)
self.bw_map_screen.selected_entities = {}
if len(select) == 1:
ent = select[0]
self.set_entity_text(ent)
self.bw_map_screen.choose_entity(ent)
else:
for ent in select:
self.bw_map_screen.selected_entities[ent] = True
self.set_entity_text_multiple(self.bw_map_screen.selected_entities)
self.bw_map_screen.update()
except:
traceback.print_exc()
def add_item_sorted(self, entity):
max_count = self.entity_list_widget.count()
entityobj = self.level.obj_map[entity]
index = 0
entity_item = BWEntityEntry(entity, "{0}[{1}]".format(entity, entityobj.type))
# Similar to loading a level, we add the entity in a sorted way by
# creating this string and comparing it for every item in the list.
entity_string = get_type(entityobj.type)+entityobj.type+entityobj.id
inserted = False
for i in range(max_count):
curritem = self.entity_list_widget.item(i)
currobj = self.level.obj_map[curritem.xml_ref]
currstring = get_type(currobj.type)+currobj.type+currobj.id
# The list is already sorted, so if we find an item bigger than
# the one we are inserting, we know the position we have to insert the item in.
# String comparison should be alpabetically.
if currstring > entity_string:
self.entity_list_widget.insertItem(i, entity_item)
inserted = True
break
# If we couldn't insert the item, i.e. there are no items at all
# or all items are smaller than the item we add, we just add it at the end.
if not inserted:
self.entity_list_widget.addItem(entity_item)
def get_entity_item_pos(self, entityid):
for i in range(self.entity_list_widget.count()):
item = self.entity_list_widget.item(i)
if item.xml_ref == entityid:
return i
return None
def action_passenger_window(self):
#if self.passenger_window.isVisible()
print("window is visible: ", self.passenger_window.isVisible())
#self.passenger_window.reset()
if not self.passenger_window.isVisible():
self.passenger_window.destroy()
self.passenger_window = BWPassengerWindow()
self.passenger_window.passengerlist.currentItemChanged.connect(self.passengerwindow_action_choose_entity)
self.passenger_window.show()
self.passenger_window.activateWindow()
if self.bw_map_screen.current_entity is not None:
self.passenger_window.reset()
entityobj = self.level.obj_map[self.bw_map_screen.current_entity]
self.passenger_window.set_title(entityobj.id)
if entityobj.has_attr("mPassenger"):
for i, passenger in enumerate(entityobj.get_attr_elements("mPassenger")):
if passenger in self.level.obj_map:
passengerobj = self.level.obj_map[passenger]
list_item_name = "{0}[{1}]".format(passenger, passengerobj.type)
elif passenger == "0":
list_item_name = "{0}<none>".format(passenger)
else:
list_item_name = "{0}<missing>".format(passenger)
self.passenger_window.add_passenger(list_item_name, passenger)
self.passenger_window.update()
def passengerwindow_action_choose_entity(self, current, previous):
try:
if current is not None and current.xml_ref in self.level.obj_map:
self.set_entity_text(current.xml_ref)
self.bw_map_screen.choose_entity(current.xml_ref)
elif current is not None:
self.statusbar.showMessage("No such entity: {0}".format(current.xml_ref), 1000*2)
except:
traceback.print_exc()
def move_entity(self):
if not self.dragging:
if not self.moving:
self.moving = True
currtext = self.button_move_entity.text()
self.button_move_entity.setText("Stop [Move Entity]")
else:
self.moving = False
currtext = "Move Entity"
self.button_move_entity.setText(currtext)
def button_load_level(self):
try:
print("ok", self.default_path)
self.xmlPath = ""
filepath, choosentype = QFileDialog.getOpenFileName(
self, "Open File",
self.default_path,
BW_LEVEL+";;"+BW_COMPRESSED_LEVEL+";;All files (*)")
print("doooone")
if filepath:
print("resetting")
self.reset()
print("done")
print("chosen type:",choosentype)
# Some BW levels are clear XML files, some are compressed with GZIP
# We decide between the two either based on user choice or end of filepath
if choosentype == BW_COMPRESSED_LEVEL or filepath.endswith(".gz"):
print("OPENING AS COMPRESSED")
file_open = gzip.open
else:
file_open = open
with file_open(filepath, "rb") as f:
try:
self.level = BattWarsLevel(f)
self.default_path = filepath
set_default_path(filepath)
self.setup_visibility_toggles()
for obj_id, obj in sorted(self.level.obj_map.items(),
key=lambda x: get_type(x[1].type)+x[1].type+x[1].id):
#print("doing", obj_id)
if get_position_attribute(obj) is None:
continue
#if not obj.has_attr("Mat"):
# continue
x, y, angle = object_get_position(self.level, obj_id)
assert type(x) != str
x, y = bw_coords_to_image_coords(x, y)
item = BWEntityEntry(obj_id, "{0}[{1}]".format(obj_id, obj.type))
self.entity_list_widget.addItem(item)
self.bw_map_screen.add_entity(x, y, obj_id, obj.type, update=False)
#if obj.type == "cMapZone":
update_mapscreen(self.bw_map_screen, obj)
print("ok")
self.bw_map_screen.update()
path_parts = path.split(filepath)
self.setWindowTitle("BW-MapEdit - {0}".format(path_parts[-1]))
except Exception as error:
print("error", error)
traceback.print_exc()
except Exception as er:
print("errrorrr", er)
traceback.print_exc()
print("loaded")
def button_save_level(self):
if self.level is not None:
filepath, choosentype = QFileDialog.getSaveFileName(
self, "Save File",
self.default_path,
BW_LEVEL+";;"+BW_COMPRESSED_LEVEL+";;All files (*)")
print(filepath, "saved")
if filepath:
# Simiar to load level
if choosentype == BW_COMPRESSED_LEVEL or filepath.endswith(".gz"):
file_open = gzip.open
else:
file_open = open
try:
with file_open(filepath, "wb") as f:
self.level._tree.write(f)
except Exception as error:
print("COULDN'T SAVE:", error)
traceback.print_exc()
self.default_path = filepath
else:
pass # no level loaded, do nothing
def entity_position(self, event, entity):
try:
# Make it possible to select objects in move mode, but don't make it too easy to lose
# a selection.
if not (self.moving and len(self.bw_map_screen.selected_entities) > 1):
print("got entity:",entity, self.bw_map_screen.entities[entity][2])
print(entity_get_model(self.level, entity))
self.set_entity_text(entity)
self.bw_map_screen.choose_entity(entity)
pos = self.get_entity_item_pos(entity)
print("searching:",pos)
try:
self.entity_list_widget.select_item(pos)
except:
traceback.print_exc()
self.bw_map_screen.selected_entities = {}
self.bw_map_screen.update()
except:
traceback.print_exc()
def remove_position(self):
#self.bw_map_screen.entities.pop()
try:
# Remove the entity from the map, the list widget and the level data
self.deleting_item = True
entities = []
if self.bw_map_screen.current_entity is not None:
entities.append(self.bw_map_screen.current_entity)
elif len(self.bw_map_screen.selected_entities) > 0:
entities.extend(self.bw_map_screen.selected_entities.keys())
self.bw_map_screen.selected_entities = {}
self.set_entity_text_multiple(self.bw_map_screen.selected_entities)
if len(entities) > 0:
for entity in entities:
pos = self.get_entity_item_pos(entity)
item = self.entity_list_widget.takeItem(pos)
assert item.xml_ref == entity
#self.entity_list_widget.clearSelection()
self.entity_list_widget.clearFocus()
self.entity_list_widget.removeItemWidget(item)
self.level.remove_object(entity)
self.bw_map_screen.remove_entity(entity)
self.bw_map_screen.update()
except:
traceback.print_exc()
raise
#@catch_exception
def get_position(self, event):
self.dragging = True
self.last_x = event.x()
self.last_y = event.y()
self.dragged_time = default_timer()
mouse_x = event.x()/self.bw_map_screen.zoom_factor
mouse_y = event.y()/self.bw_map_screen.zoom_factor
if event.buttons() == QtCore.Qt.LeftButton:
if not self.moving:
self.bw_map_screen.set_selectionbox_start((event.x(), event.y()))
else:
if self.bw_map_screen.current_entity is not None:
newx, newy = image_coords_to_bw_coords(mouse_x, mouse_y)
object_set_position(self.level, self.bw_map_screen.current_entity,
newx, newy)
self.bw_map_screen.move_entity(self.bw_map_screen.current_entity,
mouse_x, mouse_y)
self.set_entity_text(self.bw_map_screen.current_entity)
update_mapscreen(self.bw_map_screen, self.level.obj_map[self.bw_map_screen.current_entity])
elif len(self.bw_map_screen.selected_entities) > 0:
for entity in self.bw_map_screen.selected_entities:
first_entity = entity
break
#first_entity = self.bw_map_screen.selected_entities.keys()[0]
x, y, entitytype, metadata = self.bw_map_screen.entities[first_entity]
startx = endx = x
starty = endy = y
for entity in self.bw_map_screen.selected_entities:
x, y, dontneed, dontneed = self.bw_map_screen.entities[entity]
if x < startx:
startx = x
if x > endx:
endx = x
if y < starty:
starty = y
if y > endy:
endy = y
middle_x = (startx+endx) / 2
middle_y = (starty+endy) / 2
delta_x = mouse_x - middle_x
delta_y = mouse_y - middle_y
for entity in self.bw_map_screen.selected_entities:
x, y, dontneed, dontneed = self.bw_map_screen.entities[entity]
newx, newy = image_coords_to_bw_coords(x+delta_x, y+delta_y)
object_set_position(self.level, entity,
newx, newy)
self.bw_map_screen.move_entity(entity,
x+delta_x, y+delta_y)
#self.set_entity_text(self.bw_map_screen.current_entity)
update_mapscreen(self.bw_map_screen, self.level.obj_map[entity])
self.bw_map_screen.update()
@catch_exception
def mouse_move(self, event):
x, y = image_coords_to_bw_coords(event.x()/self.bw_map_screen.zoom_factor,
event.y()/self.bw_map_screen.zoom_factor)
self.statusbar.showMessage("x: {0} y: {1}".format(round(x, 5), round(y, 5)))
if self.dragging and default_timer() - self.dragged_time > 0.1:
if event.buttons() == QtCore.Qt.RightButton:
delta_x = (event.x()-self.last_x)/8
delta_y = (event.y()-self.last_y)/8
#print("hi",event.x(), event.y())
vertbar = self.scrollArea.verticalScrollBar()
horizbar = self.scrollArea.horizontalScrollBar()
vertbar.setSliderPosition(vertbar.value()-delta_y)
horizbar.setSliderPosition(horizbar.value()-delta_x)
elif event.buttons() == QtCore.Qt.LeftButton:
self.bw_map_screen.set_selectionbox_end((event.x(), event.y()))
if len(self.bw_map_screen.selected_entities) > 0 or self.bw_map_screen.current_entity is None:
self.bw_map_screen.choose_entity(None)
self.set_entity_text_multiple(self.bw_map_screen.selected_entities)
self.bw_map_screen.update()
def mouse_release(self, event):
self.dragging = False
if self.bw_map_screen.selectionbox_end is not None:
self.bw_map_screen.clear_selection_box()
self.bw_map_screen.update()
def set_entity_text_multiple(self, entities):
self.label_object_id.setText("{0} objects selected".format(len(entities)))
MAX = 15
listentities = [self.level.obj_map[x].name for x in sorted(entities.keys())][0:MAX]
listentities.sort()
if len(entities) > MAX:
listentities.append("... and {0} more".format(len(entities) - len(listentities)))
self.label_position.setText("\n".join(listentities[:5]))
self.label_model_name.setText("\n".join(listentities[5:10]))
self.label_4.setText("\n".join(listentities[10:]))#15]))
self.label_5.setText("")#("\n".join(listentities[12:16]))
def set_entity_text(self, entityid):
try:
obj = self.level.obj_map[entityid]
if obj.has_attr("mBase"):
base = self.level.obj_map[obj.get_attr_value("mBase")]
self.label_object_id.setText("{0}\n[{1}]\nBase: {2}\n[{3}]".format(
entityid, obj.type, base.id, base.type))
else:
self.label_object_id.setText("{0}\n[{1}]".format(entityid, obj.type))
self.label_model_name.setText("Model: {0}".format(entity_get_model(self.level, entityid)))
x, y, angle = object_get_position(self.level, entityid)
self.label_position.setText("x: {0}\ny: {1}".format(x, y))
self.lineedit_angle.setText(str(round(angle,2)))
self.label_4.setText("Army: {0}".format(entity_get_army(self.level, entityid)))
if not obj.has_attr("mPassenger"):
self.label_5.setText("Icon Type: \n{0}".format(entity_get_icon_type(self.level, entityid)))
else:
passengers = 0
for passenger in obj.get_attr_elements("mPassenger"):
if passenger != "0":
passengers += 1
self.label_5.setText("Icon Type: \n{0}\n\nPassengers: {1}".format(
entity_get_icon_type(self.level, entityid), passengers))
except:
traceback.print_exc()
def action_listwidget_change_selection(self, current, previous):
#QtWidgets.QListWidgetItem.
if not self.resetting and current is not None:
print("ok")
print("hi", current.text(), current.xml_ref)
self.set_entity_text(current.xml_ref)
self.bw_map_screen.choose_entity(current.xml_ref)
posx, posy, typename, metadata = self.bw_map_screen.entities[current.xml_ref]
zf = self.bw_map_screen.zoom_factor
try:
if not self.deleting_item:
x_margin = min(100, 50*zf)
y_margin = min(100, 50*zf)
self.scrollArea.ensureVisible(posx*zf, posy*zf,
xMargin=x_margin, yMargin=y_margin)
else:
self.deleting_item = False
except:
traceback.print_exc()
def zoom_out(self, noslider=False):
horizbar = self.scrollArea.horizontalScrollBar()
vertbar = self.scrollArea.verticalScrollBar()
print(horizbar.maximum(), vertbar.maximum())
if horizbar.maximum() > 0:
widthratio = horizbar.value()/horizbar.maximum()
else:
widthratio = 0
if vertbar.maximum() > 0:
heightratio = vertbar.value()/vertbar.maximum()
else:
heightratio = 0
#oldzf = self.bw_map_screen.zoom_factor / (0.1+1)
#diff = oldzf - self.bw_map_screen.zoom_factor
zf = self.bw_map_screen.zoom_factor
self.bw_map_screen.zoom(calc_zoom_out_factor(zf))#diff)
#
if not noslider:
horizbar.setSliderPosition(horizbar.maximum()*widthratio)
vertbar.setSliderPosition(vertbar.maximum()*heightratio)
self.bw_map_screen.update()
self.statusbar.showMessage("Zoom: {0}x".format(self.bw_map_screen.zoom_factor))
def zoom_in(self, noslider=False):
horizbar = self.scrollArea.horizontalScrollBar()
vertbar = self.scrollArea.verticalScrollBar()
if horizbar.maximum() > 0:
widthratio = horizbar.value()/horizbar.maximum()
else:
widthratio = 0
if vertbar.maximum() > 0:
heightratio = vertbar.value()/vertbar.maximum()
else:
heightratio = 0
zf = self.bw_map_screen.zoom_factor
self.bw_map_screen.zoom(calc_zoom_in_factor(zf))#zf)
#
print("wedidit?")
if not noslider:
horizbar.setSliderPosition(horizbar.maximum()*widthratio)
vertbar.setSliderPosition(vertbar.maximum()*heightratio)
self.bw_map_screen.update()
self.statusbar.showMessage("Zoom: {0}x".format(self.bw_map_screen.zoom_factor))
@catch_exception
def mouse_wheel_scroll_zoom(self, wheel_event):
print("scrolling", wheel_event)
print("Scroll", wheel_event.x(), wheel_event.y(), wheel_event.angleDelta().y())#, wheel_event.delta())
wheel_delta = wheel_event.angleDelta().y()
zf = self.bw_map_screen.zoom_factor
norm_x = wheel_event.x()/zf
norm_y = wheel_event.y()/zf
if wheel_delta > 0:
if zf <= 10:
self.zoom_in(True)
zf = self.bw_map_screen.zoom_factor
xmargin = self.scrollArea.viewport().width()//2 - 200
ymargin = self.scrollArea.viewport().height()//2 - 200
self.scrollArea.ensureVisible(norm_x*zf, norm_y*zf, xmargin, ymargin)
self.bw_map_screen.update()
else:
self.zoom_in()
elif wheel_delta < 0:
self.zoom_out()
def action_lineedit_changeangle(self):
if not self.resetting and self.bw_map_screen.current_entity is not None:
print("ok")
current = self.bw_map_screen.current_entity
currx, curry, angle = object_get_position(self.level, current)
newangle = self.lineedit_angle.text().strip()
print(newangle, newangle.isdecimal())
try:
angle = float(newangle)
object_set_position(self.level, current, currx, curry, angle=angle)
currentobj = self.level.obj_map[current]
update_mapscreen(self.bw_map_screen, currentobj)
self.bw_map_screen.update()
except:
traceback.print_exc()
def button_terrain_load_action(self):
try:
print("ok", self.default_path)
filepath, choosentype = QFileDialog.getOpenFileName(
self, "Open File",
self.default_path,
"BW terrain files (*.out *out.gz);;All files (*)")
print("doooone")
if filepath:
if filepath.endswith(".gz"):
file_open = gzip.open
else:
file_open = open
with file_open(filepath, "rb") as f:
try:
terrain = BWArchiveBase(f)
if self.level is not None:
waterheight = get_water_height(self.level)
else:
waterheight = None
image, light_image = parse_terrain_to_image(terrain, waterheight)
self.bw_map_screen.set_terrain(image, light_image)
except:
traceback.print_exc()
except:
traceback.print_exc()
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(820, 760)
MainWindow.setMinimumSize(QSize(720, 560))
MainWindow.setWindowTitle("BW-MapEdit")
#MainWindow.setWindowTitle("Nep-Nep")
self.centralwidget = QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
MainWindow.setCentralWidget(self.centralwidget)
self.horizontalLayout = QHBoxLayout(self.centralwidget)
self.horizontalLayout.setObjectName("horizontalLayout")
self.scrollArea = QScrollArea(self.centralwidget)
self.scrollArea.setWidgetResizable(True)
self.bw_map_screen = BWMapViewer(self.centralwidget)
self.scrollArea.setWidget(self.bw_map_screen)
self.horizontalLayout.addWidget(self.scrollArea)
#self.horizontalLayout.addWidget(self.bw_map_screen)
self.entity_list_widget = BWEntityListWidget(self.centralwidget)
self.entity_list_widget.setMaximumSize(QSize(300, 16777215))
self.entity_list_widget.setObjectName("entity_list_widget")
self.horizontalLayout.addWidget(self.entity_list_widget)
spacerItem = QSpacerItem(10, 20, QSizePolicy.Minimum, QSizePolicy.Expanding)
self.horizontalLayout.addItem(spacerItem)
self.vertLayoutWidget = QWidget(self.centralwidget)
self.vertLayoutWidget.setMaximumSize(QSize(250, 1200))
self.verticalLayout = QVBoxLayout(self.vertLayoutWidget)
self.verticalLayout.setObjectName("verticalLayout")
#self.verticalLayout.
self.button_clone_entity = QPushButton(self.centralwidget)
self.button_clone_entity.setObjectName("button_clone_entity")
self.verticalLayout.addWidget(self.button_clone_entity)
self.button_remove_entity = QPushButton(self.centralwidget)
self.button_remove_entity.setObjectName("button_remove_entity")
self.verticalLayout.addWidget(self.button_remove_entity)
self.button_move_entity = QPushButton(self.centralwidget)
self.button_move_entity.setObjectName("button_move_entity")
self.verticalLayout.addWidget(self.button_move_entity)
self.button_show_passengers = QPushButton(self.centralwidget)
self.button_show_passengers.setObjectName("button_move_entity")
self.verticalLayout.addWidget(self.button_show_passengers)
self.gridLayout = QGridLayout()
self.gridLayout.setObjectName("gridLayout")
self.button_zoom_in = QPushButton(self.centralwidget)
self.button_zoom_in.setObjectName("button_zoom_in")
self.gridLayout.addWidget(self.button_zoom_in, 0, 0, 0, 1)
self.button_zoom_out = QPushButton(self.centralwidget)
self.button_zoom_out.setObjectName("button_zoom_out")
self.gridLayout.addWidget(self.button_zoom_out, 0, 1, 0, 1)
self.button_edit_xml = QPushButton(self.centralwidget)
self.button_edit_xml.setObjectName("button_edit_xml")
self.button_edit_base_xml = QPushButton(self.centralwidget)
self.button_edit_base_xml.setObjectName("button_edit_base_xml")
self.verticalLayout.addLayout(self.gridLayout)
self.verticalLayout.addWidget(self.button_edit_xml)
self.verticalLayout.addWidget(self.button_edit_base_xml)
spacerItem1 = QSpacerItem(10, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.verticalLayout.addItem(spacerItem1)
self.verticalLayout_2 = QVBoxLayout()
self.verticalLayout_2.setObjectName("verticalLayout_2")
self.lineedit_angle = QLineEdit(self.centralwidget)
self.lineedit_angle.setObjectName("lineedit_angle")
self.lineedit_angle.setPlaceholderText("Angle")
self.label_object_id = QLabel(self.centralwidget)
self.label_object_id.setObjectName("label_object_id")
#TextSelectableByCursor
self.label_position = QLabel(self.centralwidget)
self.label_position.setObjectName("label_position")
self.label_model_name = QLabel(self.centralwidget)
self.label_model_name.setObjectName("label_model_name")
self.label_4 = QLabel(self.centralwidget)
self.label_4.setObjectName("label_4")
self.label_5 = QLabel(self.centralwidget)
self.label_5.setObjectName("label_5")
for label in (self.label_object_id, self.label_position, self.label_model_name, self.label_4, self.label_5):
label.setTextInteractionFlags(QtCore.Qt.TextSelectableByMouse)
self.verticalLayout_2.addWidget(self.lineedit_angle)
self.verticalLayout_2.addWidget(self.label_object_id)
self.verticalLayout_2.addWidget(self.label_position)
self.verticalLayout_2.addWidget(self.label_model_name)
self.verticalLayout_2.addWidget(self.label_4)
self.verticalLayout_2.addWidget(self.label_5)
self.verticalLayout.addLayout(self.verticalLayout_2)
self.horizontalLayout.addWidget(self.vertLayoutWidget)
self.menubar = QMenuBar(MainWindow)
self.menubar.setGeometry(QRect(0, 0, 820, 29))
self.menubar.setObjectName("menubar")
self.file_menu = QMenu(self.menubar)
self.file_menu.setObjectName("menuLoad")
self.file_load_action = QAction("Load", self)
self.file_load_action.triggered.connect(self.button_load_level)
self.file_menu.addAction(self.file_load_action)
self.file_save_action = QAction("Save", self)
self.file_save_action.triggered.connect(self.button_save_level)
self.file_menu.addAction(self.file_save_action)
self.visibility_menu = MenuDontClose(self.menubar)#QMenu(self.menubar)
self.visibility_menu.setObjectName("visibility")
#self.visibility_menu.addAction(self.toggle_action)
self.visibility_actions = []
self.terrain_menu = QMenu(self.menubar)
self.terrain_menu.setObjectName("terrain")
self.terrain_load_action = QAction("Load Terrain", self)
self.terrain_load_action.triggered.connect(self.button_terrain_load_action)
self.terrain_menu.addAction(self.terrain_load_action)
self.terrain_display_actions = []
self.setup_terrain_display_toggles()
#self.menuLoad_2 = QMenu(self.menubar)
#self.menuLoad_2.setObjectName("menuLoad_2")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.menubar.addAction(self.file_menu.menuAction())
#self.menubar.addAction(self.menuLoad_2.menuAction())
self.menubar.addAction(self.visibility_menu.menuAction())
self.menubar.addAction(self.terrain_menu.menuAction())
self.retranslateUi(MainWindow)
QMetaObject.connectSlotsByName(MainWindow)
def make_terrain_toggle(self, show_mode):
def terraintoggle(toggled):
print("I am", show_mode, "and I was pressed")
if toggled is True:
for action, toggle, mode in self.terrain_display_actions:
if mode != show_mode:
action.setChecked(False)
self.bw_map_screen.set_show_terrain_mode(show_mode)
elif toggled is False:
self.bw_map_screen.set_show_terrain_mode(SHOW_TERRAIN_NO_TERRAIN)
else:
print("This shouldn't be possible", toggled, type(toggled))
self.bw_map_screen.update()
return terraintoggle
def setup_terrain_display_toggles(self):
for mode, name in ((SHOW_TERRAIN_REGULAR, "Show Heightmap"),
(SHOW_TERRAIN_LIGHT, "Show Lightmap")):
toggle = self.make_terrain_toggle(mode)
toggle_action = QAction(name, self)
toggle_action.setCheckable(True)
if mode == SHOW_TERRAIN_REGULAR:
toggle_action.setChecked(True)
else:
toggle_action.setChecked(False)
toggle_action.triggered.connect(toggle)
self.terrain_menu.addAction(toggle_action)
self.terrain_display_actions.append((toggle_action, toggle, mode))
def clear_terrain_toggles(self):
try:
for action, func, mode in self.terrain_display_actions:
self.terrain_menu.removeAction(action)
self.terrain_display_actions = []
except:
traceback.print_exc()
def make_toggle_function(self, objtype):
def toggle(toggled):
print("i was pressed")
my_type = copy(objtype)
self.types_visible[my_type] = toggled
self.bw_map_screen.set_visibility(self.types_visible)
self.bw_map_screen.update()
return toggle
def setup_visibility_toggles(self):
for objtype in sorted(self.level.objtypes_with_positions):
toggle = self.make_toggle_function(objtype)
toggle_action = QAction(copy(objtype), self)
toggle_action.setCheckable(True)
toggle_action.setChecked(True)
toggle_action.triggered.connect(toggle)
self.types_visible[objtype] = True
self.visibility_menu.addAction(toggle_action)
self.visibility_actions.append((toggle_action, toggle))
toggle_all = QAction("Toggle All", self)
toggle_all.triggered.connect(self.toggle_visiblity_all)
self.visibility_menu.addAction(toggle_all)
self.visibility_actions.append((toggle_all, self.toggle_visiblity_all))
def toggle_visiblity_all(self):
for action, func in self.visibility_actions:
if action.isCheckable():
objtype = action.text()
toggle = self.types_visible[objtype]
self.types_visible[objtype] = not toggle
action.setChecked(not toggle)
self.bw_map_screen.set_visibility(self.types_visible)
self.bw_map_screen.update()
def clear_visibility_toggles(self):
try:
for action, func in self.visibility_actions:
self.visibility_menu.removeAction(action)
self.visibility_actions = []
self.types_visible = {}
except:
traceback.print_exc()
def retranslateUi(self, MainWindow):
_translate = QCoreApplication.translate
self.button_clone_entity.setText(_translate("MainWindow", "Clone Entity"))
self.button_remove_entity.setText(_translate("MainWindow", "Delete Entity"))
self.button_move_entity.setText(_translate("MainWindow", "Move Entity"))
self.button_zoom_in.setText(_translate("MainWindow", "Zoom In"))
self.button_zoom_out.setText(_translate("MainWindow", "Zoom Out"))
self.button_show_passengers.setText(_translate("MainWindow", "Show Passengers"))
self.button_edit_xml.setText("Edit Object XML")
self.button_edit_base_xml.setText("Edit Base Object XML")
self.label_model_name.setText(_translate("MainWindow", "TextLabel1"))
self.label_object_id.setText(_translate("MainWindow", "TextLabel2"))
self.label_position.setText(_translate("MainWindow", "TextLabel3"))
self.label_4.setText(_translate("MainWindow", "TextLabel4"))
self.label_5.setText(_translate("MainWindow", "TextLabel5"))
self.file_menu.setTitle(_translate("MainWindow", "File"))
self.visibility_menu.setTitle(_translate("MainWindow", "Visibility"))
self.terrain_menu.setTitle("Terrain")
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(820, 760)
MainWindow.setMinimumSize(QSize(720, 560))
MainWindow.setWindowTitle("BW-MapEdit")
#MainWindow.setWindowTitle("Nep-Nep")
self.centralwidget = QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
MainWindow.setCentralWidget(self.centralwidget)
self.horizontalLayout = QHBoxLayout(self.centralwidget)
self.horizontalLayout.setObjectName("horizontalLayout")
self.scrollArea = QScrollArea(self.centralwidget)
self.scrollArea.setWidgetResizable(True)
self.bw_map_screen = BWMapViewer(self.centralwidget)
self.scrollArea.setWidget(self.bw_map_screen)
self.horizontalLayout.addWidget(self.scrollArea)
#self.horizontalLayout.addWidget(self.bw_map_screen)
self.entity_list_widget = BWEntityListWidget(self.centralwidget)
self.entity_list_widget.setMaximumSize(QSize(300, 16777215))
self.entity_list_widget.setObjectName("entity_list_widget")
self.horizontalLayout.addWidget(self.entity_list_widget)
spacerItem = QSpacerItem(10, 20, QSizePolicy.Minimum, QSizePolicy.Expanding)
self.horizontalLayout.addItem(spacerItem)
self.vertLayoutWidget = QWidget(self.centralwidget)
self.vertLayoutWidget.setMaximumSize(QSize(250, 1200))
self.verticalLayout = QVBoxLayout(self.vertLayoutWidget)
self.verticalLayout.setObjectName("verticalLayout")
#self.verticalLayout.
self.button_clone_entity = QPushButton(self.centralwidget)
self.button_clone_entity.setObjectName("button_clone_entity")
self.verticalLayout.addWidget(self.button_clone_entity)
self.button_remove_entity = QPushButton(self.centralwidget)
self.button_remove_entity.setObjectName("button_remove_entity")
self.verticalLayout.addWidget(self.button_remove_entity)
self.button_move_entity = QPushButton(self.centralwidget)
self.button_move_entity.setObjectName("button_move_entity")
self.verticalLayout.addWidget(self.button_move_entity)
self.button_show_passengers = QPushButton(self.centralwidget)
self.button_show_passengers.setObjectName("button_move_entity")
self.verticalLayout.addWidget(self.button_show_passengers)
self.gridLayout = QGridLayout()
self.gridLayout.setObjectName("gridLayout")
self.button_zoom_in = QPushButton(self.centralwidget)
self.button_zoom_in.setObjectName("button_zoom_in")
self.gridLayout.addWidget(self.button_zoom_in, 0, 0, 0, 1)
self.button_zoom_out = QPushButton(self.centralwidget)
self.button_zoom_out.setObjectName("button_zoom_out")
self.gridLayout.addWidget(self.button_zoom_out, 0, 1, 0, 1)
self.button_edit_xml = QPushButton(self.centralwidget)
self.button_edit_xml.setObjectName("button_edit_xml")
self.button_edit_base_xml = QPushButton(self.centralwidget)
self.button_edit_base_xml.setObjectName("button_edit_base_xml")
self.verticalLayout.addLayout(self.gridLayout)
self.verticalLayout.addWidget(self.button_edit_xml)
self.verticalLayout.addWidget(self.button_edit_base_xml)
spacerItem1 = QSpacerItem(10, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.verticalLayout.addItem(spacerItem1)
self.verticalLayout_2 = QVBoxLayout()
self.verticalLayout_2.setObjectName("verticalLayout_2")
self.lineedit_angle = QLineEdit(self.centralwidget)
self.lineedit_angle.setObjectName("lineedit_angle")
self.lineedit_angle.setPlaceholderText("Angle")
self.label_object_id = QLabel(self.centralwidget)
self.label_object_id.setObjectName("label_object_id")
#TextSelectableByCursor
self.label_position = QLabel(self.centralwidget)
self.label_position.setObjectName("label_position")
self.label_model_name = QLabel(self.centralwidget)
self.label_model_name.setObjectName("label_model_name")
self.label_4 = QLabel(self.centralwidget)
self.label_4.setObjectName("label_4")
self.label_5 = QLabel(self.centralwidget)
self.label_5.setObjectName("label_5")
for label in (self.label_object_id, self.label_position, self.label_model_name, self.label_4, self.label_5):
label.setTextInteractionFlags(QtCore.Qt.TextSelectableByMouse)
self.verticalLayout_2.addWidget(self.lineedit_angle)
self.verticalLayout_2.addWidget(self.label_object_id)
self.verticalLayout_2.addWidget(self.label_position)
self.verticalLayout_2.addWidget(self.label_model_name)
self.verticalLayout_2.addWidget(self.label_4)
self.verticalLayout_2.addWidget(self.label_5)
self.verticalLayout.addLayout(self.verticalLayout_2)
self.horizontalLayout.addWidget(self.vertLayoutWidget)
self.menubar = QMenuBar(MainWindow)
self.menubar.setGeometry(QRect(0, 0, 820, 29))
self.menubar.setObjectName("menubar")
self.file_menu = QMenu(self.menubar)
self.file_menu.setObjectName("menuLoad")
self.file_load_action = QAction("Load", self)
self.file_load_action.triggered.connect(self.button_load_level)
self.file_menu.addAction(self.file_load_action)
self.file_save_action = QAction("Save", self)
self.file_save_action.triggered.connect(self.button_save_level)
self.file_menu.addAction(self.file_save_action)
self.visibility_menu = MenuDontClose(self.menubar)#QMenu(self.menubar)
self.visibility_menu.setObjectName("visibility")
#self.visibility_menu.addAction(self.toggle_action)
self.visibility_actions = []
self.terrain_menu = QMenu(self.menubar)
self.terrain_menu.setObjectName("terrain")
self.terrain_load_action = QAction("Load Terrain", self)
self.terrain_load_action.triggered.connect(self.button_terrain_load_action)
self.terrain_menu.addAction(self.terrain_load_action)
self.terrain_display_actions = []
self.setup_terrain_display_toggles()
#self.menuLoad_2 = QMenu(self.menubar)
#self.menuLoad_2.setObjectName("menuLoad_2")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.menubar.addAction(self.file_menu.menuAction())
#self.menubar.addAction(self.menuLoad_2.menuAction())
self.menubar.addAction(self.visibility_menu.menuAction())
self.menubar.addAction(self.terrain_menu.menuAction())
self.retranslateUi(MainWindow)
QMetaObject.connectSlotsByName(MainWindow)
def __init__(self,
show=True,
app=None,
shape=(1, 1),
window_size=None,
off_screen=None,
**kwargs):
if not has_pyqt:
raise AssertionError('Requires PyQt5')
self.active = True
self.saved_camera_positions = []
self.counters = []
if window_size is None:
window_size = rcParams['window_size']
# Remove notebook argument in case user passed it
kwargs.pop('notebook', None)
# ipython magic
if scooby.in_ipython(): # pragma: no cover
from IPython import get_ipython
ipython = get_ipython()
ipython.magic('gui qt')
from IPython.external.qt_for_kernel import QtGui
QtGui.QApplication.instance()
else:
ipython = None
# run within python
if app is None:
from PyQt5.QtWidgets import QApplication
app = QApplication.instance()
if not app: # pragma: no cover
app = QApplication([''])
self.app = app
self.app_window = MainWindow()
self.app_window.setWindowTitle(kwargs.get('title', rcParams['title']))
self.frame = QFrame()
self.frame.setFrameStyle(QFrame.NoFrame)
QtInteractor.__init__(self,
parent=self.frame,
shape=shape,
off_screen=off_screen,
**kwargs)
self.app_window.signal_close.connect(self.quit)
# build main menu
self.main_menu = QMenuBar(parent=self.app_window)
self.app_window.signal_close.connect(self.main_menu.clear)
file_menu = self.main_menu.addMenu('File')
file_menu.addAction('Take Screenshot', self._qt_screenshot)
file_menu.addAction('Export as VTKjs', self._qt_export_vtkjs)
file_menu.addSeparator()
file_menu.addAction('Exit', self.app_window.close)
view_menu = self.main_menu.addMenu('View')
view_menu.addAction('Toggle Eye Dome Lighting', self._toggle_edl)
view_menu.addAction('Scale Axes', self.scale_axes_dialog)
view_menu.addAction('Clear All', self.clear)
tool_menu = self.main_menu.addMenu('Tools')
tool_menu.addAction('Enable Cell Picking (through)',
self.enable_cell_picking)
tool_menu.addAction('Enable Cell Picking (visible)',
lambda: self.enable_cell_picking(through=False))
cam_menu = view_menu.addMenu('Camera')
cam_menu.addAction('Toggle Parallel Projection',
self._toggle_parallel_projection)
cam_menu.addAction('Reset Camera', self.reset_camera)
cam_menu.addAction('Isometric View', self.view_isometric)
cam_menu.addAction('View XY Plane', self.view_xy)
cam_menu.addAction('View XZ Plane', self.view_xz)
cam_menu.addAction('View YZ Plane', self.view_yz)
cam_menu.addSeparator()
cam_menu.addAction('Save Current Camera Position',
self.save_camera_position)
cam_menu.addAction('Clear Saved Positions',
self.clear_camera_positions)
view_menu.addSeparator()
# Orientation marker
orien_menu = view_menu.addMenu('Orientation Marker')
orien_menu.addAction('Show', self.show_axes)
orien_menu.addAction('Hide', self.hide_axes)
# Bounds axes
axes_menu = view_menu.addMenu('Bounds Axes')
axes_menu.addAction('Add Bounds Axes (front)', self.show_bounds)
axes_menu.addAction('Add Bounds Grid (back)', self.show_grid)
axes_menu.addAction('Add Bounding Box', self.add_bounding_box)
axes_menu.addSeparator()
axes_menu.addAction('Remove Bounding Box', self.remove_bounding_box)
axes_menu.addAction('Remove Bounds', self.remove_bounds_axes)
# A final separator to seperate OS options
view_menu.addSeparator()
self.saved_camera_menu = self.main_menu.addMenu('Camera Positions')
vlayout = QVBoxLayout()
vlayout.addWidget(self)
self.frame.setLayout(vlayout)
self.app_window.setCentralWidget(self.frame)
if off_screen is None:
off_screen = pyvista.OFF_SCREEN
if show and not off_screen: # pragma: no cover
self.app_window.show()
self.show()
self._spawn_background_rendering()
self.window_size = window_size
self._last_update_time = time.time(
) - BackgroundPlotter.ICON_TIME_STEP / 2
self._last_window_size = self.window_size
self._last_camera_pos = self.camera_position
def _setupMenu(self):
self.menubar = QMenuBar()
self.menubar.setGeometry(QRect(0, 0, 630, 22))
self.menuFile = QMenu(self.menubar)
self.menuFile.setTitle(tr("File"))
self.menuMark = QMenu(self.menubar)
self.menuMark.setTitle(tr("Mark"))
self.menuActions = QMenu(self.menubar)
self.menuActions.setTitle(tr("Actions"))
self.menuColumns = QMenu(self.menubar)
self.menuColumns.setTitle(tr("Columns"))
self.menuView = QMenu(self.menubar)
self.menuView.setTitle(tr("View"))
self.menuHelp = QMenu(self.menubar)
self.menuHelp.setTitle(tr("Help"))
self.setMenuBar(self.menubar)
self.menuActions.addAction(self.actionDeleteMarked)
self.menuActions.addAction(self.actionMoveMarked)
self.menuActions.addAction(self.actionCopyMarked)
self.menuActions.addAction(self.actionRemoveMarked)
self.menuActions.addAction(self.actionReprioritize)
self.menuActions.addSeparator()
self.menuActions.addAction(self.actionRemoveSelected)
self.menuActions.addAction(self.actionIgnoreSelected)
self.menuActions.addAction(self.actionMakeSelectedReference)
self.menuActions.addSeparator()
self.menuActions.addAction(self.actionOpenSelected)
self.menuActions.addAction(self.actionRevealSelected)
self.menuActions.addAction(self.actionInvokeCustomCommand)
self.menuActions.addAction(self.actionRenameSelected)
self.menuMark.addAction(self.actionMarkAll)
self.menuMark.addAction(self.actionMarkNone)
self.menuMark.addAction(self.actionInvertMarking)
self.menuMark.addAction(self.actionMarkSelected)
self.menuView.addAction(self.actionPowerMarker)
self.menuView.addAction(self.actionDelta)
self.menuView.addSeparator()
self.menuView.addAction(self.actionDetails)
self.menuView.addAction(self.app.actionIgnoreList)
self.menuView.addAction(self.app.actionPreferences)
self.menuHelp.addAction(self.app.actionShowHelp)
self.menuHelp.addAction(self.app.actionOpenDebugLog)
self.menuHelp.addAction(self.app.actionAbout)
self.menuFile.addAction(self.actionSaveResults)
self.menuFile.addAction(self.actionExportToHTML)
self.menuFile.addAction(self.actionExportToCSV)
self.menuFile.addSeparator()
self.menuFile.addAction(self.app.actionQuit)
self.menubar.addAction(self.menuFile.menuAction())
self.menubar.addAction(self.menuMark.menuAction())
self.menubar.addAction(self.menuActions.menuAction())
self.menubar.addAction(self.menuColumns.menuAction())
self.menubar.addAction(self.menuView.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
# Columns menu
menu = self.menuColumns
self._column_actions = []
for index, (display, visible) in enumerate(self.app.model.result_table.columns.menu_items()):
action = menu.addAction(display)
action.setCheckable(True)
action.setChecked(visible)
action.item_index = index
self._column_actions.append(action)
menu.addSeparator()
action = menu.addAction(tr("Reset to Defaults"))
action.item_index = -1
# Action menu
actionMenu = QMenu(tr("Actions"), self.menubar)
actionMenu.addAction(self.actionDeleteMarked)
actionMenu.addAction(self.actionMoveMarked)
actionMenu.addAction(self.actionCopyMarked)
actionMenu.addAction(self.actionRemoveMarked)
actionMenu.addSeparator()
actionMenu.addAction(self.actionRemoveSelected)
actionMenu.addAction(self.actionIgnoreSelected)
actionMenu.addAction(self.actionMakeSelectedReference)
actionMenu.addSeparator()
actionMenu.addAction(self.actionOpenSelected)
actionMenu.addAction(self.actionRevealSelected)
actionMenu.addAction(self.actionInvokeCustomCommand)
actionMenu.addAction(self.actionRenameSelected)
self.actionActions.setMenu(actionMenu)
self.actionsButton.setMenu(self.actionActions.menu())
class MainWindow(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
self.minimize_to_tray = True
self.alarm = AlarmDaemon(self)
# Window definitions
self.setMinimumSize(QSize(480, 320))
self.setWindowTitle("AthanPy")
self.centralwidget = QWidget(self)
self.centralwidget.setObjectName("centralwidget")
# Layout definitions
self.gridLayout_2 = QGridLayout(self.centralwidget)
self.gridLayout_2.setObjectName("gridLayout_2")
self.gridLayout = QGridLayout()
self.gridLayout.setObjectName("gridLayout")
# Widget definitions
#self.txt_times = QLabel(self.centralwidget)
#sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred)
#sizePolicy.setHorizontalStretch(0)
#sizePolicy.setVerticalStretch(0)
#sizePolicy.setHeightForWidth(self.txt_times.sizePolicy().hasHeightForWidth())
#self.txt_times.setSizePolicy(sizePolicy)
#self.txt_times.setAlignment(Qt.AlignCenter)
#self.txt_times.setObjectName("txt_times")
#self.txt_times.setText(self.get_prayertime_text())
#self.gridLayout.addWidget(self.txt_times, 1, 0, 1, 1)
self.lbl_prayername = {}
self.lbl_prayertime = {}
i = 1
for n in [
'Fajr', 'Sunrise', 'Dhuhr', 'Asr', 'Maghrib', 'Isha',
'Midnight'
]:
self.lbl_prayername[n] = QLabel(self.centralwidget)
self.lbl_prayername[n].setAlignment(Qt.AlignLeft)
self.lbl_prayername[n].setText(n)
self.gridLayout.addWidget(self.lbl_prayername[n], i, 0, 1, 1)
self.lbl_prayertime[n] = QLabel(self.centralwidget)
self.lbl_prayertime[n].setAlignment(Qt.AlignCenter)
self.gridLayout.addWidget(self.lbl_prayertime[n], i, 1, 1, 1)
i += 1
self.set_prayertime()
self.txt_date = QLabel(self.centralwidget)
self.txt_date.setObjectName("txt_date")
self.gridLayout.addWidget(self.txt_date, 0, 0, 1, 2)
self.gridLayout_2.addLayout(self.gridLayout, 0, 0, 1, 1)
# Menu bar
self.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(self)
self.menubar.setGeometry(QRect(0, 0, 480, 20))
self.menubar.setObjectName("menubar")
self.menuFile = QMenu(self.menubar)
self.menuFile.setObjectName("menuFile")
self.setMenuBar(self.menubar)
# Status bar
self.statusbar = QStatusBar(self)
self.statusbar.setObjectName("statusbar")
self.setStatusBar(self.statusbar)
self.actionSettings = QAction(self)
self.actionSettings.setObjectName("actionSettings")
self.actionSettings.setText("Settings")
self.actionSettings.triggered.connect(self.show_settings)
self.actionQuit = QAction(self)
self.actionQuit.setObjectName("actionQuit")
self.actionQuit.setText("Quit")
self.actionQuit.triggered.connect(qApp.quit)
self.menuFile.addAction(self.actionSettings)
self.menuFile.addAction(self.actionQuit)
self.menuFile.setTitle("File")
self.menubar.addAction(self.menuFile.menuAction())
# QSystemTrayIcon
self.tray_icon = QSystemTrayIcon(self)
self.tray_icon.setIcon(self.style().standardIcon(
QStyle.SP_ComputerIcon))
show_action = QAction("Show", self)
hide_action = QAction("Hide", self)
mute_action = QAction("Mute sound", self)
stopsound_action = QAction("Stop sound", self)
quit_action = QAction("Quit", self)
show_action.triggered.connect(self.show)
hide_action.triggered.connect(self.hide)
mute_action.setCheckable(True)
mute_action.changed.connect(
lambda: self.alarm.set_mute(mute_action.isChecked()))
stopsound_action.triggered.connect(self.alarm.stop_sound)
quit_action.triggered.connect(QCoreApplication.quit)
tray_menu = QMenu()
tray_menu.addAction(show_action)
tray_menu.addAction(hide_action)
tray_menu.addSeparator()
tray_menu.addAction(stopsound_action)
tray_menu.addAction(mute_action)
tray_menu.addSeparator()
tray_menu.addAction(quit_action)
self.tray_icon.setContextMenu(tray_menu)
self.tray_icon.show()
# Settings window
self.settingswindow = SettingsWindow(self)
def closeEvent(self, event):
if self.minimize_to_tray:
print('triggered 1')
event.ignore()
self.hide()
self.tray_icon.showMessage("Tray Program",
"Application was minimized to tray",
QSystemTrayIcon.Information, 2000)
def set_prayertime(self):
SettingsManager.refreshVariables(self.show_settings)
SettingsManager.calcTimes()
times = SettingsManager.times
# For testing purposes, comment/delete when not in use
#times['dhuhr'] = '08:02'
#times['asr'] = '08:03'
# times['isha'] = '16:20'
for n in [
'Fajr', 'Sunrise', 'Dhuhr', 'Asr', 'Maghrib', 'Isha',
'Midnight'
]:
self.lbl_prayertime[n].setText(times[n.lower()])
self.alarm.stop_alarm()
self.alarm.schedule_alarm(times)
self.alarm.start()
#return output
def show_settings(self):
self.settingswindow.show()
