[Mod] Set timeout for google api

konono · Feb 7, 2022 · a28c9dc · a28c9dc
1 parent 5fb2373
commit a28c9dc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/holoscope/core.py b/holoscope/core.py
@@ -4,6 +4,7 @@
 import importlib
 import json
 import logging
+import socket
 
 from apiclient.discovery import build
 from holoscope.config import ConfigLoader
@@ -16,6 +17,8 @@
 EXPOTER_PLUGIN_DIR = "holoscope.exporter_plugin"
 
 log = logging.getLogger(__name__)
+timeout_in_sec = 5
+socket.setdefaulttimeout(timeout_in_sec)
 
 
 class Holoscope(object):

diff --git a/holoscope/exporter_plugin/google_calendar.py b/holoscope/exporter_plugin/google_calendar.py
@@ -6,6 +6,7 @@
 import logging
 import os.path
 import pickle
+import socket
 import textwrap
 
 from ..datamodel import GCalEvent
@@ -18,8 +19,9 @@
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
 
-
 log = logging.getLogger(__name__)
+timeout_in_sec = 5
+socket.setdefaulttimeout(timeout_in_sec)
 
 CALENDAR_API_SERVICE_NAME = 'calendar'
 CALENDAR_API_VERSION = 'v3'