stock1114

# encoding: utf-8

import requests

import logging

import logging.config

import random

import os

import yaml

import time

import threading

import re

import datetime

import json

class Observer(object):

    open_price_last_1 = ''

    close_price_last_1 = ''

    highest_price_last_1 = ''

    minimum_price_last_1 = ''

    ding_fen_xing = []

    di_fen_xing = []

    not_ding_di = []

    def __init__(self):

        self.headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.77 Safari/537.36', 'Connection': 'close'}

        self.stop_flag = False

        self.potential_stock_list = self.get_potential_stock()

        #self.request_session = requests.session()

    def get_stock_data(self,stock_code,scale):

        """

        获取K 线数据

        :param stock_code: 代码

        :param scale: 级别

        :return: k线数据

        """

        #url = 'http://ifzq.gtimg.cn/appstock/app/kline/mkline?param=sh600030,m30,,320&_var=m30_today&r=0.1700474168906776'

        url = 'http://ifzq.gtimg.cn/appstock/app/kline/mkline'

        params = {

            'param':  '{},m{},,320'.format(stock_code,scale),

            '_var': 'm{}_today'.format(scale),

            'r': '0.1700474{}'.format("".join(random.choice("") for i in range(10)))

        }

        logging.info('url:%s \t params:%s',url,params)

        res = requests.get(url,params=params,headers=self.headers)

        res_str = res.content.decode('unicode_escape').rstrip()

        #logging.info('response:%s:',res_str)

        res_dict = eval(res_str[res_str.index("={")+1:res_str.index("}}}")+3])

        scale_data = res_dict['data'][stock_code]['m'+scale]

        #logging.info(scale_data)

        return scale_data

    def get_stock_code(self,a_type):

        """

        获取两市股票代码

        :param a_type: sh or sz

        :return: stock_code

        """

        #url = 'http://stock.gtimg.cn/data/index.php?appn=rank&t=rankash/chr&p=1&o=0&l=40&v=list_data'

        url = 'http://stock.gtimg.cn/data/index.php'

        params = {

            'appn': 'rank',

            't': 'ranka{}/chr'.format(a_type),

            'p': 1,

            'o': 0,

            'l': 3000,

            'v': 'list_data'

        }

        logging.info('url:%s \t params:%s', url, params)

        res = requests.get(url, params=params, headers=self.headers)

        res_str = res.content.decode('unicode_escape').rstrip()

        #logging.info('response:%s:',res_str)

        res_str_list = res_str[res_str.index("data:'") + 6:res_str.index("'}")].split(',')

        logging.info(res_str_list)

        return res_str_list

    def get_plate_data(self,stock_code):

        """

        获取盘中数据，如果出现大于80万的买单弹框提示

        :param stock_code:

        :return:

        """

        #url = 'http://web.sqt.gtimg.cn/q=sh601208?r=0.9884275211413494'

        url = 'http://web.sqt.gtimg.cn'

        params = {

            'q': stock_code,

             'r': '0.1700474{}'.format("".join(random.choice("") for i in range(10)))

        }

        logging.info('url:%s \t params:%s', url, params)

        res = requests.get(url, params=params, headers=self.headers)

        res_text = res.content.decode('GBK')

        logging.info('response:%s:',res_text.rstrip())

        plate_data = res_text.split('~')

        stock_name_this = plate_data[1]

        stock_code_this = plate_data[2]

        for list_data in plate_data:

            if '|' in list_data:

                plate_date_strike  = list_data.split('|')

                #logging.info(plate_date_strike )

                buy_list = []

                for plate_date_strike_detail in plate_date_strike:

                    if 'B' in plate_date_strike_detail :

                        buy_list.append(int(plate_date_strike_detail.split('/')[4]))

                logging.info(buy_list)

                if buy_list and max(buy_list) > 800000:

                    logging.info('%s %s 动了' % (stock_name_this, stock_code_this))

                    self.alert_msg('%s %s 动了' % (stock_name_this, stock_code_this))

                    #self.stop_flag = True

                    self.potential_stock_list.remove(stock_code)

                else:

                    logging.info('%s %s 没动' % (stock_name_this, stock_code_this))

    def get_potential_stock(self):

        potential_stock_list_new = []

        stock_codes = self.get_stock_code('sh') + self.get_stock_code('sz')

        logging.info(('stock_codes',stock_codes))

        stock_codes_length_div25 = int(len(stock_codes)/25)

        logging.info(stock_codes_length_div25)

        for stock_codes_length_div25_index in range(stock_codes_length_div25):

            logging.info(stock_codes_length_div25_index)

            if stock_codes_length_div25_index == stock_codes_length_div25 -1:

                req_str = ','.join(stock_codes[stock_codes_length_div25_index * 25:])

            else:

                req_str = ','.join(stock_codes[stock_codes_length_div25_index * 25:(stock_codes_length_div25_index + 1) * 25])

            url = 'http://qt.gtimg.cn/q=%s&r=9%s'%(req_str,''.join(str(i1) for i1 in random.sample(range(1,9),8)))

            logging.info(url)

            #res = requests.get(url, headers=self.headers)

            res = requests.session().get(url,headers=self.headers)

            res_str = res.content.decode('GBK').rstrip()

            stock_data_list = res_str.replace('\n','').split(';')

            for stock_data in stock_data_list:

                date_today = str(datetime.datetime.now()).split(' ')[0].replace('-','')

                day_rise_str_match = re.search(date_today + '(.*)?/',stock_data)

                logging.info(('day_rise_str_match',day_rise_str_match))

                if day_rise_str_match :

                    day_rise_str = day_rise_str_match.group(1)

                    day_rise_float = float(day_rise_str.split('~')[2])

                    logging.info(('day_rise_float',day_rise_float))

                    #当日涨幅在4%和7%之间

                    if day_rise_float >4.0 and day_rise_float < 7.0:

                        match_str = re.search('v_([s,h,z]{2}\d+)=.*',stock_data)

                        if match_str:

                            stock_code = match_str.group(1)

                            # n天内涨停次数 > 6

                            if self.get_n_days_limit_up_times(stock_code,250) > 6 :

                                potential_stock_list_new.append(stock_code)

            time.sleep(3)

        potential_stock_list_new = list(set(potential_stock_list_new))

        logging.info(('potential_stock_list_new',potential_stock_list_new))

        return potential_stock_list_new

    def is_potential_stock_list_changed(self):

        potential_stock_list_new = self.get_potential_stock()

        #如果交集等于并集则表示没有改变

        if set(potential_stock_list_new) & set(self.potential_stock_list) == set(potential_stock_list_new) | set(

                self.potential_stock_list):

            return False

        else:

            self.stop_flag = True

            return True

    def get_n_days_limit_up_times(self,stock_code,n):

        date_n_ago = str(datetime.datetime.now() - datetime.timedelta(days=n)).split(' ')[0]

        date_today = str(datetime.datetime.now()).split(' ')[0]

        rand_str = '0.1700474{}'.format("".join(random.choice("") for i in range(10)))

        url = 'http://web.ifzq.gtimg.cn/appstock/app/fqkline/get?_var=kline_dayqfq&param=%s,day,%s,%s,320,qfq&r=%s'%(stock_code,date_n_ago,date_today,rand_str)

        res = requests.get(url, headers=self.headers)

        res_str = res.content.decode('unicode_escape').rstrip().replace('kline_dayqfq=','')

        logging.info(res_str)

        logging.info(stock_code)

        #res_dict = json.loads(res_str)

        #day_kline_data = res_dict.get('data').get(stock_code).get('day')

        day_kline_data_match = re.search('\[\[(.*)?\]\]',res_str)

        if day_kline_data_match :

            day_kline_data_match_str = day_kline_data_match.group(1)

            day_kline_data_match_str = day_kline_data_match_str.replace('"','')

            day_kline_data = day_kline_data_match_str.split('],[')

            logging.info(day_kline_data)

            limit_up_times = 0

            for i in range(len(day_kline_data)-1):

                if float(day_kline_data[i+1].split(',')[2])/float(day_kline_data[i].split(',')[2]) > 1.095 :

                    limit_up_times = limit_up_times +1

        return limit_up_times

    def monitor_start(self):

        #self.stop_flag = False

        for monitor_stock in self.potential_stock_list:

            m = MonitorThread(monitor_stock,self)

            m.start()

    def alert_msg(self,msg):

        commond = 'msg * %s'%msg

        os.system(commond)

