import re
from common import util
from official.model import ProdCategory
from django.conf import settings
from common import curl_interface
import time


def get_prod_info(prod_id):
    """
    获取巴士产品信息
    :param prod_id:
    :return:
    """
    # 参数校验
    if re.match('^[1-9][0-9]{0,10}', prod_id) is None:
        return util.re_err('参数错误')

    # 数据获取
    get_info = ProdCategory.get_bus_line_prod(prod_id)
    if get_info['category_id'] != 1:
        return util.re_err('该产品不是巴士产品')

    # 获取价格日历
    date_price = get_date_price(prod_id)
    if date_price['flag'] is False:
        return date_price
    res = {
        'prod_info': get_info,
        'date_list': date_price['data']['list']
    }
    return util.re_su('', res)


def get_line_stock(prod_id, date):
    """
    获取产品库存
    :param prod_id:
    :param date:
    :return:
    """
    send_data = {
        'pro_cate_id': prod_id,
        'date': date
    }
    curl = curl_interface.CurlInterface(send_data, 4)
    get_stock = curl.exec(settings.WX_HOST + '/zzcx/home/bus/get-line-bus-stock', 'POST')
    return get_stock


def get_date_price(prod_id):
    """
    获取价格日历
    :param line_id:
    :return:
    """
    send_data = {
        'pro_cate_id': prod_id,
    }
    curl = curl_interface.CurlInterface(send_data, 4)
    ticket_arr_cs = curl.exec(settings.WX_HOST + '/zzcx/home/bus/get-line-sale-date', 'POST')
    return ticket_arr_cs


def get_prod_array(prod_id):
    """
    获取票种数组
    :param prod_id:
    :return:
    """
    get_arr = ProdCategory.get_prod_array(prod_id)
    return util.re_su('', {'list': get_arr})