endoreg-db 0.3.6__py3-none-any.whl → 0.3.7__py3-none-any.whl

endoreg_db/utils/dataloader.py

@@ -1,92 +1,92 @@
-import os
-import yaml
-from django.core.exceptions import ObjectDoesNotExist
-
-def load_model_data_from_yaml(command, model_name, metadata, verbose):
-    """
-    Load model data from YAML files.
-    
-    Args:
-        command: Command object for stdout writing.
-        model_name: Name of the model being loaded.
-        metadata: Metadata including directory and foreign key information.
-        verbose: Boolean indicating whether to print verbose output.
-    """
-    if verbose:
-        command.stdout.write(f"Start loading {model_name}")
-    model = metadata["model"]
-    dir_path = metadata["dir"]
-    foreign_keys = metadata["foreign_keys"]
-    foreign_key_models = metadata["foreign_key_models"]
-
-    _files = [f for f in os.listdir(dir_path) if f.endswith('.yaml')]
-    # sort
-    _files.sort()
-    for file in _files:
-        with open(os.path.join(dir_path, file), 'r') as file:
-            yaml_data = yaml.safe_load(file)
-        
-        load_data_with_foreign_keys(command, model, yaml_data, foreign_keys, foreign_key_models, verbose)
-
-def load_data_with_foreign_keys(command, model, yaml_data, foreign_keys, foreign_key_models, verbose):
-    """
-    Load data handling foreign keys and many-to-many relationships.
-    
-    Args:
-        command: Command object for stdout writing.
-        model: The Django model for the data.
-        yaml_data: Data loaded from YAML.
-        foreign_keys: List of foreign keys.
-        foreign_key_models: Corresponding models for the foreign keys.
-        verbose: Boolean indicating whether to print verbose output.
-    """
-    for entry in yaml_data:
-        fields = entry.get('fields', {})
-        name = fields.pop('name', None)
-        m2m_relationships = {}  # Store many-to-many relationships
-        print(entry)
-
-        # Handle foreign keys and many-to-many relationships
-        for fk_field, fk_model in zip(foreign_keys, foreign_key_models):
-            print(fk_field, fk_model)
-            target_keys = fields.pop(fk_field, None)
-            
-            # Ensure the foreign key exists
-            if target_keys is None:
-                if verbose:
-                    command.stdout.write(command.style.WARNING(f"Foreign key {fk_field} not found in fields"))
-                continue  # Skip if no foreign key provided
-
-            # Process many-to-many fields or foreign keys
-            if isinstance(target_keys, list):  # Assume many-to-many relationship
-                related_objects = []
-                for key in target_keys:
-                    obj, created = fk_model.objects.get_or_create(name=key)
-                    if created and verbose:
-                        command.stdout.write(command.style.SUCCESS(f"Created {fk_model.__name__} {key}"))
-                    related_objects.append(obj)
-                m2m_relationships[fk_field] = related_objects
-            else:  # Single foreign key relationship
-                try:
-                    if model == "endoreg_db.case_template_rule":
-                        print(fk_model, target_keys)
-                    obj = fk_model.objects.get_by_natural_key(target_keys)
-                except ObjectDoesNotExist:
-                    if verbose:
-                        command.stdout.write(command.style.WARNING(f"{fk_model.__name__} with key {target_keys} not found"))
-                    continue
-                fields[fk_field] = obj
-    
-        # Create or update the main object
-        if name is None:
-            obj, created = model.objects.get_or_create(**fields)
-        else:
-            obj, created = model.objects.update_or_create(defaults=fields, name=name)
-        if created and verbose:
-            command.stdout.write(command.style.SUCCESS(f'Created {model.__name__} {name}'))
-        elif verbose:
-            command.stdout.write(command.style.WARNING(f'Skipped {model.__name__} {name}, already exists'))
-
-        # Set many-to-many relationships
-        for field_name, related_objs in m2m_relationships.items():
-            getattr(obj, field_name).set(related_objs)
+import os
+import yaml
+from django.core.exceptions import ObjectDoesNotExist
+
+def load_model_data_from_yaml(command, model_name, metadata, verbose):
+    """
+    Load model data from YAML files.
+    
+    Args:
+        command: Command object for stdout writing.
+        model_name: Name of the model being loaded.
+        metadata: Metadata including directory and foreign key information.
+        verbose: Boolean indicating whether to print verbose output.
+    """
+    if verbose:
+        command.stdout.write(f"Start loading {model_name}")
+    model = metadata["model"]
+    dir_path = metadata["dir"]
+    foreign_keys = metadata["foreign_keys"]
+    foreign_key_models = metadata["foreign_key_models"]
+
+    _files = [f for f in os.listdir(dir_path) if f.endswith('.yaml')]
+    # sort
+    _files.sort()
+    for file in _files:
+        with open(os.path.join(dir_path, file), 'r') as file:
+            yaml_data = yaml.safe_load(file)
+        
+        load_data_with_foreign_keys(command, model, yaml_data, foreign_keys, foreign_key_models, verbose)
+
+def load_data_with_foreign_keys(command, model, yaml_data, foreign_keys, foreign_key_models, verbose):
+    """
+    Load data handling foreign keys and many-to-many relationships.
+    
+    Args:
+        command: Command object for stdout writing.
+        model: The Django model for the data.
+        yaml_data: Data loaded from YAML.
+        foreign_keys: List of foreign keys.
+        foreign_key_models: Corresponding models for the foreign keys.
+        verbose: Boolean indicating whether to print verbose output.
+    """
+    for entry in yaml_data:
+        fields = entry.get('fields', {})
+        name = fields.pop('name', None)
+        m2m_relationships = {}  # Store many-to-many relationships
+        print(entry)
+
+        # Handle foreign keys and many-to-many relationships
+        for fk_field, fk_model in zip(foreign_keys, foreign_key_models):
+            print(fk_field, fk_model)
+            target_keys = fields.pop(fk_field, None)
+            
+            # Ensure the foreign key exists
+            if target_keys is None:
+                if verbose:
+                    command.stdout.write(command.style.WARNING(f"Foreign key {fk_field} not found in fields"))
+                continue  # Skip if no foreign key provided
+
+            # Process many-to-many fields or foreign keys
+            if isinstance(target_keys, list):  # Assume many-to-many relationship
+                related_objects = []
+                for key in target_keys:
+                    obj, created = fk_model.objects.get_or_create(name=key)
+                    if created and verbose:
+                        command.stdout.write(command.style.SUCCESS(f"Created {fk_model.__name__} {key}"))
+                    related_objects.append(obj)
+                m2m_relationships[fk_field] = related_objects
+            else:  # Single foreign key relationship
+                try:
+                    if model == "endoreg_db.case_template_rule":
+                        print(fk_model, target_keys)
+                    obj = fk_model.objects.get_by_natural_key(target_keys)
+                except ObjectDoesNotExist:
+                    if verbose:
+                        command.stdout.write(command.style.WARNING(f"{fk_model.__name__} with key {target_keys} not found"))
+                    continue
+                fields[fk_field] = obj
+    
+        # Create or update the main object
+        if name is None:
+            obj, created = model.objects.get_or_create(**fields)
+        else:
+            obj, created = model.objects.update_or_create(defaults=fields, name=name)
+        if created and verbose:
+            command.stdout.write(command.style.SUCCESS(f'Created {model.__name__} {name}'))
+        elif verbose:
+            command.stdout.write(command.style.WARNING(f'Skipped {model.__name__} {name}, already exists'))
+
+        # Set many-to-many relationships
+        for field_name, related_objs in m2m_relationships.items():
+            getattr(obj, field_name).set(related_objs)

