#coding=utf-8
import datetime
import math
import db

def get_update_datas(df, count):
	datas = df.values[0:count].tolist()
	if len(datas) > 0:
		for i in xrange(0,len(datas)):
			for j in xrange(0, len(datas[i])):
				if type(datas[i][j]) is float and math.isnan(datas[i][j]):
					datas[i][j] = 0.0
		return datas

#macd过滤器
def xuangu_filter(code, df):
	
	d = db.DB.getInstance()
	basics = d.get_basics()

	#停牌的股
	today = datetime.date.today()
	lasttradeday = datetime.datetime.strptime(df.iloc[0]['date'], "%Y-%m-%d").date()
	days = today - lasttradeday
	if days.days > 3:
		return False
	#st股
	if basics.loc[code]['name'][:2].lower() == "st" or basics.loc[code]['name'][1:3].lower() == "st":
		return False
	#过滤大盘股
	#if basics.loc[code]['outstanding']*10000 > 100000000:
		#flag = False
	#市盈率为负数或过大
	#if basics.loc[code]['pe'] <= 0 or basics.loc[code]['pe'] > 1000:
		#flag = False				
	#每股收益
	#if basics.loc[code]['bvps'] <= 0:
		#flag = False
	return True

#macd过滤器
def isnew(df):
	return df.shape[0] <= 30