-
-
Notifications
You must be signed in to change notification settings - Fork 420
Get IBE's query_region_async function minimally working #1430
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
odysseus9672
wants to merge
12
commits into
astropy:main
Choose a base branch
from
odysseus9672:ibe-dev-search
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from 9 commits
Commits
Show all changes
12 commits
Select commit
Hold shift + click to select a range
7fc6dbd
Fix bug in list_missions()
odysseus9672 ff4988c
break a long line
odysseus9672 75af6f7
remove redundant backslash (PEP 8)
odysseus9672 04902fc
Fix placement of rstrip
odysseus9672 0766865
fix whitespace PEP 8 issues
odysseus9672 0c4c7c7
Fix ibe's list_datasets, list_tables, and get_columns methods
odysseus9672 516bc5e
remove extraneous comment
odysseus9672 6c61f32
get query_region_async minimally functional.
odysseus9672 a17f56b
Fix PEP8 issues
odysseus9672 62ce28b
Remove version checking hack
odysseus9672 51f2abe
Remove `dataset` parameter from `config`.
odysseus9672 b505817
Trim code
odysseus9672 File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -10,11 +10,14 @@ | |
from __future__ import print_function, division | ||
|
||
import os | ||
import sys | ||
|
||
import webbrowser | ||
from bs4 import BeautifulSoup | ||
|
||
import astropy.coordinates as coord | ||
from astropy.table import Table | ||
from astropy.io.ascii.html import HTML | ||
import six | ||
|
||
from ..exceptions import InvalidQueryError | ||
|
@@ -24,6 +27,14 @@ | |
|
||
__all__ = ['Ibe', 'IbeClass'] | ||
|
||
if sys.version_info[0] >= 3: | ||
decode = lambda x: x.decode() | ||
encode = lambda x: bytes(x, "ascii") | ||
|
||
else: | ||
decode = lambda x: x | ||
encode = lambda x: x | ||
|
||
|
||
class IbeClass(BaseQuery): | ||
URL = conf.server | ||
|
@@ -107,7 +118,8 @@ def query_region( | |
# Raise exception, if request failed | ||
response.raise_for_status() | ||
|
||
return Table.read(response.text, format='ipac', guess=False) | ||
return commons.parse_votable( | ||
encode(response.text)).get_first_table().to_table() | ||
|
||
def query_region_sia(self, coordinate=None, mission=None, | ||
dataset=None, table=None, width=None, | ||
|
@@ -156,8 +168,10 @@ def query_region_async( | |
SQL-like query string. Required if ``coordinates`` is absent. | ||
mission : str | ||
The mission to be used (if not the default mission). | ||
Note: this kwarg is not actually used here. | ||
odysseus9672 marked this conversation as resolved.
Show resolved
Hide resolved
|
||
dataset : str | ||
The dataset to be used (if not the default dataset). | ||
Note: this option is meaningless to the ibe service. | ||
table : str | ||
The table to be queried (if not the default table). | ||
columns : str, list | ||
|
@@ -195,7 +209,8 @@ def query_region_async( | |
The action to perform at the server. The default is ``'search'``, | ||
which returns a table of the available data. ``'data'`` requires | ||
advanced path construction that is not yet supported. ``'sia'`` | ||
provides access to the 'simple image access' IVOA protocol | ||
is here for legacy reasons. This function is built around IRSA's | ||
VO TAP API. IRSA's SIA API is completely different. | ||
|
||
Returns | ||
------- | ||
|
@@ -220,8 +235,16 @@ def query_region_async( | |
"The action='data' option is a placeholder for future " + | ||
"functionality.") | ||
|
||
if action == "sia": | ||
raise ValueError( | ||
"The action='sia' is not implemented for IRSA's IBE " | ||
"interface because IRSA's SIA interface is radically " | ||
"different.") | ||
|
||
args = { | ||
'INTERSECT': intersect | ||
'INTERSECT': intersect, | ||
'TABLE': table or self.TABLE, | ||
'RESPONSEFORMAT': "VOTABLE" | ||
} | ||
|
||
# Note: in IBE, if 'mcen' argument is present, it is true. | ||
|
@@ -247,14 +270,11 @@ def query_region_async( | |
columns = columns.split() | ||
args['columns'] = ','.join(columns) | ||
|
||
url = "{URL}{action}/{mission}/{dataset}/{table}".format( | ||
URL=self.URL, | ||
action=action, | ||
mission=mission or self.MISSION, | ||
dataset=dataset or self.DATASET, | ||
table=table or self.TABLE) | ||
url = self.URL.rstrip("/") | ||
|
||
response = self._request('GET', url, args, timeout=self.TIMEOUT) | ||
|
||
return self._request('GET', url, args, timeout=self.TIMEOUT) | ||
return response | ||
|
||
def list_missions(self, cache=True): | ||
""" | ||
|
@@ -276,14 +296,17 @@ def list_missions(self, cache=True): | |
|
||
root = BeautifulSoup(response.text) | ||
links = root.findAll('a') | ||
missions = [os.path.basename(a.attrs['href']) for a in links] | ||
|
||
missions = [os.path.basename(a.attrs['href'].rstrip('/')) | ||
for a in links] | ||
self._missions = missions | ||
|
||
return missions | ||
|
||
def list_datasets(self, mission=None, cache=True): | ||
""" | ||
For a given mission, list the available datasets | ||
For a given mission, list the available datasets. | ||
This level has no meaning in IRSA's ibe service. | ||
|
||
Parameters | ||
---------- | ||
|
@@ -299,24 +322,8 @@ def list_datasets(self, mission=None, cache=True): | |
datasets : list | ||
A list of dataset names | ||
""" | ||
if mission is None: | ||
mission = self.MISSION | ||
if mission not in self.list_missions(): | ||
raise ValueError("Invalid mission specified: {0}." | ||
"Must be one of: {1}" | ||
.format(mission, self.list_missions())) | ||
|
||
url = "{URL}search/{mission}/".format(URL=self.URL, mission=mission) | ||
response = self._request('GET', url, timeout=self.TIMEOUT, | ||
cache=cache) | ||
|
||
root = BeautifulSoup(response.text) | ||
links = root.findAll('a') | ||
datasets = [a.text for a in links | ||
if a.attrs['href'].count('/') >= 4 # shown as '..'; ignore | ||
] | ||
|
||
return datasets | ||
return ["images"] | ||
|
||
def list_tables(self, mission=None, dataset=None, cache=True): | ||
""" | ||
|
@@ -358,9 +365,7 @@ def list_tables(self, mission=None, dataset=None, cache=True): | |
.format(dataset, mission, | ||
self.list_datasets(mission, cache=True))) | ||
|
||
url = "{URL}search/{mission}/{dataset}/".format(URL=self.URL, | ||
mission=mission, | ||
dataset=dataset) | ||
url = "{URL}search/{mission}/".format(URL=self.URL, mission=mission) | ||
response = self._request('GET', url, timeout=self.TIMEOUT, | ||
cache=cache) | ||
|
||
|
@@ -386,10 +391,9 @@ def show_docs(self, mission=None, dataset=None, table=None): | |
The table to be queried (if not the default table). | ||
""" | ||
|
||
url = "{URL}docs/{mission}/{dataset}/{table}".format( | ||
url = "{URL}docs/{mission}/{table}".format( | ||
URL=self.URL, | ||
mission=mission or self.MISSION, | ||
dataset=dataset or self.DATASET, | ||
table=table or self.TABLE) | ||
|
||
return webbrowser.open(url) | ||
|
@@ -413,19 +417,20 @@ def get_columns(self, mission=None, dataset=None, table=None): | |
A table containing a description of the columns | ||
""" | ||
|
||
url = "{URL}search/{mission}/{dataset}/{table}".format( | ||
url = "{URL}search/{mission}/{table}/".format( | ||
URL=self.URL, | ||
mission=mission or self.MISSION, | ||
dataset=dataset or self.DATASET, | ||
table=table or self.TABLE) | ||
|
||
response = self._request( | ||
'GET', url, {'FORMAT': 'METADATA'}, timeout=self.TIMEOUT) | ||
'GET', url, {'FORMAT': 'METADATA'}, | ||
timeout=self.TIMEOUT) | ||
|
||
# Raise exception, if request failed | ||
response.raise_for_status() | ||
HTMLtoTable = HTML() | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. It would be nicer to use the |
||
|
||
return Table.read(response.text, format='ipac', guess=False) | ||
return HTMLtoTable.read(response.text) | ||
|
||
|
||
Ibe = IbeClass() |
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.