endoreg_db/utils/file_operations.py

@@ -1,30 +1,30 @@
-import shutil
-from pathlib import Path
-from endoreg_db.utils.uuid import get_uuid
-
-def get_uuid_filename(file:Path) -> tuple[str, str]:
-    """
-    Returns a new filename with a uuid
-    """
-    # Get the file extension
-    file_extension = file.suffix
-    # Generate a new file name
-    uuid = get_uuid()
-    new_file_name = f"{uuid}{file_extension}"
-    return new_file_name, uuid
-    
-def rename_file_uuid(old_file:Path):
-    """
-    Rename a file by assigning a uuid while preserving file extension. Returns new filepath and uuid
-    """
-    # Get the file extension
-    file_extension = old_file.suffix
-    # Generate a new file name
-    uuid = get_uuid()
-    new_file_name = f"{uuid}{file_extension}"
-
-    # Rename the file
-    new_file = old_file.with_name(new_file_name)
-    shutil.move(old_file.resolve().as_posix(), new_file.resolve().as_posix())
-
-    return new_file, uuid
+import shutil
+from pathlib import Path
+from endoreg_db.utils.uuid import get_uuid
+
+def get_uuid_filename(file:Path) -> tuple[str, str]:
+    """
+    Returns a new filename with a uuid
+    """
+    # Get the file extension
+    file_extension = file.suffix
+    # Generate a new file name
+    uuid = get_uuid()
+    new_file_name = f"{uuid}{file_extension}"
+    return new_file_name, uuid
+    
+def rename_file_uuid(old_file:Path):
+    """
+    Rename a file by assigning a uuid while preserving file extension. Returns new filepath and uuid
+    """
+    # Get the file extension
+    file_extension = old_file.suffix
+    # Generate a new file name
+    uuid = get_uuid()
+    new_file_name = f"{uuid}{file_extension}"
+
+    # Rename the file
+    new_file = old_file.with_name(new_file_name)
+    shutil.move(old_file.resolve().as_posix(), new_file.resolve().as_posix())
+
+    return new_file, uuid

