import botocore.session
from aws_secretsmanager_caching import SecretCache, SecretCacheConfig
import os


def upload_junit_to_datadog():
    client = botocore.session.get_session()\
        .create_client('secretsmanager', region_name='us-east-1')
    cache_config = SecretCacheConfig()
    cache = SecretCache(config=cache_config, client=client)

    secret = cache.get_secret_string('prod/datadog/DD_API_KEY')
    mobile_os = os.environ.get('OS')
    brand = os.environ.get('BRAND')
    service = f'{brand}-{mobile_os}-' \
              f'{os.environ.get("JOB_NAME", "mobile-test-mode")}'
    command = f'DD_ENV=qa DATADOG_API_KEY={secret} DATADOG_SITE=datadoghq.com \
        yarn datadog-ci junit upload \
        --service {service} \
        --tags test.environment:browserstack \
        junit'

    stream = os.popen(command)
    output = stream.read()
    print(output)