kkonganti@0: #!/usr/bin/env python
kkonganti@0: 
kkonganti@0: import os
kkonganti@0: import sys
kkonganti@0: from textwrap import dedent
kkonganti@0: 
kkonganti@0: import yaml
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def main():
kkonganti@0:     """
kkonganti@0:     Takes a tab-delimited text file with a mandatory header
kkonganti@0:     column and generates an HTML table.
kkonganti@0:     """
kkonganti@0: 
kkonganti@0:     args = sys.argv
kkonganti@0:     if len(args) < 2 or len(args) >= 4:
kkonganti@0:         print(
kkonganti@0:             f"\nAt least one argument specifying the *.tblsum file is required.\n"
kkonganti@0:             + "No more than 2 command-line arguments should be passed.\n"
kkonganti@0:         )
kkonganti@0:         exit(1)
kkonganti@0: 
kkonganti@0:     table_sum_on = str(args[1]).lower()
kkonganti@0:     table_sum_on_file = table_sum_on + ".tblsum.txt"
kkonganti@0:     cell_colors = f"{table_sum_on}.cellcolors.yml"
kkonganti@0: 
kkonganti@0:     if len(args) == 3:
kkonganti@0:         description = str(args[2])
kkonganti@0:     else:
kkonganti@0:         description = "The results table shown here is a collection from all samples."
kkonganti@0: 
kkonganti@0:     if os.path.exists(cell_colors) and os.path.getsize(cell_colors) > 0:
kkonganti@0:         with open(cell_colors, "r") as cc_yml:
kkonganti@0:             cell_colors = yaml.safe_load(cc_yml)
kkonganti@0:     else:
kkonganti@0:         cell_colors = dict()
kkonganti@0: 
kkonganti@0:     if not (os.path.exists(table_sum_on_file) and os.path.getsize(table_sum_on_file) > 0):
kkonganti@0:         exit(0)
kkonganti@0: 
kkonganti@0:     with open(table_sum_on_file, "r") as tbl:
kkonganti@0:         header = tbl.readline()
kkonganti@0:         header_cols = header.strip().split("\t")
kkonganti@0: 
kkonganti@0:         html = [
kkonganti@0:             dedent(
kkonganti@0:                 f"""<script type="text/javascript">
kkonganti@0:                     $(document).ready(function () {{
kkonganti@0:                         $('#cpipes-process-custom-res-{table_sum_on}').DataTable({{
kkonganti@0:                             scrollX: true,
kkonganti@0:                             fixedColumns: true, dom: 'Bfrtip',
kkonganti@0:                             buttons: [
kkonganti@0:                                 'copy',
kkonganti@0:                                 {{
kkonganti@0:                                     extend: 'print',
kkonganti@0:                                     title: 'CPIPES: MultiQC Report: {table_sum_on}'
kkonganti@0:                                 }},
kkonganti@0:                                 {{
kkonganti@0:                                     extend: 'excel',
kkonganti@0:                                     filename: '{table_sum_on}_results',
kkonganti@0:                                 }},
kkonganti@0:                                 {{
kkonganti@0:                                     extend: 'csv',
kkonganti@0:                                     filename: '{table_sum_on}_results',
kkonganti@0:                                 }}
kkonganti@0:                             ]
kkonganti@0:                         }});
kkonganti@0:                     }});
kkonganti@0:                 </script>
kkonganti@0:                 <div class="table-responsive">
kkonganti@0:                 <style>
kkonganti@0:                 #cpipes-process-custom-res tr:nth-child(even) {{
kkonganti@0:                     background-color: #f2f2f2;
kkonganti@0:                 }}
kkonganti@0:                 </style>
kkonganti@0:                 <table class="table" style="width:100%" id="cpipes-process-custom-res-{table_sum_on}">
kkonganti@0:                 <thead>
kkonganti@0:                 <tr>"""
kkonganti@0:             )
kkonganti@0:         ]
kkonganti@0: 
kkonganti@0:         for header_col in header_cols:
kkonganti@0:             html.append(
kkonganti@0:                 dedent(
kkonganti@0:                     f"""
kkonganti@0:                         <th> {header_col} </th>"""
kkonganti@0:                 )
kkonganti@0:             )
kkonganti@0: 
kkonganti@0:         html.append(
kkonganti@0:             dedent(
kkonganti@0:                 """
kkonganti@0:                 </tr>
kkonganti@0:                 </thead>
kkonganti@0:                 <tbody>"""
kkonganti@0:             )
kkonganti@0:         )
kkonganti@0: 
kkonganti@0:         for row in tbl:
kkonganti@0:             html.append("<tr>\n")
kkonganti@0:             data_cols = row.strip().split("\t")
kkonganti@0:             if len(header_cols) != len(data_cols):
kkonganti@0:                 print(
kkonganti@0:                     f"\nWARN: Number of header columns ({len(header_cols)}) and data "
kkonganti@0:                     + f"columns ({len(data_cols)}) are not equal!\nWill append empty columns!\n"
kkonganti@0:                 )
kkonganti@0:                 if len(header_cols) > len(data_cols):
kkonganti@0:                     data_cols += (len(header_cols) - len(data_cols)) * " "
kkonganti@0:                     print(len(data_cols))
kkonganti@0:                 else:
kkonganti@0:                     header_cols += (len(data_cols) - len(header_cols)) * " "
kkonganti@0: 
kkonganti@0:             html.append(
kkonganti@0:                 dedent(
kkonganti@0:                     f"""
kkonganti@0:                         <td><samp>{data_cols[0]}</samp></td>
kkonganti@0:                     """
kkonganti@0:                 )
kkonganti@0:             )
kkonganti@0: 
kkonganti@0:             for data_col in data_cols[1:]:
kkonganti@0:                 data_col_w_color = f"""<td>{data_col}</td>
kkonganti@0:                 """
kkonganti@0:                 if (
kkonganti@0:                     table_sum_on in cell_colors.keys()
kkonganti@0:                     and data_col in cell_colors[table_sum_on].keys()
kkonganti@0:                 ):
kkonganti@0:                     data_col_w_color = f"""<td style="background-color: {cell_colors[table_sum_on][data_col]}">{data_col}</td>
kkonganti@0:                     """
kkonganti@0:                 html.append(dedent(data_col_w_color))
kkonganti@0:             html.append("</tr>\n")
kkonganti@0:         html.append("</tbody>\n")
kkonganti@0:         html.append("</table>\n")
kkonganti@0:         html.append("</div>\n")
kkonganti@0: 
kkonganti@0:         mqc_yaml = {
kkonganti@0:             "id": f"{table_sum_on.upper()}_collated_table",
kkonganti@0:             "section_name": f"{table_sum_on.upper()}",
kkonganti@0:             "section_href": f"https://github.com/CFSAN-Biostatistics/bettercallsal",
kkonganti@0:             "plot_type": "html",
kkonganti@0:             "description": f"{description}",
kkonganti@0:             "data": ("").join(html),
kkonganti@0:         }
kkonganti@0: 
kkonganti@0:         with open(f"{table_sum_on.lower()}_mqc.yml", "w") as html_mqc:
kkonganti@0:             yaml.dump(mqc_yaml, html_mqc, default_flow_style=False)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: if __name__ == "__main__":
kkonganti@0:     main()