class LoanController:
    def __init__(self):
        self.client = DbClient.Instance()
        self.userController = UserController()
        self.bookController = BookController()

    def checkout_book(self, user, book):
        if user.currentFine is not 0:
            return 0
        elif len(user.loanedBooks) >= 4:
            return 1
        else:
            user.loanedBooks.append(book.id)
            if book.id in user.waitingBooks:
                user.waitingBooks.remove(str(book.id))
            book.isAvailable = False
            user.totalLoanedBooks = user.totalLoanedBooks + 1
            user.lastLoanedBook = book.title
            self.userController.update_member_attributes(user)
            self.bookController.update_book_attributes(book)
            new_loan = Loan(str(user.id), str(book.id))
            self.client.add_loan_record(new_loan)
            return 2

    def return_book(self, user, book):
        loan_info = self.get_loan_record(str(book.id))
        existing_loan = Loan(str(loan_info["user_id"]),
                             str(loan_info["book_id"]))
        existing_loan.startDate = loan_info["startDate"]
        existing_loan.returnDate = loan_info["returnDate"]
        existing_loan.check_status()
        self.update_loan_attributes(existing_loan)
        user.loanedBooks.remove(str(book.id))
        book.isAvailable = True
        user.formerFine = user.formerFine + existing_loan.currentFine
        self.client.delete_loan_record(str(book.id))
        self.bookController.update_book_attributes(book)
        self.userController.update_member_attributes(user)
        self.update_fines()
        self.available_notification(book)

    def available_notification(self, returned_book):
        mail_client = MailClient()
        for waiting_user_id in returned_book.waitingList:
            waiting_user = self.userController.get_user_by_id(
                str(waiting_user_id))
            _thread.start_new_thread(
                mail_client.send_email,
                (waiting_user.username, waiting_user.email_address,
                 returned_book.title))
            # mail_client.send_email(waiting_user.username,waiting_user.email_address,returned_book.title)

    def get_loan_record(self, book_id):
        return self.client.find_loan_record(str(book_id))

    def instantiate_loan(self, loan_info):
        existing_loan = Loan(str(loan_info["user_id"]),
                             str(loan_info["book_id"]))
        existing_loan.startDate = loan_info["startDate"]
        existing_loan.returnDate = loan_info["returnDate"]
        return existing_loan

    def update_fines(self):
        self.reset_fines()
        loans = self.client.db.loans.find({})
        for loan in loans:
            existing_loan = self.instantiate_loan(loan)
            existing_loan.check_status()
            self.update_loan_attributes(existing_loan)
            existing_user = self.userController.get_user_by_id(
                str(existing_loan.user_id))
            existing_user.currentFine = existing_user.currentFine + existing_loan.currentFine
            self.userController.update_member_attributes(existing_user)

    def reset_fines(self):
        current_users = self.client.search_user_db("", "", "")
        for person in current_users:
            user_op = self.userController.get_user_by_id(str(person["_id"]))
            user_op.currentFine = user_op.formerFine
            self.userController.update_member_attributes(user_op)

    def delete_user_from_loans(self, user_id):
        loans = self.client.db.loans.find({})
        for loan in loans:
            if loan["user_id"] == user_id:
                op_book = self.bookController.get_book_by_id(
                    str(loan["book_id"]))
                op_book.isAvailable = True
                self.bookController.update_book_attributes(op_book)
                self.client.delete_loan_record(str(loan["book_id"]))

    def delete_book_from_loans(self, book_id):
        loans = self.client.db.loans.find({})
        for loan in loans:
            if loan["book_id"] == book_id:
                self.client.delete_loan_record(str(loan["book_id"]))

    def update_loan_attributes(self, loan):
        self.client.update_loan_attributes_db(loan)

    def reset_user_loans(self, user_id):
        loans = self.client.db.loans.find({})
        for loan_info in loans:
            if loan_info["user_id"] == user_id and loan_info["status"] == 1:
                updated_loan = self.instantiate_loan(loan_info)
                updated_loan.startDate = datetime.datetime.now(
                ) - datetime.timedelta(days=40)
                updated_loan.check_status()
                self.update_loan_attributes(updated_loan)
