dbdicom 0.3.6__py3-none-any.whl → 0.3.7__py3-none-any.whl

dbdicom/api.py

@@ -1,10 +1,18 @@
+import os
+import shutil
+import zipfile
+from pathlib import Path
 from typing import Union
+from tqdm import tqdm
+
 
 import vreg
 
 from dbdicom.dbd import DataBaseDicom
 
 
+
+
 def open(path:str) -> DataBaseDicom:
     """Open a DICOM database
 
@@ -363,10 +371,74 @@ def unique(pars:list, entity:list) -> dict:
     return u
 
 
+def archive(path, archive_path):
+    dbd = open(path)
+    dbd.archive(archive_path)
+    dbd.close()
+
+
+def restore(archive_path, path):
+    _copy_and_extract_zips(archive_path, path)
+    dbd = open(path)
+    dbd.close()
 
 
+def _copy_and_extract_zips(src_folder, dest_folder):
+    if not os.path.exists(dest_folder):
+        os.makedirs(dest_folder)
 
+    # First pass: count total files
+    total_files = sum(len(files) for _, _, files in os.walk(src_folder))
 
-if __name__=='__main__':
+    with tqdm(total=total_files, desc="Copying and extracting") as pbar:
+        for root, dirs, files in os.walk(src_folder):
+            rel_path = os.path.relpath(root, src_folder)
+            dest_path = os.path.join(dest_folder, rel_path)
+            os.makedirs(dest_path, exist_ok=True)
 
+            for file in files:
+                src_file_path = os.path.join(root, file)
+                dest_file_path = os.path.join(dest_path, file)
+
+                if file.lower().endswith('.zip'):
+                    try:
+                        zip_dest_folder = dest_file_path[:-4]
+                        with zipfile.ZipFile(src_file_path, 'r') as zip_ref:
+                            zip_ref.extractall(zip_dest_folder)
+                        #tqdm.write(f"Extracted ZIP: {src_file_path}")
+                        #_flatten_folder(zip_dest_folder) # still needed?
+                    except zipfile.BadZipFile:
+                        tqdm.write(f"Bad ZIP file skipped: {src_file_path}")
+                else:
+                    shutil.copy2(src_file_path, dest_file_path)
+
+                pbar.update(1)
+
+
+def _flatten_folder(root_folder):
+    for dirpath, dirnames, filenames in os.walk(root_folder, topdown=False):
+        for filename in filenames:
+            src_path = os.path.join(dirpath, filename)
+            dst_path = os.path.join(root_folder, filename)
+            
+            # If file with same name exists, optionally rename or skip
+            if os.path.exists(dst_path):
+                base, ext = os.path.splitext(filename)
+                counter = 1
+                while os.path.exists(dst_path):
+                    dst_path = os.path.join(root_folder, f"{base}_{counter}{ext}")
+                    counter += 1
+
+            shutil.move(src_path, dst_path)
+
+        # Remove empty subdirectories (but skip the root folder)
+        if dirpath != root_folder:
+            try:
+                os.rmdir(dirpath)
+            except OSError:
+                print(f"Could not remove {dirpath} — not empty or in use.")
+
+
+
+if __name__=='__main__':
     pass

dbdicom/dbd.py

@@ -2,6 +2,7 @@ import os
 from datetime import datetime
 import json
 from typing import Union
+import zipfile
 
 from tqdm import tqdm
 import numpy as np
@@ -880,6 +881,34 @@ class DataBaseDicom():
         register.add_instance(self.register, attr, rel_path)
 
 
+    def archive(self, archive_path):
+        # TODO add flat=True option for zipping at patient level
+        for pt in tqdm(self.register, desc='Archiving '):
+            for st in pt['studies']:
+                zip_dir = os.path.join(
+                    archive_path,
+                    f"Patient__{pt['PatientID']}", 
+                    f"Study__{st['StudyID']}__{st['StudyDescription']}", 
+                )
+                os.makedirs(zip_dir, exist_ok=True)
+                for sr in st['series']:
+                    try:
+                        zip_file = os.path.join(
+                            zip_dir, 
+                            f"Series__{sr['SeriesNumber']}__{sr['SeriesDescription']}.zip",
+                        )
+                        with zipfile.ZipFile(zip_file, 'w') as zipf:
+                            for rel_path in sr['instances'].values():
+                                file = os.path.join(self.path, rel_path)
+                                zipf.write(file, arcname=os.path.basename(file))
+                    except Exception as e:
+                        raise RuntimeError(
+                            f"Error extracting series {sr['SeriesDescription']} "
+                            f"in study {st['StudyDescription']} of patient {pt['PatientID']}."
+                        )
+
+
+
 
 def infer_slice_spacing(vols):
     # In case spacing between slices is not (correctly) encoded in 

{dbdicom-0.3.6.dist-info → dbdicom-0.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbdicom
-Version: 0.3.6
+Version: 0.3.7
 Summary: A pythonic interface for reading and writing DICOM databases
 Author-email: Steven Sourbron <s.sourbron@sheffield.ac.uk>, Ebony Gunwhy <e.gunwhy@sheffield.ac.uk>
 Project-URL: Homepage, https://openmiblab.github.io/dbdicom/

{dbdicom-0.3.6.dist-info → dbdicom-0.3.7.dist-info}/RECORD

@@ -1,9 +1,9 @@
 dbdicom/__init__.py,sha256=dW5aezonmMc_41Dp1PuYmXQlr307RkyJxsJuetkpWso,87
-dbdicom/api.py,sha256=GiXke8CmCjfzGnRnHqKtuq68rA7yFB4BY74VsWSb0E4,11469
+dbdicom/api.py,sha256=lUNLQo3aAAIFVF7IMccNhXAlbf1hMNTU9gHibdfbKcI,14135
 dbdicom/const.py,sha256=BqBiRRjeiSqDr1W6YvaayD8WKCjG4Cny2NT0GeLM6bI,4269
 dbdicom/database.py,sha256=_LUbH7gc9l7j_63AC71DjwxgTUwbEjHSy5kuvRw75Hw,4764
 dbdicom/dataset.py,sha256=kbswoVE3-Wtu0uwcSW319KzpdWx5fZphfKxizxJePzc,22773
-dbdicom/dbd.py,sha256=_Fj9xFpx6Rsr5UH5YR6PJaJPSOl4cMfNmk9V2b1dEyo,36894
+dbdicom/dbd.py,sha256=yvj4PRj4gP1Qmh0VhibOrZ1oxy9v9qM9iOV0L6C_Ryw,38261
 dbdicom/register.py,sha256=_NyNbOEAN_hkwjxupNpr9F5DWUwARCsci8knK41-EsA,13931
 dbdicom/external/__init__.py,sha256=XNQqfspyf6vFGedXlRKZsUB8k8E-0W19Uamwn8Aioxo,316
 dbdicom/external/__pycache__/__init__.cpython-311.pyc,sha256=pXAQ35ixd92fm6YcuHgzR1t6RcASQ-cHhU1wOA5b8sw,542
@@ -47,8 +47,8 @@ dbdicom/utils/dcm4che.py,sha256=Vxq8NYWWK3BuqJkzhBQ89oMqzJlnxqTxgsgTo_Frznc,2317
 dbdicom/utils/files.py,sha256=qhWNJqeWnRjDNbERpC6Mz962_TW9mFdvd2lnBbK3xt4,2259
 dbdicom/utils/image.py,sha256=qsU_wOdleZCqU5g7LSp8OXKmoG109NauYx7wad_ZQ7Q,3839
 dbdicom/utils/variables.py,sha256=vUh5cDnmCft5hoXDYXUvfkg5Cy5WlgMAogU38Y_BKRo,5753
-dbdicom-0.3.6.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-dbdicom-0.3.6.dist-info/METADATA,sha256=cl4tN9uUuh7Cr_5V0zlBCwyTfWdJH69C4kAettD0CFg,1030
-dbdicom-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dbdicom-0.3.6.dist-info/top_level.txt,sha256=nJWxXg4YjD6QblfmhrzTMXcr8FSKNc0Yk-CAIDUsYkQ,8
-dbdicom-0.3.6.dist-info/RECORD,,
+dbdicom-0.3.7.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+dbdicom-0.3.7.dist-info/METADATA,sha256=kSsHB7e2v_JdenBx6j9hewbYIOsPucwANgvL_BZEjXw,1030
+dbdicom-0.3.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dbdicom-0.3.7.dist-info/top_level.txt,sha256=nJWxXg4YjD6QblfmhrzTMXcr8FSKNc0Yk-CAIDUsYkQ,8
+dbdicom-0.3.7.dist-info/RECORD,,