kkonganti@0: #!/usr/bin/env python
kkonganti@0: 
kkonganti@0: import platform
kkonganti@0: import subprocess
kkonganti@0: from textwrap import dedent
kkonganti@0: 
kkonganti@0: import yaml
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _make_versions_html(versions):
kkonganti@0:     html = [
kkonganti@0:         dedent(
kkonganti@0:             """\\
kkonganti@0:             <link rel="stylesheet" type="text/css" href="https://cdn.datatables.net/v/dt/jszip-2.5.0/dt-1.12.1/b-2.2.3/b-colvis-2.2.3/b-html5-2.2.3/b-print-2.2.3/fc-4.1.0/r-2.3.0/sc-2.0.6/sb-1.3.3/sp-2.0.1/datatables.min.css"/>
kkonganti@0:             <script type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/pdfmake/0.1.36/pdfmake.min.js"></script>
kkonganti@0:             <script type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/pdfmake/0.1.36/vfs_fonts.js"></script>
kkonganti@0:             <script type="text/javascript" src="https://cdn.datatables.net/v/dt/jszip-2.5.0/dt-1.12.1/b-2.2.3/b-colvis-2.2.3/b-html5-2.2.3/b-print-2.2.3/fc-4.1.0/r-2.3.0/sc-2.0.6/sb-1.3.3/sp-2.0.1/datatables.min.js"></script>
kkonganti@0:             <style>
kkonganti@0:             #cpipes-software-versions tbody:nth-child(even) {
kkonganti@0:                 background-color: #f2f2f2;
kkonganti@0:             }
kkonganti@0:             </style>
kkonganti@0:             <table class="table" style="width:100%" id="cpipes-software-versions">
kkonganti@0:                 <thead>
kkonganti@0:                     <tr>
kkonganti@0:                         <th> Process Name </th>
kkonganti@0:                         <th> Software </th>
kkonganti@0:                         <th> Version  </th>
kkonganti@0:                     </tr>
kkonganti@0:                 </thead>
kkonganti@0:             """
kkonganti@0:         )
kkonganti@0:     ]
kkonganti@0:     for process, tmp_versions in sorted(versions.items()):
kkonganti@0:         html.append("<tbody>")
kkonganti@0:         for i, (tool, version) in enumerate(sorted(tmp_versions.items())):
kkonganti@0:             html.append(
kkonganti@0:                 dedent(
kkonganti@0:                     f"""\\
kkonganti@0:                     <tr>
kkonganti@0:                         <td><samp>{process if (i == 0) else ''}</samp></td>
kkonganti@0:                         <td><samp>{tool}</samp></td>
kkonganti@0:                         <td><samp>{version}</samp></td>
kkonganti@0:                     </tr>
kkonganti@0:                     """
kkonganti@0:                 )
kkonganti@0:             )
kkonganti@0:         html.append("</tbody>")
kkonganti@0:     html.append("</table>")
kkonganti@0:     return "\\n".join(html)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: versions_this_module = {}
kkonganti@0: versions_this_module["${task.process}"] = {
kkonganti@0:     "python": platform.python_version(),
kkonganti@0:     "yaml": yaml.__version__,
kkonganti@0: }
kkonganti@0: 
kkonganti@0: with open("$versions") as f:
kkonganti@0:     versions_by_process = yaml.load(f, Loader=yaml.BaseLoader)
kkonganti@0:     versions_by_process.update(versions_this_module)
kkonganti@0: 
kkonganti@0: # aggregate versions by the module name (derived from fully-qualified process name)
kkonganti@0: versions_by_module = {}
kkonganti@0: for process, process_versions in versions_by_process.items():
kkonganti@0:     module = process.split(":")[-1]
kkonganti@0:     try:
kkonganti@0:         assert versions_by_module[module] == process_versions, (
kkonganti@0:             "We assume that software versions are the same between all modules. "
kkonganti@0:             "If you see this error-message it means you discovered an edge-case "
kkonganti@0:             "and should open an issue in nf-core/tools. "
kkonganti@0:         )
kkonganti@0:     except KeyError:
kkonganti@0:         versions_by_module[module] = process_versions
kkonganti@0: 
kkonganti@0: versions_by_module["CPIPES"] = {
kkonganti@0:     "Nextflow": "$workflow.nextflow.version",
kkonganti@0:     "$workflow.manifest.name": "$workflow.manifest.version",
kkonganti@0:     "${params.pipeline}": "${params.workflow_version}",
kkonganti@0: }
kkonganti@0: 
kkonganti@0: versions_mqc = {
kkonganti@0:     "id": "software_versions",
kkonganti@0:     "section_name": "${workflow.manifest.name} Software Versions",
kkonganti@0:     "section_href": "https://cfsan-git.fda.gov/Kranti.Konganti/${workflow.manifest.name.toLowerCase()}",
kkonganti@0:     "plot_type": "html",
kkonganti@0:     "description": "Collected at run time from the software output (STDOUT/STDERR).",
kkonganti@0:     "data": _make_versions_html(versions_by_module),
kkonganti@0: }
kkonganti@0: 
kkonganti@0: with open("software_versions.yml", "w") as f:
kkonganti@0:     yaml.dump(versions_by_module, f, default_flow_style=False)
kkonganti@0: 
kkonganti@0: # print('sed -i -e "' + "s%'%%g" + '" *.yml')
kkonganti@0: subprocess.run('sed -i -e "' + "s%'%%g" + '" software_versions.yml', shell=True)
kkonganti@0: 
kkonganti@0: with open("software_versions_mqc.yml", "w") as f:
kkonganti@0:     yaml.dump(versions_mqc, f, default_flow_style=False)
kkonganti@0: 
kkonganti@0: with open("versions.yml", "w") as f:
kkonganti@0:     yaml.dump(versions_this_module, f, default_flow_style=False)