transmart-packer

Data transformation jobs for TranSMART


Keywords
transmart-packer, transmart, export
Licenses
GPL-3.0/GPL-3.0+
Install
pip install transmart-packer==0.7.0

Documentation

transmart-packer

Build status codecov PyPI PyPI - Downloads

Run data transformation jobs for TranSMART.

Install

First make virtual environment to install dependencies using Python 3.7+

pip install transmart-packer

Or from source:

git clone https://github.com/thehyve/transmart-packer.git
cd transmart-packer
pip install .

Dependencies

  • a Redis server running on localhost (or update packer/config.py)

Running

From root dir run:

redis-server

celery -A packer.tasks worker --loglevel=info

transmart-packer

Environment variables:

Variable Description
TRANSMART_URL The URL of the TranSMART API server
KEYCLOAK_SERVER_URL Keycloak server URL, e.g., https://keycloak-dwh-test.thehyve.net/auth
KEYCLOAK_REALM The Keycloak realm (default: transmart)
KEYCLOAK_CLIENT_ID The Keycloak client ID (default: transmart-client)
KEYCLOAK_OFFLINE_TOKEN The Keycloak offline token.
REDIS_URL Redis server URL (default: redis://localhost:6379)
DATA_DIR Directory to write export data (default: /tmp/packer/)
LOG_CFG Logging configuration (default: packer/logging.yaml)
CLIENT_ORIGIN_URL URLs to restrict cross-origin requests to (CORS) (default: *)

An optional variable VERIFY_CERT can be used to specify the path of a certificate collection file (.pem) used to verify HTTP requests.

KEYCLOAK_OFFLINE_TOKEN should be generated for a system user that has the following roles:

  • realm role offline_access – to be able to get the offline token.
  • client role impersonation on the realm-management client – to support running tranSMART queries on behalf of task users.

To get the token, run:

KEYCLOAK_CLIENT_ID=transmart-client
SYSTEM_USERNAME=system
SYSTEM_PASSWORD=choose-a-strong-system-password # CHANGE ME
KEYCLOAK_SERVER_URL=https://keycloak.example.com/auth
KEYCLOAK_REALM=example
curl -f --no-progress-meter \
  -d "client_id=${KEYCLOAK_CLIENT_ID}" \
  -d "username=${SYSTEM_USERNAME}" \
  -d "password=${SYSTEM_PASSWORD}" \
  -d "grant_type=password" \
  -d "scope=offline_access" \
  "${KEYCLOAK_SERVER_URL}/realms/${KEYCLOAK_REALM}/protocol/openid-connect/token" | jq -r '.refresh_token'

The value of the refresh_token field in the response is the offline token.

To run the stack using docker-compose follow the commands below:

# Downloads redis image and creates image with project dependencies.
docker-compose build

# After build is complete, start via:
docker-compose up

On code change the webserver will automatically restart, but the Celery workers will not. After updating the Celery task logic, you will need to restart the Docker container.

Usage

Available handlers:

Path Description
GET /jobs List all known jobs for this user.
POST /jobs/create Create a new job by providing job_type and job_parameters, creates the job and returns a task_id.
GET /jobs/status/<task_id> Get status details for a specific task.
GET /jobs/cancel/<task_id> Cancel scheduled or abort a running task.
GET /jobs/data/<task_id> Download the data that this task produced.
WS /jobs/subscribe Open websocket connection to get live updates on job progress.

To start the toy job "add" on the localhost machine make call to http://localhost:8999/jobs/create?job_type=add&job_parameters={%22x%22:500,%22y%22:1501}.

Development

Components

Overview of the components of transmart-packer and the interaction with external components.

Overview of the components of transmart-packer and the interaction with external components.

Testing

To run the test suite, we have to start redis-server and celery workers with the commands above. Then you can run:

python setup.py test

tests/csr_observation.json - test data retrieved from TranSMART using the following API call:

curl -X POST -H 'Content-type: application/json' -H 'Accept: application/json' -d \
'{
    "type":"clinical",
    "constraint": {
        "type":"study_name",
        "studyId":"CSR"
    }
}' \
'<transmart_api_url>/v2/observations'

Current file is created based on clinical test data of python_csr2transmart, with ontology_config.json and sources_config.json as configuration. Note! Do not change csr_observation.json file manually.

Extending

New jobs can be added by adding a new Celery function to the jobs folder and adding the function to the jobs registry. See the packer/jobs/example.py to learn how.

Existing jobs

Basic export job

Export transmart api client observation dataframe to tsv file

{
    "job_type":"basic_export",
    "job_parameters": {
        "constraint": {
            "type":"study_name",
            "studyId":"CSR"

        },
        "custom_name":"name of the export"
    }
}

CSR export

The Central Subject Registry (CSR) data model specific export. The model contains individual, diagnosis, biosource, biomaterial, radiology and study entities, following the hierarchy for sample data: patient > diagnosis > biosource > biomaterial. Studies are orthogonal to samples, i.e., patients are linked to studies independent of samples. Radiology, same as samples, is linked to patient, but can be also linked to diagnosis (optional). The entities IDs are first 6 columns of the export file. The rest of the columns are concepts. Higher level concepts (e.g., Age that is specific to Patient level) get distributed to all rows specific to lower levels (e.g. Diagnosis)

See the CSR test study as an example or latest sources dataset that can be used for e2e testing.

{
    "job_type":"csr_export",
    "job_parameters": {
        "constraint": {
            "type":"study_name",
            "studyId":"CSR"

        },
        "custom_name":"name of the export",
        "row_filter": {
            "type":"patient_set",
            "subjectIds": ["P2", "P6"]
        }
    }
}

where:

  • job_parameters.constraint - any transmart v2 api constraint or composition of them that used to get data from transmart.
  • job_parameters.custom_name (optional) - name of the export job and the output tsv file.
  • job_parameters.row_filter (optional) - any transmart v2 api constraint or composition of them to fetch keys ([[[[patient], diagnosis], biosource], biomaterial]) that will make it to the end result. E.g., given the CSR study and query above only rows specific to P2 and P6 patients will end up to the result table such as P2, D2, BS2, BM2, ... row. Please note that keys do not have to be equal in length. A row gets also selected if only part of keys matches. e.g. P1 vs P1, D1

Adding new entity to CSR data model:

When the CSR data model is extended with new sample related entities, the export transformation code has to be changed as well in order to include a column with the ID of the new entity as one of the identifying columns.

In order to do this, packer/table_transformations/csr_transformations.py file has to be modified. The ID_COLUMN_MAPPING map needs to be extended with the new dimension name of the new entity as key and the column name that should appear in the export as value.

If the new entity is not a part of the sample hierarchy, but only linked to one or more entities, the merging logic has to be added in transform_obs_df function in packer/table_transformations/csr_transformations.py (see the example of Radiology and Sample entities).

License

Copyright © 2019 The Hyve B.V.

This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License along with this program. If not, see https://www.gnu.org/licenses/.