Refactor/google drive module

.devcontainer/devcontainer.json

@@ -11,7 +11,6 @@
 				"ms-python.python",
 				"ms-python.vscode-pylance",
 				"ms-python.flake8",
-				"ms-python.pylint",
 				"ms-python.mypy-type-checker",
 				"mtxr.sqltools",
 				"mtxr.sqltools-driver-pg",

.vscode/settings.json

@@ -1,3 +1,9 @@
 {
     "python.pylintPath": "/usr/local/py-utils/bin/pylint",
+    "mypy-type-checker.cwd": "${workspaceFolder}/app",
+    "pylint.cwd": "${workspaceFolder}/app",
+    "pylint.path": [
+        "[\"../../usr/local/py-utils/bin/pylint\"]"
+    ],
+    "flake8.cwd": "${workspaceFolder}/app"
 }

app/commands/google_service.py

@@ -1,17 +1,19 @@
 """Testing new google service (will be removed)"""
 import os
 
-from integrations.google_workspace.google_service import get_google_service
+from integrations.google_workspace import google_drive
 from dotenv import load_dotenv
 
 load_dotenv()
 
 SRE_DRIVE_ID = os.environ.get("SRE_DRIVE_ID")
 SRE_INCIDENT_FOLDER = os.environ.get("SRE_INCIDENT_FOLDER")
+INCIDENT_TEMPLATE = os.environ.get("INCIDENT_TEMPLATE")
 
 
-def open_modal(client, body):
-    folders = list_folders()
+def open_modal(client, body, folders):
+    if not folders:
+        return
     folder_names = [i["name"] for i in folders]
     blocks = [
         {"type": "section", "text": {"type": "mrkdwn", "text": f"*{name}*"}}
@@ -25,25 +27,10 @@ def open_modal(client, body):
     client.views_open(trigger_id=body["trigger_id"], view=view)
 
 
-def google_service_command(client, body):
-    open_modal(client, body)
-
-
-def list_folders():
-    service = get_google_service("drive", "v3")
-    results = (
-        service.files()
-        .list(
-            pageSize=25,
-            supportsAllDrives=True,
-            includeItemsFromAllDrives=True,
-            corpora="drive",
-            q="parents in '{}' and mimeType = 'application/vnd.google-apps.folder' and trashed=false and not name contains '{}'".format(
-                SRE_INCIDENT_FOLDER, "Templates"
-            ),
-            driveId=SRE_DRIVE_ID,
-            fields="nextPageToken, files(id, name)",
-        )
-        .execute()
-    )
-    return results.get("files", [])
+def google_service_command(client, body, respond):
+    respond(f"Healthcheck status: {google_drive.healthcheck()}")
+    folders = google_drive.list_folders_in_folder(SRE_INCIDENT_FOLDER)
+    if not folders:
+        respond("The folder ID is invalid. Please check the environment variables.")
+        return
+    open_modal(client, body, folders)

app/commands/sre.py

@@ -21,6 +21,8 @@
 \n      - show the version of the SRE Bot
 \n      - montre la version du bot SRE"""
 
+PREFIX = os.environ.get("PREFIX", "")
+
 
 def sre_command(ack, command, logger, respond, client, body):
     ack()
@@ -47,8 +49,12 @@ def sre_command(ack, command, logger, respond, client, body):
             webhook_helper.handle_webhook_command(args, client, body, respond)
         case "version":
             respond(f"SRE Bot version: {os.environ.get('GIT_SHA', 'unknown')}")
-        case "google-service":
-            google_service.google_service_command(client, body)
+        case "google":
+            if PREFIX == "dev-":
+                google_service.google_service_command(client, body, respond)
+            else:
+                respond("This command is only available in the dev environment.")
+            return
         case _:
             respond(
                 f"Unknown command: `{action}`. Type `/sre help` to see a list of commands. \nCommande inconnue: `{action}`. Entrez `/sre help` pour une liste des commandes valides"

app/integrations/google_workspace/google_docs.py

@@ -0,0 +1,4 @@
+"""Google Docs module.
+
+This module provides functions to create and manipulate Google Docs.
+"""