From f56bde877de071e8d2901b74de7a4fbd7a3ba162 Mon Sep 17 00:00:00 2001 From: Anas <77075674+anasty17@users.noreply.github.com> Date: Tue, 29 Jun 2021 20:54:56 +0300 Subject: [PATCH] Random SA at bot startup (#249) --- bot/helper/mirror_utils/upload_utils/gdriveTools.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/bot/helper/mirror_utils/upload_utils/gdriveTools.py b/bot/helper/mirror_utils/upload_utils/gdriveTools.py index 8b47404..93e2615 100644 --- a/bot/helper/mirror_utils/upload_utils/gdriveTools.py +++ b/bot/helper/mirror_utils/upload_utils/gdriveTools.py @@ -8,6 +8,7 @@ import re import json import requests import logging +from random import randrange from google.auth.transport.requests import Request from google.oauth2 import service_account @@ -27,7 +28,8 @@ from bot.helper.ext_utils.fs_utils import get_mime_type, get_path_size LOGGER = logging.getLogger(__name__) logging.getLogger('googleapiclient.discovery').setLevel(logging.ERROR) -SERVICE_ACCOUNT_INDEX = 0 +if USE_SERVICE_ACCOUNTS: + SERVICE_ACCOUNT_INDEX = randrange(len(os.listdir("accounts"))) TELEGRAPHLIMIT = 80 class GoogleDriveHelper: