"""Function logic."""

from src.connectors import get_graphql_connector

with open('./src/query.gql') as f:
    GQL_QUERY = f.read()


def fetch_product_status_data(product_id):
    """Fetch product status data by product id."""
    response = get_graphql_connector().execute(GQL_QUERY, {'id': str(product_id)})
    return response['data']['product']


def get_formatted_product_status_data(product_id):
    """Get formatted product status data."""
    result = fetch_product_status_data(product_id)
    error_correction_status = ''
    release_correction_id = ''
    if result.get('releaseCorrection'):
        if 'status' in result['releaseCorrection']:
            error_correction_status = result['releaseCorrection']['status']
        if 'releaseCorrectionId' in result['releaseCorrection']:
            release_correction_id = result['releaseCorrection']['releaseCorrectionId']
    return {
        'release_status': result.get('status'),
        'error_correction_status': error_correction_status,
        'release_correction_id': release_correction_id
    }