Beispiel #2
0
class LibrarianHomeView(Ui_libraryMainWindow):
    def __init__(self, user):
        self.userRegisterScreen = userRegisterFormView.UserRegisterFormView()
        self.bookRegisterScreen = bookRegisterFormView.BookRegisterFormView()
        self.librarianHome = QtWidgets.QMainWindow()
        self.ui = Ui_libraryMainWindow()
        self.ui.setupUi(self.librarianHome)
        self.ui.logoutButton.clicked.connect(self.logout)
        self.ui.userResetButton.clicked.connect(self.reset_user_filters)
        self.ui.bookResetButton.clicked.connect(self.reset_book_filters)
        self.ui.userDeleteButton.clicked.connect(self.delete_selected_user)
        self.ui.userCreateButton.clicked.connect(self.register_user_form)
        self.ui.userUpdateButton.clicked.connect(self.update_user_form)
        self.ui.userSearchButton.clicked.connect(self.search_users)
        self.ui.bookSearchButton.clicked.connect(self.search_books)
        self.ui.bookDeleteButton.clicked.connect(self.delete_selected_book)
        self.ui.bookCreateButton.clicked.connect(self.register_book_form)
        self.ui.bookUpdateButton.clicked.connect(self.update_book_form)
        self.ui.userTableWidget.cellDoubleClicked.connect(
            self.update_user_form)
        self.ui.bookTableWidget.cellDoubleClicked.connect(
            self.update_book_form)
        self.currentUser = user
        self.confirm = confirmView.ConfirmView()
        self.error = errorView.ErrorView()
        self.userController = UserController()
        self.bookController = BookController()
        self.loanController = LoanController()
        self.customize_scene()
        self.current_user_queries = ["", "", ""]
        self.current_book_queries = ["", "", ""]
        self.set_button_effects()
        self.book_ids = []
        self.user_ids = []

        self.list_users()
        self.list_books()

    def customize_scene(self):
        self.ui.greetingLabel.setText("Hi, " + str(self.currentUser.name) +
                                      " " + str(self.currentUser.surname))
        self.ui.dateLabel.setText(datetime.datetime.now().strftime("%Y-%m-%d"))
        self.ui.userTableWidget.setColumnCount(4)
        self.ui.bookTableWidget.setColumnCount(4)
        self.ui.userTableWidget.setHorizontalHeaderLabels(
            ["Name", "Surname", "Username", "User ID"])
        self.ui.bookTableWidget.setHorizontalHeaderLabels(
            ["Title", "Author", "Published Year", "Book ID"])

    def reset_user_filters(self):
        self.current_user_queries = ["", "", ""]
        self.list_users()

    def reset_book_filters(self):
        self.current_book_queries = ["", "", ""]
        self.list_books()

    def delete_selected_user(self):
        selected_row = self.ui.userTableWidget.currentRow()
        if selected_row > -1:
            selected_user_id = self.ui.userTableWidget.item(selected_row,
                                                            3).text()
            self.confirm.confirm_flag = 0
            self.confirm.confirmScreen.exec_()
            if self.confirm.confirm_flag is 1:
                self.userController.delete_one_user(str(selected_user_id))
                self.loanController.delete_user_from_loans(
                    str(selected_user_id))
                self.bookController.delete_from_all_waiting_lists(
                    str(selected_user_id))

            self.list_users()

    def delete_selected_book(self):
        selected_row = self.ui.bookTableWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.bookTableWidget.item(selected_row,
                                                            3).text()
            self.confirm.confirm_flag = 0
            self.confirm.confirmScreen.exec_()
            if self.confirm.confirm_flag is 1:
                self.bookController.delete_one_book_by_id(
                    str(selected_book_id))
                self.userController.delete_from_all_owned_books(
                    str(selected_book_id))
                self.userController.delete_from_all_waiting_lists(
                    str(selected_book_id))
                self.loanController.delete_book_from_loans(
                    str(selected_book_id))
            self.list_books()

    def list_users(self):
        query_result = self.userController.search_users(
            self.current_user_queries[0], self.current_user_queries[1],
            self.current_user_queries[2])
        i = 0
        current_row_count = query_result.count()
        self.ui.userTableWidget.setRowCount(current_row_count)
        for record in query_result:
            self.ui.userTableWidget.setItem(
                i, 0, QtWidgets.QTableWidgetItem(record["name"]))
            self.ui.userTableWidget.setItem(
                i, 1, QtWidgets.QTableWidgetItem(record["surname"]))
            self.ui.userTableWidget.setItem(
                i, 2, QtWidgets.QTableWidgetItem(record["username"]))
            self.ui.userTableWidget.setItem(
                i, 3, QtWidgets.QTableWidgetItem(str(record["_id"])))
            i = i + 1
        self.show()

    def list_books(self):
        query_result = self.bookController.search_books(
            self.current_book_queries[0], self.current_book_queries[1],
            self.current_book_queries[2])
        i = 0
        current_row_count = query_result.count()
        self.ui.bookTableWidget.setRowCount(current_row_count)
        for record in query_result:
            self.ui.bookTableWidget.setItem(
                i, 0, QtWidgets.QTableWidgetItem(record["title"]))
            self.ui.bookTableWidget.setItem(
                i, 1, QtWidgets.QTableWidgetItem(record["author"]))
            self.ui.bookTableWidget.setItem(
                i, 2, QtWidgets.QTableWidgetItem(record["year"]))
            self.ui.bookTableWidget.setItem(
                i, 3, QtWidgets.QTableWidgetItem(str(record["_id"])))
            if record["isAvailable"] is False:
                self.ui.bookTableWidget.item(i, 0).setBackground(
                    QtGui.QColor(255, 68, 68))
                self.ui.bookTableWidget.item(i, 1).setBackground(
                    QtGui.QColor(255, 68, 68))
                self.ui.bookTableWidget.item(i, 2).setBackground(
                    QtGui.QColor(255, 68, 68))
            i = i + 1
        self.show()

    def logout(self):
        import views.LoginView as LoginView
        self.librarianHome.hide()
        self.ui = LoginView.LoginView()
        self.ui.show()

    def show(self):
        self.librarianHome.show()

    def update_user_form(self):
        selected_row = self.ui.userTableWidget.currentRow()
        if selected_row > -1:
            selected_user_id = self.ui.userTableWidget.item(selected_row,
                                                            3).text()
            self.userRegisterScreen.type = 1
            self.userRegisterScreen.currentUser = self.userController.get_user_by_id(
                str(selected_user_id))
            self.userRegisterScreen.prepare_update(
                self.userRegisterScreen.currentUser)
            self.userRegisterScreen.userRegisterFormScreen.exec_()
            self.list_users()

    def update_book_form(self):
        selected_row = self.ui.bookTableWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.bookTableWidget.item(selected_row,
                                                            3).text()
            self.bookRegisterScreen.type = 1
            self.bookRegisterScreen.currentBook = self.bookController.get_book_by_id(
                str(selected_book_id))
            self.bookRegisterScreen.prepare_update(
                self.bookRegisterScreen.currentBook)
            self.bookRegisterScreen.bookRegisterFormScreen.exec_()
            self.list_books()

    def register_user_form(self):
        self.userRegisterScreen.type = 0
        self.userRegisterScreen.clear_forms()
        self.userRegisterScreen.userRegisterFormScreen.exec_()
        self.list_users()

    def register_book_form(self):
        self.bookRegisterScreen.type = 0
        self.bookRegisterScreen.clear_forms()
        self.bookRegisterScreen.bookRegisterFormScreen.exec_()
        self.list_books()

    def search_users(self):
        search_username = self.ui.usernameLineEdit.text()
        search_name = self.ui.nameLineEdit.text()
        search_surname = self.ui.surnameLineEdit.text()
        self.current_user_queries = [
            search_username, search_name, search_surname
        ]
        self.list_users()

    def search_books(self):
        search_title = self.ui.titleLineEdit.text()
        search_author = self.ui.authorLineEdit.text()
        search_year = self.ui.yearLineEdit.text()
        self.current_book_queries = [search_title, search_author, search_year]
        self.list_books()

    def released_color_change(self):
        self.librarianHome.sender().setStyleSheet(
            "QPushButton {\n"
            "color: white;\n"
            "background-color: QLinearGradient( x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #dd7a23, stop: 0.1 #e87919, stop: 0.49 #ce650a, stop: 0.5 #c45d03, stop: 1 #d16304);\n"
            "font-size: 30px;\n"
            "border-radius:10px;\n"
            "}")

    def pressed_color_change(self):
        self.librarianHome.sender().setStyleSheet("QPushButton {\n"
                                                  "color: white;\n"
                                                  "background-color: red;\n"
                                                  "font-size: 30px;\n"
                                                  "border-radius:10px;\n"
                                                  "}")

    def set_button_effects(self):
        self.ui.logoutButton.released.connect(self.released_color_change)
        self.ui.logoutButton.pressed.connect(self.pressed_color_change)
        self.ui.userSearchButton.released.connect(self.released_color_change)
        self.ui.userSearchButton.pressed.connect(self.pressed_color_change)
        self.ui.userCreateButton.released.connect(self.released_color_change)
        self.ui.userCreateButton.pressed.connect(self.pressed_color_change)
        self.ui.userDeleteButton.released.connect(self.released_color_change)
        self.ui.userDeleteButton.pressed.connect(self.pressed_color_change)
        self.ui.userUpdateButton.released.connect(self.released_color_change)
        self.ui.userUpdateButton.pressed.connect(self.pressed_color_change)
        self.ui.bookSearchButton.released.connect(self.released_color_change)
        self.ui.bookSearchButton.pressed.connect(self.pressed_color_change)
        self.ui.bookDeleteButton.released.connect(self.released_color_change)
        self.ui.bookDeleteButton.pressed.connect(self.pressed_color_change)
        self.ui.bookCreateButton.released.connect(self.released_color_change)
        self.ui.bookCreateButton.pressed.connect(self.pressed_color_change)
        self.ui.bookResetButton.released.connect(self.released_color_change)
        self.ui.bookResetButton.pressed.connect(self.pressed_color_change)
        self.ui.bookUpdateButton.released.connect(self.released_color_change)
        self.ui.bookUpdateButton.pressed.connect(self.pressed_color_change)
        self.ui.userResetButton.released.connect(self.released_color_change)
        self.ui.userResetButton.pressed.connect(self.pressed_color_change)
