from official.logic import reserve_bus_logic as logic
from common import util


def get_prod_info(request):
    """
    巴士产品获取
    :param request:
    :return:
    """
    # 获取参数
    prod_id = request.POST.get('prod_id', '')  # 产品ID
    get_info = logic.get_prod_info(prod_id)  # 获取产品信息
    return util.re_json(get_info)


def get_line_stock(request):
    """
    获取主线路库存信息
    :param request:
    :return:
    """
    prod_id = request.POST.get('prod_id', '')  # 产品ID
    date = request.POST.get('date', '')  # 日期
    get_stock = logic.get_line_stock(prod_id, date)
    return util.re_json(get_stock)


def get_prod_array(request):
    """
    获取产品数组
    :param request:
    :return:
    """
    prod_id = request.POST.get('prod_id', '')  # 子线路产品ID
    get_prod = logic.get_prod_array(prod_id)
    return util.re_json(get_prod)