"""DataDog Trace."""
# Patch as soon as possible to ensure modules report to datadog apm
# We manually patch to avoid dd-trace run dupping logs until this is resolved
# via https://github.com/DataDog/dd-trace-py/pull/1690
import os

import ddtrace
from owslogger.utils import datadog

from backend import config

if config.ENVIRONMENT not in (config.QA_ENVIRONMENT, config.PROD_ENVIRONMENT):
    ddtrace.tracer.enabled = False

ddtrace.patch_all(futures=True, httplib=True, SQLAlchemy=True)
if config.ENVIRONMENT in (config.QA_ENVIRONMENT, config.PROD_ENVIRONMENT):
    config.SERVICE_VERSION = datadog.set_app_version()
ddtrace.tracer.set_tags(
    {ddtrace.constants.ENV_KEY: os.environ.get('DATADOG_ENV'),
     ddtrace.constants.SERVICE_KEY: os.environ.get('DD_SERVICE'),
     ddtrace.constants.VERSION_KEY: os.environ.get('DD_VERSION')})