Searching of objects from an array of objects using ID in Python

Program to illustrate the working of our solution

class Student:
    def getStudentInfo(self):
        self.__rollno = input("Enter Roll Number : ")
        self.__name = input("Enter Name : ")
        self.__marks = int(input("Enter Marks : "))
        return self.__rollno
        
    def putStudentInfo(self):
        print(self.__rollno,self.__name,self.__marks)
        
StudentList = {}
while(True):
    studentInfo=Student()
    key = studentInfo.getStudentInfo()
    StudentList.setdefault(key,studentInfo)
    ch=input("Add More y/n?")
    if(ch=='n'):break

print("All Student's Information : ")
L = StudentList.values()
for studentInfo in L:
    studentInfo.putStudentInfo()

roll = input("Enter Roll Number you Want to Search: ")
studentInfo = StudentList.get(roll,"Not Found....")
if isinstance(studentInfo,Student):
    studentInfo.putStudentInfo()

Output:

Enter Roll Number : 1
Enter Name : John
Enter Marks : 87
Add More y/n?y
Enter Roll Number : 4
Enter Name : Joe
Enter Marks : 54
Add More y/n?n
All Student's Information : 
1 John 87
4 Joe 54
Enter Roll Number you Want to Search: 1
1 John 87