"""DataDog Trace."""

# DataDog APM bootstrap code
import ddtrace

from users import (
    config,
    handlers,  # noqa
)

# patch as soon as possible to ensure modules report to datadog apm
ddtrace.patch_all(futures=True, httplib=True, sqlalchemy=True)  # noqa
# Add Orchard Headers to DataDog APM

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

if ddtrace.tracer.enabled:
    ddtrace.config.trace_headers(
        [
            'Grass-Account-Type',
            'Grass-Account-Id',
            'Orchard-User-Id',
            'Orchard-Profile-Type',
            'Orchard-Profile-Id',
            'Orchard-Identity-Id',
            'Correlation-Id',
            'User-Agent',  # necessary for the core-load-tester service to work
        ]
    )