Source code for AFL.automation.shared.utilities

import types
import datetime
from AFL.automation.shared.units import has_units

try:
    import matplotlib.pyplot as plt
except ModuleNotFoundError:
    print('importing Matplotlib failed, do you want plotting to work?')
import io




def listify(obj):
    if isinstance(obj, str) or isinstance(obj, dict) or not hasattr(obj, "__iter__"):
        obj = [obj]
    elif has_units(obj):
        #special handling for pint quanitites whch, for some reason
        #have __iter__ defined for single values...
        try:
            len(obj)
        except TypeError:
            obj = [obj]
    return obj




def tprint(in_str):
    now = datetime.datetime.now().strftime('%y%m%d-%H:%M:%S')
    print(f'[{now}] {in_str}')

    


def makeRegistar():
    registry = {}
    def registrarDecorator(prepType):#the actual decorator
        def registrar(cls):#function that interacts with cls
            registry[prepType] = cls
            return cls #return unwrapped class after registering
        return registrar
    registrarDecorator.registry = registry
    return registrarDecorator




def mpl_plot_to_bytes(fig=None,format='svg'):
    if fig is None:
        fig = plt.gcf()
    byte_str  = io.BytesIO()
    fig.savefig(byte_str,format=format)
    byte_str.seek(0) # need to rewind
    return byte_str





def xarray_to_bytes(ds,format='svg'):
    byteobj = ds.to_netcdf()
    byte_str  = io.BytesIO(byteobj)
    byte_str.seek(0) # need to rewind
    return byte_str