Python41

2023-10-12  本文已影响0人  rong酱
#!/usr/bin/env python
##coding=utf-8

import os
import sys
import argparse
import vcf

parser = argparse.ArgumentParser(description="")
parser.add_argument('-i', '--invcf', help = 'the input vcf file', required = True)
parser.add_argument('-o', '--outvcf', help = 'the output vcf file', required = True)
argv = vars(parser.parse_args())
inF = os.path.abspath(argv['invcf'].strip())
ouF = os.path.abspath(argv['outvcf'].strip())

def parser_vcf(infile,outfile):
    samplename=str(str(str(infile).strip().split('/')[-1]).strip().split("_")[0])
    vcf_r=vcf.Reader(filename=infile)
    for record in vcf_r:
        if len(record.FILTER) == 0 :
            print("samplename: "+str(samplename))
            sample1=samplename+".Mutect2"
            sample2=samplename+".vardict"
            print("sample1: "+str(record.genotype(sample1)['GT']))
            break
#            if record.samples[sample2]['DP'] == "." and record.samples[sample2]['GT'] == ".":

parser_vcf(inF,ouF)
上一篇 下一篇

猜你喜欢

热点阅读