1
import sys
2
import datetime
3
from dateutil import parser
4

5
import pandas
6
from pandas.tseries import offsets
7

8
from PySide6.QtWidgets import QApplication, QDialog, QMainWindow, QPushButton, QDialogButtonBox, QVBoxLayout, QLabel, QTableWidget, QMenu,QAbstractItemView
9
from PySide6.QtGui import QPalette, QColor
10
from PySide6.QtWidgets import QApplication, QTableWidgetItem
11
import PySide6.QtCore
12
from PySide6.QtCore import QDate, QRect
13
from PySide6 import QtWidgets
14
from PySide6.QtUiTools import QUiLoader
15

16
expenses = pandas.read_excel("FamilyBudget.xlsx")
17
expenses = expenses.sort_values("Дата", ascending=False)
18
expenses["Оставшаяся сумма"] = expenses["Сумма дохода"] - \
19
    expenses["Сумма расхода"]
20

21
expenses = expenses.groupby('Дата').sum().reset_index()
22
print(expenses.groupby('Дата').sum())
23
this_year = datetime.date.today() - offsets.YearBegin()
24
this_month = datetime.date.today() - offsets.MonthBegin()
25

26
today = datetime.datetime.today()+datetime.timedelta(days=1)
27
first = today.replace(day=1)
28
last_month = first - datetime.timedelta(days=1)
29

30
first_day_of_current_month = datetime.date.today().replace(day=1)
31
last_day_of_previous_month = first_day_of_current_month - \
32
    datetime.timedelta(days=1)
33

34
previous_year = datetime.datetime(today.year - 1, 1, 1)
35

36
def update_set_ui(currentRow, currentColumn, previousRow, previousColumn):
37
    print(currentRow, currentColumn, previousRow, previousColumn)
38

39

40
def button_clicked(window):
41
    start = datetime.datetime.combine(
42
        window.StartDate.date().toPython(), datetime.datetime.min.time())
43
    end = datetime.datetime.combine(
44
        window.EndDate.date().toPython(), datetime.datetime.min.time())
45
    summ = expenses.loc[(start < expenses["Дата"]) & (
46
        expenses["Дата"] < end), "Сумма дохода"].sum()
47

48
    summ_expens = expenses.loc[(start < expenses["Дата"]) & (
49
        expenses["Дата"] < end), "Сумма расхода"].sum()
50
    summ_res = expenses.loc[(start < expenses["Дата"]) & (
51
        expenses["Дата"] < end), "Оставшаяся сумма"].sum()
52
    dlg = CustomDialog(start, end, window, summ, summ_expens, summ_res)
53
    if dlg.exec():
54
        print("Success!")
55
    else:
56
        print("Cancel!")
57

58

59
def redact_element(window,):
60
    pass
61

62

63
class MyWindow(QTableWidget):
64
    def __init__(self,parent):
65
        super().__init__(parent)
66

67
        # Create the context menu and add some actions
68
        self.context_menu = QMenu(self)
69
        action1 = self.context_menu.addAction("Изменить")
70
        action2 = self.context_menu.addAction("Удалить")
71

72
        # Connect the actions to methods
73
        action1.triggered.connect(self.action1_triggered)
74
        action2.triggered.connect(self.action2_triggered)
75
        self.show()
76

77
    def contextMenuEvent(self, event):
78
        # Show the context menu
79
        self.context_menu.exec(event.globalPos())
80

81
    def action1_triggered(self):
82
        # Handle the "Action 1" action
83
        pass
84

85
    def action2_triggered(self):
86
        global  expenses
87
        print(super().currentRow())
88
        print(parser.parse(super().itemAt(0, super().currentRow()).text()))
89
        expenses = expenses.drop(expenses.isin([parser.parse(super().itemAt(0, super().currentRow()).text())]).any(axis=1).idxmax())
90
        self.render_table()
91
        # Handle the "Action 2" action
92
        pass
93

94
    def render_table(self):
95
        global expenses
96
        expenses_p = expenses.values.tolist()
97

98
        super().setRowCount(len(expenses_p))
99

100
        for y in range(len(expenses_p)):
101
            for x in range(4):
102
                super().setItem(
103
                    y, x, QTableWidgetItem(str(expenses_p[y][x])))
104

105
class CustomDialog(QDialog):
106
    def __init__(self, start, end, parent=None, incomes=0, expenses=0, results=0):
107
        super().__init__(parent)
108

109
        self.setWindowTitle("Отчёт по дате")
110

111
        QBtn = (
112
            QDialogButtonBox.StandardButton.Ok)
113

114
        self.buttonBox = QDialogButtonBox(QBtn)
115
        self.buttonBox.accepted.connect(self.accept)
116
        self.buttonBox.rejected.connect(self.reject)
117

118
        layout = QVBoxLayout()
119
        message = QLabel(f"За промежуток с {start} до {end}:\n доходов на {incomes} расходов на {
120
                         expenses}\nи итоговый бюджет на {results} рублей.")
121
        layout.addWidget(message)
