From 77175cbb3a189e0250bbc01a1e08bac2d0767e97 Mon Sep 17 00:00:00 2001
From: h7x4 <h7x4@nani.wtf>
Date: Tue, 14 May 2024 03:42:33 +0200
Subject: [PATCH] cli/bookcase_item: add command to extend borrowing

---
 README.md                              |  2 +-
 worblehat/cli/subclis/bookcase_item.py | 54 +++++++++++++++++++++++++-
 2 files changed, 53 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 5382fec..e773b1b 100644
--- a/README.md
+++ b/README.md
@@ -64,7 +64,7 @@ Run `poetry run worblehat --help` for more info
 - [X] Ability to borrow and deliver items
 - [ ] Ability to borrow and deliver multiple items at a time
 - [X] Ability to enter the queue for borrowing an item
-- [ ] Ability to extend a borrowing, only if no one is behind you in the queue
+- [x] Ability to extend a borrowing, only if no one is behind you in the queue
 - [X] Ability to list borrowed items which are overdue
 - [~] Ability to search for items
 - [ ] Ability to print PVV-specific labels for items missing a label, or which for any other reason needs a custom one
diff --git a/worblehat/cli/subclis/bookcase_item.py b/worblehat/cli/subclis/bookcase_item.py
index 2672a50..40e3a22 100644
--- a/worblehat/cli/subclis/bookcase_item.py
+++ b/worblehat/cli/subclis/bookcase_item.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timedelta
 from textwrap import dedent
 from sqlalchemy import select
 
@@ -7,6 +7,7 @@ from sqlalchemy.orm import Session
 from worblehat.cli.prompt_utils import (
     InteractiveItemSelector,
     NumberedCmd,
+    NumberedItemSelector,
     format_date,
     prompt_yes_no,
 )
@@ -22,6 +23,7 @@ from worblehat.services.bookcase_item import (
     create_bookcase_item_from_isbn,
     is_valid_isbn,
 )
+from worblehat.services.config import Config
 
 from .bookcase_shelf_selector import select_bookcase_shelf
 
@@ -174,6 +176,50 @@ class BookcaseItemCli(NumberedCmd):
         print(f'Successfully delivered the item for {borrowing.username}')
 
 
+    def do_extend_borrowing(self, _: str):
+        borrowings = self.sql_session.scalars(
+            select(BookcaseItemBorrowing)
+            .join(BookcaseItem, BookcaseItem.uid == BookcaseItemBorrowing.fk_bookcase_item_uid)
+            .where(BookcaseItem.isbn == self.bookcase_item.isbn)
+            .order_by(BookcaseItemBorrowing.username)
+        ).all()
+
+        if len(borrowings) == 0:
+            print('No one seems to have borrowed this item')
+            return
+
+        borrowing_queue = self.sql_session.scalars(
+            select(BookcaseItemBorrowingQueue)
+            .where(
+                BookcaseItemBorrowingQueue.item == self.bookcase_item,
+                BookcaseItemBorrowingQueue.item_became_available_time == None,
+            )
+            .order_by(BookcaseItemBorrowingQueue.entered_queue_time)
+        ).all()
+
+        if len(borrowing_queue) != 0:
+          print('Sorry, you cannot extend the borrowing because there are people waiting in the queue')
+          print('Borrowing queue:')
+          for i, b in enumerate(borrowing_queue):
+              print(f'  {i + 1}) {b.username}')
+          return
+
+        print('Who are you?')
+        selector = NumberedItemSelector(
+            items = list(borrowings),
+            stringify = lambda b: f'{b.username} - Until {format_date(b.end_time)}',
+        )
+        selector.cmdloop()
+        if selector.result is None:
+            return
+        borrowing = selector.result
+
+        borrowing.end_time = datetime.now() + timedelta(days=int(Config['deadline_daemon.days_before_queue_position_expires']))
+        self.sql_session.flush()
+
+        print(f'Successfully extended the borrowing for {borrowing.username} until {format_date(borrowing.end_time)}')
+
+
     def do_done(self, _: str):
         return True
 
@@ -188,10 +234,14 @@ class BookcaseItemCli(NumberedCmd):
             'doc': 'Deliver',
         },
         3: {
+            'f': do_extend_borrowing,
+            'doc': 'Extend borrowing',
+        },
+        4: {
             'f': do_edit,
             'doc': 'Edit',
         },
-        4: {
+        5: {
             'f': do_update_data,
             'doc': 'Pull updated data from online databases',
         },