CODEWITHSUNDEEP
pythonos.path
fredooms
fredooms

Reputation: 123

file renaming adding a suffix

How could I add from the following variable fIN = T1_r.nii.gz the following suffix _brain and create the following output filename?

fOut = T1_r_brain.nii.gz

When I use the following command line

fIn2, file_extension = os.path.splitext(fIn)

it only removes the .gz extension. Thank you for your help Fred

Upvotes: 1

Views: 472

Answers (2)

revmatcher
revmatcher

Reputation: 930

split_path = 'T1_r.nii.gz'.split('.')
    
split_path[0] += '_brain'
    
final_path = ".".join(split_path)

Upvotes: 1

jkr
jkr

Reputation: 19250

I had to write a utility for this, and here's what I came up with.

from pathlib import Path

def add_str_before_suffixes(filepath, string: str) -> Path:
    """Append a string to a filename immediately before extension(s).
    
    Parameters
    ----------
    filepath : Path-like
        Path to modify. Can contain multiple extensions like `.bed.gz`.
    string : str
        String to append to filename.
    
    Returns
    -------
    Instance of `pathlib.Path`.
    
    Examples
    --------
    >>> add_str_before_suffixes("foo", "_baz")
    PosixPath('foo_baz')
    >>> add_str_before_suffixes("foo.bed", "_baz")
    PosixPath('foo_baz.bed')
    >>> add_str_before_suffixes("foo.bed.gz", "_baz")
    PosixPath('foo_baz.bed.gz')
    """
    filepath = Path(filepath)
    suffix = "".join(filepath.suffixes)
    orig_name = filepath.name.replace(suffix, "")
    new_name = f"{orig_name}{string}{suffix}"
    return filepath.with_name(new_name)

Here is an example:

>>> f_in = "T1_r.nii.gz"
>>> add_str_before_suffixes(f_in, "_brain")
PosixPath('T1_r_brain.nii.gz')

Upvotes: 1

Related Questions