fileSaveScripts.py

import os
import numpy as np
import shutil

def file_exists(file_name, extension, *args):
    # Construct the directory path
    directory_path = os.path.join(".", *args)
    file_name = os.path.splitext(file_name)[0]
    file_name = f"{file_name}.{extension}"
    file_path = os.path.join(directory_path, file_name)
    
    if os.path.exists(file_path):
        print(f"File {file_path} already exists, skipping...")
        return True
    else:
        return False

def save_numpy_file(data, file_name, *args):
    """
    Save a numpy array to a specified directory structure.

    Parameters:
    - data: numpy array to be saved
    - file_name: name of the file (without extension)
    - *args: additional arguments representing directory structure
    """
    # Construct the directory path
    directory_path = os.path.join(".", *args)
    
    # Ensure the directory exists
    os.makedirs(directory_path, exist_ok=True)
    
    # Construct the full file path
    file_path = os.path.join(directory_path, f"{file_name}.npy")
    
    # Save the numpy array to the file
    try:
        np.save(file_path, data)
    except Exception as e:
        print(f"An error occurred: {e}")
    
    print(f"File saved to {file_path}")

def save_dataframe_to_csv(df, file_path):
    # Ensure the path is relative to the current working directory
    if file_path.startswith('/'):
        file_path = '.' + file_path
    
    directory = os.path.dirname(file_path)

    # Check if the directory exists and create it if it doesn't
    if not os.path.exists(directory):
        os.makedirs(directory)

    # Save the DataFrame to a CSV file
    df.to_csv(file_path, index=False)
    
    return file_path

def save_temp_file(temp_file_path, desired_name, *args):
    """
    Move a temporary file to a specified directory structure with a new name.

    Parameters:
    - temp_file_path: str, path to the temporary file
    - desired_name: str, desired name of the file (without extension)
    - *args: additional arguments representing directory structure
    """
    # Extract the file extension from the temp file
    _, file_extension = os.path.splitext(temp_file_path)
    
    # Construct the directory path
    directory_path = os.path.join(".", *args)
    
    # Ensure the directory exists
    os.makedirs(directory_path, exist_ok=True)
    
    # Construct the full file path with the desired name and original extension
    file_path = os.path.join(directory_path, f"{desired_name}{file_extension}")
    
    # Move the temporary file to the new location with the new name
    shutil.move(temp_file_path, file_path)
    
    print(f"File moved to {file_path}")

# Example usage
if __name__ == "__main__":
    # Example numpy data
    data = np.array([1, 2, 3, 4, 5])
    
    # Example function call
    save_numpy_file(data, "test", "data", "example")