main.py

import random
import classes
import storydata

# Initialize variables
guard_count = 10

# Initialize prison
prison = classes.Prison(["cells", "cafeteria", "yard", "laundry", "library", "solitary", "office"])

# Initialize player and characters
player = classes.Player("Andy", prison)
guards = {number:classes.Guard(number, prison, random.randint(0,100)) for number in range(1, guard_count + 1)}
inmates = {name:classes.Inmate(name, prison, storydata.inmates[name]["strength"], storydata.inmates[name]["friendliness"], storydata.inmates[name]["dialogues"]) for name in storydata.inmates.keys()}

# Initialize controller
controller = classes.Controller(player)

# Initialize character positions
prison.rooms["cells"].occupants.append(player)
prison.rooms["cells"].occupants.extend(list(inmates.values()))

while True:
	controller.get_input()
	print(f"You are in the {player.get_location()}, with {', '.join([player.name for player in player.get_location().other_occupants(player)])}")