IO module

Filesystem manipulation.

physicslab.io.gather_files(extension, folder, key_edit=None, trim_extension=True)[source]

Gather all files of the given extension located in or under folder.

Parameters

  • extension (str) – File extension to look for

  • folder (str) – Look in that folder and all its subfolders

  • key_edit (callable, optional) – If supplied, this function will be applied to the filename stem, defaults to None

  • trim_extension (bool, optional) – Cut the extension from filename to be used as key, defaults to True

Returns

Dictionary form {filename : path}

Return type

dict

physicslab.io.subfolder(folder, look_for)[source]

Look for a subfolder containing look_for in its name.

Parameters

  • folder (str) – Search there

  • look_for (str) – Part of the folder name to look for. Can be RegEx.

Raises

  • OSError – If no subfolder found

  • OSError – If multiple subfolders found

Returns

Path to the found subfolder

Return type

str