Command line interface (CLI) and application program interface (API) to allow users to search, download from and upload to HEPData.
The code is compatible with both Python 2 and Python 3. Inspiration from arxiv-cli.
Install from PyPI using pip
:
$ pip install --user hepdata-cli
$ hepdata-cli --help
With Python 3 (<3.7), if the LANG
environment variable is not set, you might get an error like:
RuntimeError: Click will abort further execution because Python 3 was configured to use ASCII as encoding for the environment. Consult https://click.palletsprojects.com/python3/ for mitigation steps.
In this case, you will need to export a Unicode locale as described in the Click documentation.
Install from GitHub in a virtual environment:
$ git clone https://github.com/HEPData/hepdata-cli.git
$ cd hepdata-cli
$ python3 -m venv ~/venv/hepdata-cli
$ source ~/venv/hepdata-cli/bin/activate
(hepdata-cli) $ pip install -e '.[tests]'
(hepdata-cli) $ hepdata-cli --help
(hepdata-cli) $ pytest --cov=hepdata_cli
You can use HEPData-CLI both as a command-line interface (CLI) to search, download and upload records from/to the HEPData database, or as a Python library to perform the same operations via its application program interface (API).
$ hepdata-cli [-v/--version, --help]
$ hepdata-cli [--verbose] find [QUERY] [-kw/--keyword KEYWORD] [-i/--ids IDTYPE]
$ hepdata-cli [--verbose] download [IDS] [-f/--file-format FORMAT] [-i/--ids IDTYPE] [-t/--table-name TABLE-NAME] [-d/--download-dir DOWNLOAD-DIR]
$ hepdata-cli [--verbose] fetch-names [IDS] [-i/--ids IDTYPE]
$ hepdata-cli [--verbose] upload [PATH-TO-FILE-ARCHIVE] [-e/--email YOUR-EMAIL] [-r/--recid RECORD-ID] [-i/--invitation-cookie COOKIE] [-s/--sandbox TRUE/FALSE] [-p/--password PASSWORD]
The command find
searches the HEPData database for matches of QUERY
. The advanced search syntax from the website can be used.
The command download
downloads records from the database (see options below).
The command fetch-names
returns the names of the data tables in the records whose ids are supplied.
The command upload
uploads a file to the HEPData web site as either a sandbox or normal record.
The argument [-kw/--keyword KEYWORD]
filters the search result dictionary for specific keywords.
An exact match of the keyword is first attempted, otherwise partial matches are accepted.
The argument [-i/--ids IDTYPE]
accepts IDTYPE
equal to arxiv
, hepdata
orinspire
.
The argument [-f/--file-format FORMAT]
accepts FORMAT
equal to csv
, root
, yaml
, yoda
, yoda1
, or json
.
In the first four cases a .tar.gz archive is downloaded and unpacked as a directory, whereas in the last case a .json file is downloaded.
The argument [-t/--table-name TABLE-NAME]
accepts a string giving the table name as input.
In this case only the specified table is downloaded as a .csv, .root, .yaml, .yoda, .yoda1 or .json file.
The argument [-d/--download-dir DOWNLOAD-DIR]
specifies the directory to download the files.
If not specified, the default download directory is ./hepdata-downloads
.
The argument [-e/--email YOUR-EMAIL]
is the uploader's email, needed to associate the submission to their HEPData account.
The argument [-i/--invitation-cookie COOKIE]
must be supplied for non-sandbox submissions.
This can be found in the Uploader invitation email received at the beginning of the submission process.
The argument [-s/--sandbox TRUE/FALSE]
is a boolean to decide whether to upload to the sandbox or not.
The argument [-p/--password PASSWORD
is the password for the uploader's HEPData account (prompt if not specified).
Warning: do not store your password unencrypted in any code intended for shared use.
The hepdata-cli download/fetch-names
and hepdata-cli find
commands can be concatenated, if a IDTYPE
is specified for find
.
It is also possible to concatenate arxiv download
, form pypi/arxiv-cli, with hepdata-cli find
, if arxiv
is used as IDTYPE
.
Equivalently to the above, these commands can be invoked by the API (in fact, the CLI is just a wrapper around the API).
from hepdata_cli.api import Client
client = Client(verbose=True)
client.find(query, keyword, ids)
client.download(id_list, file_format, ids, table_name, download_dir)
client.fetch_names(id_list, ids)
client.upload(path_to_file, email, recid, invitation_cookie, sandbox, password)
$ hepdata-cli --verbose find 'reactions:"P P--> LQ LQ X"'
or equivalently
client.find('reactions:"P P--> LQ LQ X"')
matches a single entry and returns full metadata dictionary.
$ hepdata-cli --verbose find 'reactions:"P P--> LQ LQ"' -kw year
or equivalently
client.find('reactions:"P P--> LQ LQ"', keyword='year')
matches four entries and returns their publication years, as a dictionary.
$ hepdata-cli --verbose find 'reactions:"P P--> LQ LQ"' -i hepdata
or equivalently
client.find('reactions:"P P--> LQ LQ"', ids='hepdata')
matches four entries and returns their hepdata ids, as a plain list.
$ hepdata-cli --verbose download $(hepdata-cli find 'reactions:"P P--> LQ LQ"' -i inspire) -i inspire -f csv
or equivalently
id_list = client.find('reactions:"P P--> LQ LQ"', ids='inspire')
client.download(id_list, ids='inspire', file_format='csv')
downloads four .tar.gz archives containing csv files and unpacks them in the default ./hepdata-downloads
directory.
$ hepdata-cli fetch-names $(hepdata-cli find 'reactions:"P P--> LQ LQ"' -i hepdata) -i hepdata
or equivalently
id_list = client.find('reactions:"P P--> LQ LQ"', ids='hepdata')
client.fetch_names(id_list, ids='hepdata')
returns all table names in the four matching records.
This example requires arxiv-cli to be installed, which is easily done via:
$ pip install --user arxiv-cli
Note that arxiv-cli installs an older version of click which changes the CLI command
in Example 5 above from fetch-names
to fetch_names
.
Then,
$ arxiv download $(hepdata-cli find 'reactions:"P P--> LQ LQ"' -i arxiv)
or equivalently
import arxiv_cli
import hepdata_cli
arxiv_client = arxiv_cli.Client()
hepdata_client = hepdata_cli.Client()
id_list = hepdata_client.find('reactions:"P P--> LQ LQ"', ids='arxiv')
arxiv_client.download(id_list)
downloads two pdfs from the arXiv.
$ hepdata-cli upload /path/to/TestHEPSubmission.tar.gz -e [email protected] -s True
or equivalently
client.upload('/path/to/TestHEPSubmission.tar.gz', email='[email protected]', sandbox=True)
The uploaded submission can then be found from your sandbox. You will be prompted for the password associated with your HEPData account. If your account was created with CERN or ORCID authentication, you will first need to set a password.
$ hepdata-cli upload /path/to/TestHEPSubmission.tar.gz -e [email protected] -r 1234567890 -s True
or equivalently
client.upload('/path/to/TestHEPSubmission.tar.gz', email='[email protected]', recid='1234567890', sandbox=True)
Note that you must have uploaded the original sandbox record yourself and that you will be prompted for a password.
$ hepdata-cli upload /path/to/TestHEPSubmission.tar.gz -e [email protected] -r 123456 -i 8232e07f-d1d8-4883-bb1d-77fd9994ce4f -s False
or equivalently
client.upload('/path/to/TestHEPSubmission.tar.gz', email='[email protected]', recid='123456', invitation_cookie='8232e07f-d1d8-4883-bb1d-77fd9994ce4f', sandbox=False)
The uploaded submission can then be found from your Dashboard.
The invitation cookie is sent in your original invitation email.
You must have already claimed permissions by clicking the link in that email or from your Dashboard.
Again, you will be prompted for a password, which must be set if using CERN/ORCID login.
The password can alternatively be passed as an argument to the CLI (-p PASSWORD
) or API (password=PASSWORD
).
However, please be careful to keep your password secure, for example, by defining an encrypted environment variable for a CI/CD workflow.