util.py

import re


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