Library Management System.py

class Library:
    def __init__(self , listOfBooks):
        self.books = listOfBooks
    def displayAvailableBooks(self):
        print('Books present in this Library are: ')
        for book in self.books:
            print('\t *' , book)
    def borrowBook(self , bookName):
        if bookName in self.books:
            print(f'You have been issued {bookName}.Please keep it safe and return it within 30 days.')
            self.books.remove(bookName)
        else:
            print('Sorry! This book is either not available or has already been issued to someone else. Please wait until the book is available')
    def returnBook(self , bookName):
        self.books.append(bookName)
        print('Thanks for adding/returning this book.Hope you enjoyed reading it.Have a great day ahead!')

class Student:
    def requestBook(self):
        self.book = input('Enter the name of the book you want to borrow : ')
        return self.book
    def returnBook(self):
        self.book = input('Enter the name of the book you want to add/return : ')
        return self.book
        
if __name__ == "__main__":
    centralLibrary = Library(['Let us C', 'Algorithms', 'Django', 'The Alchemist', 'Da VinciCode'])
    varsha = Student()
    while True:
        print('''\n===== WELCOME TO CENTRAL LIBRARY ===== 
        Please choose an option
        1.List all the books
        2.Request a book
        3.Add/Return a book
        4.Exit the library''')
        try:
            a = int(input('Enter a choice : ')) 
            if a==1:
                centralLibrary.displayAvailableBooks()
            elif a==2:
                centralLibrary.borrowBook(varsha.requestBook())
            elif a==3:
                centralLibrary.returnBook(varsha.returnBook())
            elif a==4:
                print('Thanks for choosing Central Library. Have a great day!')
                exit()
            else:
                print('Invalid Choice!')
        except Exception as e:
            print(e)