class MonitorThread(threading.Thread):

    def __init__(self,stock_code,observer):

        threading.Thread.__init__(self)

        self.name = stock_code

        self.stock_code = stock_code

        self.observer = observer

    def run(self):

        logging.info(self.name + '监控开始...')

        while 1:

            self.observer.get_plate_data(self.stock_code)

            if self.observer.stop_flag :

                break

            if self.observer.is_potential_stock_list_changed():

                Observer().monitor_start()

                break

            time.sleep(5)

        logging.info(self.name + '监控结束!')

if __name__ == '__main__':

    path = 'logging.yaml'

    value = os.getenv('LOG_CFG', None)

    if value:

        path = value

    if os.path.exists(path):

        with open(path, "r") as f:

            config = yaml.load(f)

            logging.config.dictConfig(config)

    else:

        print('log config file not found!')

    # monitor_list = ['sh603590','sz300702']

    # for stock_code in monitor_list:

    #     thread_a = MonitorThread(stock_code)

    #     thread_a.start()

    # o = Observer()

    # code_list_sh = o.get_stock_code('sh')

    # code_list_sz = o.get_stock_code('sz')

    # code_list = code_list_sh + code_list_sz

    # code_list_keguanzhu = []

    # for stock_code in code_list:

    #     m30_data = o.get_stock_data(stock_code, '30')

    #     #倒数第二个k线最低点是最后三根k线最低点中最低且最后一根k线收盘价高于中间一个k线的最高价

    #     if min(float(m30_data[-1][4]), float(m30_data[-2][4]), float(m30_data[-3][4])) == float(m30_data[-2][4]) and float(m30_data[-1][2]) > float(m30_data[-2][3]):

    #         logging.info('%s可以关注'%stock_code)

    #         code_list_keguanzhu.append(stock_code)

    # logging.info('可关注的%d个票:%s'%(len(code_list_keguanzhu),str(code_list_keguanzhu)))

    Observer().monitor_start()

    # a,b,c='aa,bb,cc'.split(',')

    # print((a,b,c))