o joi@sdZddlmZmZddlmZddlmZddlZddlm Z e dddZ ej Z ee e jfZd e d ee e ffd d Zd eegdfd dfddZdS)aInterface for generating multiple FAR files from a Pynini file. Given a number of output FAR files (FST archive) via --output in the form designator1=file1,designator2=file2,... a Pynini file produces the FAR files file1, file2,... by exporting a number of FSTs into these FAR files. To this end, a Pynini file creates a generator_main file with the signature: def generator_main(exporter_map: multi_grm.ExporterMapping): which can contain lines of the form my_exporter = exporter_map[designator] my_exporter[fst_name] = fst to export a given FST under a corresponding name into the file identified by the given designator. The generator main function should be have a: if __name__ == '__main__': multi_grm.run(generator_main) For an example, see multi_grm_example.py. )CallableMapping)app)flagsN)exportoutputszHThe output FAR files in the form designator1=file1,designator2=file2,...target_file_pairsreturncCs^i}|dD]%}|d}t|dkrtd|d||d<td|d|dq|S)aaGenerates a map from a list designator1=file1,designator2=file2,... Args: target_file_pairs: A string of the form designator1=file1,designator2=file2,... Returns: A map from designators to file names. Raises: app.UsageError: If there is a parsing error. Does not check whether the files can be written. ,=zA--outputs must be of form designator1=file1,designator2=file2,...rz1Setting up target designator [%s] with file [%s].)splitlenr UsageErrorlogginginfo)rresulttarget_file_pairtarget_file_listrK/home/ubuntu/.local/lib/python3.10/site-packages/pynini/export/multi_grm.py_get_target_file_map6s     rgenerator_maincs*fdd}tddt_t|dS)a%Executes the multi_grm FAR export program to export a number of FARs. Args: generator_main: The FAR generator_main function to execute. It takes a single argument "exporter_map", which is a map from string designators to export.Exporter objects with a __setitem__ method. cs|z6|d|rtd|ttj}|stddd|D}||D]}|q-WdSdt _ )Nrz#Unexpected command line arguments: z3--outputs must specify at least one name=file pair.cSsi|] \}}|t|qSr)rExporter).0 designatorfilenamerrr as z%run..main..fatal) poprrr_OUTPUTSvalueitemsvaluescloseFLAGSstderrthreshold) unused_argvr exporter_mapexrrrmainWs*    zrun..mainrwarningN)rmark_flag_as_requiredr&r'rrun)rr,rr+rr/Ns r/)__doc__typingrrabslrrr pynini.exportr DEFINE_stringr!r&strrExporterMappingrr/rrrrs