TiBillet-Ticket-Nonfree/DjangoFiles/Administration/admin_tenant.py

from django.contrib import admin
from django.contrib.admin import AdminSite
from django.contrib.auth.models import Group
from solo.admin import SingletonModelAdmin

from AuthBillet.models import HumanUser, SuperHumanUser, TermUser
from BaseBillet.models import Configuration, Event, OptionGenerale, Article, TarifBillet, Reservation, LigneArticle
from django.contrib.auth.admin import UserAdmin

from Customers.models import Client
from PaiementStripe.models import Paiement_stripe


class StaffAdminSite(AdminSite):
    site_header = "TiBillet Staff Admin"
    site_title = "TiBillet Staff Admin"
    site_url = '/'

    def has_permission(self, request):
        """
        Removed check for is_staff.
        """
        try:
            if request.tenant in request.user.client_admin.all():
                return request.user.is_staff
            elif request.user.client_source == Client.objects.get(schema_name="public"):
                return request.user.is_superuser
            else:
                return False
        except AttributeError:
            # AttributeError: 'AnonymousUser' object has no attribute 'client_source'
            return False
        except Exception as e:
            raise e


staff_admin_site = StaffAdminSite(name='staff_admin')


# USER
# -------------------------------------/
class UserAdminTibillet(UserAdmin):
    # list_display = ('email', 'client_source', 'achat')
    list_display = ('email', 'is_active')
    list_filter = ('email', 'is_active',)
    fieldsets = (
        (None, {'fields': ('email', 'password')}),
        # ('Permissions', {'fields': ('is_staff', 'is_active')}),
    )

    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('email', 'password1', 'password2', 'is_active')}
         ),
    )

    search_fields = ('email',)
    ordering = ('email',)

    def save_model(self, request, obj, form, change):
        if not obj.client_source:
            obj.client_source = request.tenant
            obj.username = obj.email
            obj.save()
        obj.client_achat.add(request.tenant)


class HumanUserAdmin(UserAdminTibillet):
    pass


staff_admin_site.register(HumanUser, HumanUserAdmin)


class SuperHumanUserAdmin(UserAdminTibillet):
    def save_model(self, request, obj, form, change):

        super(SuperHumanUserAdmin, self).save_model(request, obj, form, change)

        staff_group = Group.objects.get_or_create(name="staff")[0]
        obj.groups.add(staff_group)
        obj.client_achat.add(request.tenant)
        obj.client_admin.add(request.tenant)

        # execution du save de la classe orginale user admin ( heritage de l'hérité )
        super(UserAdminTibillet, self).save_model(request, obj, form, change)


staff_admin_site.register(SuperHumanUser, SuperHumanUserAdmin)


class TermUserAdmin(UserAdminTibillet):
    pass


staff_admin_site.register(TermUser, TermUserAdmin)


########################################################################
class ConfigurationAdmin(SingletonModelAdmin):
    # readonly_fields = []

    fieldsets = (
        (None, {
            'fields': (
                'organisation',
                'short_description',
                'adresse',
                'phone',
                'email',
                'site_web',
                'twitter',
                'facebook',
                'instagram',
                'img',
                'carte_restaurant',
            )
        }),
        ('Adhésions', {
            'fields': (
                'adhesion_obligatoire',
                'cadeau_adhesion',
            ),
        }),
        ('Paiements', {
            'fields': (
                'mollie_api_key',
                'stripe_api_key',
                'stripe_test_api_key',
                'stripe_mode_test',
            ),
        }),
        ('Billetterie', {
            'fields': (
                'activer_billetterie',
                'jauge_max',
                'option_generale_radio',
                'option_generale_checkbox',
            ),
        }),
        ('Cashless', {
            'fields': (
                'server_cashless',
                'key_cashless',
            ),
        }),
    )


staff_admin_site.register(Configuration, ConfigurationAdmin)



class EventAdmin(admin.ModelAdmin):
    list_display = (
        'name',
        'reservations',
        'datetime',
    )
    readonly_fields = (
        'reservations',
    )

staff_admin_site.register(Event, EventAdmin)


class OptionGeneraleAdmin(admin.ModelAdmin):
    list_display = (
        'name',
        'poids',
    )
    list_editable = (
        'poids',
    )

class ReservationAdmin(admin.ModelAdmin):
    list_display = (
        'user_mail',
        'total_billet',
        '_options_',
        'total_prix',
        'status'
    )
    readonly_fields = list_display

staff_admin_site.register(Reservation, ReservationAdmin)

class ArticleAdmin(admin.ModelAdmin):
    list_display = (
        'name',
        'prix',
        'stock',
        'reservation_par_user_max',
        'vat',
        'publish',
    )
    list_editable = (
        'prix',
        'stock',
        'reservation_par_user_max',
        'vat',
        'publish',
    )

staff_admin_site.register(Article, ArticleAdmin)



staff_admin_site.register(LigneArticle, admin.ModelAdmin)



staff_admin_site.register(OptionGenerale, OptionGeneraleAdmin)

staff_admin_site.register(TarifBillet, admin.ModelAdmin)


class PaiementStripeAdmin(admin.ModelAdmin):
    list_display = (
        'uuid_8',
        'user',
        'total',
        'order_date',
        'status',
        'articles',
    )
    ordering = ('-order_date',)
    # readonly_fields = (
    #     'reservations',
    # )


staff_admin_site.register(Paiement_stripe, PaiementStripeAdmin)