evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Fix /list if fileName string contains multiple quotes (#141)

This commit is contained in:
Shubham dubey 2020-07-29 20:53:34 +05:30 committed by lzzy12
parent 121fb52e03
commit 10593d8a66
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
bot/helper/mirror_utils/upload_utils/gdriveTools.py
View File

@ -407,8 +407,15 @@ class GoogleDriveHelper:
scopes=self.__OAUTH_SCOPE) scopes=self.__OAUTH_SCOPE)
return build('drive', 'v3', credentials=credentials, cache_discovery=False) 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): def drive_list(self, fileName):
msg = "" msg = ""
fileName = self.escapes(str(fileName))
# Create Search Query for API request. # Create Search Query for API request.
query = f"'{parent_id}' in parents and (name contains '{fileName}')" query = f"'{parent_id}' in parents and (name contains '{fileName}')"
response = self.__service.files().list(supportsTeamDrives=True, response = self.__service.files().list(supportsTeamDrives=True,