This library covers the UNICORE REST API, making common tasks like file access, job submission and management, workflow submission and management more convenient, and integrating UNICORE features better with typical Python usage.
The full, up-to-date documentation of the REST API can be found here
In addition, this library contains code for using UFTP (UNICORE FTP) for filesystem mounts with FUSE, a UFTP driver for PyFilesystem and a UNICORE implementation of a Dask Cluster
This project has received funding from the European Union’s Horizon 2020 Framework Programme for Research and Innovation under the Specific Grant Agreement Nos. 720270, 785907 and 945539 (Human Brain Project SGA 1, 2 and 3)
See LICENSE file for licensing information
The complete documentation of PyUNICORE can be viewed here
Install from PyPI with
pip install -U pyunicore
Additional extra packages may be required for your use case:
- Using the UFTP fuse driver requires "fusepy"
- Using UFTP with pyfilesystem requires "fs"
- Creating JWT tokens signed with keys requires the "cryptography" package
You can install (one or more) extras with pip:
pip install -U pyunicore[crypto,fs,fuse]
import pyunicore.client as uc_client
import pyunicore.credentials as uc_credentials
import json
base_url = "https://localhost:8080/DEMO-SITE/rest/core"
# authenticate with username/password
credential = uc_credentials.UsernamePassword("demouser", "test123")
client = uc_client.Client(credential, base_url)
print(json.dumps(client.properties, indent = 2))
PyUNICORE supports a variety of authentication options.
my_job = {'Executable': 'date'}
job = client.new_job(job_description=my_job, inputs=[])
print(json.dumps(job.properties, indent = 2))
job.poll() # wait for job to finish
work_dir = job.working_dir
print(json.dumps(work_dir.properties, indent = 2))
stdout = work_dir.stat("/stdout")
print(json.dumps(stdout.properties, indent = 2))
content = stdout.raw().read()
print(content)
registry_url = "https://localhost:8080/REGISTRY/rest/registries/default_registry"
# authenticate with username/password
credential = uc_credentials.UsernamePassword("demouser", "test123")
r = uc_client.Registry(credential, registry_url)
print(r.site_urls)
More examples for using PyUNICORE can be found in the "integration-tests" folder in the source code repository.
You can create a PyFilesystem FS
object either directly in code, or implicitely via a URL.
The convenient way is via URL:
from fs import open_fs
fs_url = "uftp://demouser:test123@localhost:9000/rest/auth/TEST:/data"
uftp_fs = open_fs(fs_url)
PyUNICORE contains a FUSE driver based on fusepy, allowing you to mount a remote filesystem via UFTP. Mounting is a two step process,
- authenticate to an Auth server, giving you the UFTPD host/port and one-time password
- run the FUSE driver
Opens a local server socket for clients to connect to, where traffic gets forwarded to a service on a HPC cluster login (or compute) node. This feature requires UNICORE 9.1.0 or later on the server side.
You can use this feature in two ways
- in your own applications via the
pyunicore.client.Job
class. - you can also open a tunnel from the command line using the 'pyunicore.forwarder' module
PyUNICORE provides an implementation of a Dask Cluster, allowing to run the Dask client on your local host (or in a Jupyter notebook in the Cloud), and have the Dask scheduler and workers running remotely on the HPC site.
PyUNICORE provides a tool to convert a CWL CommanLineTool and input into a UNICORE job file. Given the following YAML files that describe a CommandLineTool wrapper for the echo command and an input file:
# echo.cwl
cwlVersion: v1.2
class: CommandLineTool
baseCommand: echo
inputs:
message:
type: string
inputBinding:
position: 1
outputs: []
# hello_world.yml
message: "Hello World"
A UNICORE job file can be generated using the following command:
unicore-cwl-runner echo.cwl hello_world.yml > hello_world.u
The pyunicore.helpers
module provides helper code for:
- Connecting to
- a Registry (
pyunicore.helpers.connect_to_registry
). - a site via a Registry URL (
pyunicore.helpers.connect_to_site_from_registry
). - a site via its core URL (
pyunicore.helpers.connect_to_site
).
- a Registry (
- Defining descriptions as a dataclass and easily converting to a
dict
as required bypyunicore.client.Client.new_job
via ato_dict()
method:-
pyunicore.helpers.jobs.Description
forpyunicore.client.Client.new_job()
-
pyunicore.helpers.workflows.Description
forpyunicore.client.WorkflowService.new_workflow()
-
- Defining a workflow description
import json
import pyunicore.credentials as uc_credentials
import pyunicore.helpers as helpers
registry_url = "https://localhost:8080/REGISTRY/rest/registries/default_registry"
credentials = uc_credentials.UsernamePassword("demouser", "test123")
client = helpers.connection.connect_to_registry(
registry_url=registry_url,
credentials=credentials,
)
print(json.dumps(client.properties, indent=2))
import json
import pyunicore.credentials as uc_credentials
import pyunicore.helpers as helpers
registry_url = "https://localhost:8080/REGISTRY/rest/registries/default_registry"
site = "DEMO-SITE"
credentials = uc_credentials.UsernamePassword("demouser", "test123")
client = helpers.connection.connect_to_site_from_registry(
registry_url=registry_url,
site_name=site,
credentials=credentials,
)
print(json.dumps(client.properties, indent=2))
import json
import pyunicore.credentials as uc_credentials
import pyunicore.helpers as helpers
site_url = "https://localhost:8080/DEMO-SITE/rest/core"
credentials = uc_credentials.UsernamePassword("demouser", "test123")
client = helpers.connection.connect_to_site(
site_api_url=site_url ,
credentials=credentials,
)
print(json.dumps(client.properties, indent=2))
from pyunicore import helpers
client = ...
resources = helpers.jobs.Resources(nodes=4)
job = helpers.jobs.Description(
executable="ls",
project="demoproject",
resources=resources
)
client.new_job(job.to_dict())
This works analogously for pyunicore.helpers.workflows
.
-
Fork the repository
-
Install the development dependencies
pip install -r requirements-dev.txt
-
Install pre-commit hooks
pre-commit install