pset6 code

Author: rohanrdy

week6/pizza/pizza.py

@@ -0,0 +1,26 @@
+from sys import exit, argv
+import csv
+import tabulate
+
+
+if len(argv) < 2:
+    exit("Too few command-line arguments")
+elif len(argv) == 2:
+    file_path = argv[1]
+else:
+    exit("Too many command-line arguments")
+
+if file_path[-4:] == ".csv":
+    try:
+        f = open(file_path)
+    except FileNotFoundError:
+        exit("File does not exist")
+    else:
+        menu_list = []
+        with f as file:
+            reader = csv.reader(file)
+            for row in reader:
+                menu_list.append(row)
+        print(tabulate.tabulate(menu_list, headers="firstrow", tablefmt="grid"))
+else:
+    exit("Not a CSV file")

week6/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

week6/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

week6/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

week6/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

week6/scourgify/scourgify.py

@@ -0,0 +1,28 @@
+from sys import exit, argv
+import csv
+
+
+if len(argv) < 3:
+    exit("Too few command-line arguments")
+elif len(argv) > 3:
+    exit("Too many command-line arguments")
+else:
+    if argv[1][-4:] == ".csv" and argv[2][-4:] == ".csv":
+        try:
+            f = open(argv[1])
+        except FileNotFoundError:
+            exit(f"Could not read {argv[1]}")
+        else:
+            listofdict = []
+            with f as file:
+                reader = csv.DictReader(file)
+                for row in reader:
+                    x, y = row["name"].split(', ')
+                    listofdict.append({"first": y, "last": x, "house": row["house"]})
+            with open(argv[2], "w") as file:
+                writer = csv.DictWriter(file, fieldnames=["first", "last", "house"])
+                writer.writeheader()
+                for i in range(len(listofdict)):
+                    writer.writerow({"first": listofdict[i]["first"], "last": listofdict[i]["last"], "house": listofdict[i]["house"]})
+    else:
+        exit("Not a CSV file")

week6/shirt/after1.jpg

@@ -0,0 +1,35 @@
+from sys import exit, argv
+from PIL import Image, ImageOps
+
+
+def main():
+    if len(argv) < 3:
+        exit("Too few command-line arguments")
+    elif len(argv) > 3:
+        exit("Too many command-line arguments")
+    else:
+        if file_extension_check(1) != file_extension_check(2):
+            exit("Input and output have different extensions")
+
+    try:
+        shirt = Image.open("shirt.png")
+        before = Image.open(argv[1])
+    except FileNotFoundError:
+        exit("Could not find the image file")
+    else:
+        size = shirt.size # it is a tuple (600, 600)
+        before = ImageOps.fit(before, size) # resize and crop the input
+        before.paste(shirt, box = (0, 0), mask = shirt) # https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.paste
+        before.save(argv[2], format=None) # saving the image
+
+
+def file_extension_check(i):
+    match endpart := argv[i][(argv[i].index('.')):].lower(): # find period symbol location and slice from there to the end and finally making it lowercase. ":=" is walrus operator an assignment expression
+        case ".png" | ".jpg" | ".jpeg":
+            return endpart
+        case _:
+            exit("Invalid output")
+
+
+if __name__ == "__main__":
+    main()