From 10593d8a66a614e5452207e7151c0b39993df24f Mon Sep 17 00:00:00 2001 From: Shubham dubey Date: Wed, 29 Jul 2020 20:53:34 +0530 Subject: [PATCH] Fix /list if fileName string contains multiple quotes (#141) --- bot/helper/mirror_utils/upload_utils/gdriveTools.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/bot/helper/mirror_utils/upload_utils/gdriveTools.py b/bot/helper/mirror_utils/upload_utils/gdriveTools.py index e918506..554c789 100644 --- a/bot/helper/mirror_utils/upload_utils/gdriveTools.py +++ b/bot/helper/mirror_utils/upload_utils/gdriveTools.py @@ -407,8 +407,15 @@ class GoogleDriveHelper: scopes=self.__OAUTH_SCOPE) return build('drive', 'v3', credentials=credentials, cache_discovery=False) + def escapes(self, str): + chars = ['\\', "'", '"', r'\a', r'\b', r'\f', r'\n', r'\r', r'\t'] + for char in chars: + str = str.replace(char, '\\'+char) + return str + def drive_list(self, fileName): msg = "" + fileName = self.escapes(str(fileName)) # Create Search Query for API request. query = f"'{parent_id}' in parents and (name contains '{fileName}')" response = self.__service.files().list(supportsTeamDrives=True,