endoreg_db/utils/hashs.py

@@ -1,34 +1,34 @@
-import hashlib
-from pathlib import Path
-
-def get_video_hash(video_path):
-    """
-    Get the hash of a video file.
-    """
-    # Open the video file in read-binary mode:
-    with open(video_path, 'rb') as f:
-        # Create the hash object, passing in the video contents for hashing:
-        hash_object = hashlib.sha256(f.read())
-        # Get the hexadecimal representation of the hash
-        video_hash = hash_object.hexdigest()
-        assert len(video_hash) <= 255, "Hash length exceeds 255 characters"
-
-    return video_hash
-
-def get_pdf_hash(pdf_path:Path):
-    """
-    Get the hash of a pdf file.
-    """
-    pdf_hash = None
-
-    # Open the file in binary mode and read its contents
-    with open(pdf_path, 'rb') as f:
-        pdf_contents = f.read()
-        # Create a hash object using SHA-256 algorithm
-
-    hash_object = hashlib.sha256(pdf_contents, usedforsecurity=False)
-    # Get the hexadecimal representation of the hash
-    pdf_hash = hash_object.hexdigest()
-    assert len(pdf_hash) <= 255, "Hash length exceeds 255 characters"
-
+import hashlib
+from pathlib import Path
+
+def get_video_hash(video_path):
+    """
+    Get the hash of a video file.
+    """
+    # Open the video file in read-binary mode:
+    with open(video_path, 'rb') as f:
+        # Create the hash object, passing in the video contents for hashing:
+        hash_object = hashlib.sha256(f.read())
+        # Get the hexadecimal representation of the hash
+        video_hash = hash_object.hexdigest()
+        assert len(video_hash) <= 255, "Hash length exceeds 255 characters"
+
+    return video_hash
+
+def get_pdf_hash(pdf_path:Path):
+    """
+    Get the hash of a pdf file.
+    """
+    pdf_hash = None
+
+    # Open the file in binary mode and read its contents
+    with open(pdf_path, 'rb') as f:
+        pdf_contents = f.read()
+        # Create a hash object using SHA-256 algorithm
+
+    hash_object = hashlib.sha256(pdf_contents, usedforsecurity=False)
+    # Get the hexadecimal representation of the hash
+    pdf_hash = hash_object.hexdigest()
+    assert len(pdf_hash) <= 255, "Hash length exceeds 255 characters"
+
     return pdf_hash