U Kv fQ@s@dZddlmZddlmZGdddZddZd d d Zd S) z$ Handle file opening for read/write )Path)_is_string_likec@s0eZdZdZddZddZddZdd Zd S) EmptyContextManagerzw This class is needed to allow file-like object to be used as context manager, but without getting closed. cCs ||_dSN_obj)selfobjr >/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/iolib/openfile.py__init__szEmptyContextManager.__init__cCs|jS)z)When entering, return the embedded objectr)rr r r __enter__szEmptyContextManager.__enter__cGsdS)zDo not hide anythingFr )rargsr r r __exit__szEmptyContextManager.__exit__cCs t|j|Sr)getattrr)rnamer r r __getattr__szEmptyContextManager.__getattr__N)__name__ __module__ __qualname____doc__r r rrr r r r r s rcCs4|dr"ddl}|j|||dSt|||dSdS)Nz.gzrencoding)endswithgzipopen)fnamemoderrr r r _opens rrNc Cst|rt|}t|tr(|j||dSt|dr@|j||dSzt|||dWStk rz(d|krp|jd|ksd|kr|jWntk rt dYnXt |YSXdS) aT Light wrapper to handle strings, path objects and let files (anything else) pass through. It also handle '.gz' files. Parameters ---------- fname : str, path object or file-like object File to open / forward mode : str Argument passed to the 'open' or 'gzip.open' function encoding : str For Python 3 only, specify the encoding of the file Returns ------- A file-like object that is always a context-manager. If the `fname` was already a file-like object, the returned context manager *will not close the file*. )rrrrrwaz,fname must be a string or a file-like objectN) rr isinstancerhasattr TypeErrorreadwriteAttributeError ValueErrorr)rrrr r r get_file_obj's"   r))rN)rpathlibrZnumpy.lib._iotoolsrrrr)r r r r s