o joi @sHdZddlZddlmZddlZddlZeeejfZGdddZ dS)z4Helper classes for generating FAR files with Pynini.N)Unionc @sTeZdZdZ  ddededejddfd d Zd ed ej ddfd dZ dddZ dS)ExporteraHelper class to collect and export FSTs into an FAR archive file. For example, the following code collects 2 FSTs and writes them upon destruction of the involved exporter instance. exporter = grm.Exporter(filename) exporter.export("FST1", fst1) exporter.export("FST2", fst2) Typically, instead of explicitly creating an Exporter, a client will use the exporter provided as the argument to the generator_main function. standarddefaultfilenamearc_typefar_typereturnNcCs4td|i|_t||_||_||_d|_dS)a:Creates an exporter that writes a FAR archive file upon destruction. Args: filename: A string with the filename. arc_type: A string with the arc type; one of: "standard", "log", "log64". far_type: A string with the file type; one of: "default", "sstable", "sttable", "stlist". zSetting up exporter for %rTN) logginginfo_fstsosfspath _filename _arc_type _far_type_is_open)selfrrrrH/home/ubuntu/.local/lib/python3.10/site-packages/pynini/export/export.py__init__(s   zExporter.__init__namefstcCs(|jsJtd||j||j|<dS)zRegister an FST under a given name to be saved into the FST archive. Args: name: A string with the name of the fst. fst: An FST to be stored. zAdding FST %r to archive %rN)rr r rr )rrrrrr __setitem__;s zExporter.__setitem__cCs|jsJtd|jtj|jd|j|jd}t|j D]}td||j|j |||<qWdn1s:wYtd|jd|_dS)z=Writes the registered FSTs into the given file and closes it.zWriting FSTs into %rw)rrzWriting FST %r to %rNzWriting FSTs to %r completeF) rr r rpyniniFarrrsortedr )rsinkrrrrcloseFs   zExporter.close)rr)r N) __name__ __module__ __qualname____doc__ _FilenamestrrFarTyperFstrrrrrrrs  r) r#r typingrr rr%PathLiker$rrrrrs