dna.py for CS50

dna.py

# first arg = database csv
# second arg = sequence
import csv
from sys import argv, exit
# if user does not adhere to usage, quit
if len(argv) != 3:
    print("Usage: python dna.py data.csv sequence.txt")
    exit(1)
# open the passed files and close them after work done
with open(argv[1]) as dataf, open(argv[2]) as seqf:
    datao = csv.reader(dataf)   # open database as csv object
    seq = seqf.read()           # read entire sequence into one string
    data = []                   # two dimensional array to store database
    for row in datao:           # add all the rows of database to 2D array
        data.append(row)
    header = data[0:1][0]       # copy the first row (column headings) to 'header'
    data = data[1:]             # remove header from data itself
    for row in data:            # iterate over each row of data
        this = True             # boolean flag where true = this person is match
        for i in range(1, len(row)):    # for all the STR mentioned in database, check current person with seq
            this = this and header[i]*int(row[i]) in seq and header[i]*(int(row[i])+1) not in seq
        if this:                # if all the STR matches for this person, print and exit
            print(row[0])
            exit(0)
    print("No match")           # if no match then we come here, so print no match

DAMN!!