uploading pset6

Author: aviiciii

pset6/lines.py

@@ -0,0 +1,28 @@
+from sys import argv, exit
+
+if len(argv) != 2:
+    if len(argv) < 2:
+        exit('Too few command-line arguments')
+    else:
+        exit('Too mant command-line arguments')
+
+filename = argv[1]
+
+dot = filename.find('.')
+if filename[dot:] != '.py':
+    exit('Not a Python file')
+
+lines = 0
+
+try:
+    with open(filename,'r') as file:
+        for line in file:
+            if line.isspace():
+                continue
+            if line.strip().startswith('#'):
+                continue
+            lines +=1
+except FileNotFoundError:
+    exit('File does not exist')
+
+print(lines)

pset6/pizza/pizza.py

@@ -0,0 +1,36 @@
+from sys import argv, exit
+from tabulate import tabulate
+import csv
+
+def main():
+    filename = get_filename(argv)
+    check_csv(filename)
+
+    try:
+        with open(filename,'r') as file:
+            reader=csv.reader(file)
+            table=[]
+            headers=list(next(reader))
+            for row in reader:
+                table.append([row[0], row[1], row[2]])
+                print(tabulate(table, headers, tablefmt="grid"))
+    except FileNotFoundError:
+        exit('File does not exist')
+    except Exception as arg:
+        exit(arg)
+
+def get_filename(arguments):
+    if len(arguments) != 2:
+        if len(arguments) < 2:
+            exit('Too few command-line arguments')
+        else:
+            exit('Too mant command-line arguments')
+    return argv[1]
+
+def check_csv(filename):
+    dot = filename.find('.')
+    if filename[dot:] != '.csv':
+        exit('Not a CSV file')
+
+if __name__ == '__main__':
+    main()

pset6/pizza/regular.csv

@@ -0,0 +1,6 @@
+Regular Pizza,Small,Large
+Cheese,$13.50,$18.95
+1 topping,$14.75,$20.95
+2 toppings,$15.95,$22.95
+3 toppings,$16.95,$24.95
+Special,$18.50,$26.95

pset6/pizza/sicilian.csv

@@ -0,0 +1,6 @@
+Sicilian Pizza,Small,Large
+Cheese,$25.50,$39.95
+1 item,$27.50,$41.95
+2 items,$29.50,$43.95
+3 items,$31.50,$45.95
+Special,$33.50,$47.95

pset6/scourgify/after.csv

@@ -0,0 +1,54 @@
+first,last,house
+Hannah,Abbott,Hufflepuff
+Katie,Bell,Gryffindor
+Susan,Bones,Hufflepuff
+Terry,Boot,Ravenclaw
+Lavender,Brown,Gryffindor
+Millicent,Bulstrode,Slytherin
+Cho,Chang,Ravenclaw
+Penelope,Clearwater,Ravenclaw
+Vincent,Crabbe,Slytherin
+Colin,Creevey,Gryffindor
+Dennis,Creevey,Gryffindor
+Cedric,Diggory,Hufflepuff
+Marietta,Edgecombe,Ravenclaw
+Justin,Finch-Fletchley,Hufflepuff
+Seamus,Finnigan,Gryffindor
+Anthony,Goldstein,Ravenclaw
+Gregory,Goyle,Slytherin
+Hermione,Granger,Gryffindor
+Angelina,Johnson,Gryffindor
+Lee,Jordan,Gryffindor
+Neville,Longbottom,Gryffindor
+Luna,Lovegood,Ravenclaw
+Remus,Lupin,Gryffindor
+Draco,Malfoy,Slytherin
+Scorpius,Malfoy,Slytherin
+Ernie,Macmillan,Hufflepuff
+Minerva,McGonagall,Gryffindor
+Eloise,Midgen,Gryffindor
+Cormac,McLaggen,Gryffindor
+Graham,Montague,Slytherin
+Theodore,Nott,Slytherin
+Pansy,Parkinson,Slytherin
+Padma,Patil,Gryffindor
+Parvati,Patil,Gryffindor
+Harry,Potter,Gryffindor
+Tom,Riddle,Slytherin
+Demelza,Robins,Gryffindor
+Newt,Scamander,Hufflepuff
+Horace,Slughorn,Slytherin
+Zacharias,Smith,Hufflepuff
+Severus,Snape,Slytherin
+Alicia,Spinnet,Gryffindor
+Pomona,Sprout,Hufflepuff
+Dean,Thomas,Gryffindor
+Romilda,Vane,Gryffindor
+Myrtle,Warren,Ravenclaw
+Fred,Weasley,Gryffindor
+George,Weasley,Gryffindor
+Ginny,Weasley,Gryffindor
+Percy,Weasley,Gryffindor
+Ron,Weasley,Gryffindor
+Oliver,Wood,Gryffindor
+Blaise,Zabini,Slytherin

