shafafiyat/wells/views.py

318 lines
13 KiB
Python

from django.shortcuts import render, get_object_or_404
from django.http import JsonResponse
from django.urls import reverse
from django.views.decorators.http import require_http_methods, require_GET, require_POST
from django.core.paginator import Paginator
from django.db.models import Q
from django.contrib import messages
from django import forms
from .models import Well, WaterMeterManufacturer
from .forms import WellForm, WaterMeterManufacturerForm
from django.contrib.auth.decorators import login_required
from common.decorators import allowed_roles
from common.consts import UserRoles
from processes.utils import scope_wells_queryset
from processes.models import ProcessInstance
@login_required
@allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT])
def well_list(request):
"""نمایش لیست چاه‌ها"""
base = Well.objects.select_related(
'representative',
'water_meter_manufacturer',
'affairs',
'county',
'broker'
).filter(is_deleted=False)
wells = scope_wells_queryset(request.user, base)
# فرم برای افزودن چاه جدید
form = WellForm()
context = {
'wells': wells,
'form': form,
}
return render(request, 'wells/well_list.html', context)
@require_POST
@login_required
@allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT])
def add_well_ajax(request):
"""AJAX endpoint for adding wells"""
try:
print(f"POST data: {request.POST}") # Debug log
print(f"FILES data: {request.FILES}") # Debug log
form = WellForm(request.POST, request.FILES)
print(f"Form is valid: {form.is_valid()}") # Debug log
if form.is_valid():
well = form.save(commit=False)
# تنظیم فیلدهای affairs, county, broker از profile کاربر
if hasattr(request.user, 'profile'):
well.affairs = request.user.profile.affairs
well.county = request.user.profile.county
well.broker = request.user.profile.broker
well.save()
return JsonResponse({
'success': True,
'message': 'چاه با موفقیت ایجاد شد',
'well': {
'id': well.id,
'water_subscription_number': well.water_subscription_number,
'representative': str(well.representative) if well.representative else '',
'electricity_subscription_number': well.electricity_subscription_number or '',
'water_meter_manufacturer': str(well.water_meter_manufacturer) if well.water_meter_manufacturer else '',
'affairs': str(well.affairs) if well.affairs else '',
'county': str(well.county) if well.county else '',
'broker': str(well.broker) if well.broker else '',
}
})
else:
print(f"Form errors: {form.errors}") # Debug log
return JsonResponse({
'success': False,
'message': 'خطا در اعتبارسنجی فرم',
'errors': form.errors
})
except forms.ValidationError as e:
return JsonResponse({
'success': False,
'message': str(e)
})
except Exception as e:
print(f"Exception in add_well_ajax: {str(e)}") # Debug log
return JsonResponse({
'success': False,
'message': f'خطا در ذخیره چاه: {str(e)}',
'errors': {}
})
@require_POST
@login_required
@allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT])
def edit_well_ajax(request, well_id):
"""AJAX endpoint for editing wells"""
well = get_object_or_404(Well, id=well_id)
form = WellForm(request.POST, request.FILES, instance=well)
if form.is_valid():
try:
well = form.save(commit=False)
# تنظیم فیلدهای affairs, county, broker از profile کاربر (در ویرایش هم می‌تواند به‌روزرسانی شود)
if hasattr(request.user, 'profile'):
well.affairs = request.user.profile.affairs
well.county = request.user.profile.county
well.broker = request.user.profile.broker
# حذف فایل اگر درخواست شده باشد
if request.POST.get('remove_file') == 'true':
well.representative_letter_file.delete(save=False)
well.save()
return JsonResponse({
'success': True,
'message': 'چاه با موفقیت ویرایش شد',
'well': {
'id': well.id,
'water_subscription_number': well.water_subscription_number,
'representative': str(well.representative) if well.representative else '',
'electricity_subscription_number': well.electricity_subscription_number or '',
'water_meter_manufacturer': str(well.water_meter_manufacturer) if well.water_meter_manufacturer else '',
'affairs': str(well.affairs) if well.affairs else '',
'county': str(well.county) if well.county else '',
'broker': str(well.broker) if well.broker 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 delete_well(request, well_id):
"""حذف چاه"""
well = get_object_or_404(Well, id=well_id)
water_subscription_number = well.water_subscription_number
well.delete()
print(f"Well deleted: {well_id}")
return JsonResponse({
'success': True,
'message': f'چاه {water_subscription_number} با موفقیت حذف شد'
})
@require_GET
@login_required
def get_well_data(request, well_id):
"""دریافت اطلاعات چاه برای ویرایش"""
well = get_object_or_404(Well, id=well_id)
return JsonResponse({
'success': True,
'well': {
'id': well.id,
'representative': well.representative.id if well.representative else '',
'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_old_serial_number': well.water_meter_old_serial_number or '',
'water_meter_manufacturer': well.water_meter_manufacturer.id if well.water_meter_manufacturer else '',
'utm_x': str(well.utm_x) if well.utm_x else '',
'utm_y': str(well.utm_y) if well.utm_y else '',
'utm_zone': well.utm_zone or '',
'utm_hemisphere': well.utm_hemisphere or '',
'well_power': well.well_power or '',
'reference_letter_number': well.reference_letter_number or '',
'reference_letter_date': well.reference_letter_date.strftime('%Y-%m-%d') if well.reference_letter_date else '',
'representative_letter_file_url': well.representative_letter_file.url if well.representative_letter_file else '',
'representative_letter_file_name': well.representative_letter_file.name.split('/')[-1] if well.representative_letter_file else '',
# affairs, county, broker are auto-filled from user profile, so not needed in edit form
}
})
@require_GET
@login_required
@allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT])
def get_well_details(request, well_id):
"""جزئیات کامل چاه برای نمایش در مدال"""
well = get_object_or_404(
Well.objects.select_related(
'representative', 'water_meter_manufacturer', 'affairs', 'county', 'broker'
),
id=well_id
)
lat_long = None
try:
lat_long_val = well.lat_long()
if lat_long_val:
# utm.to_latlon returns (lat, lon)
lat_long = {
'lat': round(float(lat_long_val[0]), 6),
'lon': round(float(lat_long_val[1]), 6),
}
except Exception:
lat_long = None
data = {
'id': well.id,
'water_subscription_number': well.water_subscription_number,
'electricity_subscription_number': well.electricity_subscription_number or '',
'representative': {
'id': well.representative.id if well.representative else None,
'full_name': well.representative.get_full_name() if well.representative else '',
'username': well.representative.username if well.representative else '',
},
'water_meter_serial_number': well.water_meter_serial_number or '',
'water_meter_old_serial_number': well.water_meter_old_serial_number or '',
'water_meter_manufacturer': str(well.water_meter_manufacturer) if well.water_meter_manufacturer else '',
'utm': {
'x': str(well.utm_x) if well.utm_x is not None else '',
'y': str(well.utm_y) if well.utm_y is not None else '',
'zone': well.utm_zone or '',
'hemisphere': well.utm_hemisphere or '',
},
'lat_long': lat_long,
'well_power': well.well_power or '',
'reference_letter_number': well.reference_letter_number or '',
'reference_letter_date': well.reference_letter_date.strftime('%Y-%m-%d') if well.reference_letter_date else '',
'representative_letter_file_url': well.representative_letter_file.url if well.representative_letter_file else '',
'affairs': str(well.affairs) if well.affairs else '',
'county': str(well.county) if well.county else '',
'broker': str(well.broker) if well.broker else '',
}
# تعداد درخواست‌ها برای نمایش سریع
try:
total_requests = ProcessInstance.objects.filter(well_id=well.id, is_deleted=False).count()
except Exception:
total_requests = 0
return JsonResponse({'success': True, 'well': data, 'total_requests': total_requests})
@require_GET
@login_required
@allowed_roles([UserRoles.ADMIN, UserRoles.BROKER, UserRoles.MANAGER, UserRoles.ACCOUNTANT])
def get_well_requests(request, well_id):
"""سوابق درخواست‌های مرتبط با یک چاه"""
# Scoped access: reuse base scoping by filtering on ProcessInstance via broker/affairs of current user if needed
qs = ProcessInstance.objects.select_related(
'process', 'current_step', 'requester', 'representative'
).filter(well_id=well_id, 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(),
'priority': inst.priority,
'priority_display': inst.get_priority_display(),
'current_step': inst.current_step.name if inst.current_step else '',
'requester': inst.requester.get_full_name() if inst.requester else '',
'representative': inst.representative.get_full_name() if inst.representative else '',
'created': inst.jcreated_date() if hasattr(inst, 'created') and inst.created else '',
'url': url,
})
return JsonResponse({'success': True, 'requests': items})
@require_POST
@login_required
def create_water_meter_manufacturer(request):
"""ایجاد شرکت سازنده کنتور آب جدید"""
form = WaterMeterManufacturerForm(request.POST)
if form.is_valid():
manufacturer = form.save()
return JsonResponse({
'success': True,
'message': 'شرکت سازنده با موفقیت ایجاد شد',
'manufacturer': {
'id': manufacturer.id,
'name': manufacturer.name
}
})
else:
return JsonResponse({
'success': False,
'message': 'خطا در اعتبارسنجی فرم',
'errors': form.errors
})
# Create your views here.