B âêõbCã@sxdZddlmZmZmZdgZddlmZmZddl Z ddl m Z ddl m Z ddlmZmZdd lmZd d d„ZdS) z8 Tools to aid in input/output within the parmed package é)Úprint_functionÚdivisionÚabsolute_importÚgenopen)Ú TextIOWrapperÚBytesION)ÚPY2)Úurlopen)Ú HTTPErrorÚURLError)ÚDEFAULT_ENCODINGÚrc Csr|dkrtdƒ‚d}| d¡r,|dd…}nv| d¡sJ| d¡sJ| d ¡r¢d }|d kr^td ƒ‚y t|ƒ}Wn6ttfk r }ztd ||fƒ‚Wdd}~XYnX| d¡r¤ddl}trò|ròt ƒ}|  |  |  ¡¡¡|  ¡| d¡t|ƒS|dkrnt ƒ}tj |¡r@| |d¡}|  |  ¡¡WdQRX| d¡| |d¡}|  |  ¡¡~trf|St|ƒStr„| ||d¡S|rŽ|}t| ||d¡ƒSn’| d¡r6ddl} tr |røt ƒ}|  |  ¡¡| d¡|  ¡| j|ddS|  ||d¡Sn,|r"t| j|ddƒSt|  ||d¡ƒS|rPtrF|St|ƒSntr`t||ƒSt||tdSdS)aì Opens a file, automatically detecting compression schemes by filename extension. Note, these files are opened in a way that *always* returns a string. This is an important distinction in Python 3 where many file-like objects return bytes instead of strings. This is detected and handled properly so that the object returned from this function is always string-based (so you cannot write or read bytes directly from a file opened via ``genopen``). This routine also recognizes URLs and will read remote files when given a URL starting with either http:// or https://. Like with standard local file names, compression is automatically detected by filename extension, and both gzip and bzip2 files are supported. Parameters ---------- name : str Name of the file to open or URL to a remote file to access mode : str, optional Whether to open the file to 'r'ead, 'w'rite, or 'a'ppend. Default is 'r' Returns ------- file : file-like A file-like object in the requested mode Notes ----- Python's BZ2File does not support writing to ``append`` mode (mode='a'), so it is faked here. The entire file contents are read into memory and then written into a 'new' file with the same name as the original. As such, it is noticeably slower and more resource-intensive (particularly for large files) than using gzipped files. In Python 2, opened URLs are not file-like *enough* for GzipFile or BZ2File to read directly from them, so they must first be loaded entirely into memory. With Python 3, this limitation is not present, so reading remote Gzipped or Bzipped files is significantly cheaper with respect to memory requirements. )Úwr Úaz"open mode must be "w", "r", or "a"Fzfile:///éNzhttp://zhttps://zftp://T)rrz Cannot write or append a webpagezCould not open %s: %sz.bz2rrÚrbÚwbÚbz.gzr )ÚfileobjÚmode)Úencoding)Ú ValueErrorÚ startswithr r r ÚIOErrorÚendswithÚbz2rrÚwriteZ decompressÚreadÚcloseÚseekrÚosÚpathÚexistsZBZ2FileÚgzipZGzipFileÚopenr ) ÚnamerZis_urlZopen_urlÚerrZtmpÚfr#©r(ú.lib/python3.7/site-packages/parmed/utils/io.pyrsv)   "         )r )Ú__doc__Z __future__rrrÚ__all__Úiorrr Zparmed.utils.sixrZ%parmed.utils.six.moves.urllib.requestr Z#parmed.utils.six.moves.urllib.errorr r Zparmed.constantsr rr(r(r(r)Ús