minigo/cloud_logging.py at eval_server · gms2009/minigo

Repository navigation

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"""Helpers to configure cloud logging."""

import logging

import contextlib

import io

import sys

import os

import google.cloud.logging as glog

LOGGING_PROJECT = os.environ.get('LOGGING_PROJECT', '')

def configure(project=LOGGING_PROJECT):

"""Configures cloud logging

This is called for all main calls. If a $LOGGING_PROJECT is environment

variable configured, then STDERR and STDOUT are redirected to cloud

logging.

"""

if not project:

sys.stderr.write('!! Error: The $LOGGING_PROJECT enviroment '

'variable is required in order to set up cloud logging. '

'Cloud logging is disabled.\n')

return

try:

# if this fails, redirect stderr to /dev/null so no startup spam.

with contextlib.redirect_stderr(io.StringIO()):

client = glog.Client(project)

client.setup_logging(logging.INFO)

except:

logging.basicConfig(level=logging.INFO)

sys.stderr.write('!! Cloud logging disabled\n')