122
        layout.addWidget(self.buttonBox)
123
        self.setLayout(layout)
124

125

126
def submit_info(window, date, plus, minus):
127
    add_new_thing(date, plus, minus)
128
    render_table(window)
129

130

131
def add_new_thing(date, plus, minus):
132
    global expenses
133

134
    expenses.loc[len(expenses)] = [datetime.datetime.combine(
135
        date.toPython(), datetime.datetime.min.time()), plus, minus, 0]
136

137

138
def render_table(window):
139
    expenses_p = expenses.values.tolist()
140

141
    window.budgetTable.setRowCount(len(expenses_p))
142

143
    for y in range(len(expenses_p)):
144
        for x in range(4):
145
            window.budgetTable.setItem(
146
                y, x, QTableWidgetItem(str(expenses_p[y][x])))
147

148

149
def set_time_using_choice(window):
150
    times = {0: (this_month, today),
151
             1: (last_month, this_month),
152
             2: (this_year, today),
153
             3: (previous_year, this_year),
154
             4: (expenses["Дата"].min(), expenses["Дата"].max())}
155

156
    choice = window.ChosedPeriod.currentIndex()
157

158
    window.StartDate.setDate(QDate(times[choice][0]))
159
    window.EndDate.setDate(QDate(times[choice][1]))
160

161

162

163
if __name__ == "__main__":
164
    app = QApplication(sys.argv)
165

166
    ui_file_name = "main.ui"
167
    ui_file = PySide6.QtCore.QFile(ui_file_name)
168
    if not ui_file.open(PySide6.QtCore.QFile.ReadOnly):
169
        print(f"Cannot open {ui_file_name}: {ui_file.errorString()}")
170
        sys.exit(-1)
171
    loader = QUiLoader()
172
    window = loader.load(ui_file)
173
    ui_file.close()
174
    window.setFixedSize(900, 700)
175

176
    set_time_using_choice(window)
177
    window.dateEdit.setDate(QDate(datetime.date.today()))
178
    window.AddValue.clicked.connect(lambda: submit_info(window,
179
                                                        window.dateEdit.date(), window.ExpenseValue.value(), window.IncomeValue.value()))
180
    window.ApplyPeriodButton.clicked.connect(
181
        lambda: set_time_using_choice(window))
182
    window.MakeReport.clicked.connect(lambda: button_clicked(window))
183
    window.DeleteButton.clicked.connect(lambda: window.budgetTable.action2_triggered())
184
    window.budgetTable = MyWindow(window.centralwidget)
185
    if (window.budgetTable.columnCount() < 4):
186
        window.budgetTable.setColumnCount(4)
187
    __qtablewidgetitem = QTableWidgetItem("Дата")
188
    window.budgetTable.setHorizontalHeaderItem(0, __qtablewidgetitem)
189
    __qtablewidgetitem1 = QTableWidgetItem("Доходы")
190
    window.budgetTable.setHorizontalHeaderItem(1, __qtablewidgetitem1)
191
    __qtablewidgetitem2 = QTableWidgetItem("Расходы")
192
    window.budgetTable.setHorizontalHeaderItem(2, __qtablewidgetitem2)
193
    __qtablewidgetitem3 = QTableWidgetItem("Разница")
194
    window.budgetTable.setHorizontalHeaderItem(3, __qtablewidgetitem3)
195
    window.budgetTable.setObjectName(u"budgetTable")
196
    window.budgetTable.setGeometry(QRect(10, 150, 891, 501))
197
    window.budgetTable.setSelectionMode(QAbstractItemView.SelectionMode.SingleSelection)
198
    window.budgetTable.setSortingEnabled(True)
199
    window.budgetTable.setColumnCount(4)
200
    window.budgetTable.horizontalHeader().setCascadingSectionResizes(True)
201
    window.budgetTable.horizontalHeader().setDefaultSectionSize(180)
202
    window.budgetTable.horizontalHeader().setProperty(u"showSortIndicator", True)
203
    window.budgetTable.horizontalHeader().setStretchLastSection(True)
204
    window.budgetTable.verticalHeader().setVisible(False)
205
    window.budgetTable.verticalHeader().setCascadingSectionResizes(False)
206
    window.budgetTable.verticalHeader().setMinimumSectionSize(40)
207
    window.budgetTable.verticalHeader().setHighlightSections(False)
208
    window.budgetTable.verticalHeader().setProperty(u"showSortIndicator", False)
209
    window.budgetTable.verticalHeader().setStretchLastSection(False)
210
    window.budgetTable.setAlternatingRowColors(True)
211

212
    window.budgetTable.setColumnCount(4)
213
    window.budgetTable.setSelectionBehavior(QtWidgets.QTableView.SelectRows)
214
    app.aboutToQuit.connect(lambda:
215
                            expenses.to_excel("FamilyBudget.xlsx", index=False))
216

217
    render_table(window)
218

219
    if not window:
220
        print(loader.errorString())
221
        sys.exit(-1)
222

223
    window.show()
224

225
    sys.exit(app.exec())