i+ddlZddlmZmZ ddlZddlmZn#ddlmZddlm ZYnxYwddl Z ddl m Z ddl mZmZdZd Zd Zdd Zdd ZdZdZdZdZdZdS)N)BASE_URLBASE_URL_REDUCED)StringIO)BytesIO) MMTFDecoder) MMTFEncoderTemplateEncodercn tj|ddS#tj$r }Yd}~dSd}~wwxYw)z Check to see if the internet is on by pinging a set address. :param address: the IP or address to hit :return: a boolean - true if can be reached, false if not. r)timeoutTNF)urllib2urlopenURLError)addresserrs ^/srv/www/vhosts/g4struct/public_html/venv/lib/python3.11/site-packages/mmtf/api/default_api.py _internet_onrsP ++++t  uuuuus 44cnt}t|||||dS)aAPI function to write data as MMTF to a file :param file_path the path of the file to write :param input_data the input data in any user format :param input_function a function to converte input_data to an output format. Must contain all methods in TemplateEncoder N)r pass_data_on write_file) file_path input_datainput_function mmtf_encoders r write_mmtfrs9==L^\:::I&&&&&c||||S)a0Helper to pass data from one data structure to another. :param input_data the input data in any user format :param input_function a function to converte input_data to an output format. Must contain all methods in TemplateEncoder. :param output_data a data holder for the data to be put into.)rr output_datas rrr&s N:{+++ rFc~t||}tj|}|ddtj|}|ddkr"t|}n|}t|S)z|" Get the msgpack unpacked data given a PDB id. :param pdb_id: the input PDB id :return the unpacked data (a dict) zAccept-encodinggzipzContent-Encoding) get_urlr Request add_headerrinfoget ungzip_dataread_unpack)pdb_idreducedurlrequestresponsedatas rget_raw_data_from_urlr00s & ! !Coc""G (&111w''H}}-..&888==??++}} 4==rc.|r t|zSt|zS)zuGet the URL for the data for a given PDB id. :param pdb_id: the input PDB id :return the URL for this PDB id)rr)r*r+s rr"r"?s" !&((&  rcVtj|d}|S)NF)raw)msgpackunpackbr()r/out_datas rr)r)Is#tyy{{666H Orcft}|t||S)zxReturn a decoded API to the data from a PDB id. :param pdb_id: the input PDB id :return an API to decoded data )r decode_datar0)r*decoders rfetchr:Ns/ mmG -f55666 Nrct}t|d5}|t|dddn #1swxYwY|S)zReturn a decoded API to the data from a file path. :param file_path: the input file path. Data is not entropy compressed (e.g. gzip) :return an API to decoded data rbN)ropenr8r))r newDecoderfhs rparser@Ws J i  ,"wr{{+++,,,,,,,,,,,,,,, s#AAAct}|ttj|d|S)zReturn a decoded API to the data from a file path. File is gzip compressed. :param file_path: the input file path. Data is gzip compressed. :return an API to decoded datar<)rr8r)r!r=)rr>s r parse_gziprBbs;J749Y#=#=>>??? rcNt|}tj|}|S)zpReturn a string of data after gzip decoding :param the input gziped data :return the gzip decoded data)fileobj)rr!GzipFile)rbuffs rr'r'ks( :  C c"""A Hr)F)r!mmtf.utils.constantsrrr rurllib.requestr-iorr4 mmtf_readerr mmtf_writerr r rrrr0r"r)r:r@rBr'rrrrMsY ;;;;;;;;'NNN!!!!!!!'$$$$$$&&&&&&&&$$$$$$44444444    ' ' '    !!!!      s )