ross/penracourses
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Ross
2023-12-26 18:05:50 +00:00
parent c933038b83
commit 21e44dcaff
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generic/models.py
+3 -4
View File
@@ -242,7 +242,6 @@ class SeriesImageBase(models.Model):
with pydicom.dcmread(self.image) as ds: with pydicom.dcmread(self.image) as ds:
image_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="md5", direct_pixel_data=False) image_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="md5", direct_pixel_data=False)
self.is_dicom = is_dicom
self.image_md5_hash = image_hash self.image_md5_hash = image_hash
image_blake3_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="blake3", direct_pixel_data=True) image_blake3_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="blake3", direct_pixel_data=True)
@@ -254,15 +253,15 @@ class SeriesImageBase(models.Model):
def save(self, *args, **kwargs): def save(self, *args, **kwargs):
"""Override save method to add image hash""" """Override save method to add image hash"""
if self.image: if self.image:
print(self.pk, self.image) with pydicom.dcmread(self.image) as ds:
if not self.image_md5_hash: if not self.image_md5_hash:
image_hash, is_dicom = get_image_hash(self.image, hash_type="md5", direct_pixel_data=False) image_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="md5", direct_pixel_data=False)
self.is_dicom = is_dicom self.is_dicom = is_dicom
self.image_md5_hash = image_hash self.image_md5_hash = image_hash
if not self.image_blake3_hash: if not self.image_blake3_hash:
image_blake3_hash, is_dicom = get_image_hash(self.image, hash_type="blake3", direct_pixel_data=True) image_blake3_hash, is_dicom = get_image_hash(self.image, dataset=ds, hash_type="blake3", direct_pixel_data=True)
self.is_dicom = is_dicom self.is_dicom = is_dicom
self.image_blake3_hash = image_blake3_hash self.image_blake3_hash = image_blake3_hash