App: Type to go to folder

[skip ci]

Author: khemarato

scripts/gdrive.py

@@ -169,16 +169,19 @@ def add_tracked_folder(slug, public, private, gfolders=None):
   FOLDERS_DATA_FILE.write_text(json.dumps(gfolders, sort_keys=True, indent=2))
   return gfolders
 
-def get_gfolders_for_course(course):
+def get_gfolders_for_course(course, invite_to_add=True):
   """Returns a (public, private) tuple of GIDs given a human course string"""
   gfolders = FOLDERS_DATA()
   parts = course.split('/')
   course = parts[0]
   if course not in gfolders:
-    print("Hmmm... I don't know that Google Drive folder! Let's add it:")
-    publicurl = input("Public link: ") or None
-    privateurl = input("Private link: ") or None
-    gfolders = add_tracked_folder(course, publicurl, privateurl, gfolders=gfolders)
+    if invite_to_add:
+      print("Hmmm... I don't know that Google Drive folder! Let's add it:")
+      publicurl = input("Public link: ") or None
+      privateurl = input("Private link: ") or None
+      gfolders = add_tracked_folder(course, publicurl, privateurl, gfolders=gfolders)
+    else:
+      raise FileNotFoundError("I don't know that one!")
 
   private_folder = folderlink_to_id(gfolders[course]['private'])
   public_folder = folderlink_to_id(gfolders[course]['public'])
@@ -208,7 +211,10 @@ def get_gfolders_for_course(course):
     if found:
       del parts[0]
       continue
-    print(f"No subfolder found matching \"{q}\"")
+    err = f"No subfolder found matching \"{q}\""
+    if not invite_to_add:
+      raise FileNotFoundError(err)
+    print(err)
     q = input_with_prefill("Create new subfolder: ", titlecase(parts[0]))
     if not q:
       if public_folder:

scripts/gdrive_app.py

@@ -7,7 +7,7 @@
 
 from PySide6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
                                QHBoxLayout, QListWidget, QListWidgetItem,
-                               QPushButton, QLineEdit, QSplitter,
+                               QPushButton, QLineEdit, QSplitter, QMessageBox,
                                QListView, QMenu, QProgressDialog)
 from PySide6.QtCore import Qt, QSize
 from PySide6.QtGui import QIcon, QPixmap, QShortcut, QKeySequence
@@ -301,7 +301,7 @@ def init_ui(self):
         self.fwd_btn.clicked.connect(self.go_forward)
 
         self.address_bar = QLineEdit()
-        self.address_bar.setReadOnly(True)
+        self.address_bar.returnPressed.connect(self.on_address_bar_return)
 
         top_bar.addWidget(self.back_btn)
         top_bar.addWidget(self.fwd_btn)
@@ -433,6 +433,36 @@ def on_nav_clicked(self, item: QListWidgetItem):
         root_type = item.data(Qt.UserRole)
         self.load_root(root_type)
 
+    def on_address_bar_return(self):
+        if not self.gcache:
+            QMessageBox.information(self, "Loading", "Please wait for the cache to finish loading.")
+            return
+
+        query = self.address_bar.text()
+        if not query:
+            return
+        
+        import gdrive
+        try:
+            # resulting_tuple is (public_folder, private_folder)
+            resulting_tuple = gdrive.get_gfolders_for_course(query, invite_to_add=False)
+            folder_id = resulting_tuple[1] or resulting_tuple[0]
+            if not folder_id:
+                return
+            
+            # Use gcache to get the folder's name
+            folder_item = self.gcache.get_item(folder_id)
+            if folder_item:
+                folder_name = folder_item.get('name', query)
+            else:
+                folder_name = query
+                
+            self.load_folder(folder_id, folder_name)
+        except FileNotFoundError as e:
+            QMessageBox.warning(self, "Not Found", str(e))
+        except Exception as e:
+            QMessageBox.critical(self, "Error", f"An error occurred: {e}")
+
     def populate_files(self, items: List[Dict[str, Any]]):
         if hasattr(self, 'current_cancel_flag'):
             self.current_cancel_flag[0] = True