Initial commit

core/apps/management/views/common/edit.py

@@ -2,7 +2,8 @@ from django.contrib.auth.decorators import login_required
 from core.apps.accounts.models import User
 
 from core.apps.management.forms import DeviceForm, IncomeForm, ExpenseForm, WarehouseForm, UserCreateForm, \
-    ToyMovementEmployeeForm, ToyMovementForm, ExpenseFormEmployee, ExpenseFormManager, ExpenseFormBusinessman
+    ToyMovementEmployeeForm, ToyMovementForm, ExpenseFormEmployee, ExpenseFormManager, ExpenseFormBusinessman, \
+    DevicePaymentForm
 from django.shortcuts import render, redirect, get_object_or_404
 from core.apps.management.models import Device, Income, Expense, Warehouse, ToyMovement
 from django.db import transaction
@@ -30,11 +31,11 @@ def edit_income(request, pk):
         form = IncomeForm(request.POST, instance=income)
         if form.is_valid():
             form.save()
-            return redirect("income_list")
+            return redirect("common/create/income_create.html")
     else:
         form = IncomeForm(instance=income)
 
-    return render(request, "common/edit/income_edit.html", {
+    return render(request, "common/create/income_create.html", {
         "form": form,
         "title": "Kirimni tahrirlash"
     })
@@ -125,6 +126,22 @@ def edit_user(request, pk):
         "title": "Foydalanuvchini tahrirlash",
     })
 
+
+@login_required
+@role_required(["employee"])
+def mark_device_paid(request, pk):
+    device = get_object_or_404(Device, pk=pk)
+
+    # Security check
+    if device.district.region != request.user.region:
+        return redirect("device_payment_list")
+
+    if request.method == "POST":
+        device.is_paid = True
+        device.save()
+
+    return redirect("device_payment_list")
+
 # @role_required(["businessman"])
 # @login_required
 # def edit_toy_movement(request, pk):