|
11 | 11 | from io import BytesIO |
12 | 12 | from reportlab.lib.pagesizes import A4 |
13 | 13 | from reportlab.pdfgen import canvas |
| 14 | +from calendar import month_name |
| 15 | +from sqlalchemy import extract |
14 | 16 |
|
15 | 17 |
|
16 | 18 |
|
@@ -117,118 +119,44 @@ def enroll(): |
117 | 119 |
|
118 | 120 | return render_template('enroll.html', form=form) |
119 | 121 |
|
120 | | -# @app.route('/expenses', methods=['GET', 'POST']) |
121 | | -# def expenses(): |
122 | | -# form = ExpenseForm() |
123 | | -# if form.validate_on_submit(): |
124 | | -# # Save the new expense to the database |
125 | | -# new_expense = Expense( |
126 | | -# item_name=form.item_name.data, |
127 | | -# price=form.price.data, |
128 | | -# date=form.date.data |
129 | | -# ) |
130 | | -# db.session.add(new_expense) |
131 | | -# db.session.commit() |
132 | | -# flash('Expense added successfully!', 'success') |
133 | | -# return redirect(url_for('expenses')) |
134 | | - |
135 | | -# # Retrieve all expenses from the database |
136 | | -# expenses = Expense.query.all() |
137 | | -# total = sum(expense.price for expense in expenses) |
138 | | - |
139 | | -# return render_template('expenses.html', form=form, expenses=expenses, total=total) |
140 | 122 |
|
141 | 123 | @app.route('/expenses', methods=['GET', 'POST']) |
| 124 | +@login_required |
142 | 125 | def expenses(): |
143 | 126 | form = ExpenseForm() |
144 | | - current_year = datetime.today().year |
145 | | - current_month = datetime.today().month |
| 127 | + |
| 128 | + # Get filter parameters |
| 129 | + month = request.args.get('month', datetime.now().month, type=int) |
| 130 | + year = request.args.get('year', datetime.now().year, type=int) |
| 131 | + |
| 132 | + # Query expenses with filters |
| 133 | + expenses = Expense.query.filter( |
| 134 | + extract('year', Expense.date) == year, |
| 135 | + extract('month', Expense.date) == month |
| 136 | + ).order_by(Expense.date.desc()).all() |
| 137 | + |
| 138 | + # Calculate total |
| 139 | + total = sum(expense.price for expense in expenses) |
146 | 140 |
|
147 | | - # Adding a new expense if the form is submitted |
148 | 141 | if form.validate_on_submit(): |
149 | | - new_expense = Expense( |
| 142 | + expense = Expense( |
150 | 143 | item_name=form.item_name.data, |
151 | 144 | price=form.price.data, |
152 | | - date=form.date.data |
| 145 | + date=form.date.data, |
| 146 | + user_id=current_user.id |
153 | 147 | ) |
154 | | - db.session.add(new_expense) |
| 148 | + db.session.add(expense) |
155 | 149 | db.session.commit() |
156 | 150 | flash('Expense added successfully!', 'success') |
157 | 151 | return redirect(url_for('expenses')) |
158 | 152 |
|
159 | | - # Group expenses by month and year |
160 | | - grouped_expenses = db.session.query( |
161 | | - db.extract('year', Expense.date).label('year'), |
162 | | - db.extract('month', Expense.date).label('month'), |
163 | | - db.func.sum(Expense.price).label('total') |
164 | | - ).group_by('year', 'month').order_by('year', 'month').all() |
165 | | - |
166 | | - # Retrieve expenses for a specific month if selected |
167 | | - year = request.args.get('year', current_year, type=int) |
168 | | - month = request.args.get('month', current_month, type=int) |
169 | | - expenses = Expense.query.filter( |
170 | | - db.extract('year', Expense.date) == year, |
171 | | - db.extract('month', Expense.date) == month |
172 | | - ).all() |
173 | | - total = sum(expense.price for expense in expenses) |
174 | | - |
175 | | - return render_template('expenses.html', form=form, expenses=expenses, total=total, |
176 | | - grouped_expenses=grouped_expenses, year=year, month=month, |
177 | | - current_year=current_year, current_month=current_month) |
178 | | -# @app.route('/export_pdf/<int:year>/<int:month>', methods=['GET']) |
179 | | -# def export_pdf(year, month): |
180 | | -# # Get expenses for the specified month and year |
181 | | -# expenses = Expense.query.filter( |
182 | | -# db.extract('year', Expense.date) == year, |
183 | | -# db.extract('month', Expense.date) == month |
184 | | -# ).all() |
185 | | - |
186 | | -# # Calculate total expenses for the month |
187 | | -# total = sum(expense.price for expense in expenses) |
188 | | - |
189 | | -# # Create a BytesIO buffer for the PDF |
190 | | -# buffer = BytesIO() |
191 | | -# pdf = canvas.Canvas(buffer, pagesize=A4) |
192 | | -# width, height = A4 |
193 | | - |
194 | | -# # PDF Title |
195 | | -# pdf.setFont("Helvetica-Bold", 16) |
196 | | -# pdf.drawString(200, height - 50, f"Expense Report - {month}/{year}") |
197 | | - |
198 | | -# # Table headers |
199 | | -# pdf.setFont("Helvetica-Bold", 12) |
200 | | -# pdf.drawString(50, height - 100, "Item Name") |
201 | | -# pdf.drawString(250, height - 100, "Price (Rs)") |
202 | | -# pdf.drawString(400, height - 100, "Date") |
203 | | - |
204 | | -# # Table content |
205 | | -# y_position = height - 130 |
206 | | -# pdf.setFont("Helvetica", 10) |
207 | | -# for expense in expenses: |
208 | | -# pdf.drawString(50, y_position, expense.item_name) |
209 | | -# pdf.drawString(250, y_position, f"Rs {expense.price}") |
210 | | -# pdf.drawString(400, y_position, expense.date.strftime('%Y-%m-%d')) |
211 | | -# y_position -= 20 |
212 | | - |
213 | | -# # Add a new page if y_position is too low |
214 | | -# if y_position < 50: |
215 | | -# pdf.showPage() |
216 | | -# y_position = height - 50 |
217 | | - |
218 | | -# # Total expenses |
219 | | -# pdf.setFont("Helvetica-Bold", 12) |
220 | | -# pdf.drawString(50, y_position - 30, f"Total Expenses for {month}/{year}: Rs {total}") |
221 | | - |
222 | | -# pdf.save() |
223 | | - |
224 | | -# # Move the buffer's position back to the start |
225 | | -# buffer.seek(0) |
226 | | - |
227 | | -# # Send the PDF as a file download |
228 | | -# return send_file(buffer, as_attachment=True, download_name=f"Expense_Report_{month}_{year}.pdf", mimetype='application/pdf') |
229 | | - |
| 153 | + return render_template('expenses.html', |
| 154 | + form=form, |
| 155 | + expenses=expenses, |
| 156 | + total=total, |
| 157 | + current_month=month, |
| 158 | + current_year=year) |
230 | 159 |
|
231 | | -# Constants |
232 | 160 | INCOME = 345000 |
233 | 161 |
|
234 | 162 | @app.route('/export_pdf/<int:year>/<int:month>', methods=['GET']) |
@@ -394,6 +322,9 @@ def view_all_fee_records(): |
394 | 322 | fee_records = FeeRecord.query.join(Student).all() |
395 | 323 | return render_template('view_all_fee_records.html', fee_records=fee_records) |
396 | 324 |
|
| 325 | +@app.template_filter('month_name') |
| 326 | +def month_name_filter(month_number): |
| 327 | + return month_name[month_number] |
397 | 328 |
|
398 | 329 | if __name__ == '__main__': |
399 | 330 | app.run(debug=True,port=5051) |
0 commit comments