"""Model for the graphql-router service."""

from product_review.util.graphql import make_request, strip_query
from product_review.constants import context

def get_vendor_product_manager(vendor_id):
    """Get vendor product manager details."""
    query_string = """
    query vendorProductManager($vendorId: Int!) {
        vendor(vendorId: $vendorId) {
            productManager
        }
    }
    """
    params = {"vendorId": vendor_id}
    response = make_request(
        strip_query(query_string), params, context.GRAPHQL_USER_HEADERS)
    return response['vendor']['productManager']


def get_vendor_company_brand(vendor_id):
    """Get vendor company brand details."""
    query_string = """
    query vendorCompanyBrand($vendorId: Int!) {
        vendor(vendorId: $vendorId) {
            companyBrand {
                name
            }
        }
    }
    """
    params = {"vendorId": vendor_id}
    response = make_request(
        strip_query(query_string), params, context.GRAPHQL_USER_HEADERS)
    return response['vendor']['companyBrand']