endoreg-db 0.5.3__py3-none-any.whl → 0.6.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of endoreg-db might be problematic. Click here for more details.

Files changed (320) hide show
  1. endoreg_db/admin.py +90 -1
  2. endoreg_db/urls.py +173 -0
  3. endoreg_db/views.py +0 -3
  4. endoreg_db-0.6.0.dist-info/METADATA +151 -0
  5. endoreg_db-0.6.0.dist-info/RECORD +11 -0
  6. {endoreg_db-0.5.3.dist-info → endoreg_db-0.6.0.dist-info}/WHEEL +2 -1
  7. endoreg_db-0.6.0.dist-info/top_level.txt +1 -0
  8. endoreg_db/data/__init__.py +0 -72
  9. endoreg_db/data/active_model/data.yaml +0 -3
  10. endoreg_db/data/agl_service/data.yaml +0 -19
  11. endoreg_db/data/case_template/rule/00_patient_lab_sample_add_default_value.yaml +0 -167
  12. endoreg_db/data/case_template/rule/01_patient-set-age.yaml +0 -8
  13. endoreg_db/data/case_template/rule/01_patient-set-gender.yaml +0 -9
  14. endoreg_db/data/case_template/rule/11_create_patient_lab_sample.yaml +0 -23
  15. endoreg_db/data/case_template/rule/12_create-patient_medication-anticoagulation.yaml +0 -19
  16. endoreg_db/data/case_template/rule/13_create-patient_medication_schedule-anticoagulation.yaml +0 -19
  17. endoreg_db/data/case_template/rule/19_create_patient.yaml +0 -17
  18. endoreg_db/data/case_template/rule_type/base_types.yaml +0 -35
  19. endoreg_db/data/case_template/rule_value/.init +0 -0
  20. endoreg_db/data/case_template/rule_value_type/base_types.yaml +0 -59
  21. endoreg_db/data/case_template/template/base.yaml +0 -8
  22. endoreg_db/data/case_template/template_type/pre_endoscopy.yaml +0 -3
  23. endoreg_db/data/case_template/tmp/_rule_value +0 -13
  24. endoreg_db/data/case_template/tmp/rule/01_atrial_fibrillation.yaml +0 -21
  25. endoreg_db/data/case_template/tmp/rule/02_create_object.yaml +0 -10
  26. endoreg_db/data/case_template/tmp/template/atrial_fibrillation_low_risk.yaml +0 -7
  27. endoreg_db/data/center/data.yaml +0 -60
  28. endoreg_db/data/center_resource/green_endoscopy_dashboard_CenterResource.yaml +0 -144
  29. endoreg_db/data/center_waste/green_endoscopy_dashboard_CenterWaste.yaml +0 -48
  30. endoreg_db/data/disease/cardiovascular.yaml +0 -37
  31. endoreg_db/data/disease/hepatology.yaml +0 -5
  32. endoreg_db/data/disease/misc.yaml +0 -6
  33. endoreg_db/data/disease/renal.yaml +0 -5
  34. endoreg_db/data/disease_classification/chronic_kidney_disease.yaml +0 -6
  35. endoreg_db/data/disease_classification/coronary_vessel_disease.yaml +0 -6
  36. endoreg_db/data/disease_classification_choice/chronic_kidney_disease.yaml +0 -41
  37. endoreg_db/data/disease_classification_choice/coronary_vessel_disease.yaml +0 -20
  38. endoreg_db/data/distribution/date/patient.yaml +0 -7
  39. endoreg_db/data/distribution/multiple_categorical/.init +0 -0
  40. endoreg_db/data/distribution/numeric/.init +0 -0
  41. endoreg_db/data/distribution/single_categorical/patient.yaml +0 -7
  42. endoreg_db/data/emission_factor/green_endoscopy_dashboard_EmissionFactor.yaml +0 -132
  43. endoreg_db/data/endoscope_type/data.yaml +0 -11
  44. endoreg_db/data/endoscopy_processor/data.yaml +0 -47
  45. endoreg_db/data/event/cardiology.yaml +0 -28
  46. endoreg_db/data/event/neurology.yaml +0 -14
  47. endoreg_db/data/event/surgery.yaml +0 -13
  48. endoreg_db/data/event/thrombembolism.yaml +0 -20
  49. endoreg_db/data/examination/examinations/data.yaml +0 -66
  50. endoreg_db/data/examination/time/data.yaml +0 -48
  51. endoreg_db/data/examination/time-type/data.yaml +0 -8
  52. endoreg_db/data/examination/type/data.yaml +0 -5
  53. endoreg_db/data/gender/data.yaml +0 -18
  54. endoreg_db/data/information_source/data.yaml +0 -30
  55. endoreg_db/data/information_source/medication.yaml +0 -6
  56. endoreg_db/data/lab_value/cardiac_enzymes.yaml +0 -31
  57. endoreg_db/data/lab_value/coagulation.yaml +0 -49
  58. endoreg_db/data/lab_value/electrolytes.yaml +0 -190
  59. endoreg_db/data/lab_value/gastrointestinal_function.yaml +0 -121
  60. endoreg_db/data/lab_value/hematology.yaml +0 -169
  61. endoreg_db/data/lab_value/hormones.yaml +0 -53
  62. endoreg_db/data/lab_value/lipids.yaml +0 -44
  63. endoreg_db/data/lab_value/misc.yaml +0 -30
  64. endoreg_db/data/lab_value/renal_function.yaml +0 -11
  65. endoreg_db/data/label/label/data.yaml +0 -62
  66. endoreg_db/data/label/label-set/data.yaml +0 -18
  67. endoreg_db/data/label/label-type/data.yaml +0 -7
  68. endoreg_db/data/log_type/data.yaml +0 -57
  69. endoreg_db/data/material/material.yaml +0 -91
  70. endoreg_db/data/medication/anticoagulation.yaml +0 -65
  71. endoreg_db/data/medication/tah.yaml +0 -70
  72. endoreg_db/data/medication_indication/anticoagulation.yaml +0 -120
  73. endoreg_db/data/medication_indication_type/data.yaml +0 -11
  74. endoreg_db/data/medication_indication_type/thrombembolism.yaml +0 -41
  75. endoreg_db/data/medication_intake_time/base.yaml +0 -31
  76. endoreg_db/data/medication_schedule/apixaban.yaml +0 -95
  77. endoreg_db/data/medication_schedule/ass.yaml +0 -12
  78. endoreg_db/data/medication_schedule/enoxaparin.yaml +0 -26
  79. endoreg_db/data/model_type/data.yaml +0 -7
  80. endoreg_db/data/network_device/data.yaml +0 -29
  81. endoreg_db/data/network_device_type/data.yaml +0 -12
  82. endoreg_db/data/patient_lab_sample_type/generic.yaml +0 -6
  83. endoreg_db/data/pdf_type/data.yaml +0 -28
  84. endoreg_db/data/product/green_endoscopy_dashboard_Product.yaml +0 -66
  85. endoreg_db/data/product_group/green_endoscopy_dashboard_ProductGroup.yaml +0 -33
  86. endoreg_db/data/product_material/green_endoscopy_dashboard_ProductMaterial.yaml +0 -308
  87. endoreg_db/data/product_weight/green_endoscopy_dashboard_ProductWeight.yaml +0 -88
  88. endoreg_db/data/profession/data.yaml +0 -70
  89. endoreg_db/data/reference_product/green_endoscopy_dashboard_ReferenceProduct.yaml +0 -55
  90. endoreg_db/data/report_reader_flag/ukw-examination-generic.yaml +0 -26
  91. endoreg_db/data/report_reader_flag/ukw-histology-generic.yaml +0 -19
  92. endoreg_db/data/resource/green_endoscopy_dashboard_Resource.yaml +0 -15
  93. endoreg_db/data/tmp/chronic_kidney_disease.yaml +0 -0
  94. endoreg_db/data/tmp/congestive_heart_failure.yaml +0 -0
  95. endoreg_db/data/transport_route/green_endoscopy_dashboard_TransportRoute.yaml +0 -12
  96. endoreg_db/data/unit/concentration.yaml +0 -92
  97. endoreg_db/data/unit/data.yaml +0 -17
  98. endoreg_db/data/unit/length.yaml +0 -31
  99. endoreg_db/data/unit/misc.yaml +0 -20
  100. endoreg_db/data/unit/rate.yaml +0 -6
  101. endoreg_db/data/unit/time.yaml +0 -13
  102. endoreg_db/data/unit/volume.yaml +0 -35
  103. endoreg_db/data/unit/weight.yaml +0 -38
  104. endoreg_db/data/waste/data.yaml +0 -12
  105. endoreg_db/forms/__init__.py +0 -3
  106. endoreg_db/forms/questionnaires/__init__.py +0 -1
  107. endoreg_db/forms/questionnaires/tto_questionnaire.py +0 -23
  108. endoreg_db/forms/settings/__init__.py +0 -8
  109. endoreg_db/forms/unit.py +0 -6
  110. endoreg_db/management/commands/_load_model_template.py +0 -41
  111. endoreg_db/management/commands/delete_all.py +0 -18
  112. endoreg_db/management/commands/delete_legacy_images.py +0 -19
  113. endoreg_db/management/commands/delete_legacy_videos.py +0 -17
  114. endoreg_db/management/commands/extract_legacy_video_frames.py +0 -18
  115. endoreg_db/management/commands/fetch_legacy_image_dataset.py +0 -32
  116. endoreg_db/management/commands/fix_auth_permission.py +0 -20
  117. endoreg_db/management/commands/import_legacy_images.py +0 -94
  118. endoreg_db/management/commands/import_legacy_videos.py +0 -76
  119. endoreg_db/management/commands/load_active_model_data.py +0 -45
  120. endoreg_db/management/commands/load_ai_model_data.py +0 -45
  121. endoreg_db/management/commands/load_base_db_data.py +0 -136
  122. endoreg_db/management/commands/load_center_data.py +0 -43
  123. endoreg_db/management/commands/load_disease_classification_choices_data.py +0 -41
  124. endoreg_db/management/commands/load_disease_classification_data.py +0 -41
  125. endoreg_db/management/commands/load_disease_data.py +0 -40
  126. endoreg_db/management/commands/load_distribution_data.py +0 -66
  127. endoreg_db/management/commands/load_endoscope_type_data.py +0 -45
  128. endoreg_db/management/commands/load_endoscopy_processor_data.py +0 -45
  129. endoreg_db/management/commands/load_event_data.py +0 -41
  130. endoreg_db/management/commands/load_examination_data.py +0 -75
  131. endoreg_db/management/commands/load_g_play_data.py +0 -113
  132. endoreg_db/management/commands/load_gender_data.py +0 -44
  133. endoreg_db/management/commands/load_green_endoscopy_wuerzburg_data.py +0 -133
  134. endoreg_db/management/commands/load_information_source.py +0 -45
  135. endoreg_db/management/commands/load_lab_value_data.py +0 -50
  136. endoreg_db/management/commands/load_label_data.py +0 -67
  137. endoreg_db/management/commands/load_logging_data.py +0 -39
  138. endoreg_db/management/commands/load_medication_data.py +0 -41
  139. endoreg_db/management/commands/load_medication_indication_data.py +0 -63
  140. endoreg_db/management/commands/load_medication_indication_type_data.py +0 -41
  141. endoreg_db/management/commands/load_medication_intake_time_data.py +0 -41
  142. endoreg_db/management/commands/load_medication_schedule_data.py +0 -55
  143. endoreg_db/management/commands/load_network_data.py +0 -57
  144. endoreg_db/management/commands/load_pdf_type_data.py +0 -61
  145. endoreg_db/management/commands/load_profession_data.py +0 -44
  146. endoreg_db/management/commands/load_report_reader_flag.py +0 -46
  147. endoreg_db/management/commands/load_unit_data.py +0 -46
  148. endoreg_db/management/commands/load_user_groups.py +0 -28
  149. endoreg_db/management/commands/register_ai_model.py +0 -64
  150. endoreg_db/management/commands/reset_celery_schedule.py +0 -9
  151. endoreg_db/migrations/0001_initial.py +0 -1567
  152. endoreg_db/migrations/0002_anonymizedimagelabel_anonymousimageannotation_and_more.py +0 -55
  153. endoreg_db/migrations/0003_anonymousimageannotation_original_image_url_and_more.py +0 -39
  154. endoreg_db/migrations/0004_alter_rawpdffile_file.py +0 -20
  155. endoreg_db/migrations/0005_uploadedfile_alter_rawpdffile_file_anonymizedfile.py +0 -40
  156. endoreg_db/migrations/0006_alter_rawpdffile_file.py +0 -20
  157. endoreg_db/migrations/0007_networkdevicelogentry_datetime_and_more.py +0 -43
  158. endoreg_db/migrations/0008_networkdevicelogentry_aglnet_ip_and_more.py +0 -28
  159. endoreg_db/migrations/0009_alter_networkdevicelogentry_vpn_service_status.py +0 -18
  160. endoreg_db/migrations/0010_remove_networkdevicelogentry_hostname.py +0 -17
  161. endoreg_db/migrations/__init__.py +0 -0
  162. endoreg_db/models/__init__.py +0 -73
  163. endoreg_db/models/ai_model/__init__.py +0 -3
  164. endoreg_db/models/ai_model/active_model.py +0 -9
  165. endoreg_db/models/ai_model/model_meta.py +0 -24
  166. endoreg_db/models/ai_model/model_type.py +0 -26
  167. endoreg_db/models/ai_model/utils.py +0 -8
  168. endoreg_db/models/annotation/__init__.py +0 -3
  169. endoreg_db/models/annotation/anonymized_image_annotation.py +0 -60
  170. endoreg_db/models/annotation/binary_classification_annotation_task.py +0 -80
  171. endoreg_db/models/annotation/image_classification.py +0 -27
  172. endoreg_db/models/case_template/__init__.py +0 -6
  173. endoreg_db/models/case_template/case_template.py +0 -81
  174. endoreg_db/models/case_template/case_template_rule.py +0 -276
  175. endoreg_db/models/case_template/case_template_rule_value.py +0 -73
  176. endoreg_db/models/case_template/case_template_type.py +0 -28
  177. endoreg_db/models/center/__init__.py +0 -4
  178. endoreg_db/models/center/center.py +0 -25
  179. endoreg_db/models/center/center_product.py +0 -34
  180. endoreg_db/models/center/center_resource.py +0 -19
  181. endoreg_db/models/center/center_waste.py +0 -11
  182. endoreg_db/models/data_file/__init__.py +0 -6
  183. endoreg_db/models/data_file/base_classes/__init__.py +0 -2
  184. endoreg_db/models/data_file/base_classes/abstract_frame.py +0 -51
  185. endoreg_db/models/data_file/base_classes/abstract_video.py +0 -201
  186. endoreg_db/models/data_file/frame.py +0 -45
  187. endoreg_db/models/data_file/import_classes/__init__.py +0 -32
  188. endoreg_db/models/data_file/import_classes/processing_functions/__init__.py +0 -35
  189. endoreg_db/models/data_file/import_classes/processing_functions/pdf.py +0 -28
  190. endoreg_db/models/data_file/import_classes/processing_functions/video.py +0 -260
  191. endoreg_db/models/data_file/import_classes/raw_pdf.py +0 -188
  192. endoreg_db/models/data_file/import_classes/raw_video.py +0 -343
  193. endoreg_db/models/data_file/metadata/__init__.py +0 -3
  194. endoreg_db/models/data_file/metadata/pdf_meta.py +0 -70
  195. endoreg_db/models/data_file/metadata/sensitive_meta.py +0 -31
  196. endoreg_db/models/data_file/metadata/video_meta.py +0 -133
  197. endoreg_db/models/data_file/report_file.py +0 -89
  198. endoreg_db/models/data_file/video/__init__.py +0 -7
  199. endoreg_db/models/data_file/video/import_meta.py +0 -25
  200. endoreg_db/models/data_file/video/video.py +0 -25
  201. endoreg_db/models/data_file/video_segment.py +0 -107
  202. endoreg_db/models/disease.py +0 -56
  203. endoreg_db/models/emission/__init__.py +0 -1
  204. endoreg_db/models/emission/emission_factor.py +0 -20
  205. endoreg_db/models/event.py +0 -22
  206. endoreg_db/models/examination/__init__.py +0 -4
  207. endoreg_db/models/examination/examination.py +0 -26
  208. endoreg_db/models/examination/examination_time.py +0 -27
  209. endoreg_db/models/examination/examination_time_type.py +0 -24
  210. endoreg_db/models/examination/examination_type.py +0 -18
  211. endoreg_db/models/hardware/__init__.py +0 -2
  212. endoreg_db/models/hardware/endoscope.py +0 -44
  213. endoreg_db/models/hardware/endoscopy_processor.py +0 -143
  214. endoreg_db/models/information_source.py +0 -29
  215. endoreg_db/models/label/__init__.py +0 -1
  216. endoreg_db/models/label/label.py +0 -84
  217. endoreg_db/models/laboratory/__init__.py +0 -1
  218. endoreg_db/models/laboratory/lab_value.py +0 -102
  219. endoreg_db/models/legacy_data/__init__.py +0 -3
  220. endoreg_db/models/legacy_data/image.py +0 -34
  221. endoreg_db/models/logging/__init__.py +0 -4
  222. endoreg_db/models/logging/agl_service.py +0 -19
  223. endoreg_db/models/logging/base.py +0 -22
  224. endoreg_db/models/logging/log_type.py +0 -23
  225. endoreg_db/models/logging/network_device.py +0 -27
  226. endoreg_db/models/medication/__init__.py +0 -1
  227. endoreg_db/models/medication/medication.py +0 -148
  228. endoreg_db/models/network/__init__.py +0 -3
  229. endoreg_db/models/network/agl_service.py +0 -38
  230. endoreg_db/models/network/network_device.py +0 -53
  231. endoreg_db/models/network/network_device_type.py +0 -23
  232. endoreg_db/models/other/__init__.py +0 -5
  233. endoreg_db/models/other/distribution.py +0 -215
  234. endoreg_db/models/other/material.py +0 -16
  235. endoreg_db/models/other/resource.py +0 -18
  236. endoreg_db/models/other/transport_route.py +0 -21
  237. endoreg_db/models/other/waste.py +0 -20
  238. endoreg_db/models/patient_examination/__init__.py +0 -35
  239. endoreg_db/models/permissions/__init__.py +0 -44
  240. endoreg_db/models/persons/__init__.py +0 -7
  241. endoreg_db/models/persons/examiner/__init__.py +0 -2
  242. endoreg_db/models/persons/examiner/examiner.py +0 -16
  243. endoreg_db/models/persons/examiner/examiner_type.py +0 -2
  244. endoreg_db/models/persons/first_name.py +0 -18
  245. endoreg_db/models/persons/gender.py +0 -22
  246. endoreg_db/models/persons/last_name.py +0 -20
  247. endoreg_db/models/persons/patient/__init__.py +0 -8
  248. endoreg_db/models/persons/patient/case/__init__.py +0 -0
  249. endoreg_db/models/persons/patient/case/case.py +0 -30
  250. endoreg_db/models/persons/patient/patient.py +0 -216
  251. endoreg_db/models/persons/patient/patient_disease.py +0 -16
  252. endoreg_db/models/persons/patient/patient_event.py +0 -22
  253. endoreg_db/models/persons/patient/patient_lab_sample.py +0 -106
  254. endoreg_db/models/persons/patient/patient_lab_value.py +0 -176
  255. endoreg_db/models/persons/patient/patient_medication.py +0 -44
  256. endoreg_db/models/persons/patient/patient_medication_schedule.py +0 -28
  257. endoreg_db/models/persons/person.py +0 -31
  258. endoreg_db/models/persons/portal_user_information.py +0 -27
  259. endoreg_db/models/prediction/__init__.py +0 -2
  260. endoreg_db/models/prediction/image_classification.py +0 -37
  261. endoreg_db/models/prediction/video_prediction_meta.py +0 -244
  262. endoreg_db/models/product/__init__.py +0 -5
  263. endoreg_db/models/product/product.py +0 -97
  264. endoreg_db/models/product/product_group.py +0 -19
  265. endoreg_db/models/product/product_material.py +0 -24
  266. endoreg_db/models/product/product_weight.py +0 -26
  267. endoreg_db/models/product/reference_product.py +0 -99
  268. endoreg_db/models/questionnaires/__init__.py +0 -114
  269. endoreg_db/models/quiz/__init__.py +0 -2
  270. endoreg_db/models/quiz/quiz_answer.py +0 -41
  271. endoreg_db/models/quiz/quiz_question.py +0 -54
  272. endoreg_db/models/report_reader/__init__.py +0 -2
  273. endoreg_db/models/report_reader/report_reader_config.py +0 -53
  274. endoreg_db/models/report_reader/report_reader_flag.py +0 -20
  275. endoreg_db/models/rules/__init__.py +0 -5
  276. endoreg_db/models/rules/rule.py +0 -24
  277. endoreg_db/models/rules/rule_applicator.py +0 -224
  278. endoreg_db/models/rules/rule_attribute_dtype.py +0 -19
  279. endoreg_db/models/rules/rule_type.py +0 -22
  280. endoreg_db/models/rules/ruleset.py +0 -19
  281. endoreg_db/models/unit.py +0 -22
  282. endoreg_db/queries/__init__.py +0 -5
  283. endoreg_db/queries/annotations/__init__.py +0 -3
  284. endoreg_db/queries/annotations/legacy.py +0 -158
  285. endoreg_db/queries/get/__init__.py +0 -6
  286. endoreg_db/queries/get/annotation.py +0 -0
  287. endoreg_db/queries/get/center.py +0 -42
  288. endoreg_db/queries/get/model.py +0 -13
  289. endoreg_db/queries/get/patient.py +0 -14
  290. endoreg_db/queries/get/patient_examination.py +0 -20
  291. endoreg_db/queries/get/prediction.py +0 -0
  292. endoreg_db/queries/get/report_file.py +0 -33
  293. endoreg_db/queries/get/video.py +0 -31
  294. endoreg_db/queries/get/video_import_meta.py +0 -0
  295. endoreg_db/queries/get/video_prediction_meta.py +0 -0
  296. endoreg_db/queries/sanity/__init_.py +0 -0
  297. endoreg_db/serializers/__init__.py +0 -10
  298. endoreg_db/serializers/ai_model.py +0 -19
  299. endoreg_db/serializers/annotation.py +0 -17
  300. endoreg_db/serializers/center.py +0 -11
  301. endoreg_db/serializers/examination.py +0 -33
  302. endoreg_db/serializers/frame.py +0 -13
  303. endoreg_db/serializers/hardware.py +0 -21
  304. endoreg_db/serializers/label.py +0 -22
  305. endoreg_db/serializers/patient.py +0 -10
  306. endoreg_db/serializers/prediction.py +0 -15
  307. endoreg_db/serializers/report_file.py +0 -7
  308. endoreg_db/serializers/video.py +0 -27
  309. endoreg_db/utils/__init__.py +0 -1
  310. endoreg_db/utils/cropping.py +0 -29
  311. endoreg_db/utils/dataloader.py +0 -92
  312. endoreg_db/utils/file_operations.py +0 -30
  313. endoreg_db/utils/hashs.py +0 -34
  314. endoreg_db/utils/legacy_ocr.py +0 -201
  315. endoreg_db/utils/ocr.py +0 -190
  316. endoreg_db/utils/uuid.py +0 -4
  317. endoreg_db/utils/video_metadata.py +0 -87
  318. endoreg_db-0.5.3.dist-info/METADATA +0 -28
  319. endoreg_db-0.5.3.dist-info/RECORD +0 -319
  320. {endoreg_db-0.5.3.dist-info/licenses → endoreg_db-0.6.0.dist-info}/LICENSE +0 -0
