django_code/apps/cloudmanager/admin.py

@@ -4,6 +4,7 @@ from .models import Deployment_Data,Schedule_Day_Data,Schedule_Week_Data,Namespa
 from .utils import import_namespace,import_deployment,generate_default_day_schedule,generate_default_week_schedule,Check_server_supposed_XXX,retrieve_and_apply
 from apps.kubernetes_connection.utils import scale_up,scale_down
 from admin_extra_buttons.api import ExtraButtonsMixin, button
+from import_export.admin import ImportExportModelAdmin
 
 
 DB = 'cloudmanager_db'
@@ -85,7 +86,7 @@ class MultiDBModelAdmin(admin.ModelAdmin):
 
 
 @admin.register(Deployment_Data)
-class DeploymentDataAdmin(ExtraButtonsMixin, MultiDBModelAdmin):
+class DeploymentDataAdmin(ImportExportModelAdmin,ExtraButtonsMixin, MultiDBModelAdmin):
     list_display = ('name', 'namespace', 'schedule_week', 'schedule_day', 'replicas')
     list_filter = (ActiveAtHourFilter,)
     actions = [Check_server_supposed_False,Check_server_supposed_true,scale_up_deploy,scale_down_deploy]
@@ -102,7 +103,7 @@ class DeploymentDataAdmin(ExtraButtonsMixin, MultiDBModelAdmin):
         self.message_user(request, 'Deployments imported')
 
 @admin.register(Schedule_Day_Data)
-class ScheduleDayAdmin(ExtraButtonsMixin,MultiDBModelAdmin):
+class ScheduleDayAdmin(ImportExportModelAdmin,ExtraButtonsMixin,MultiDBModelAdmin):
     list_display = ['name_schedule'] + [f'state_at_{h}' for h in range(24)]
     @button()
     def generate_default_day_schedule_admin(self, request):
@@ -111,7 +112,7 @@ class ScheduleDayAdmin(ExtraButtonsMixin,MultiDBModelAdmin):
         self.message_user(request, 'Created default schedules')
 
 @admin.register(Schedule_Week_Data)
-class ScheduleWeekAdmin(ExtraButtonsMixin,MultiDBModelAdmin):
+class ScheduleWeekAdmin(ImportExportModelAdmin,ExtraButtonsMixin,MultiDBModelAdmin):
     list_display = (
         'name_schedule',
         'state_Monday', 'state_Tuesday', 'state_Wednesday',
@@ -125,7 +126,7 @@ class ScheduleWeekAdmin(ExtraButtonsMixin,MultiDBModelAdmin):
 
     
 @admin.register(Namespace_Data)
-class NamespaceAdmin(ExtraButtonsMixin, MultiDBModelAdmin):
+class NamespaceAdmin(ImportExportModelAdmin,ExtraButtonsMixin, MultiDBModelAdmin):
     list_display = ('namespace',)