view completemask.py @ 8:2d9926ce62be draft

"planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/ivar/ commit 6dae6f97a45a61b1f10be4227d978584624c3b3d"
author iuc
date Thu, 05 Aug 2021 12:45:47 +0000
parents 78bbd17d0703
children
line wrap: on
line source

#!/usr/bin/env python

import sys


if __name__ == '__main__':
    with open(sys.argv[1]) as i:
        getmasked_output = i.readline().strip()

    if not getmasked_output:
        print()
        print('No affected primer binding sites found!')
    else:
        masked_primers = getmasked_output.split('\t')
        with open(sys.argv[2]) as i:
            amplicon_data = [line.strip().split('\t') for line in i]

        masked_complete = []
        for primer in masked_primers:
            for amplicon in amplicon_data:
                if primer in amplicon:
                    masked_complete += amplicon
        result = '\t'.join(sorted(set(masked_complete)))
        print()
        print('Removing reads primed with any of:')
        print(result)
        with open(sys.argv[1], 'w') as o:
            o.write(result + '\n')