GoogleDrive: fix parent_id always None if input not url

Change-Id: I8532f357b4770a15f09033d185a073db16b2a695
deepak46810 · Sep 9, 2021 · 11602d7 · 11602d7
1 parent 4097aff
commit 11602d7
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/bot/plugins/gdrive.py b/bot/plugins/gdrive.py
@@ -49,9 +49,12 @@
                     r"([a-zA-Z0-9][a-zA-Z0-9-]+[a-zA-Z0-9])\.[^\s]{2,}")
 
 
-def getIdFromUrl(url: str) -> Optional[str]:
+def getIdFromUrl(url: Optional[str]) -> Optional[str]:
+    if not url:
+        return None
+
     match = PATTERN.search(url)
-    return match[0] if match else None
+    return match[0] if match else url
 
 
 class GoogleDrive(plugin.Plugin):
@@ -652,9 +655,7 @@ async def cmd_gdsearch(self, ctx: command.Context) -> Optional[str]:
             return "__Invalid parameters of input.__"
 
         name = options.get("name")
-        parent = options.get("parent")
-        if parent:
-            parent = getIdFromUrl(parent)
+        parent = getIdFromUrl(options.get("parent"))
 
         limit = int(options.get("limit", 15))
         if limit > 1000: