diff --git a/konova/management/commands/kspMigrater/compensation_migrater.py b/konova/management/commands/kspMigrater/compensation_migrater.py index c9da16a6..69b4a7f3 100644 --- a/konova/management/commands/kspMigrater/compensation_migrater.py +++ b/konova/management/commands/kspMigrater/compensation_migrater.py @@ -4,7 +4,7 @@ from django.db import transaction from codelist.models import KonovaCode from codelist.settings import CODELIST_BIOTOPES_ID, CODELIST_BIOTOPES_EXTRA_CODES_ID, \ CODELIST_COMPENSATION_ACTION_DETAIL_ID, CODELIST_COMPENSATION_ACTION_ID -from compensation.models import Compensation, CompensationState, CompensationAction, UnitChoices +from compensation.models import Compensation, CompensationState, CompensationAction, UnitChoices, CompensationDocument from intervention.models import Responsibility, Handler, Intervention from konova.management.commands.kspMigrater.base_migrater import BaseMigrater from konova.models import Deadline, DeadlineType @@ -63,6 +63,7 @@ class CompensationMigrater(BaseMigrater): compensation = self._migrate_states(compensation, kom) compensation = self._migrate_deadlines(compensation, kom) compensation = self._migrate_actions(compensation, kom) + compensation = self._migrate_documents(compensation, CompensationDocument, kom) try: compensation = self._migrate_interventions_reference(compensation, kom) compensation.save()