a h@s.ddlZddlmZGdddZddZdS)N)pathc@seZdZdZddZdS)dataSourcesInfoz A class to parse and store information from a YAML file containing observations data. Attributes: tab (dict): A dictionary to store the parsed observations data. cCstttd}t|d}t||_Wdn1s>0Yi|_|jdD]:}|d}i|j|<|dD]}|d}||j||<qvqXdS)zx The constructor for dataSourcesInfo class. Reads from a YAML file and initializes the 'tab' attribute. z table.ymlrNZ observationskxdetailsvar) rjoindirname__file__openyamlZ safe_loaddatatab)selfZ yaml_filefileZ observationrZdetailrr5/home/caroline.viezel/readDiag/gsidiag/datasources.py__init__ s *  zdataSourcesInfo.__init__N)__name__ __module__ __qualname____doc__rrrrrrsrcCst}||jvr||j|vrt||j||vrB|j|||Std|td|j||D]}td|qbqtd|d|td|j|D]}td|qn(td|td|jD]}td|qdS)aU Retrieves information for a specified feature based on the kx and variable. Args: kx (int): The kx value to look up. var (str): The variable to look up. feature (str): The specific feature to retrieve information about. Returns: The value of the requested feature if found, otherwise None. zInvalid Feature request:zTry using one of:z *ZVariablezdoesn't exist in kxzInvalid kx request:N)rrprint)rrfeatureZdataInfoitemrrr getVarInfos"     r)r osrrrrrrrs