Mercurial > repos > estrain > lissero
diff variant4b.py @ 0:bc556481a1fb draft default tip
planemo upload commit e12eb56d1744da8f7af8ca1819e2617c83fb17a8
| author | estrain |
|---|---|
| date | Fri, 13 Mar 2026 12:01:17 +0000 |
| parents | |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/variant4b.py Fri Mar 13 12:01:17 2026 +0000 @@ -0,0 +1,54 @@ +import sys + +def identify_variants_with_genes(input_file_path, output_file_path, simple_name): + # Define the genes of interest + genes_of_interest = ['LMO0737', 'ORF2110', 'ORF2819'] + + # Open the input file and read its lines + with open(input_file_path, 'r') as file: + lines = file.readlines() + + # Check if the file has more than just the header + if len(lines) <= 1: + print("Input file does not contain enough data.") + return + + # Extract the column headers and find the indices of the genes of interest + headers = lines[0].strip().split('\t') + gene_indices = [headers.index(gene) for gene in genes_of_interest] + serotype_index = headers.index('SEROTYPE') + + # Modify the header to include the new first column + modified_header = f"FileName\t{lines[0]}" + + # Initialize a list to hold the modified lines, starting with the modified header + modified_lines = [modified_header] + + # Process each data line in the input file + for line in lines[1:]: + data = line.strip().split('\t') + # Check if the genes of interest are all present (marked as "FULL") + if all(data[index] == 'FULL' for index in gene_indices): + # Modify the SEROTYPE column to "4b variant" + data[serotype_index] = "4b variant" + # Prepend the simple name to the line + modified_line = f"{simple_name}\t" + '\t'.join(data) + '\n' + # Add the modified line to the list + modified_lines.append(modified_line) + + # Write the modified lines to the output file + with open(output_file_path, 'w') as file: + file.writelines(modified_lines) + + print(f'Results written to {output_file_path}') + +if __name__ == "__main__": + if len(sys.argv) != 4: + print("Usage: python script.py <input_file_path> <output_file_path> <simple_name>") + sys.exit(1) + + input_file_path = sys.argv[1] + output_file_path = sys.argv[2] + simple_name = sys.argv[3] + identify_variants_with_genes(input_file_path, output_file_path, simple_name) +
