from django import forms from django.contrib import admin from .models import ( Hotel, UserHotel, APIConfiguration, Reservation, FraudLog ) from django.urls import path from django.shortcuts import redirect from django.utils.html import format_html from pms_integration.api_client import APIClient # Custom form for Hotel to filter APIConfiguration class HotelForm(forms.ModelForm): class Meta: model = Hotel fields = "__all__" def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) # Exclude APIs already linked to another hotel used_apis = Hotel.objects.exclude(api__isnull=True).values_list('api', flat=True) self.fields['api'].queryset = APIConfiguration.objects.exclude(id__in=used_apis) @admin.register(Hotel) class HotelAdmin(admin.ModelAdmin): list_display = ['name', 'hotel_id', 'pms', 'timezone', 'description'] list_filter = ['name', 'pms', 'timezone'] def sync_button(self, obj): return format_html( 'Синхронизировать', f"/admin/hotels/sync/{obj.id}/" ) def get_urls(self): urls = super().get_urls() custom_urls = [ path('sync//', self.sync_hotel_data), ] return custom_urls + urls def sync_hotel_data(self, request, hotel_id): try: hotel = Hotel.objects.get(id=hotel_id) client = APIClient(hotel.pms) client.run(hotel) self.message_user(request, f"Данные отеля {hotel.name} успешно синхронизированы.") except Exception as e: self.message_user(request, f"Ошибка: {str(e)}", level="error") return redirect("..") @admin.register(UserHotel) class UserHotelAdmin(admin.ModelAdmin): list_display = ('user', 'hotel') search_fields = ('user__username', 'hotel__name') # list_filter = ('hotel',) # ordering = ('-hotel',) @admin.register(Reservation) class ReservationAdmin(admin.ModelAdmin): list_display = ('reservation_id', 'hotel', 'room_number', 'room_type', 'check_in', 'check_out', 'status', 'price', 'discount') search_fields = ('hotel', 'room_number', 'room_type', 'check_in', 'check_out', 'status', 'price', 'discount') list_filter = ('hotel', 'room_number', 'room_type', 'check_in', 'check_out', 'status', 'price', 'discount') ordering = ('-check_in',)