util.py

import re


def get_wl(fn):
    """Extract wavelength from string"""
    return int(re.search(r"WL(\d+)", fn).group(1))