74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
import re
|
|
from django.conf import settings
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
import pydicom
|
|
import pydicom.errors
|
|
|
|
import six
|
|
import os
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Checks dicom files for anonymisation'
|
|
|
|
#def add_arguments(self, parser):
|
|
# parser.add_argument('poll_ids', nargs='+', type=int)
|
|
|
|
def handle(self, *args, **options):
|
|
#for poll_id in options['poll_ids']:
|
|
# try:
|
|
# poll = Poll.objects.get(pk=poll_id)
|
|
# except Poll.DoesNotExist:
|
|
# raise CommandError('Poll "%s" does not exist' % poll_id)
|
|
|
|
# poll.opened = False
|
|
# poll.save()
|
|
|
|
# self.stdout.write(self.style.SUCCESS('Successfully closed poll "%s"' % poll_id))
|
|
files = get_all_media()
|
|
|
|
for file in files:
|
|
try:
|
|
ds = pydicom.dcmread(file, force=False)
|
|
except pydicom.errors.InvalidDicomError:
|
|
continue
|
|
|
|
site_codes = ("ref", "rk9", "ra9", "rh8", "rbz", "rba")
|
|
|
|
for code in site_codes:
|
|
# TODO rewrite
|
|
try:
|
|
if code in ds.PatientID.lower():
|
|
self.stdout.write(self.style.WARNING(f"Identifiable dicom [{ds.PatientID.lower()}] {file}"))
|
|
except AttributeError:
|
|
self.stdout.write(self.style.WARNING(f"Error parsing {file}"))
|
|
pass
|
|
try:
|
|
if code in ds.AccessionNumber.lower():
|
|
self.stdout.write(self.style.WARNING(f"Error parsing [{ds.AccessionNumber.lower()}] {file}"))
|
|
except AttributeError:
|
|
pass
|
|
|
|
|
|
|
|
def get_all_media(exclude=None):
|
|
"""
|
|
Get all media from MEDIA_ROOT
|
|
"""
|
|
|
|
if not exclude:
|
|
exclude = []
|
|
|
|
media = set()
|
|
|
|
for root, dirs, files in os.walk(six.text_type(settings.MEDIA_ROOT)):
|
|
for name in files:
|
|
path = os.path.abspath(os.path.join(root, name))
|
|
relpath = os.path.relpath(path, settings.MEDIA_ROOT)
|
|
|
|
for e in exclude:
|
|
if re.match(r'^%s$' % re.escape(e).replace('\\*', '.*'), relpath):
|
|
break
|
|
else:
|
|
media.add(path)
|
|
|
|
return media |