From b4d4172c991da073fa0fa8eabd62e9d2c14de5bb Mon Sep 17 00:00:00 2001
From: Levi <57452819+l3v11@users.noreply.github.com>
Date: Mon, 30 May 2022 14:29:17 +0600
Subject: [PATCH] Fix bot not cloning data into destination drives

---
 bot/helper/drive_utils/gdriveTools.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/bot/helper/drive_utils/gdriveTools.py b/bot/helper/drive_utils/gdriveTools.py
index 6921faee..76e61be9 100644
--- a/bot/helper/drive_utils/gdriveTools.py
+++ b/bot/helper/drive_utils/gdriveTools.py
@@ -14,6 +14,7 @@
 from telegram import InlineKeyboardMarkup
 from telegraph.exceptions import RetryAfterError
 
+from google.auth.transport.requests import Request
 from google.oauth2 import service_account
 from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
@@ -66,8 +67,10 @@ def authorize(self):
         if not USE_SERVICE_ACCOUNTS:
             if os.path.exists('token.json'):
                 creds = Credentials.from_authorized_user_file('token.json', self.__OAUTH_SCOPE)
+                if creds and creds.expired and creds.refresh_token:
+                    creds.refresh(Request())
             else:
-                LOGGER.error("token.json file is missing")
+                LOGGER.error("The token.json file is missing")
         else:
             LOGGER.info(f"Authorizing with {SERVICE_ACCOUNT_INDEX}.json file")
             creds = service_account.Credentials.from_service_account_file(
@@ -81,7 +84,11 @@ def alt_authorize(self):
             if os.path.exists('token.json'):
                 LOGGER.info("Authorizing with token.json file")
                 creds = Credentials.from_authorized_user_file('token.json', self.__OAUTH_SCOPE)
+                if creds and creds.expired and creds.refresh_token:
+                    creds.refresh(Request())
                 return build('drive', 'v3', credentials=creds, cache_discovery=False)
+            else:
+                LOGGER.error("The token.json file is missing")
         return None
 
     @staticmethod
@@ -274,7 +281,7 @@ def clone(self, link, key):
             name = meta.get("name")
             mime_type = meta.get("mimeType")
             if mime_type == self.__G_DRIVE_DIR_MIME_TYPE:
-                dir_id = self.create_directory(meta.get('name'), PARENT_ID)
+                dir_id = self.create_directory(meta.get('name'), parent_id)
                 self.cloneFolder(meta.get('name'), meta.get('name'), meta.get('id'), dir_id)
                 durl = self.__G_DRIVE_DIR_BASE_DOWNLOAD_URL.format(dir_id)
                 if self.is_cancelled:
@@ -292,7 +299,7 @@ def clone(self, link, key):
                     url = f'{index_url}/{url_path}/'
                     msg += f'<b> | <a href="{url}">Index Link</a></b>'
             else:
-                file = self.copyFile(meta.get('id'), PARENT_ID)
+                file = self.copyFile(meta.get('id'), parent_id)
                 msg += f'<b>Name:</b> <code>{file.get("name")}</code>'
                 if mime_type is None:
                     mime_type = 'File'