@@ -1,94 +0,0 @@
1
- # command expects a directory as an argument
2
- # directory should contain a img_dicts.jsonl file, a settings.json and an "images" directory
3
- # images should be in jpeg format
4
- # reads the img_dicts.jsonl file and creates a new database entry for each image
5
- # img_dicts contains a line for each image in the following format:
6
- # {"id": 30604292, "path": "images/30604292.jpg", "labels": ["appendix", "polyp"]}
7
-
8
- # import_settings.json example:
9
- # {
10
- # "annotator": "legacy",
11
- # "labelset_name": "multilabel_classification",
12
- # "labelset_version": 0
13
- # }
14
-
15
- import os
16
- import json
17
- from tqdm import tqdm
18
- from django.core.management.base import BaseCommand
19
- from endoreg_db.models import LegacyImage, LabelSet, Label, ImageClassificationAnnotation
20
- from django.core.files import File
21
- from uuid import uuid4
22
-
23
- class Command(BaseCommand):
24
- """
25
- Imports images from a directory into the database
26
-
27
- Usage:
28
- python manage.py import_legacy_images <directory>
29
- """
30
- help = 'Imports images from a directory into the database'
31
-
32
- def add_arguments(self, parser):
33
- parser.add_argument('directory', type=str)
34
-
35
- def handle(self, *args, **options):
36
- directory = options['directory']
37
- if not os.path.isdir(directory):
38
- raise Exception(f"Directory {directory} does not exist")
39
-
40
- # read settings.json
41
- with open(os.path.join(directory, 'import_settings.json'), 'r') as f:
42
- settings = json.load(f)
43
-
44
- # read img_dicts.jsonl
45
- with open(os.path.join(directory, 'img_dicts.jsonl'), 'r') as f:
46
- img_dicts = [json.loads(line) for line in f.readlines()]
47
-
48
- # get labelset
49
- labelset_name = settings['labelset_name']
50
- labelset_version = settings['labelset_version']
51
- try:
52
- labelset = LabelSet.objects.get(name=labelset_name, version=labelset_version)
53
- except LabelSet.DoesNotExist:
54
- raise Exception(f"No labelset found with the name {labelset_name} and version {labelset_version}")
55
-
56
-
57
- # get labels in dict to lookup by name
58
- labels = {label.name: label for label in labelset.labels.all()}
59
-
60
- # create images and image_classification_annotations
61
- for img_dict in tqdm(img_dicts):
62
-
63
- # Open the image file
64
- with open(os.path.join(directory, img_dict['path']), 'rb') as f:
65
- # Create the Django File object
66
- django_file = File(f)
67
-
68
- # Extract only the filename
69
- filename = os.path.basename(img_dict['path'])
70
- img_suffix = os.path.splitext(filename)[1]
71
- # assign uuid as new filename
72
- filename = str(uuid4()) + img_suffix
73
-
74
-
75
- # Create a new LegacyImage instance and save the image
76
- image = LegacyImage(image=filename, suffix=img_suffix)
77
- image.image.save(filename, django_file, save=True)
78
-
79
- image_annotations = []
80
- for label_name, label in labels.items():
81
- if label_name in img_dict['labels']:
82
- value = True
83
- else:
84
- value = False
85
-
86
- image_annotations.append(ImageClassificationAnnotation(
87
- legacy_image=image,
88
- label=label,
89
- value=value,
90
- annotator=settings['annotator']
91
- ))
92
-
93
- ImageClassificationAnnotation.objects.bulk_create(image_annotations)
94
-
@@ -1,76 +0,0 @@
1
- from endoreg_db.utils.hashs import get_video_hash
2
- import subprocess
3
- from django.core.files import File
4
- from django.core.management.base import BaseCommand
5
- from endoreg_db.models.data_file.video import LegacyVideo
6
- import os
7
- from tqdm import tqdm
8
- # import cv2
9
-
10
- def convert_mkv_to_mp4(source_path, target_path):
11
- """
12
- Convert a .mkv file to a .mp4 file using FFmpeg.
13
-
14
- Parameters:
15
- mov_file_path (str): The file path of the input .MOV file.
16
- mp4_file_path (str): The file path where the output .mp4 file should be saved.
17
-
18
- Returns:
19
- None
20
- """
21
- cmd = ["ffmpeg", "-y", "-i", source_path, "-vcodec", "h264", "-acodec", "aac", target_path]
22
- subprocess.run(cmd)
23
-
24
-
25
- class Command(BaseCommand):
26
- """
27
- Imports videos from a directory into the database
28
-
29
- Usage:
30
- python manage.py import_legacy_videos <directory>
31
- """
32
- help = 'Imports videos from a directory into the database'
33
-
34
- def add_arguments(self, parser):
35
- parser.add_argument('directory', type=str)
36
-
37
- def handle(self, *args, **options):
38
- directory = options['directory']
39
- if not os.path.isdir(directory):
40
- raise Exception(f"Directory {directory} does not exist")
41
-
42
- # iterate over all subdirectories and gather all .mkv files
43
- mkv_files = []
44
- for root, dirs, files in os.walk(directory):
45
- for file in files:
46
- if file.endswith(".mkv"):
47
- mkv_files.append(os.path.join(root, file))
48
-
49
- # create videos
50
- for mkv_file in tqdm(mkv_files):
51
- # get the name of the video
52
- video_name = os.path.basename(mkv_file).replace(".mkv", "")
53
-
54
- # convert the .mkv file to a .mp4 file
55
- mp4_file = os.path.join(directory, video_name + ".mp4")
56
- if not os.path.isfile(mp4_file):
57
- convert_mkv_to_mp4(mkv_file, mp4_file)
58
-
59
- # create the video object
60
- video_hash = get_video_hash(mp4_file)
61
-
62
- # check if the video already exists
63
- if LegacyVideo.objects.filter(video_hash=video_hash).exists():
64
- print(f"Video with hash {video_hash} already exists. Skipping...")
65
- continue
66
-
67
- video = LegacyVideo.objects.create(video_hash=video_hash, suffix = "mp4")
68
-
69
- # open the .mp4 file
70
- with open(mp4_file, 'rb') as f:
71
- # create the Django File object
72
- django_file = File(f)
73
- # save the file to the video
74
- video.file.save(video_name + ".mp4", django_file)
75
- raise Exception("Stop here: NEED TO FIX OPENCV DEPENDENCY")
76
- video.initialize_video_specs(cv2.VideoCapture(video.file.path))
@@ -1,45 +0,0 @@
1
- from django.conf import settings
2
- import os
3
- from django.core.management.base import BaseCommand
4
- from ...models import ActiveModel
5
- from ...utils import load_model_data_from_yaml
6
- from ...data import ACTIVE_MODEL_DATA_DIR
7
-
8
- SOURCE_DIR = ACTIVE_MODEL_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
9
-
10
- MODEL_0 = ActiveModel
11
-
12
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
13
- MODEL_0.__name__,
14
- ]
15
-
16
- IMPORT_METADATA = {
17
- MODEL_0.__name__: {
18
- "dir": SOURCE_DIR, # e.g. "interventions"
19
- "model": MODEL_0, # e.g. Intervention
20
- "foreign_keys": [], # e.g. ["intervention_types"]
21
- "foreign_key_models": [] # e.g. [InterventionType]
22
- }
23
- }
24
-
25
- class Command(BaseCommand):
26
- help = """Load all .yaml files in the data/intervention directory
27
- into the Intervention and InterventionType model"""
28
-
29
- def add_arguments(self, parser):
30
- parser.add_argument(
31
- '--verbose',
32
- action='store_true',
33
- help='Display verbose output',
34
- )
35
-
36
- def handle(self, *args, **options):
37
- verbose = options['verbose']
38
- for model_name in IMPORT_MODELS:
39
- _metadata = IMPORT_METADATA[model_name]
40
- load_model_data_from_yaml(
41
- self,
42
- model_name,
43
- _metadata,
44
- verbose
45
- )
@@ -1,45 +0,0 @@
1
- from django.conf import settings
2
- import os
3
- from django.core.management.base import BaseCommand
4
- from ...models import ModelType
5
- from ...utils import load_model_data_from_yaml
6
- from ...data import MODEL_TYPE_DATA_DIR
7
-
8
- SOURCE_DIR = MODEL_TYPE_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
9
-
10
- MODEL_0 = ModelType
11
-
12
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
13
- MODEL_0.__name__,
14
- ]
15
-
16
- IMPORT_METADATA = {
17
- MODEL_0.__name__: {
18
- "dir": SOURCE_DIR, # e.g. "interventions"
19
- "model": MODEL_0, # e.g. Intervention
20
- "foreign_keys": [], # e.g. ["intervention_types"]
21
- "foreign_key_models": [] # e.g. [InterventionType]
22
- }
23
- }
24
-
25
- class Command(BaseCommand):
26
- help = """Load all .yaml files in the data/intervention directory
27
- into the Intervention and InterventionType model"""
28
-
29
- def add_arguments(self, parser):
30
- parser.add_argument(
31
- '--verbose',
32
- action='store_true',
33
- help='Display verbose output',
34
- )
35
-
36
- def handle(self, *args, **options):
37
- verbose = options['verbose']
38
- for model_name in IMPORT_MODELS:
39
- _metadata = IMPORT_METADATA[model_name]
40
- load_model_data_from_yaml(
41
- self,
42
- model_name,
43
- _metadata,
44
- verbose
45
- )
@@ -1,136 +0,0 @@
1
- from django.core.management import BaseCommand, call_command
2
-
3
- class Command(BaseCommand):
4
- help = 'Run all data loading commands in the correct order'
5
-
6
- def add_arguments(self, parser):
7
- parser.add_argument(
8
- '--verbose',
9
- action='store_true',
10
- help='Display verbose output for all commands',
11
- )
12
-
13
- def handle(self, *args, **options):
14
- # verbose = options['verbose']
15
- verbose = True
16
-
17
- self.stdout.write(self.style.SUCCESS("Populating base db models with data..."))
18
-
19
- # Run the load_network_data command
20
- self.stdout.write(self.style.SUCCESS("Running load_network_data..."))
21
- call_command('load_network_data', verbose=verbose)
22
-
23
- # Run the load_logging_data command
24
- self.stdout.write(self.style.SUCCESS("Running load_logging_data..."))
25
- call_command('load_logging_data', verbose=verbose)
26
-
27
- # Run the load_profession_data command
28
- self.stdout.write(self.style.SUCCESS("Running load_profession_data..."))
29
- call_command('load_profession_data', verbose=verbose)
30
-
31
- # run the load_gender_data command
32
- self.stdout.write(self.style.SUCCESS("Running load_gender_data..."))
33
- call_command('load_gender_data', verbose=verbose)
34
-
35
- # Run the load_disease data command
36
- self.stdout.write(self.style.SUCCESS("Running load_disease_data..."))
37
- call_command('load_disease_data', verbose=verbose)
38
-
39
- # Run the load_disease_classification data command
40
- self.stdout.write(self.style.SUCCESS("Running load_disease_classification_data..."))
41
- call_command('load_disease_classification_data', verbose=verbose)
42
-
43
- # Run the load_disease_classification_choices data command
44
- self.stdout.write(self.style.SUCCESS("Running load_disease_classification_choices_data..."))
45
- call_command('load_disease_classification_choices_data', verbose=verbose)
46
-
47
- # rund the load_user_groups command
48
- self.stdout.write(self.style.SUCCESS("Running load_user_groups..."))
49
- call_command('load_user_groups', verbose=verbose)
50
-
51
- # run the load_report_reader_flag command
52
- self.stdout.write(self.style.SUCCESS("Running load_report_reader_flag..."))
53
- call_command('load_report_reader_flag', verbose=verbose)
54
-
55
- # Run the load_pdf_type_data command
56
- self.stdout.write(self.style.SUCCESS("Running load_pdf_type_data..."))
57
- call_command('load_pdf_type_data', verbose=verbose)
58
-
59
- # Run the load_endoscopy_processor_data command
60
- self.stdout.write(self.style.SUCCESS("Running load_endoscopy_processor_data..."))
61
- call_command('load_endoscopy_processor_data', verbose=verbose)
62
-
63
- # Run the load_endoscope_type_data command
64
- self.stdout.write(self.style.SUCCESS("Running load_endoscope_type_data..."))
65
- call_command('load_endoscope_type_data', verbose=verbose)
66
-
67
- # Run the load_unit_data command
68
- self.stdout.write(self.style.SUCCESS("Running load_unit_data..."))
69
- call_command('load_unit_data', verbose=verbose)
70
-
71
- # Run the load_lab_value_data command
72
- self.stdout.write(self.style.SUCCESS("Running load_lab_value_data..."))
73
- call_command("load_lab_value_data", verbose=verbose)
74
-
75
- # Run the load_information_source command
76
- self.stdout.write(self.style.SUCCESS("Running load_information_source..."))
77
- call_command('load_information_source', verbose=verbose)
78
-
79
- # Run the load_center_data command
80
- self.stdout.write(self.style.SUCCESS("Running load_center_data..."))
81
- call_command('load_center_data', verbose=verbose)
82
-
83
- # Run the load_examination_data command
84
- self.stdout.write(self.style.SUCCESS("Running load_examination_data..."))
85
- call_command('load_examination_data', verbose=verbose)
86
-
87
- # Run the load_label_data command
88
- self.stdout.write(self.style.SUCCESS("Running load_label_data..."))
89
- call_command('load_label_data', verbose=verbose)
90
-
91
- # Run the load_ai_model_data command
92
- self.stdout.write(self.style.SUCCESS("Running load_ai_model_data..."))
93
- call_command('load_ai_model_data', verbose=verbose)
94
-
95
- # Run the load_event_data command
96
- self.stdout.write(self.style.SUCCESS("Running load_event_data..."))
97
- call_command('load_event_data', verbose=verbose)
98
-
99
- # run the load_medication_data command
100
- self.stdout.write(self.style.SUCCESS("Running load_medication_data..."))
101
- call_command('load_medication_data', verbose=verbose)
102
-
103
- # Run the load_medication_indication_type_data command
104
- self.stdout.write(self.style.SUCCESS("Running load_medication_indication_type_data..."))
105
- call_command('load_medication_indication_type_data', verbose=verbose)
106
-
107
- # Run the load_medication_intake_time_data command
108
- self.stdout.write(self.style.SUCCESS("Running load_medication_intake_time_data..."))
109
- call_command('load_medication_intake_time_data', verbose=verbose)
110
-
111
- # Run the load_medication_schedule_data command
112
- self.stdout.write(self.style.SUCCESS("Running load_medication_schedule_data..."))
113
- call_command('load_medication_schedule_data', verbose=verbose)
114
-
115
- # Run the load_medication_indication_data command
116
- self.stdout.write(self.style.SUCCESS("Running load_medication_indication_data..."))
117
- call_command('load_medication_indication_data', verbose=verbose)
118
-
119
- # Run the load_green_endoscopy_wuerzburg_data command
120
- self.stdout.write(self.style.SUCCESS("Running load_green_endoscopy_wuerzburg_data..."))
121
- call_command('load_green_endoscopy_wuerzburg_data', verbose=verbose)
122
-
123
- # Load G-Play Data
124
- # run the load_distribution_data command
125
- self.stdout.write(self.style.SUCCESS("Running load_distribution_data..."))
126
- call_command('load_distribution_data', verbose=verbose)
127
- # Run the load_g_play_data command
128
- self.stdout.write(self.style.SUCCESS("Running load_g_play_data..."))
129
- call_command('load_g_play_data', verbose=verbose)
130
-
131
- # Necessary? Migrate? FIXME
132
- # # Run the load_active_model_data command
133
- # self.stdout.write(self.style.SUCCESS("Running load_active_model_data..."))
134
- # call_command('load_active_model_data', verbose=verbose)
135
-
136
- self.stdout.write(self.style.SUCCESS("All data loading commands executed successfully."))
@@ -1,43 +0,0 @@
1
- from django.core.management.base import BaseCommand
2
- from ...utils import load_model_data_from_yaml
3
- from ...models import Center, FirstName, LastName
4
- from ...data import CENTER_DATA_DIR
5
-
6
- SOURCE_DIR = CENTER_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
7
-
8
- MODEL_0 = Center
9
-
10
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
11
- MODEL_0.__name__,
12
- ]
13
-
14
- IMPORT_METADATA = {
15
- MODEL_0.__name__: {
16
- "dir": SOURCE_DIR, # e.g. "interventions"
17
- "model": MODEL_0, # e.g. Intervention
18
- "foreign_keys": ["first_names", "last_names"], # e.g. ["intervention_types"]
19
- "foreign_key_models": [FirstName, LastName] # e.g. [InterventionType]
20
- }
21
- }
22
-
23
- class Command(BaseCommand):
24
- help = """Load all .yaml files in the data/intervention directory
25
- into the Intervention and InterventionType model"""
26
-
27
- def add_arguments(self, parser):
28
- parser.add_argument(
29
- '--verbose',
30
- action='store_true',
31
- help='Display verbose output',
32
- )
33
-
34
- def handle(self, *args, **options):
35
- verbose = options['verbose']
36
- for model_name in IMPORT_MODELS:
37
- _metadata = IMPORT_METADATA[model_name]
38
- load_model_data_from_yaml(
39
- self,
40
- model_name,
41
- _metadata,
42
- verbose
43
- )
@@ -1,41 +0,0 @@
1
- from django.core.management.base import BaseCommand
2
- from endoreg_db.models import DiseaseClassificationChoice as MODEL_0
3
- from ...utils import load_model_data_from_yaml
4
- from ...data import DISEASE_CLASSIFICATION_CHOICE_DATA_DIR as SOURCE_DIR
5
-
6
- from endoreg_db.models import DiseaseClassification
7
-
8
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
9
- MODEL_0.__name__,
10
- ]
11
-
12
- IMPORT_METADATA = {
13
- MODEL_0.__name__: {
14
- "dir": SOURCE_DIR, # e.g. "interventions"
15
- "model": MODEL_0, # e.g. Intervention
16
- "foreign_keys": ["disease_classification"], # e.g. ["intervention_types"]
17
- "foreign_key_models": [DiseaseClassification] # e.g. [InterventionType]
18
- }
19
- }
20
-
21
- class Command(BaseCommand):
22
- help = """Load all .yaml files in the data/intervention directory
23
- into the Intervention and InterventionType model"""
24
-
25
- def add_arguments(self, parser):
26
- parser.add_argument(
27
- '--verbose',
28
- action='store_true',
29
- help='Display verbose output',
30
- )
31
-
32
- def handle(self, *args, **options):
33
- verbose = options['verbose']
34
- for model_name in IMPORT_MODELS:
35
- _metadata = IMPORT_METADATA[model_name]
36
- load_model_data_from_yaml(
37
- self,
38
- model_name,
39
- _metadata,
40
- verbose
41
- )
@@ -1,41 +0,0 @@
1
- from django.core.management.base import BaseCommand
2
- from endoreg_db.models import DiseaseClassification as MODEL_0
3
- from ...utils import load_model_data_from_yaml
4
- from ...data import DISEASE_CLASSIFICATION_DATA_DIR as SOURCE_DIR
5
-
6
- from endoreg_db.models import Disease
7
-
8
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
9
- MODEL_0.__name__,
10
- ]
11
-
12
- IMPORT_METADATA = {
13
- MODEL_0.__name__: {
14
- "dir": SOURCE_DIR, # e.g. "interventions"
15
- "model": MODEL_0, # e.g. Intervention
16
- "foreign_keys": ["disease"], # e.g. ["intervention_types"]
17
- "foreign_key_models": [Disease] # e.g. [InterventionType]
18
- }
19
- }
20
-
21
- class Command(BaseCommand):
22
- help = """Load all .yaml files in the data/intervention directory
23
- into the Intervention and InterventionType model"""
24
-
25
- def add_arguments(self, parser):
26
- parser.add_argument(
27
- '--verbose',
28
- action='store_true',
29
- help='Display verbose output',
30
- )
31
-
32
- def handle(self, *args, **options):
33
- verbose = options['verbose']
34
- for model_name in IMPORT_MODELS:
35
- _metadata = IMPORT_METADATA[model_name]
36
- load_model_data_from_yaml(
37
- self,
38
- model_name,
39
- _metadata,
40
- verbose
41
- )
@@ -1,40 +0,0 @@
1
- from django.core.management.base import BaseCommand
2
- from endoreg_db.models import Disease as MODEL_0
3
- from ...utils import load_model_data_from_yaml
4
- from ...data import DISEASE_DATA_DIR as SOURCE_DIR
5
-
6
-
7
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
8
- MODEL_0.__name__,
9
- ]
10
-
11
- IMPORT_METADATA = {
12
- MODEL_0.__name__: {
13
- "dir": SOURCE_DIR, # e.g. "interventions"
14
- "model": MODEL_0, # e.g. Intervention
15
- "foreign_keys": [], # e.g. ["intervention_types"]
16
- "foreign_key_models": [] # e.g. [InterventionType]
17
- }
18
- }
19
-
20
- class Command(BaseCommand):
21
- help = """Load all .yaml files in the data/intervention directory
22
- into the Intervention and InterventionType model"""
23
-
24
- def add_arguments(self, parser):
25
- parser.add_argument(
26
- '--verbose',
27
- action='store_true',
28
- help='Display verbose output',
29
- )
30
-
31
- def handle(self, *args, **options):
32
- verbose = options['verbose']
33
- for model_name in IMPORT_MODELS:
34
- _metadata = IMPORT_METADATA[model_name]
35
- load_model_data_from_yaml(
36
- self,
37
- model_name,
38
- _metadata,
39
- verbose
40
- )
@@ -1,66 +0,0 @@
1
- from django.core.management.base import BaseCommand
2
- from endoreg_db.models import (
3
- NumericValueDistribution,
4
- SingleCategoricalValueDistribution,
5
- MultipleCategoricalValueDistribution,
6
- DateValueDistribution
7
-
8
- )
9
- from collections import OrderedDict
10
-
11
- from ...utils import load_model_data_from_yaml
12
- from ...data import (
13
- DISTRIBUTION_NUMERIC_DATA_DIR,
14
- DISTRIBUTION_SINGLE_CATEGORICAL_DATA_DIR,
15
- DISTRIBUTION_MULTIPLE_CATEGORICAL_DATA_DIR,
16
- DISTRIBUTION_DATE_DATA_DIR,
17
- )
18
-
19
- IMPORT_METADATA = OrderedDict({
20
- NumericValueDistribution.__name__: {
21
- "dir": DISTRIBUTION_NUMERIC_DATA_DIR,
22
- "model": NumericValueDistribution,
23
- "foreign_keys": [],
24
- "foreign_key_models": []
25
- },
26
- SingleCategoricalValueDistribution.__name__: {
27
- "dir": DISTRIBUTION_SINGLE_CATEGORICAL_DATA_DIR,
28
- "model": SingleCategoricalValueDistribution,
29
- "foreign_keys": [],
30
- "foreign_key_models": []
31
- },
32
- MultipleCategoricalValueDistribution.__name__: {
33
- "dir": DISTRIBUTION_MULTIPLE_CATEGORICAL_DATA_DIR,
34
- "model": MultipleCategoricalValueDistribution,
35
- "foreign_keys": [],
36
- "foreign_key_models": []
37
- },
38
- DateValueDistribution.__name__: {
39
- "dir": DISTRIBUTION_DATE_DATA_DIR,
40
- "model": DateValueDistribution,
41
- "foreign_keys": [],
42
- "foreign_key_models": []
43
- },
44
- })
45
-
46
- class Command(BaseCommand):
47
- help = """Load all .yaml files in the data/intervention directory
48
- into the Intervention and InterventionType model"""
49
-
50
- def add_arguments(self, parser):
51
- parser.add_argument(
52
- '--verbose',
53
- action='store_true',
54
- help='Display verbose output',
55
- )
56
-
57
- def handle(self, *args, **options):
58
- verbose = options['verbose']
59
- for model_name in IMPORT_METADATA.keys():
60
- _metadata = IMPORT_METADATA[model_name]
61
- load_model_data_from_yaml(
62
- self,
63
- model_name,
64
- _metadata,
65
- verbose
66
- )
@@ -1,45 +0,0 @@
1
- from django.conf import settings
2
- import os
3
- from django.core.management.base import BaseCommand
4
- from ...models import EndoscopeType
5
- from ...utils import load_model_data_from_yaml
6
- from ...data import ENDOSCOPE_TYPE_DATA_DIR
7
-
8
- SOURCE_DIR = ENDOSCOPE_TYPE_DATA_DIR # e.g. settings.DATA_DIR_INTERVENTION
9
-
10
- MODEL_0 = EndoscopeType
11
-
12
- IMPORT_MODELS = [ # string as model key, serves as key in IMPORT_METADATA
13
- MODEL_0.__name__,
14
- ]
15
-
16
- IMPORT_METADATA = {
17
- MODEL_0.__name__: {
18
- "dir": SOURCE_DIR, # e.g. "interventions"
19
- "model": MODEL_0, # e.g. Intervention
20
- "foreign_keys": [], # e.g. ["intervention_types"]
21
- "foreign_key_models": [] # e.g. [InterventionType]
22
- }
23
- }
24
-
25
- class Command(BaseCommand):
26
- help = """Load all .yaml files in the data/intervention directory
27
- into the Intervention and InterventionType model"""
28
-
29
- def add_arguments(self, parser):
30
- parser.add_argument(
31
- '--verbose',
32
- action='store_true',
33
- help='Display verbose output',
34
- )
35
-
36
- def handle(self, *args, **options):
37
- verbose = options['verbose']
38
- for model_name in IMPORT_MODELS:
39
- _metadata = IMPORT_METADATA[model_name]
40
- load_model_data_from_yaml(
41
- self,
42
- model_name,
43
- _metadata,
44
- verbose
45
- )