Sun, May 5, 2024, 11:45 PM -07:00

1 files changed, 11 insertions, 11 deletions

diff --git a/wk6/pset/dna/dna.py b/wk6/pset/dna/dna.py
index a7d0295..cde7aef 100644
--- a/wk6/pset/dna/dna.py
+++ b/wk6/pset/dna/dna.py
@@ -5,28 +5,28 @@ import sys
 def main():
 
     # TODO: Check for command-line usage
-    if len(sys.argv) != 1 and len(sys.argv) != 2:
+    if len(sys.argv) < 2 and len(sys.argv) > 3:
         print("Oopsie you used the wrong number of arguments D: you should use two next time :3")
         return
-    # check to see if arg 0 is a csv file
-    if not sys.argv[0].endswith('.csv'):
-        print("Oh no, that isn't a csv file :( that won't work can you fix that please?")
-        return
-    # check to see if arg 1 is a txt file
-    if not sys.argv[1].endswith('.txt'):
-        print("Oh no, that isn't a txt file :( that won't work can you fix that please?")
-        return
+    # # check to see if arg 0 is a csv file
+    # if not sys.argv[0].endswith('.csv'):
+    #     print("Oh no, that isn't a csv file :( that won't work can you fix that please?")
+    #     return
+    # # check to see if arg 1 is a txt file
+    # if not sys.argv[1].endswith('.txt'):
+    #     print("Oh no, that isn't a txt file :( that won't work can you fix that please?")
+    #     return
 
     # TODO: Read database file into a variable
     rows = []
-    with open(argv[0]) as file:
+    with open(sys.argv[0]) as file:
         reader = csv.DictReader(file)
         for row in reader:
             rows.append(row)
 
     # TODO: Read DNA sequence file into a variable
     sequence = ''
-    with open(argv[1]) as file:
+    with open(sys.argv[1]) as file:
         sequence = sequence.join(file.readlines())
 
     # TODO: Find longest match of each STR in DNA sequence