diff --git a/django_code/apps/cloudmanager/admin.py b/django_code/apps/cloudmanager/admin.py index 253caf2..4f0ff4b 100644 --- a/django_code/apps/cloudmanager/admin.py +++ b/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',)