• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• sroka-master
  • images
  • config.sample.ini
  • LICENSE
  • Test APIs.ipynb
  • setup.py
  • .githooks
    • pre-commit
    • isort_hooks.py
  • setup.cfg
  • README.md
  • requirements.txt
  • sroka
    • api
      • s3_connection
        • s3_connection_api.py
        • __init__.py
        • README.md
      • qubole
        • qubole_api.py
        • query_result_file.py
        • __init__.py
        • README.md
      • google_drive
        • google_drive_api.py
        • __init__.py
        • README.md
      • google_ad_manager
        • gam_api.py
        • __init__.py
        • README.md
      • neo4j
        • __init__.py
        • README.md
        • neo4j_api.py
      • mysql
        • mysql.py
        • mysql_helpers.py
        • __init__.py
        • README.md
      • ga
        • ga.py
        • __init__.py
        • README.md
      • athena
        • athena_api.py
        • athena_api_helpers.py
        • __init__.py
        • README.md
      • __init__.py
      • moat
        • moat_api_helpers.py
        • __init__.py
        • README.md
        • moat_api.py
      • google_bigquery
        • __init__.py
        • bigquery_api.py
        • README.md
      • rubicon
        • __init__.py
        • README.md
        • rubicon_api.py
    • config
      • config.py
      • __init__.py
    • __init__.py
  • CODEOWNERS
  • .gitignore

import configparser
import json
import os
import stat

from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow

default_config_filepath = os.path.expanduser('~/.sroka_config/')


def set_google_credentials(authorized_user_file,
                           key_file_location,
                           scope):

    try:
        credentials = Credentials.from_authorized_user_file(authorized_user_file, scopes=[scope])
    except FileNotFoundError:
        flow = InstalledAppFlow.from_client_secrets_file(key_file_location, [scope])
        credentials = flow.run_console()
        os.makedirs(os.path.expanduser('~/.cache/'), exist_ok=True)
        with open(authorized_user_file, 'w') as file:
            json.dump({
                'client_id': credentials._client_id,
                'client_secret': credentials._client_secret,
                'refresh_token': credentials._refresh_token
            }, file)
        os.chmod(authorized_user_file, 0o600)
    return credentials


def get_value(group, key):
    """Function that gets configuration from config file."""

    config_path = default_config_filepath + "config.ini"
    if 'CONFIG_FILE_PATH' in os.environ:
        config_path = os.environ['CONFIG_FILE_PATH']

    config = configparser.ConfigParser()
    readable_by_others = stat.S_IRGRP | stat.S_IROTH
    is_protected = not bool(os.stat(config_path).st_mode & readable_by_others)
    if not is_protected:
        raise Exception("Configuration file {config} is not protected, " +
                        "make sure you're the only one allowed to read " +
                        "it by executing `chmod 600 {config}`".format(config=config_path))
    config.read(config_path)
    return config.get(group, key)


def has_value(group, key):
    try:
        get_value(group, key)
        return True
    except (configparser.NoSectionError, configparser.NoOptionError):
        return False


def get_file_path(group):
    """Function that gets configuration files for APIs."""
    if group == 'google_ad_manager':
        return os.environ.get('CONFIG_AD_MANAGER', os.path.join(default_config_filepath, 'ad_manager.json'))
    if group == 'google_analytics':
        return os.environ.get('CONFIG_GOOGLE_ANALYTICS', os.path.join(default_config_filepath, 'client_secrets.json'))
    if group == 'google_drive':
        return os.environ.get('CONFIG_GOOGLE_DRIVE', os.path.join(default_config_filepath, 'credentials.json'))
    if group == 'google_bigquery':
        return os.environ.get('GOOGLE_APPLICATION_CREDENTIALS',
                              os.path.join(default_config_filepath, 'bigquery_credentials.json'))


def setup_env_variables(filepath=None):
    """Function that sets configuration file path."""
    config_filepath = filepath
    if not os.path.exists(config_filepath):
        raise Exception('Config file {} was not found!'.format(config_filepath))
    os.environ['CONFIG_FILE_PATH'] = config_filepath


def setup_admanager_config(filepath=None):
    """Function that sets configuration for Ad Manager file path."""
    config_filepath = filepath
    if not os.path.exists(config_filepath):
        raise Exception('Config file {} was not found!'.format(config_filepath))
    os.environ['CONFIG_AD_MANAGER'] = config_filepath


def setup_client_secret(filepath=None):
    """Function that sets configuration for Google Analytics file path."""
    config_filepath = filepath
    if not os.path.exists(config_filepath):
        raise Exception('Config file {} was not found!'.format(config_filepath))
    os.environ['CONFIG_GOOGLE_ANALYTICS'] = config_filepath


def setup_bigquery_config(filepath=None):
    """Function that sets configuration for BigQuery file path."""
    config_filepath = filepath
    if not os.path.exists(config_filepath):
        raise Exception('Config file {} was not found!'.format(config_filepath))
    os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = config_filepath


def setup_google_sheets_credentials(filepath=None):
    """Function that sets configuration for Google Sheets file path."""
    config_filepath = filepath
    if not os.path.exists(config_filepath):
        raise Exception('Config file {} was not found!'.format(config_filepath))
    os.environ['CONFIG_GOOGLE_DRIVE'] = config_filepath