# -*- coding: utf-8 -*-

from __future__ import division

from django.shortcuts import HttpResponse

from stock.models import StockInfo
from utils.redis.connect import r
from utils.redis.rkeys import JDJOS_STOCK_CHANGED_SET
from utils.stock_utils import refresh_stock_info, send_stock_info


def refresh_stockinfo(request):
    refresh_stock_info()
    return HttpResponse('Refresh Success')


def send_stockinfo(request):
    pks = r.smembers(JDJOS_STOCK_CHANGED_SET)
    for pk in pks:
        try:
            stock = StockInfo.objects.get(pk=pk, status=True)
        except StockInfo.DoesNotExist:
            continue
        send_stock_info(stock)
    return HttpResponse('Send Success')