from django.contrib import messages from django.contrib.auth import login, authenticate, logout from django.shortcuts import render, redirect, get_object_or_404 from django.http import JsonResponse from django.views.decorators.http import require_POST, require_GET from django.views.decorators.csrf import csrf_exempt from django import forms from django.contrib.auth.decorators import login_required from django.urls import reverse from accounts.models import Profile from accounts.forms import CustomerForm from processes.utils import scope_customers_queryset from common.consts import UserRoles from common.decorators import allowed_roles # Create your views here. def login_view(request): """ renders login page and authenticating user POST requests to log user in """ # If already authenticated, go straight to request list if request.user.is_authenticated: return redirect("processes:request_list") if request.method == "POST": username = request.POST.get("username") password = request.POST.get("password") user = authenticate(request, username=username, password=password) if user is not None: login(request, user) return redirect("processes:request_list") else: messages.error(request, "کاربری با این مشخصات یافت نشد!") return redirect("accounts:login") return render(request, "accounts/login.html") def dashboard(request): return render(request, "accounts/dashboard.html") @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def customer_list(request): # Get all profiles that have customer role base = Profile.objects.filter(roles__slug=UserRoles.CUSTOMER.value, is_deleted=False).select_related('user') customers = scope_customers_queryset(request.user, base) form = CustomerForm() return render(request, "accounts/customer_list.html", { "customers": customers, "form": form }) @require_POST @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def add_customer_ajax(request): """AJAX endpoint for adding customers""" form = CustomerForm(request.POST, request.FILES) form.request = request # Pass request to form if form.is_valid(): try: customer = form.save() return JsonResponse({ 'success': True, 'message': 'مشترک با موفقیت اضافه شد!', 'customer': { 'id': customer.id, 'name': customer.user.get_full_name(), 'username': customer.user.username, 'phone': customer.phone_number_1 or 'ثبت نشده', 'national_code': customer.national_code or 'ثبت نشده', 'status': 'تکمیل شده' if customer.is_completed else 'ناقص' } }) except forms.ValidationError as e: return JsonResponse({ 'success': False, 'message': str(e) }) except Exception as e: return JsonResponse({ 'success': False, 'message': f'خطا در ذخیره مشترک: {str(e)}' }) else: return JsonResponse({ 'success': False, 'message': 'خطا در اعتبارسنجی فرم', 'errors': form.errors }) @require_POST @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def edit_customer_ajax(request, customer_id): customer = get_object_or_404(Profile, id=customer_id) form = CustomerForm(request.POST, request.FILES, instance=customer) form.request = request # Pass request to form if form.is_valid(): try: customer = form.save() return JsonResponse({ 'success': True, 'message': 'مشترک با موفقیت ویرایش شد!', 'customer': { 'id': customer.id, 'name': customer.user.get_full_name(), 'username': customer.user.username, 'phone': customer.phone_number_1 or 'ثبت نشده', 'national_code': customer.national_code or 'ثبت نشده', 'status': 'تکمیل شده' if customer.is_completed else 'ناقص' } }) except forms.ValidationError as e: return JsonResponse({ 'success': False, 'message': str(e) }) except Exception as e: return JsonResponse({ 'success': False, 'message': f'خطا در ویرایش مشترک: {str(e)}' }) else: return JsonResponse({ 'success': False, 'message': 'خطا در اعتبارسنجی فرم', 'errors': form.errors }) @require_GET @login_required def get_customer_data(request, customer_id): customer = get_object_or_404(Profile, id=customer_id) # Create form with existing customer data form = CustomerForm(instance=customer, initial={ 'first_name': customer.user.first_name, 'last_name': customer.user.last_name, }) # Render form fields as HTML form_html = { 'first_name': str(form['first_name']), 'last_name': str(form['last_name']), 'phone_number_1': str(form['phone_number_1']), 'phone_number_2': str(form['phone_number_2']), 'national_code': str(form['national_code']), 'card_number': str(form['card_number']), 'account_number': str(form['account_number']), 'address': str(form['address']), 'bank_name': str(form['bank_name']), } return JsonResponse({ 'success': True, 'customer': { 'id': customer.id, 'first_name': customer.user.first_name, 'last_name': customer.user.last_name, 'phone_number_1': customer.phone_number_1 or '', 'phone_number_2': customer.phone_number_2 or '', 'national_code': customer.national_code or '', 'card_number': customer.card_number or '', 'account_number': customer.account_number or '', 'address': customer.address or '', 'bank_name': customer.bank_name or '', }, 'form_html': form_html }) @require_GET @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def get_customer_details(request, customer_id): """جزئیات کامل مشترک برای نمایش در مدال""" customer = get_object_or_404( Profile.objects.select_related('user', 'affairs', 'county', 'broker'), id=customer_id ) data = { 'id': customer.id, 'user': { 'username': customer.user.username, 'first_name': customer.user.first_name or '', 'last_name': customer.user.last_name or '', 'full_name': customer.user.get_full_name() or customer.user.username, 'email': customer.user.email or '', 'date_joined': customer.jcreated_date() if customer.user.date_joined else '', }, 'national_code': customer.national_code or '', 'phone_number_1': customer.phone_number_1 or '', 'phone_number_2': customer.phone_number_2 or '', 'card_number': customer.card_number or '', 'account_number': customer.account_number or '', 'bank_name': customer.get_bank_name_display() or '', 'address': customer.address or '', 'pic_url': customer.pic.url if customer.pic else '', 'affairs': str(customer.affairs) if customer.affairs else '', 'county': str(customer.county) if customer.county else '', 'broker': str(customer.broker) if customer.broker else '', 'is_completed': customer.is_completed, } # تعداد چاه‌ها و درخواست‌ها برای نمایش سریع try: from wells.models import Well from processes.models import ProcessInstance total_wells = Well.objects.filter(representative=customer.user, is_deleted=False).count() total_requests = ProcessInstance.objects.filter(representative=customer.user, is_deleted=False).count() except Exception: total_wells = 0 total_requests = 0 return JsonResponse({ 'success': True, 'customer': data, 'total_wells': total_wells, 'total_requests': total_requests }) @require_GET @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def get_customer_wells(request, customer_id): """چاه‌های مرتبط با یک مشترک""" customer = get_object_or_404(Profile, id=customer_id) try: from wells.models import Well qs = Well.objects.select_related( 'water_meter_manufacturer', 'affairs', 'county', 'broker' ).filter(representative=customer.user, is_deleted=False).order_by('-created') items = [] for well in qs[:100]: # محدودسازی برای عملکرد items.append({ 'id': well.id, 'water_subscription_number': well.water_subscription_number, 'electricity_subscription_number': well.electricity_subscription_number or '', 'water_meter_serial_number': well.water_meter_serial_number or '', 'water_meter_manufacturer': str(well.water_meter_manufacturer) if well.water_meter_manufacturer else '', 'well_power': well.well_power or '', 'affairs': str(well.affairs) if well.affairs else '', 'county': str(well.county) if well.county else '', 'broker': str(well.broker) if well.broker else '', 'created': well.jcreated_date() if hasattr(well, 'created') and well.created else '', }) except Exception: items = [] return JsonResponse({'success': True, 'wells': items}) @require_GET @login_required @allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT]) def get_customer_requests(request, customer_id): """درخواست‌های مرتبط با یک مشترک""" customer = get_object_or_404(Profile, id=customer_id) try: from processes.models import ProcessInstance qs = ProcessInstance.objects.select_related( 'process', 'current_step', 'requester', 'well' ).filter(representative=customer.user, is_deleted=False).order_by('-created') items = [] for inst in qs[:100]: # محدودسازی برای عملکرد try: url = reverse('processes:instance_summary', args=[inst.id]) if inst.status == 'completed' else reverse('processes:instance_steps', args=[inst.id]) except Exception: url = '' items.append({ 'id': inst.id, 'code': inst.code, 'process': inst.process.name if inst.process else '', 'status': inst.status, 'status_display': inst.get_status_display(), 'current_step': inst.current_step.name if inst.current_step else '', 'requester': inst.requester.get_full_name() if inst.requester else '', 'well_subscription': inst.well.water_subscription_number if inst.well else '', 'created': inst.jcreated_date() if hasattr(inst, 'created') and inst.created else '', 'url': url, }) except Exception: items = [] return JsonResponse({'success': True, 'requests': items}) @login_required def logout_view(request): """Log out current user and redirect to login page.""" logout(request) return redirect("accounts:login")