from django.contrib import admin
from mainApp.models import *
# Register your models here.
admin.site.register(User)
admin.site.register(ClinicType)
admin.site.register(ExpenseItem)
admin.site.register(Expense)
admin.site.register(Clinic)
#admin.site.register(Patient)
admin.site.register(PatientInfo)
admin.site.register(PatientExtraInfo)


admin.site.register(Nationality)
admin.site.register(Location)
admin.site.register(City)


admin.site.register(MedicalService)
#admin.site.register(PatientVisit)
admin.site.register(MedicalServiceDoctor)
# admin.site.register(Company)
# admin.site.register(Organization)
# admin.site.register(MedicalServiceOrganization)
admin.site.register(PatientExternalPayment)
admin.site.register(Drug)
admin.site.register(MedicalTest)
admin.site.register(TimeTodo)
admin.site.register(Radiology)
admin.site.register(PresDrugItem)
admin.site.register(PresRadiologyItem)
admin.site.register(PresTestItem)
admin.site.register(PtAttachment)
admin.site.register(NextVisit)
admin.site.register(AdvanceReservation)
admin.site.register(ExtraPtAttachment)


#admin.site.register(WeekDay)


### ADD USERS TO GROUP DJANGO ADMIN CODE 
from django import forms
from django.contrib import admin
from django.contrib.auth import get_user_model
from django.contrib.admin.widgets import FilteredSelectMultiple    
from django.contrib.auth.models import Group


User = get_user_model()


# Create ModelForm based on the Group model.
class GroupAdminForm(forms.ModelForm):
    class Meta:
        model = Group
        exclude = []

    # Add the users field.
    users = forms.ModelMultipleChoiceField(
         queryset=User.objects.all(), 
         required=False,
         # Use the pretty 'filter_horizontal widget'.
         widget=FilteredSelectMultiple('users', False)
    )

    def __init__(self, *args, **kwargs):
        # Do the normal form initialisation.
        super(GroupAdminForm, self).__init__(*args, **kwargs)
        # If it is an existing group (saved objects have a pk).
        if self.instance.pk:
            # Populate the users field with the current Group users.
            self.fields['users'].initial = self.instance.user_set.all()

    def save_m2m(self):
        # Add the users to the Group.
        self.instance.user_set.set(self.cleaned_data['users'])

    def save(self, *args, **kwargs):
        # Default save
        instance = super(GroupAdminForm, self).save()
        # Save many-to-many data
        self.save_m2m()
        return instance

admin.site.unregister(Group)

# Create a new Group admin.
class GroupAdmin(admin.ModelAdmin):
    # Use our custom form.
    form = GroupAdminForm
    # Filter permissions horizontal as well.
    filter_horizontal = ['permissions']

# Register the new Group ModelAdmin.
admin.site.register(Group, GroupAdmin)


class PatientVisitAdmin(admin.ModelAdmin):
    list_display = ("id",'patient', 'medicalservice',"visit_day","entry_datetime","entry_order")

admin.site.register(PatientVisit, PatientVisitAdmin)

class PatientAdmin(admin.ModelAdmin):
    list_display = ("id",'fullname', 'balance',)

admin.site.register(Patient, PatientAdmin)


class WeekDayAdmin(admin.ModelAdmin):
    list_display = ('name', 'order',"max_patients","code")

admin.site.register(WeekDay, WeekDayAdmin)