docker-python/patches/sitecustomize.py at master · HpcDataLab/docker-python · GitHub

History

74 lines (64 loc) · 2.57 KB

Raw

import os

from log import Log

import sys

import importlib

import importlib.machinery

class GcpModuleFinder(importlib.abc.MetaPathFinder):

_MODULES = [

'google.cloud.bigquery',

'google.cloud.storage',

'google.cloud.automl_v1beta1',

'google.cloud.translate',

'google.cloud.translate_v2',

'google.cloud.translate_v3',

'google.cloud.language',

'google.cloud.language_v1',

'google.cloud.videointelligence',

'google.cloud.videointelligence_v1',

'google.cloud.vision',

'google.cloud.vision_v1',

]

_KAGGLE_GCP_PATH = 'kaggle_gcp.py'

def __init__(self):

pass

def _is_called_from_kaggle_gcp(self):

import inspect

for frame in inspect.stack():

if os.path.basename(frame.filename) == self._KAGGLE_GCP_PATH:

return True

return False

def find_spec(self, fullname, path, target=None):

if fullname in self._MODULES:

# If being called from kaggle_gcp, don't return our

# monkeypatched module to avoid circular dependency,

# since we call kaggle_gcp to load the module.

if self._is_called_from_kaggle_gcp():

return None

return importlib.machinery.ModuleSpec(fullname, GcpModuleLoader())

class GcpModuleLoader(importlib.abc.Loader):

def __init__(self):

pass

def create_module(self, spec):

"""Create the gcp module from the spec.

"""

import kaggle_gcp

_LOADERS = {

'google.cloud.bigquery': kaggle_gcp.init_bigquery,

'google.cloud.storage': kaggle_gcp.init_gcs,

'google.cloud.automl_v1beta1': kaggle_gcp.init_automl,

'google.cloud.translate': kaggle_gcp.init_translation_v3,

'google.cloud.translate_v2': kaggle_gcp.init_translation_v2,

'google.cloud.translate_v3': kaggle_gcp.init_translation_v3,

'google.cloud.language': kaggle_gcp.init_natural_language,

'google.cloud.language_v1': kaggle_gcp.init_natural_language,

'google.cloud.videointelligence': kaggle_gcp.init_video_intelligence,

'google.cloud.videointelligence_v1': kaggle_gcp.init_video_intelligence,

'google.cloud.vision': kaggle_gcp.init_vision,

'google.cloud.vision_v1': kaggle_gcp.init_vision

}

monkeypatch_gcp_module = _LOADERS[spec.name]()

return monkeypatch_gcp_module

def exec_module(self, module):

pass

if not hasattr(sys, 'frozen'):

sys.meta_path.insert(0, GcpModuleFinder())

docker-python/patches/sitecustomize.py at master · HpcDataLab/docker-python · GitHub

Expand file tree

Breadcrumbs

sitecustomize.py

Latest commit

History

Breadcrumbs

sitecustomize.py

File metadata and controls