File IO in python

python_file_IO_1.py

import csv
from cs50 import get_string
from student import Student

# Space for students
students = []

# Prompt for students' names and dorms
for i in range(3):
	name = get_string("name: ")
	dorm = get_string("dorm: ")
	students.append(Student(name, dorm))


with open("students.csv", "w") as file:
	writer = csv.writer(file)
	writer.writerow(("Name", "Dorm"))
	for student in students:
		writer.writerow((student.name, student.dorm))

student.py

class Student:
	def __init__(self, name, dorm):
		self.name = name
		self.dorm = dorm
		print(f"created a new student object {self.name} living in {self.dorm}!")