def __init__(self, actualize_visual_state_callback):
GLog.debug("ui.UsersForm.__init__")
super().__init__()
self.add_edit_dialog = None
self.actualize_visual_state_callback = actualize_visual_state_callback
uic.loadUi('ui/usersForm.ui', self)
self.actionExit.triggered.connect(self.close)
self.actionExit.setShortcut('Ctrl+W')
self.btnClose.clicked.connect(self.close)
self.actionAdd_item.triggered.connect(self.add_item)
self.btnAdd.clicked.connect(self.add_item)
self.actionRemove_item.triggered.connect(self.remove_item)
self.btnRemove.clicked.connect(self.remove_item)
self.actionEdit_item.triggered.connect(self.edit_item)
self.btnEdit.clicked.connect(self.edit_item)
self.table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.table.setSelectionMode(QAbstractItemView.SingleSelection)
self.table.setColumnWidth(0, 80)
self.table.setColumnWidth(1, 150)
self.table.setColumnWidth(2, 100)
self.table.setColumnWidth(3, 350)
self.init_data()
def remove_item(self):
entity = Expense if self.tabWidget.currentIndex() == 0 else Income
table = self.tableExpenses if self.tabWidget.currentIndex(
) == 0 else self.tableIncomes
rows = table.selectionModel().selectedRows()
if len(rows) == 0:
# todo: говнокодство конечно... надо дисейблить кнопку пока ничего не выбрано в таблице... но времени нет
msg = 'Пользователь не выбран, удалять нечего'
QMessageBox.information(self, 'Remove item', msg)
return
msg = f'Вы действительно хотете удалить {"пользователя" if len(rows) == 1 else "пользователей"}?'
reply = QMessageBox.question(
self, f'Remove item{"" if len(rows) == 1 else "s"}', msg,
QMessageBox.Yes, QMessageBox.No)
if reply == QMessageBox.Yes:
for index in sorted(rows, reverse=True):
GLog.debug('remove_item :: row: %s => User.ID:%s' %
(index.row(), table.item(index.row(), 0).text()))
try:
entity = Expense if self.tabWidget.currentIndex(
) == 0 else Income
entity.delete(table.item(index.row(), 0).text())
except Exception as e:
# todo: хорошо бы конечно оповщение о том что пошло не так - но не в этой жизни...
GLog.error("ui.MainForm.remove_item :: error :%s" % e)
return
table.model().removeRow(index.row())
def __init__(self, actualize_visual_state_callback):
GLog.debug("ui.MainForm.__init__")
super().__init__()
self.add_edit_dialog = None
self.actualize_visual_state_callback = actualize_visual_state_callback
uic.loadUi('ui/mainForm.ui', self)
self.actionExit.triggered.connect(qApp.quit)
self.btnExit.clicked.connect(qApp.quit)
self.actionAdminUsers.triggered.connect(self.open_users)
self.actionAdd_item.triggered.connect(self.add_item)
self.btnAdd.clicked.connect(self.add_item)
self.actionRemove_item.triggered.connect(self.remove_item)
self.btnRemove.clicked.connect(self.remove_item)
self.actionEdit_item.triggered.connect(self.edit_item)
self.btnEdit.clicked.connect(self.edit_item)
self.tableIncomes.setSelectionBehavior(QAbstractItemView.SelectRows)
self.tableIncomes.setSelectionMode(QAbstractItemView.SingleSelection)
self.tableExpenses.setSelectionBehavior(QAbstractItemView.SelectRows)
self.tableExpenses.setSelectionMode(QAbstractItemView.SingleSelection)
self.tableIncomes.setColumnWidth(0, 80)
self.tableIncomes.setColumnWidth(1, 150)
self.tableIncomes.setColumnWidth(2, 100)
self.tableIncomes.setColumnWidth(3, 350)
self.tableExpenses.setColumnWidth(0, 80)
self.tableExpenses.setColumnWidth(1, 150)
self.tableExpenses.setColumnWidth(2, 100)
self.tableExpenses.setColumnWidth(3, 350)
def __init__(self, *args):
GLog.debug("ui.UserAddEditDialog.__init__")
super().__init__(*args)
uic.loadUi('ui/userAddEditDialog.ui', self)
self.bntSave.clicked.connect(self.accept)
self.btnCancel.clicked.connect(self.close)
self.init_value()
def _delete(self):
"""."""
GLog.debug('Удаление дохода из БД')
self.db().execute(
'''
DELETE FROM "incomes"
WHERE "id" = ?;
''', [self.ID])
def _delete(self):
"""."""
GLog.debug('Удаление пользователя из БД')
self.db().execute(
'''
DELETE FROM "users"
WHERE "id" = ?;
''', [self.ID])
def _create(self):
"""."""
GLog.debug('Вставка дохода в БД')
self.ID = self.db().execute('''
INSERT INTO "incomes" (
"date", "value", "note"
) VALUES (?, ?, ?);
''', [self.date, self.value, self.note],
with_result=True)
def _update(self):
"""."""
GLog.debug('Изменениее дохода в БД')
self.db().execute(
'''
UPDATE "incomes" SET
"date" = ?,
"value" = ?,
"note" = ?
WHERE "id" = ?;
''', [self.date, self.value, self.note, self.ID])
def _create(self):
"""."""
GLog.debug('Вставка пользователя в БД')
passw = self.check_str(self.password, arg='Пароль пользователя')
passw = sha256(passw.encode('utf-8')).hexdigest()
self.ID = self.db().execute('''
INSERT INTO "users" (
"login", "passw", "name", "is_admin"
) VALUES (?, ?, ?, ?);
''', [self.login, passw, self.name, self.is_admin],
with_result=True)
def _update(self):
"""."""
GLog.debug('Изменениее пользователя в БД')
passw = self.check_str(self.password, arg='Пароль пользователя')
passw = sha256(passw.encode('utf-8')).hexdigest()
self.db().execute(
'''
UPDATE "users" SET
"login" = ?,
"passw" = ?,
"name" = ?,
"is_admin" = ?
WHERE "id" = ?;
''', [self.login, passw, self.name, self.is_admin, self.ID])
def on_login(self):
GLog.debug("ui.LoginForm.on_login")
login = self.inputLogin.text()
password = self.inputPass.text()
GLog.info(f"ui.LoginForm.on_login :: try to login as: {login}")
try:
user = User.select_by_credentials(login, password)
if user.ID > 0:
state.State[state.STATE_KEY_USER] = user
state.State[
state.
STATE_KEY_SECTION] = state.STATE_SECTION_MAIN_EXPENSES
self.on_login_success()
except Exception as e:
GLog.error("ui.LoginForm.on_login :: error: %s", e)
def edit_item(self):
entity = Expense if self.tabWidget.currentIndex() == 0 else Income
table = self.tableExpenses if self.tabWidget.currentIndex(
) == 0 else self.tableIncomes
rows = table.selectionModel().selectedRows()
if len(rows) != 1:
# todo: говнокодство конечно... надо дисейблить кнопку пока ничего не выбрано или выбрано >1 в таблице... но времени нет
msg = 'Пользователь не выбран/выбрано больше 1 пользователя, изменение невозможно'
QMessageBox.information(self, 'Edit item', msg)
return
index = rows[0]
GLog.debug('edit_item :: row: %s => User.ID:%s' %
(index.row(), table.item(index.row(), 0).text()))
State[STATE_KEY_ELEMENT] = {
'id':
table.item(index.row(), 0).text(),
'value':
float(table.item(index.row(), 2).text()),
'date':
datetime.strptime(
table.item(index.row(), 1).text(),
'%Y-%m-%d %H:%M:%S').timestamp(),
'note':
table.item(index.row(), 3).text(),
}
self.add_edit_dialog.init_value()
if not self.add_edit_dialog.exec():
return
try:
res = entity.edit(
State[STATE_KEY_ELEMENT]['id'], {
'value':
self.add_edit_dialog.valueDoubleSpinBox.value(),
'date':
self.add_edit_dialog.dateTimeEdit.dateTime().
toSecsSinceEpoch(),
'note':
self.add_edit_dialog.noteTextEdit.toPlainText(),
})
except Exception as e:
# todo: хорошо бы конечно оповщение о том что пошло не так - но не в этой жизни...
GLog.error("ui.MainForm.edit_item :: error :%s" % e)
return
table.setItem(index.row(), 0, QTableWidgetItem(str(res.ID)))
table.setItem(index.row(), 1, QTableWidgetItem(res.date_str))
table.setItem(index.row(), 2, QTableWidgetItem(str(res.value)))
table.setItem(index.row(), 3, QTableWidgetItem(res.note))
def edit_item(self):
rows = self.table.selectionModel().selectedRows()
if len(rows) != 1:
# todo: говнокодство конечно... надо дисейблить кнопку пока ничего не выбрано или выбрано >1 в таблице... но времени нет
msg = 'Пользователь не выбран/выбрано больше 1 пользователя, изменение невозможно'
QMessageBox.information(self, 'Edit item', msg)
return
index = rows[0]
GLog.debug('edit_item :: row: %s => User.ID:%s' %
(index.row(), self.table.item(index.row(), 0).text()))
State[STATE_KEY_ELEMENT] = {
'id': self.table.item(index.row(), 0).text(),
'name': self.table.item(index.row(), 1).text(),
'login': self.table.item(index.row(), 2).text(),
'is_admin': self.table.item(index.row(), 3).text() == "Да",
}
self.add_edit_dialog.init_value()
if not self.add_edit_dialog.exec():
return
try:
user = User.edit(
State[STATE_KEY_ELEMENT]['id'], {
'is_admin': self.add_edit_dialog.isAdmin.isChecked(),
'name': self.add_edit_dialog.userName.text(),
'login': self.add_edit_dialog.userLogin.text(),
'password': self.add_edit_dialog.userPassword.text(),
})
except Exception as e:
# todo: хорошо бы конечно оповщение о том что пошло не так - но не в этой жизни...
GLog.error("ui.UsersForm.edit_item :: error :%s" % e)
return
self.table.setItem(index.row(), 0, QTableWidgetItem(str(user.ID)))
self.table.setItem(index.row(), 1, QTableWidgetItem(user.name))
self.table.setItem(index.row(), 2, QTableWidgetItem(user.login))
self.table.setItem(index.row(), 3,
QTableWidgetItem('Да' if user.admin else 'Нет'))
def recalc_elements_availability(self):
cur_user = State[STATE_KEY_USER]
GLog.debug("ui.MainForm.recalc_elements_availability: %s",
cur_user.admin)
self.menuAdmin.setEnabled(cur_user.admin)
self.actionAdminUsers.setEnabled(cur_user.admin)
def __init__(self, on_login_success):
GLog.debug("ui.LoginForm.__init__")
super().__init__()
self.on_login_success = on_login_success
uic.loadUi('ui/loginForm.ui', self)
self.btnLogin.clicked.connect(self.on_login)