aboutsummaryrefslogtreecommitdiff
path: root/scripts/maintenance/delete_genotypes.py
blob: b7f837581283497536db357d08d277a417e191c1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import sys
import csv

import utilities
import datastructure
import genotypes

def main(argv):
    # config
    config = utilities.get_config(argv[1])
    print("config:")
    for item in config.items('config'):
        print(("\t%s" % (str(item))))
    # var
    print("variable:")
    inbredsetid = config.get('config', 'inbredsetid')
    print(("\tinbredsetid: %s" % inbredsetid))
    # datafile
    datafile = open(config.get('config', 'datafile'), 'r')
    datafile = csv.reader(datafile, delimiter='\t', quotechar='"')
    next(datafile)
    delrowcount = 0
    for row in datafile:
        if len(row) == 0:
            continue
        genoname = row[0]
        delrowcount += genotypes.delete(genoname, inbredsetid)
    print(("deleted %d genotypes" % (delrowcount)))

if __name__ == "__main__":
    print(("command line arguments:\n\t%s" % sys.argv))
    main(sys.argv)
    print("exit successfully")