IMPORT

from sys import argv
from csv import reader
from cs50 import SQL

if len(argv) != 2:
    print("Usage: python import.py characters.csv")
    exit()

db = SQL("sqlite:///students.db")

with open(argv[1], newline='') as characters:
    reader = reader(characters)

    for character in reader:
        if character[0] == 'name':
            continue

        first = character[0].split()[0]
        last = character[0].split()[-1]

        if character[0].count(' ') == 2:
            middle = character[0].split()[1]
        else:
            middle = "NULL"

        db.execute("INSERT INTO students (first, middle, last, house, birth) VALUES(?, ?, ?, ?, ?)", first, middle, last, character[1], character[2])

ROSTER

from sys import argv
from cs50 import SQL

if len(argv) != 2:
    print("Usage: python roster.py House")
    exit()

db = SQL("sqlite:///students.db")

characteres = db.execute("SELECT * FROM students WHERE house = (?) ORDER BY last", argv[1])

for c in characteres:
    if c['middle'] != "NULL":
        print("{} {} {}, born {}" .format(c['first'], c['middle'], c['last'], c['birth']))
    else:
        print("{} {}, born {}" .format(c['first'], c['last'], c['birth']))