pset6/scourgify/before.csv

@@ -0,0 +1,54 @@
+name,house
+"Abbott, Hannah",Hufflepuff
+"Bell, Katie",Gryffindor
+"Bones, Susan",Hufflepuff
+"Boot, Terry",Ravenclaw
+"Brown, Lavender",Gryffindor
+"Bulstrode, Millicent",Slytherin
+"Chang, Cho",Ravenclaw
+"Clearwater, Penelope",Ravenclaw
+"Crabbe, Vincent",Slytherin
+"Creevey, Colin",Gryffindor
+"Creevey, Dennis",Gryffindor
+"Diggory, Cedric",Hufflepuff
+"Edgecombe, Marietta",Ravenclaw
+"Finch-Fletchley, Justin",Hufflepuff
+"Finnigan, Seamus",Gryffindor
+"Goldstein, Anthony",Ravenclaw
+"Goyle, Gregory",Slytherin
+"Granger, Hermione",Gryffindor
+"Johnson, Angelina",Gryffindor
+"Jordan, Lee",Gryffindor
+"Longbottom, Neville",Gryffindor
+"Lovegood, Luna",Ravenclaw
+"Lupin, Remus",Gryffindor
+"Malfoy, Draco",Slytherin
+"Malfoy, Scorpius",Slytherin
+"Macmillan, Ernie",Hufflepuff
+"McGonagall, Minerva",Gryffindor
+"Midgen, Eloise",Gryffindor
+"McLaggen, Cormac",Gryffindor
+"Montague, Graham",Slytherin
+"Nott, Theodore",Slytherin
+"Parkinson, Pansy",Slytherin
+"Patil, Padma",Gryffindor
+"Patil, Parvati",Gryffindor
+"Potter, Harry",Gryffindor
+"Riddle, Tom",Slytherin
+"Robins, Demelza",Gryffindor
+"Scamander, Newt",Hufflepuff
+"Slughorn, Horace",Slytherin
+"Smith, Zacharias",Hufflepuff
+"Snape, Severus",Slytherin
+"Spinnet, Alicia",Gryffindor
+"Sprout, Pomona",Hufflepuff
+"Thomas, Dean",Gryffindor
+"Vane, Romilda",Gryffindor
+"Warren, Myrtle",Ravenclaw
+"Weasley, Fred",Gryffindor
+"Weasley, George",Gryffindor
+"Weasley, Ginny",Gryffindor
+"Weasley, Percy",Gryffindor
+"Weasley, Ron",Gryffindor
+"Wood, Oliver",Gryffindor
+"Zabini, Blaise",Slytherin

pset6/scourgify/scourgify.py

@@ -0,0 +1,48 @@
+from sys import argv, exit
+import csv
+
+def main():
+    readfile, writefile = get_filenames(argv)
+    check_csv(readfile)
+    check_csv(writefile)
+
+    firstnames=[]
+    lastnames=[]
+    house=[]
+    try:
+        with open(readfile,'r') as file:
+            reader=csv.DictReader(file)
+            for row in reader:
+                last, first=row['name'].split(',')
+                lastnames.append(last.strip())
+                firstnames.append(first.strip())
+                house.append(row['house'])
+    except FileNotFoundError:
+        exit('File does not exist')
+    except Exception as arg:
+        exit(arg)
+
+    with open(writefile,'w') as file:
+        fieldnames=['first','last','house']
+        writer=csv.DictWriter(file, fieldnames=fieldnames)
+        writer.writeheader()
+        for i in range(len(firstnames)):
+            writer.writerow({'first': firstnames[i],'last': lastnames[i],'house': house[i],})
+
+
+
+def get_filenames(arguments):
+    if len(arguments) != 3:
+        if len(arguments) < 3:
+            exit('Too few command-line arguments')
+        else:
+            exit('Too mant command-line arguments')
+    return [argv[1],argv[2]]
+
+def check_csv(filename):
+    dot = filename.find('.')
+    if filename[dot:] != '.csv':
+        exit('Not a CSV file')
+
+if __name__ == '__main__':
+    main()