def __init__(self, parent=None):
QWidget.__init__(self, parent)
self.model = None
self.table_name = ''
self.mapper = None
self.modified = False
self.name = "N/A"
self.operation_type = None
self.layout = QGridLayout(self)
self.layout.setContentsMargins(2, 2, 2, 2)
self.bold_font = QFont()
self.bold_font.setBold(True)
self.main_label = QLabel(self)
self.main_label.setFont(self.bold_font)
self.layout.addWidget(self.main_label, 0, 0, 1, 1, Qt.AlignLeft)
self.commit_button = QPushButton(load_icon("accept.png"), '', self)
self.commit_button.setToolTip(self.tr("Commit changes"))
self.commit_button.setEnabled(False)
self.revert_button = QPushButton(load_icon("cancel.png"), '', self)
self.revert_button.setToolTip(self.tr("Cancel changes"))
self.revert_button.setEnabled(False)
self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum,
QSizePolicy.Expanding)
self.horizontalSpacer = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
def __init__(self):
QDialog.__init__(self)
self.setupUi(self)
self.model = None
self._previous_row = -1
self.selected_id = 0
self.p_selected_name = ''
self._filter_text = ''
self.selection_enabled = False
self.group_id = None
self.group_key_field = None
self.group_key_index = None
self.group_fkey_field = None
self.filter_field = None
self._filter_value = ''
self.toggle_state = False
self.toggle_field = None
self.search_field = None
self.search_text = ""
self.tree_view = False
self.toolbar = None
self.custom_editor = False
self.AddChildBtn.setVisible(False)
self.GroupLbl.setVisible(False)
self.GroupCombo.setVisible(False)
self.SearchFrame.setVisible(False)
self.AddBtn.setIcon(load_icon("add.png"))
self.AddChildBtn.setIcon(load_icon("add_child.png"))
self.RemoveBtn.setIcon(load_icon("delete.png"))
self.CommitBtn.setIcon(load_icon("accept.png"))
self.RevertBtn.setIcon(load_icon("cancel.png"))
self.SearchString.textChanged.connect(self.OnSearchChange)
self.GroupCombo.currentIndexChanged.connect(self.OnGroupChange)
self.Toggle.stateChanged.connect(self.OnToggleChange)
self.AddBtn.clicked.connect(self.OnAdd)
self.AddChildBtn.clicked.connect(self.OnChildAdd)
self.RemoveBtn.clicked.connect(self.OnRemove)
self.CommitBtn.clicked.connect(self.OnCommit)
self.RevertBtn.clicked.connect(self.OnRevert)
self.DataView.doubleClicked.connect(self.OnDoubleClicked)
self.DataView.clicked.connect(self.OnClicked)
self.TreeView.doubleClicked.connect(self.OnDoubleClicked)
self.TreeView.clicked.connect(self.OnClicked)
def createStatementsImportMenu(self):
for i, source in enumerate(self.statements.sources):
if 'icon' in source:
source_icon = load_icon(source['icon'])
action = QAction(source_icon, source['name'], self)
else:
action = QAction(source['name'], self)
action.setData(i)
self.menuStatement.addAction(action)
self.statementGroup.addAction(action)
def createStatementsImportMenu(self):
for i, statement in enumerate(self.statements.items):
statement_name = statement['name'].replace('&', '&&') # & -> && to prevent shortcut creation
if statement['icon']:
statement_icon = load_icon(statement['icon'])
action = QAction(statement_icon, statement_name, self)
else:
action = QAction(statement_name, self)
action.setData(i)
self.menuStatement.addAction(action)
self.statementGroup.addAction(action)
def __init__(self, parent=None):
MdiWidget.__init__(self, parent)
self.setupUi(self)
self.current_index = None # this is used in onOperationContextMenu() to track item for menu
# Set icons
self.NewOperationBtn.setIcon(load_icon("new.png"))
self.CopyOperationBtn.setIcon(load_icon("copy.png"))
self.DeleteOperationBtn.setIcon(load_icon("delete.png"))
# Operations view context menu
self.contextMenu = QMenu(self.OperationsTableView)
self.actionReconcile = QAction(load_icon("reconcile.png"),
self.tr("Reconcile"), self)
self.actionCopy = QAction(load_icon("copy.png"), self.tr("Copy"), self)
self.actionDelete = QAction(load_icon("delete.png"), self.tr("Delete"),
self)
self.contextMenu.addAction(self.actionReconcile)
self.contextMenu.addSeparator()
self.contextMenu.addAction(self.actionCopy)
self.contextMenu.addAction(self.actionDelete)
# Customize UI configuration
self.balances_model = BalancesModel(self.BalancesTableView)
self.BalancesTableView.setModel(self.balances_model)
self.balances_model.configureView()
self.operations_model = OperationsModel(self.OperationsTableView)
self.OperationsTableView.setModel(self.operations_model)
self.operations_model.configureView()
self.OperationsTableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.connect_signals_and_slots()
self.NewOperationMenu = QMenu()
for i in range(self.OperationsTabs.count()):
if hasattr(self.OperationsTabs.widget(i), "isCustom"):
self.OperationsTabs.widget(i).dbUpdated.connect(self.dbUpdated)
self.OperationsTabs.widget(i).dbUpdated.connect(
self.operations_model.refresh)
self.NewOperationMenu.addAction(
self.OperationsTabs.widget(i).name,
partial(self.createOperation, i))
self.NewOperationBtn.setMenu(self.NewOperationMenu)
# Setup balance and holdings parameters
current_time = QDateTime.currentDateTime()
current_time.setTimeSpec(
Qt.UTC) # We use UTC everywhere so need to force TZ info
self.BalanceDate.setDateTime(current_time)
self.BalancesCurrencyCombo.setIndex(
JalSettings().getValue('BaseCurrency'))
self.OperationsTabs.setCurrentIndex(LedgerTransaction.NA)
self.OperationsTableView.selectRow(0)
self.DateRange.setCurrentIndex(0)
def __init__(self, language):
QMainWindow.__init__(self, None)
self.running = False
self.setupUi(self)
self.restoreGeometry(base64.decodebytes(JalSettings().getValue('WindowGeometry', '').encode('utf-8')))
self.restoreState(base64.decodebytes(JalSettings().getValue('WindowState', '').encode('utf-8')))
self.ledger = Ledger()
# Customize Status bar and logs
self.ProgressBar = QProgressBar(self)
self.StatusBar.addPermanentWidget(self.ProgressBar)
self.ProgressBar.setVisible(False)
self.ledger.setProgressBar(self, self.ProgressBar)
self.Logs.setStatusBar(self.StatusBar)
self.logger = logging.getLogger()
self.logger.addHandler(self.Logs)
log_level = os.environ.get('LOGLEVEL', 'INFO').upper()
self.logger.setLevel(log_level)
self.currentLanguage = language
self.downloader = QuoteDownloader()
self.statements = Statements(self)
self.reports = Reports(self, self.mdiArea)
self.backup = JalBackup(self, get_dbfilename(get_app_path()))
self.estimator = None
self.price_chart = None
self.actionImportSlipRU.setEnabled(dependency_present(['pyzbar', 'PIL']))
self.actionAbout = QAction(text=self.tr("About"), parent=self)
self.MainMenu.addAction(self.actionAbout)
self.langGroup = QActionGroup(self.menuLanguage)
self.createLanguageMenu()
self.statementGroup = QActionGroup(self.menuStatement)
self.createStatementsImportMenu()
self.reportsGroup = QActionGroup(self.menuReports)
self.createReportsMenu()
self.setWindowIcon(load_icon("jal.png"))
self.connect_signals_and_slots()
self.actionOperations.trigger()
def __init__(self):
QDialog.__init__(self)
self.setupUi(self)
self._asset_id = -1
# Custom model to allow common submit errors handling and error message display
self._model = AssetsListModel("assets", self)
self._mapper = QDataWidgetMapper(self._model)
self._mapper.setModel(self._model)
self._mapper.setSubmitPolicy(QDataWidgetMapper.AutoSubmit)
self._mapper.addMapping(self.NameEdit,
self._model.fieldIndex("full_name"))
self._mapper.addMapping(self.isinEdit, self._model.fieldIndex("isin"))
self._mapper.addMapping(self.TypeCombo,
self._model.fieldIndex("type_id"))
self._mapper.addMapping(self.CountryCombo,
self._model.fieldIndex("country_id"))
self._mapper.addMapping(self.BaseAssetSelector,
self._model.fieldIndex("base_asset"))
self._model.select()
self._symbols_model = SymbolsListModel("asset_tickers",
self.SymbolsTable)
self.SymbolsTable.setModel(self._symbols_model)
self._symbols_model.select()
self._symbols_model.configureView()
self._data_model = ExtraDataModel("asset_data", self.DataTable)
self.DataTable.setModel(self._data_model)
self._data_model.select()
self._data_model.configureView()
self.AddSymbolButton.setIcon(load_icon("add.png"))
self.RemoveSymbolButton.setIcon(load_icon("delete.png"))
self.AddDataButton.setIcon(load_icon("add.png"))
self.RemoveDataButton.setIcon(load_icon("delete.png"))
self.OkButton.setIcon(load_icon("accept.png"))
self.CancelButton.setIcon(load_icon("cancel.png"))
self.TypeCombo.currentIndexChanged.connect(self.onTypeUpdate)
self.AddSymbolButton.clicked.connect(self.onAddSymbol)
self.RemoveSymbolButton.clicked.connect(self.onRemoveSymbol)
self.AddDataButton.clicked.connect(self.onAddData)
self.RemoveDataButton.clicked.connect(self.onRemoveData)
def __init__(self, language):
QMainWindow.__init__(self, None)
self.setupUi(self)
self.currentLanguage = language
self.current_index = None # this is used in onOperationContextMenu() to track item for menu
self.ledger = Ledger()
self.downloader = QuoteDownloader()
self.taxes = TaxesRus()
self.statements = StatementLoader()
self.backup = JalBackup(self, get_dbfilename(get_app_path()))
self.estimator = None
self.price_chart = None
self.actionImportSlipRU.setEnabled(
dependency_present(['pyzbar', 'PIL']))
self.actionAbout = QAction(text=self.tr("About"), parent=self)
self.MainMenu.addAction(self.actionAbout)
self.langGroup = QActionGroup(self.menuLanguage)
self.createLanguageMenu()
self.statementGroup = QActionGroup(self.menuStatement)
self.createStatementsImportMenu()
# Set icons
self.setWindowIcon(load_icon("jal.png"))
self.NewOperationBtn.setIcon(load_icon("new.png"))
self.CopyOperationBtn.setIcon(load_icon("copy.png"))
self.DeleteOperationBtn.setIcon(load_icon("delete.png"))
# Operations view context menu
self.contextMenu = QMenu(self.OperationsTableView)
self.actionReconcile = QAction(load_icon("reconcile.png"),
self.tr("Reconcile"), self)
self.actionCopy = QAction(load_icon("copy.png"), self.tr("Copy"), self)
self.actionDelete = QAction(load_icon("delete.png"), self.tr("Delete"),
self)
self.contextMenu.addAction(self.actionReconcile)
self.contextMenu.addSeparator()
self.contextMenu.addAction(self.actionCopy)
self.contextMenu.addAction(self.actionDelete)
# Customize Status bar and logs
self.ProgressBar = QProgressBar(self)
self.StatusBar.addWidget(self.ProgressBar)
self.ProgressBar.setVisible(False)
self.ledger.setProgressBar(self, self.ProgressBar)
self.NewLogEventLbl = QLabel(self)
self.StatusBar.addWidget(self.NewLogEventLbl)
self.Logs.setNotificationLabel(self.NewLogEventLbl)
self.Logs.setFormatter(
logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
self.logger = logging.getLogger()
self.logger.addHandler(self.Logs)
log_level = os.environ.get('LOGLEVEL', 'INFO').upper()
self.logger.setLevel(log_level)
# Setup reports tab
self.reports = Reports(self.ReportTableView, self.ReportTreeView)
# Customize UI configuration
self.balances_model = BalancesModel(self.BalancesTableView)
self.BalancesTableView.setModel(self.balances_model)
self.balances_model.configureView()
self.holdings_model = HoldingsModel(self.HoldingsTableView)
self.HoldingsTableView.setModel(self.holdings_model)
self.holdings_model.configureView()
self.HoldingsTableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.operations_model = OperationsModel(self.OperationsTableView)
self.OperationsTableView.setModel(self.operations_model)
self.operations_model.configureView()
self.OperationsTableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.connect_signals_and_slots()
self.NewOperationMenu = QMenu()
for i in range(self.OperationsTabs.count()):
if hasattr(self.OperationsTabs.widget(i), "isCustom"):
self.OperationsTabs.widget(i).dbUpdated.connect(
self.ledger.rebuild)
self.OperationsTabs.widget(i).dbUpdated.connect(
self.operations_model.refresh)
self.NewOperationMenu.addAction(
self.OperationsTabs.widget(i).name,
partial(self.createOperation, i))
self.NewOperationBtn.setMenu(self.NewOperationMenu)
# Setup balance and holdings parameters
current_time = QDateTime.currentDateTime()
current_time.setTimeSpec(
Qt.UTC) # We use UTC everywhere so need to force TZ info
self.BalanceDate.setDateTime(current_time)
self.BalancesCurrencyCombo.setIndex(
JalSettings().getValue('BaseCurrency'))
self.HoldingsDate.setDateTime(current_time)
self.HoldingsCurrencyCombo.setIndex(
JalSettings().getValue('BaseCurrency'))
self.OperationsTabs.setCurrentIndex(TransactionType.NA)
self.OperationsTableView.selectRow(0)
self.OnOperationsRangeChange(0)
def __init__(self, parent=None):
AbstractOperationDetails.__init__(self, parent)
self.name = "Income/Spending"
self.details_model = None
self.category_delegate = CategorySelectorDelegate()
self.tag_delegate = TagSelectorDelegate()
self.float_delegate = FloatDelegate(2)
self.date_label = QLabel(self)
self.details_label = QLabel(self)
self.account_label = QLabel(self)
self.peer_label = QLabel(self)
self.main_label.setText(self.tr("Income / Spending"))
self.date_label.setText(self.tr("Date/Time"))
self.details_label.setText(self.tr("Details"))
self.account_label.setText(self.tr("Account"))
self.peer_label.setText(self.tr("Peer"))
self.timestamp_editor = QDateTimeEdit(self)
self.timestamp_editor.setCalendarPopup(True)
self.timestamp_editor.setTimeSpec(Qt.UTC)
self.timestamp_editor.setFixedWidth(
self.timestamp_editor.fontMetrics().horizontalAdvance(
"00/00/0000 00:00:00") * 1.25)
self.timestamp_editor.setDisplayFormat("dd/MM/yyyy hh:mm:ss")
self.account_widget = AccountSelector(self)
self.peer_widget = PeerSelector(self)
self.a_currency = OptionalCurrencyComboBox(self)
self.a_currency.setText(self.tr("Paid in foreign currency:"))
self.add_button = QPushButton(load_icon("add.png"), '', self)
self.add_button.setToolTip(self.tr("Add detail"))
self.del_button = QPushButton(load_icon("remove.png"), '', self)
self.del_button.setToolTip(self.tr("Remove detail"))
self.copy_button = QPushButton(load_icon("copy.png"), '', self)
self.copy_button.setToolTip(self.tr("Copy detail"))
self.details_table = QTableView(self)
self.details_table.horizontalHeader().setFont(self.bold_font)
self.details_table.setAlternatingRowColors(True)
self.details_table.verticalHeader().setVisible(False)
self.details_table.verticalHeader().setMinimumSectionSize(20)
self.details_table.verticalHeader().setDefaultSectionSize(20)
self.layout.addWidget(self.date_label, 1, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.details_label, 2, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.timestamp_editor, 1, 1, 1, 4)
self.layout.addWidget(self.add_button, 2, 1, 1, 1)
self.layout.addWidget(self.copy_button, 2, 2, 1, 1)
self.layout.addWidget(self.del_button, 2, 3, 1, 1)
self.layout.addWidget(self.account_label, 1, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.peer_label, 2, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.account_widget, 1, 6, 1, 1)
self.layout.addWidget(self.peer_widget, 2, 6, 1, 1)
self.layout.addWidget(self.a_currency, 1, 7, 1, 1)
self.layout.addWidget(self.commit_button, 0, 9, 1, 1)
self.layout.addWidget(self.revert_button, 0, 10, 1, 1)
self.layout.addWidget(self.details_table, 4, 0, 1, 11)
self.layout.addItem(self.horizontalSpacer, 1, 8, 1, 1)
self.add_button.clicked.connect(self.addChild)
self.copy_button.clicked.connect(self.copyChild)
self.del_button.clicked.connect(self.delChild)
super()._init_db("actions")
self.model.beforeInsert.connect(self.before_record_insert)
self.model.beforeUpdate.connect(self.before_record_update)
self.mapper.setItemDelegate(IncomeSpendingWidgetDelegate(self.mapper))
self.details_model = DetailsModel(self.details_table, db_connection())
self.details_model.setTable("action_details")
self.details_model.setEditStrategy(QSqlTableModel.OnManualSubmit)
self.details_table.setModel(self.details_model)
self.details_model.dataChanged.connect(self.onDataChange)
self.account_widget.changed.connect(self.mapper.submit)
self.peer_widget.changed.connect(self.mapper.submit)
self.a_currency.changed.connect(self.mapper.submit)
self.a_currency.name_updated.connect(self.details_model.setAltCurrency)
self.mapper.addMapping(self.timestamp_editor,
self.model.fieldIndex("timestamp"))
self.mapper.addMapping(self.account_widget,
self.model.fieldIndex("account_id"))
self.mapper.addMapping(self.peer_widget,
self.model.fieldIndex("peer_id"))
self.mapper.addMapping(self.a_currency,
self.model.fieldIndex("alt_currency_id"))
self.details_table.setItemDelegateForColumn(2, self.category_delegate)
self.details_table.setItemDelegateForColumn(3, self.tag_delegate)
self.details_table.setItemDelegateForColumn(4, self.float_delegate)
self.details_table.setItemDelegateForColumn(5, self.float_delegate)
self.model.select()
self.details_model.select()
self.details_model.configureView()