Awesome
pbcommand High Level Overview
co-owners:
PacBio Officially Supported Library. Note the APIs are still in flux and not yet 1.0.0.
Components
PacBio library for common utils, models, and tools to interface with pbsmrtpipe workflow engine.
- Common Models and Schemas
- Service client layer to the SMRTLink services
- Tool Contract and Resolved Tool Contract interface for integrating with pbsmrtpipe and SMRT Link
PacBio Core Models and Schemas
- ToolContract : Used in define task interfaces and used in pipeline definitions (i.e., Resolved Pipeline Templates)
- ResolvedToolContract: Used in pipeline running
- PacBio Report Core data model for plots, tables and metrics (i.e., attributes) displayed in SMRTLink or available as output from SMRT Link web services) Examples
- PacBio DataStore: JSON file of datastore file(s) that are emitted from an analysis, import-dataset or other job type.
- DataStore View Rules Custom views of datastore files in SMRTLink)
- [TODO] Pipeline Template View Rules: Custom hiding and renaming of pipeline task options for SMRTLink UI) Examples
- [TODO] Resolved Pipeline Template: Custom description of pipeline, bindings and task options for SMRTLink UI Examples
- [TODO] Report View Rules Examples
An HTML view of these models can be generated by using AvroDoc
The Avro schema format can be converted to JSON Schema using Avro To Json Schema
Service Client Layer to SMRT Link
pbcommand provides a high level interface to the SMRT Link services. See pbcommand.services for more details.
Here's a terse example of getting a Job by id and fetching the report metrics from the results.
IPython 5.1.0 -- An enhanced Interactive Python.
? -> Introduction and overview of IPython's features.
%quickref -> Quick reference.
help -> Python's own help system.
object? -> Details about 'object', use 'object??' for extra details.
In [1]: from pbcommand.services import ServiceAccessLayer
In [2]: s = ServiceAccessLayer("smrtlink-beta", 8081)
In [3]: j = s.get_analysis_job_by_id(22270)
In [4]: j.id, j.name, j.path
Out[4]:
(22270,
'm54088_160819_000654_resequencing',
'/pbi/dept/secondary/siv/smrtlink/smrtlink-beta/smrtsuite_166987/userdata/jobs_root/022/022270')
In [5]: j
Out[5]: ServiceJob(id=22270, uuid=u'4443d0b6-899c-40b9-98f2-2e2b4f889b53', name='m54088_160819_000654_resequencing', state='SUCCESSFUL', path='/pbi/dept/secondary/siv/smrtlink/smrtlink-beta/smrtsuite_166987/userdata/jobs_root/022/022270', job_type='pbsmrtpipe', created_at=datetime.datetime(2016, 8, 20, 1, 25, 50, 874000, tzinfo=<iso8601.Utc>), settings={u'workflowOptions': [], u'entryPoints': [{u'fileTypeId': u'PacBio.DataSet.ReferenceSet', u'entryId': u'eid_ref_dataset', u'datasetId': 9898}, {u'fileTypeId': u'PacBio.DataSet.SubreadSet', u'entryId': u'eid_subread', u'datasetId': 35547}], u'pipelineId': u'pbsmrtpipe.pipelines.sa3_ds_resequencing_fat', u'taskOptions': [], u'name': u'm54088_160819_000654_resequencing'})
In [6]: report_metrics = s.get_analysis_job_report_attrs(22270)
In [7]: report_metrics
Out[7]:
{u'coverage.depth_coverage_mean': 11251.832147121406,
u'coverage.missing_bases_pct': 0.004559547692868867,
u'mapping_stats.mapped_readlength_max': 15478,
u'mapping_stats.mapped_readlength_mean': 3899,
u'mapping_stats.mapped_readlength_n50': 5844,
u'mapping_stats.mapped_readlength_q95': 9560,
u'mapping_stats.mapped_reads_n': 207986,
u'mapping_stats.mapped_subread_bases_n': 810462596,
u'mapping_stats.mapped_subread_concordance_mean': 0.8354,
u'mapping_stats.mapped_subread_readlength_max': 12846.0,
u'mapping_stats.mapped_subread_readlength_mean': 3823,
u'mapping_stats.mapped_subreadlength_n50': 5799,
u'mapping_stats.mapped_subreadlength_q95': 9530,
u'mapping_stats.mapped_subreads_n': 212005,
u'variants.longest_contig_name': u'11k_pbell_H1-6_ScaI_circular_3x_l65796',
u'variants.mean_contig_length': 65796.0,
u'variants.weighted_mean_bases_called': 0.9999544045230713,
u'variants.weighted_mean_concordance': 0.9999696016293528,
u'variants.weighted_mean_coverage': 11251.832147121406}
In [8]:
Tool Contract and Resolved Tool Contracts
To integrate with the pbsmrtpipe workflow engine you must to be able to generate a Tool Contract and to be able to run from a Resolved Tool Contract.
A Tool Contract contains the metadata of the exe, such as the file types of inputs, outputs and options.
Example Tool Contract Json (and Avro Schema)
Example Resolved Tool Contract Json (and Avro Schema)
There are two principle use cases, first wrapping/calling python functions that have been defined in external python packages, or scripts. Second, creating a CLI tool that supports emitting tool contracts, running resolved tool contracts and complete argparse style CLI.
Example from pbcommand.cli.examples
import sys
import logging
from pbcommand.models import FileTypes
from pbcommand.cli import registry_builder, registry_runner
log = logging.getLogger(__name__)
registry = registry_builder("pbcommand", "python -m pbcommand.cli.examples.dev_quick_hello_world ")
def _example_main(input_files, output_files, **kwargs):
# Simple Function that should imported from your library code
# write mock output files for testing purposes, otherwise the End-to-End test will fail
xs = output_files if isinstance(output_files, (list, tuple)) else [output_files]
for x in xs:
with open(x, 'w') as writer:
writer.write("Mock data\n")
return 0
@registry("dev_qhello_world", "0.2.1", FileTypes.FASTA, FileTypes.FASTA, nproc=1, options=dict(alpha=1234))
def run_rtc(rtc):
return _example_main(rtc.task.input_files[0], rtc.task.output_files[0], nproc=rtc.task.nproc)
@registry("dev_fastq2fasta", "0.1.0", FileTypes.FASTQ, FileTypes.FASTA)
def run_rtc(rtc):
return _example_main(rtc.task.input_files[0], rtc.task.output_files[0])
if __name__ == '__main__':
sys.exit(registry_runner(registry, sys.argv[1:]))
A driver is the commandline interface that the workflow engine will call.
The driver will be called with "${exe} /path/to/resolved_tool_contract.json"
The tool contracts can be emitted to a directory and used in pbsmrtpipe.
$> python -m pbcommand.cli.examples.dev_quick_hello_world -o /path/to/my-tool-contracts
Creating a Full Commandline Tool with TC/RTC and argparse support
Three Steps
- define Parser using
get_pbparser - add running from argparse and running from Resolved ToolContract funcs to call your main
- add call to driver
Import or define your main function from your library.
def run_my_main(fasta_in, fasta_out, min_length):
# do work. Main should return an int exit code and be completely independent of argparse
return 0
Define a function that will add inputs, outputs and options to your parser.
from pbcommand.models import FileTypes
def add_args_and_options(p):
# FileType, label, name, description
p.add_input_file_type(FileTypes.FASTA, "fasta_in", "Fasta File", "PacBio Spec'ed fasta file")
# File Type, label, name, description, default file name
p.add_output_file_type(FileTypes.FASTA, "fasta_out", "Filtered Fasta file", "Filtered Fasta file", "filter.fasta")
# Option id, label, default value, name, description
# for the argparse, the read-length will be translated to --read-length and (accessible via args.read_length)
p.add_int("pbcommand.task_options.dev_read_length", "read-length", 25, "Length filter", "Min Sequence Length filter")
return p
Define Parser
from pbcommand.models import TaskTypes, SymbolTypes, get_pbparser
def get_contract_parser():
tool_id = "example_namespace.tasks.my_id"
version = "0.1.0" # or reuse __version__
display_name = "My Example Tool"
# Number of processors to use, can also be SymbolTypes.MAX_NPROC
nproc = 1
# Log file, tmp dir, tmp file. See ResourceTypes in models, ResourceTypes.TMP_DIR
resource_types = ()
# Commandline exe to call "{exe}" /path/to/resolved-tool-contract.json
driver_exe = "python -m pbcommand.cli.example.dev_app --resolved-tool-contract "
desc = "Dev app for Testing that supports emitting tool contracts"
is_distributed = False
# set to True if you want your task to be submitted to the cluster manager (e.g., SGE) if
# one is provided to the workflow engine.
p = get_pbparser(tool_id, version, display_name, desc, driver_exe, is_distributed=is_distributed, nproc=nproc, resource_types=resource_types)
add_args_and_options(p)
return p
Define a Wrapping IO layer to call your main function from both the tool contract and raw argparse IO layer
def _args_runner(args):
# this is the args from parser.parse_args() using the python stdlib argparse model
# the properties of args are defined as "labels" in the add_args_and_options func.
return run_my_main(args.fasta_in, args.fasta_out, args.read_length)
def _resolved_tool_contract_runner(resolved_tool_contract):
"""
:type resolved_tool_contract: pbcommand.models.ResolvedToolContract"""
rtc = resolved_tool_contract
# all options are referenced by globally namespaced id. This allows tools to use other tools options
# e.g., pbalign to use blasr defined options.
return run_my_main(rtc.task.input_files[0], rtc.task.outputs[0], rtc.task.options["pbcommand.task_options.dev_read_length"])
Add running layer
import sys
import logging
from pbcommand.utils import setup_log
from pbcommand.cli import pbparser_runner
log = logging.getLogger(__name__)
def main(argv=sys.argv):
# New interface that supports running resolved tool contracts
log.info("Starting {f} version {v} pbcommand example dev app".format(f=__file__, v="0.1.0"))
return pbparser_runner(argv[1:],
get_contract_parser(),
_args_runner, # argparse runner func
_resolved_tool_contract_runner, # tool contract runner func
log, # log instance
setup_log # setup log func
)
if __name__ == '__main__':
sys.exit(main())
Now you can run your tool via the argparse standard interface as well as emitting a Tool Contract to stdout from the commandline interface.
> python -m 'pbcommand.cli.examples.dev_app' --emit-tool-contract
And you can run the tool from a Resolved Tool Contract
> python -m pbcommand.cli.examples.dev_app --resolved-tool-contract /path/to/resolved_contract.json
See the dev apps in "pbcommand.cli.examples" for a complete application (They require pbcore to be installed).
In addition to TC/RTC support, there's a complete argparse support for the task options. An example of help is shown below.
$> python -m pbcommand.cli.examples.dev_app --help
usage: dev_app.py [-h] [-v] [--versions] [--emit-tool-contract]
[--resolved-tool-contract RESOLVED_TOOL_CONTRACT]
[--log-level LOG_LEVEL] [--debug]
[--read-length READ_LENGTH]
fasta_in fasta_out
Dev app for Testing that supports emitting tool contracts
positional arguments:
fasta_in PacBio Spec'ed fasta file
fasta_out Filtered Fasta file
optional arguments:
-h, --help show this help message and exit
-v, --version show program's version number and exit
--versions Show versions of individual components (default: None)
--emit-tool-contract Emit Tool Contract to stdout (default: False)
--resolved-tool-contract RESOLVED_TOOL_CONTRACT
Run Tool directly from a PacBio Resolved tool contract
(default: None)
--log-level LOG_LEVEL
Set log level (default: 10)
--debug Debug to stdout (default: False)
--read-length READ_LENGTH
Min Sequence Length filter (default: 25)
DISCLAIMER
THIS WEBSITE AND CONTENT AND ALL SITE-RELATED SERVICES, INCLUDING ANY DATA, ARE PROVIDED "AS IS," WITH ALL FAULTS, WITH NO REPRESENTATIONS OR WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTIES OF MERCHANTABILITY, SATISFACTORY QUALITY, NON-INFRINGEMENT OR FITNESS FOR A PARTICULAR PURPOSE. YOU ASSUME TOTAL RESPONSIBILITY AND RISK FOR YOUR USE OF THIS SITE, ALL SITE-RELATED SERVICES, AND ANY THIRD PARTY WEBSITES OR APPLICATIONS. NO ORAL OR WRITTEN INFORMATION OR ADVICE SHALL CREATE A WARRANTY OF ANY KIND. ANY REFERENCES TO SPECIFIC PRODUCTS OR SERVICES ON THE WEBSITES DO NOT CONSTITUTE OR IMPLY A RECOMMENDATION OR ENDORSEMENT BY PACIFIC BIOSCIENCES.