Beispiel #3
0
class MemberHomeView(Ui_memberMainWindow):
    def __init__(self, user):
        self.memberHome = QtWidgets.QMainWindow()
        self.ui = Ui_memberMainWindow()
        self.ui.setupUi(self.memberHome)
        self.ui.logoutButton.clicked.connect(self.logout)
        self.confirm = confirmView.ConfirmView()
        self.error = errorView.ErrorView()
        self.notification = notificationView.NotificationView()
        self.userController = UserController()
        self.bookController = BookController()
        self.loanController = LoanController()
        self.loanController.update_fines()
        self.ui.searchButton.clicked.connect(self.search_books)
        self.ui.waitingListButton.clicked.connect(self.add_to_waiting_list)
        self.ui.checkoutButton.clicked.connect(self.checkout_book)
        self.ui.cancelSelected.clicked.connect(self.cancel_waiting)
        self.ui.returnBookButton.clicked.connect(self.return_operation)
        self.ui.payFineButton.clicked.connect(self.payment_operation)
        self.ui.resetButton.clicked.connect(self.reset_book_filters)
        self.ui.waitingListWidget.cellDoubleClicked.connect(self.show_book_info_waiting)
        self.ui.viewBookWidget.cellDoubleClicked.connect(self.show_book_info_view)
        self.ui.searchBookWidget.cellDoubleClicked.connect(self.show_book_info_search)
        self.currentUser = user
        self.payment = paymentView.PaymentView(0)
        self.prepare_scene()
        self.current_book_queries = ["", "", ""]
        self.update_lists()
        self.update_scene()
        self.set_button_effects()
        self.book_info = None
        self.check_notifications()

    def prepare_scene(self):
        self.ui.greetingLabel.setText("Hi, " + str(self.currentUser.name) + " " + str(self.currentUser.surname))
        self.ui.dateLabel.setText(datetime.datetime.now().strftime("%Y-%m-%d"))
        self.ui.searchBookWidget.setColumnCount(4)
        self.ui.viewBookWidget.setColumnCount(4)
        self.ui.waitingListWidget.setColumnCount(4)
        self.ui.searchBookWidget.setHorizontalHeaderLabels(["Title", "Author", "Year", "User ID"])
        self.ui.viewBookWidget.setHorizontalHeaderLabels(["Title", "Author", "Return Date", "User ID"])
        self.ui.waitingListWidget.setHorizontalHeaderLabels(["Title", "Author", "Published Year", "User ID"])

    def update_scene(self):
        self.currentUser = self.userController.get_user_by_id(self.currentUser.id)
        self.ui.lastBookLabel.setText("Last Loaned Book : " + self.currentUser.lastLoanedBook)
        self.ui.fineLabel.setText("Current Fine Amount : ₺" + str(self.currentUser.currentFine))
        self.ui.totalBooksLabel.setText("Total Loaned Books : " + str(self.currentUser.totalLoanedBooks))

    def logout(self):
        import views.LoginView as LoginView
        self.memberHome.hide()
        self.ui = LoginView.LoginView()
        self.ui.show()

    def show(self):
        self.memberHome.show()

    def reset_book_filters(self):
        self.current_book_queries = ["", "", ""]
        self.update_lists()

    def payment_operation(self):
        self.payment = paymentView.PaymentView(self.currentUser.currentFine)
        self.payment.paymentPrompt.exec_()
        if self.payment.paymentFlag:
            self.loanController.reset_user_loans(self.currentUser.id)
            self.currentUser.currentFine = 0
            self.currentUser.formerFine = 0
            self.userController.update_member_attributes(self.currentUser)
            self.loanController.update_fines()
        self.update_scene()

    def show_book_info_view(self):
        selected_row = self.ui.viewBookWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.viewBookWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))
            self.book_info = bookInfoView.BookInfoView(selected_book)
            self.book_info.update_scene()
            self.book_info.show()

    def show_book_info_search(self):
        selected_row = self.ui.searchBookWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.searchBookWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(selected_book_id)
            self.book_info = bookInfoView.BookInfoView(selected_book)
            self.book_info.update_scene()
            self.book_info.show()

    def show_book_info_waiting(self):
        selected_row = self.ui.waitingListWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.waitingListWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))
            self.book_info = bookInfoView.BookInfoView(selected_book)
            self.book_info.update_scene()
            self.book_info.show()

    def list_books(self):
        query_result = self.bookController.search_books(self.current_book_queries[0], self.current_book_queries[1],
                                                        self.current_book_queries[2])
        i = 0
        current_row_count = query_result.count()
        self.ui.searchBookWidget.setRowCount(current_row_count)

        for record in query_result:
            self.ui.searchBookWidget.setItem(i, 0, QtWidgets.QTableWidgetItem(record["title"]))
            self.ui.searchBookWidget.setItem(i, 1, QtWidgets.QTableWidgetItem(record["author"]))
            self.ui.searchBookWidget.setItem(i, 2, QtWidgets.QTableWidgetItem(record["year"]))
            self.ui.searchBookWidget.setItem(i, 3, QtWidgets.QTableWidgetItem(str(record["_id"])))
            if record["isAvailable"] is False:
                self.ui.searchBookWidget.item(i, 0).setBackground(QtGui.QColor(255, 68, 68))
                self.ui.searchBookWidget.item(i, 1).setBackground(QtGui.QColor(255, 68, 68))
                self.ui.searchBookWidget.item(i, 2).setBackground(QtGui.QColor(255, 68, 68))
            i = i + 1
        self.show()

    def search_books(self):
        search_title = self.ui.titleLineEdit.text()
        search_author = self.ui.authorLineEdit.text()
        search_year = self.ui.yearLineEdit.text()
        self.current_book_queries = [search_title, search_author, search_year]
        self.update_lists()

    def checkout_book(self):
        selected_row = self.ui.searchBookWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.searchBookWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))
            if selected_book.isAvailable is False:
                self.error.set_error_text("You can't checkout an unavailable book.")
                self.error.errorScreen.exec_()
            else:
                indicator = self.loanController.checkout_book(self.currentUser, selected_book)
                if indicator == 0:
                    self.error.set_error_text("You cannot checkout with existing fines.")
                    self.error.errorScreen.exec_()
                elif indicator == 1:
                    self.error.set_error_text("You cannot borrow more than 4 books.")
                    self.error.errorScreen.exec_()
                elif indicator == 2:
                    self.notification.set_notif_text("\""+selected_book.title+"\" is added to your books.")
                    self.notification.notificationScreen.exec_()
                self.update_lists()
                self.update_scene()

    def add_to_waiting_list(self):
        selected_row = self.ui.searchBookWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.searchBookWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))

            if selected_book.isAvailable is True:
                self.error.set_error_text("You can't add an available book to your waiting list.")
                self.error.errorScreen.exec_()
            elif selected_book.id in self.currentUser.loanedBooks:
                self.error.set_error_text("You already loaned this book.")
                self.error.errorScreen.exec_()
            elif selected_book.isAvailable is False:
                if str(selected_book.id) not in self.currentUser.waitingBooks:
                    self.currentUser.waitingBooks.append(str(selected_book.id))
                if str(self.currentUser.id) not in selected_book.waitingList:
                    selected_book.waitingList.append(str(self.currentUser.id))
                self.userController.update_member_attributes(self.currentUser)
                self.bookController.update_book_attributes(selected_book)
                self.update_lists()

    def update_lists(self):
        self.list_user_waiting_list()
        self.list_books()
        self.list_loaned_books()

    def check_notifications(self):
        for book in self.currentUser.waitingBooks:
            if self.bookController.get_book_by_id(book).isAvailable:
                self.notification.set_notif_text("\"" + self.bookController.get_book_by_id(book).title + "\"" +
                                                 " is available to rent.")
                self.notification.show()

    def list_user_waiting_list(self):
        i = 0
        current_row_count = len(self.currentUser.waitingBooks)
        self.ui.waitingListWidget.setRowCount(current_row_count)

        for record in self.currentUser.waitingBooks:
            waited_book = self.bookController.get_book_by_id(record)
            self.ui.waitingListWidget.setItem(i, 0, QtWidgets.QTableWidgetItem(waited_book.title))
            self.ui.waitingListWidget.setItem(i, 1, QtWidgets.QTableWidgetItem(waited_book.author))
            self.ui.waitingListWidget.setItem(i, 2, QtWidgets.QTableWidgetItem(waited_book.publishedYear))
            self.ui.waitingListWidget.setItem(i, 3, QtWidgets.QTableWidgetItem(str(waited_book.id)))
            if waited_book.isAvailable is False:
                self.ui.waitingListWidget.item(i, 0).setBackground(QtGui.QColor(255, 68, 68))
                self.ui.waitingListWidget.item(i, 1).setBackground(QtGui.QColor(255, 68, 68))
                self.ui.waitingListWidget.item(i, 2).setBackground(QtGui.QColor(255, 68, 68))
            i = i + 1
        self.show()

    def list_loaned_books(self):
        i = 0
        current_row_count = len(self.currentUser.loanedBooks)
        self.ui.viewBookWidget.setRowCount(current_row_count)

        for record in self.currentUser.loanedBooks:
            loaned_book = self.bookController.get_book_by_id(record)
            return_date = self.loanController.get_loan_record(loaned_book.id)["returnDate"]
            self.ui.viewBookWidget.setItem(i, 0, QtWidgets.QTableWidgetItem(loaned_book.title))
            self.ui.viewBookWidget.setItem(i, 1, QtWidgets.QTableWidgetItem(loaned_book.author))
            self.ui.viewBookWidget.setItem(i, 2, QtWidgets.QTableWidgetItem(return_date.strftime("%d %b %Y")))
            self.ui.viewBookWidget.setItem(i, 3, QtWidgets.QTableWidgetItem(str(loaned_book.id)))
            if datetime.datetime.now() > return_date:
                self.ui.viewBookWidget.item(i, 0).setBackground(QtGui.QColor(206,190,190))
                self.ui.viewBookWidget.item(i, 1).setBackground(QtGui.QColor(206,190,190))
                self.ui.viewBookWidget.item(i, 2).setBackground(QtGui.QColor(206,190,190))
            i = i + 1
        self.show()

    def cancel_waiting(self):
        selected_row = self.ui.waitingListWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.waitingListWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))
            self.confirm.confirm_flag = 0
            self.confirm.set_confirm_text("Are you sure you want to cancel waiting for this book?")
            self.confirm.confirmScreen.exec_()
            if self.confirm.confirm_flag is 1:
                self.currentUser.waitingBooks.remove(str(selected_book.id))
                selected_book.waitingList.remove(str(self.currentUser.id))
                self.userController.update_member_attributes(self.currentUser)
                self.bookController.update_book_attributes(selected_book)
        self.update_lists()

    def return_operation(self):
        selected_row = self.ui.viewBookWidget.currentRow()
        if selected_row > -1:
            selected_book_id = self.ui.viewBookWidget.item(selected_row, 3).text()
            selected_book = self.bookController.get_book_by_id(str(selected_book_id))
            self.confirm.confirm_flag = 0
            self.confirm.set_confirm_text("Are you sure you want to return this book?")
            self.confirm.confirmScreen.exec_()
            if self.confirm.confirm_flag is 1:
                self.loanController.return_book(self.currentUser, selected_book)
                self.update_lists()
                self.update_scene()

    def set_button_effects(self):
        self.ui.returnBookButton.released.connect(self.released_color_change)
        self.ui.returnBookButton.pressed.connect(self.pressed_color_change)
        self.ui.searchButton.released.connect(self.released_color_change)
        self.ui.searchButton.pressed.connect(self.pressed_color_change)
        self.ui.cancelSelected.released.connect(self.released_color_change)
        self.ui.cancelSelected.pressed.connect(self.pressed_color_change)
        self.ui.checkoutButton.released.connect(self.released_color_change)
        self.ui.checkoutButton.pressed.connect(self.pressed_color_change)
        self.ui.logoutButton.released.connect(self.released_color_change)
        self.ui.logoutButton.pressed.connect(self.pressed_color_change)
        self.ui.waitingListButton.released.connect(self.released_color_change)
        self.ui.waitingListButton.pressed.connect(self.pressed_color_change)
        self.ui.payFineButton.released.connect(self.released_color_change)
        self.ui.payFineButton.pressed.connect(self.pressed_color_change)
        self.ui.resetButton.released.connect(self.released_color_change)
        self.ui.resetButton.pressed.connect(self.pressed_color_change)
    def released_color_change(self):
        if self.memberHome.sender().text() == 'Add To Waiting List':
            self.memberHome.sender().setStyleSheet("QPushButton {\n"
                                                   "color: white;\n"
                                                   "background-color: QLinearGradient( x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #dd7a23, stop: 0.1 #e87919, stop: 0.49 #ce650a, stop: 0.5 #c45d03, stop: 1 #d16304);\n"
                                                   "font-size: 20px;\n"
                                                  "border-radius:10px;\n"

                                                   "}")
        else:
            self.memberHome.sender().setStyleSheet("QPushButton {\n"
                                                   "color: white;\n"
                                                   "background-color:  QLinearGradient( x1: 0, y1: 0, x2: 0, y2: 1, stop: 0 #dd7a23, stop: 0.1 #e87919, stop: 0.49 #ce650a, stop: 0.5 #c45d03, stop: 1 #d16304);\n"
                                                   "font-size: 30px;\n"
                                                  "border-radius:10px;\n"

                                                   "}")

    def pressed_color_change(self):
        if self.memberHome.sender().text() == 'Add To Waiting List':
            self.memberHome.sender().setStyleSheet("QPushButton {\n"
                                                   "color: white;\n"
                                                   "background-color: red;\n"
                                                   "font-size: 20px;\n"
                                                  "border-radius:10px;\n"

                                                   "}")
        else:
            self.memberHome.sender().setStyleSheet("QPushButton {\n"
                                                   "color: white;\n"
                                                   "background-color:red;\n"
                                                   "font-size: 30px;\n"
                                                  "border-radius:10px;\n"
                                                   "}")