htsworkflow/submission/trackhub.py

   1 import logging
   2 import os
   3
   4 import RDF
   5
   6 from htsworkflow.submission.submission import Submission
   7
   8 from htsworkflow.util.rdfhelp import \
   9      fromTypedNode, \
  10      geoSoftNS, \
  11      stripNamespace, \
  12      submissionOntology
  13
  14 from django.conf import settings
  15 from django.template import Context, loader
  16
  17 LOGGER = logging.getLogger(__name__)
  18
  19 class TrackHubSubmission(Submission):
  20     def __init__(self, name, model, host):
  21         super(TrackHubSubmission, self).__init__(name, model, host)
  22
  23     def make_hub(self, result_map):
  24         samples = []
  25         for lib_id, result_dir in result_map.items():
  26             an_analysis = self.get_submission_node(result_dir)
  27             metadata = self.get_sample_metadata(an_analysis)
  28             if len(metadata) == 0:
  29                 errmsg = 'No metadata found for {0}'
  30                 LOGGER.error(errmsg.format(str(an_analysis),))
  31                 continue
  32             elif len(metadata) > 1:
  33                 errmsg = 'Confused there are more than one sample for %s'
  34                 LOGGER.debug(errmsg % (str(an_analysis),))
  35             metadata = metadata[0]
  36             samples.append(metadata)
  37
  38         soft_template = loader.get_template('trackDb.txt')
  39         context = Context({
  40             'samples': samples,
  41         })
  42         return str(soft_template.render(context))
  43
  44     def make_mainifest(self, result_map):
  45         pass
  46
  47     def get_sample_metadata(self, analysis_node):
  48         """Gather information for filling out sample section of a SOFT file
  49         """
  50         query_template = loader.get_template('trackhub_samples.sparql')
  51
  52         context = Context({
  53             'submission': str(analysis_node.uri),
  54             'submissionSet': str(self.submissionSetNS[''].uri),
  55             })
  56
  57         results = self.execute_query(query_template, context)
  58         return results