def getRooms(self, buildingName, username = None): checkData(locals()) building = Building(buildingName = buildingName) building.retrieve() roomList = [] if username: from app.backend.model.user import User user = User(username = username) user.retrieve() for buildingRoom in building.getRooms(): for userRoom in user.getRooms(): if buildingRoom.buildingName == userRoom.buildingName and buildingRoom.roomName == userRoom.roomName: roomList.append(userRoom.getDict()) else: for room in building.getRooms(): roomList.append(room.getDict()) return {"rooms" : roomList}