import sys
import csv

def main():

    # Ensure correct usage
    if len(sys.argv) != 3:
        sys.exit("Usage: python dna.py data.csv sequence.txt")

    # open the csv file
    data = []
    with open(sys.argv[1]) as file:
        reader = csv.DictReader(file)
        for name in reader:
            data.append(name)
        header = reader.fieldnames
        header = header[1:]
    print(header)

    # open the txt file
    with open(sys.argv[2]) as f:
        sequence = f.read()
    #print(sequence)

# computing STR counts
    counts = {'str' : 'count'}
    for strr in header:
        counter = 0;
        for i in range(len(sequence) - len(strr)):
            for j in range(len(sequence)):
                if sequence[i:j] == strr:
                    counter = counter + 1
                    #print(i, j)
        counts[strr] = str(counter)
    print(counts)

if __name__ == "__main__":
    main()