B zd1@sdZddlZddlZddlZddlmZddlmZddlZe dZ e Z e eeje e e eedejdZd ZiZed Zd d Zefd dZefddZefddZdeefddZefddZefddZ defddZ!efddZ"efddZ#efd d!Z$defd"d#Z%efd$d%Z&defd&d'Z'dddefd(d)Z(efd*d+Z)efd,d-Z*efd.d/Z+efd0d1Z,efd2d3Z-ddefd4d5Z.ddefd6d7Z/dS)8zl :mod:`gwosc.api` provides the low-level interface functions that handle direct requests to the GWOSC host. N) urlencode) __version__z gwosc.apiZGWOSC_LOG_LEVELlg]zhttps://gwosc.org)z>=z=>z<=z=._matchcss|]}|ddVqdS)rr.Nr).0xrrr sz._fetch_allevents_event_json..cs g|]}|ddkr|qS)rr.r)r5r6)r3rr sz/_fetch_allevents_event_json..css|]}|ddVqdS)rr.Nr)r5r6rrrr7srcSs |ddS)Nrr0r)r6rrr z-_fetch_allevents_event_json..)keyz&multiple events matched for {!r}: '{}'z', 'z$failed to identify {} for event '{}'Nr'z at version {}r0z in catalog '{}') r,listfilteritemssetlensorted ValueErrorr join) r3r'r0r#rZ alleventsZmatchedr4namesr;metamsgr)r'r3r0r_fetch_allevents_event_jsons4       rHc Cst|d}|}|d|d}}t||fts>td|d|d|d}}}||krvtd|d |g} d |kr| d ||fd |kr| d ||fd |kr| d ||fd |kr| d ||f| S)N)rop1op2z1Could not parse select string. Unknown operators.paramval1val2z7Could not parse select string. Unrecognized parameter: z. Use one of: >zmax-[\d.+-eE]+)\s*(?P[<>=]{2})\s*(?P[\w-]+)\s*(?P[<>=]+)\s*(?P[\d.+-eE]+)\s*$zB^\s*(?P[\w-]+)\s*(?P[<>=]{2})\s*(?P[\d.+-eE]+)\s*$N)rzCould not parse select string: )recompiler[rXmatchextendrCr) selectrZtwo_opsZone_oprWsregex_parsemrrr_select_to_queryTs  recCsd|t||dS)Nz{}/eventapi/json/query/show?{})r)r re)r`rrrr_query_events_urlpsrfcCstt||dS)a"Return the JSON metadata for the events constrained by select Parameters ---------- select : `list-like` a list of range constrains for the events. All ranges should have inclusive ends (<= and >= operators). host : `str`, optional the URL of the GWOSC host to query, defaults to https://gwosc.org Returns ------- data : `dict` or `list` the JSON data retrieved from GWOSC and returnend by :meth:`requests.Response.json` Example ------- >>> fetch_filtered_events_json( ... select=[ ... "mass-1-source <= 5", ... "mass-2-source =< 10", ... "10 <= luminosity-distance <= 100", ... ] ... ) )r)rrf)r`rrrrfetch_filtered_events_jsonvsrgcCs&tt|||d|ddddS)N)r'r0r#rr-rZjsonurl)r=rHvalues)r3r'r0rrrr _event_urlsricCstt||||dS)aReturns the JSON metadata for the given event. By default, this function queries across all catalogs and all data-release versions, returning the highest available version, unless the ``version`` and/or ``catalog`` keywords are specified. Parameters ---------- event : `str` the name of the event to query catalog : `str`, optional name of catalogue that hosts this event version : `int`, `None`, optional restrict query to a given data-release version host : `str`, optional the URL of the GWOSC host to query, defaults to https://gwosc.org Returns ------- data : `dict` or `list` the JSON data retrieved from GWOSC and returned by `json.loads` )r'r0r)rri)r3r'r0rrrrfetch_event_jsons rj)0__doc__loggingosr\ urllib.parserrr getLoggerr StreamHandlerZ _loghandler setFormatter Formatter BASIC_FORMAT addHandlersetLevelintgetenvNOTSETZ_MAX_GPSZ DEFAULT_URLrr@rRrrrrr r!r"r%r&r(r)r*r+r,rHrXr[rerfrgrirjrrrrs\    %           :     "