From 137cfb8b67ebb0f1c543b8bb8d84ee04092c3c51 Mon Sep 17 00:00:00 2001 From: Shivank Kacker Date: Thu, 6 Jun 2024 15:50:13 +0530 Subject: [PATCH] remove types from migrations --- ...ntconsultation_consent_records_and_more.py | 23 ++++--------------- 1 file changed, 5 insertions(+), 18 deletions(-) diff --git a/care/facility/migrations/0443_remove_patientconsultation_consent_records_and_more.py b/care/facility/migrations/0443_remove_patientconsultation_consent_records_and_more.py index 788bf8f88e..d6f72ca5ef 100644 --- a/care/facility/migrations/0443_remove_patientconsultation_consent_records_and_more.py +++ b/care/facility/migrations/0443_remove_patientconsultation_consent_records_and_more.py @@ -8,24 +8,13 @@ from django.utils import timezone import care.facility.models.mixins.permissions.patient -from care.facility.models.file_upload import FileUpload as FileUploadModel -from care.facility.models.patient_consultation import ( - PatientConsent as PatientConsentModel, -) -from care.facility.models.patient_consultation import ( - PatientConsultation as PatientConsultationModel, -) class Migration(migrations.Migration): def migrate_consents(apps, schema_editor): - PatientConsultation: PatientConsultationModel = apps.get_model( - "facility", "PatientConsultation" - ) - PatientConsent: PatientConsentModel = apps.get_model( - "facility", "PatientConsent" - ) - FileUpload: FileUploadModel = apps.get_model("facility", "FileUpload") + PatientConsultation = apps.get_model("facility", "PatientConsultation") + PatientConsent = apps.get_model("facility", "PatientConsent") + FileUpload = apps.get_model("facility", "FileUpload") consultations = PatientConsultation.objects.filter( consent_records__isnull=False ) @@ -44,7 +33,7 @@ def migrate_consents(apps, schema_editor): files = FileUpload.objects.filter( associating_id=old_id, - file_type=FileUploadModel.FileType.CONSENT_RECORD, + file_type=7, ) kwargs = { @@ -63,9 +52,7 @@ def migrate_consents(apps, schema_editor): files.update(**kwargs) def reverse_migrate(apps, schema_editor): - PatientConsent: PatientConsentModel = apps.get_model( - "facility", "PatientConsent" - ) + PatientConsent = apps.get_model("facility", "PatientConsent") for consent in PatientConsent.objects.all(): consultation = consent.consultation consultation.consent_records.append(