Selecter.py 文件源码

def init(self, setting):
        #self.mysql = sMysql(MYSQL_DB['host'], MYSQL_DB['user'], MYSQL_DB['password'], MYSQL_DB['dbname'])
        limit = 100
        if 'limit' in setting.keys():
            limit = setting['limit']

        _where = []
        s_keys_list = setting.keys()

        if 'start' not in s_keys_list and 'end' not in s_keys_list:
            print u"StartTime OR EndTime is Error"
            sys.exit()

        _today = self.tools.d_date('%Y%m%d')
        if 'end' not in setting.keys():
            setting['end'] = _today

        if 'start' not in setting.keys():
            setting['start'] = setting['end']

        if setting['start'] == setting['end']:
            _where.append(" dateline = %s" % setting['end'])
        else:
            _where.append(" dateline <= %s" % setting['end'])
            _where.append(" dateline >= %s" % setting['start'])

        if 'universe' in setting.keys():
            s_codes = " s_code in(%s)" % self.___set_universe(setting['universe'])
            _where.append(s_codes)

        _wheres = ' AND '.join(_where)

        print u"=======????===%s====" % setting['end']

        date_sql = "select dateline FROM s_opening_day WHERE dateline <=%s order by dateline desc limit %s" % (setting['end'], limit)
        print date_sql
        temp = self.mysql.getRecord(date_sql)
        self.today = _today
        self.lastDay = temp[0]['dateline']
        self.yestoday = temp[1]['dateline']
        pandas.set_option('display.width', 200)
        sql_data = "select s_code,code,dateline,chg_m,chg,open,close,high,low,last_close,name,amount,run_market FROM s_stock_trade WHERE %s " % _wheres
        #print sql_data
        #sys.exit()
        tmpdf = pandas.read_sql(sql_data, self.mysql.db)
        #print tmpdf
        #sys.exit()
        #????????
        if ('is_open_chuquan' in setting.keys()) and setting['is_open_chuquan']:
            self._chQ = self.getChuQuan()
            #print self._chQ
            #sys.exit()
            self.df = tmpdf.apply(self.format_chuquan_hanlder, axis=1)
        else:
            self.df = tmpdf
        #print self.df
        #sys.exit()
        self.todayDF = self.df[self.df.dateline == int(self.lastDay)]
        self.yestodayDF = self.df[self.df.dateline == int(self.yestoday)]
        #sys.exit()
        print "========init Days & init stock trader Done."