python 简约策略

#!/usr/bin/env python
# coding: utf-8
import datetime

import csv
import sys
#from datetime import datetime
#from datetime import timedelta

global stocks
global k_ri
global list_code
global conn_list, conn_k

quotes = []

def read_file(name, quotes):
try:
csvfile = file(name, 'rb')
reader = csv.reader(csvfile)

for line in reader:
try:
# 忽略第一行
if reader.line_num == 1:
continue
# list_code.append(line[0])
# print(' said: ', '')
i = 0
#print line[i], line[i + 1], line[i + 2], line[i + 3], line[i + 4]
#deal_data(line[i], line[i + 1], line[i + 2], line[i + 3], line[i + 4])
quotes.append((line[i], line[i + 1], line[i + 2], line[i + 3], line[i + 4]))
except ValueError:
pass
#print "name --", name, locals()
# csvfile.close()
except IOError as err:  # 使用as将异常对象，并将其赋值给一个标识符
print('File Error:' + str(err))  # ‘+’用于字符串直接的连接

finally:
if 'csvfile' in locals():
csvfile.close()
print "close"
def judge_rise(price):
#print price
if (0 == price):
return 0.00
price *= 1.1
price += 0.005
return float('{:.2f}'.format(price))

def plot(quotes):
date_old, open_old, high_old, low_old, close_old = 0, 0, 0, 0, 0
flg_rise, high_max, flg_buy, days, cnt_win, cnt_lost = 0, 0, 0, 0, 0, 0
total = 100

for q in quotes:
date, open, high, low, close = q[0], float(q[1]), float(q[2]), float(q[3]), float(q[4])
rise = judge_rise(close_old)
#print rise, close, close_old
if (0 == flg_buy):
if (close == rise):
print "hi", rise, close, close_old, date
flg_rise += 1
high_max = high
else:
if (2 < flg_rise):
if (high_max <= high):
high_max = high
print "3 dan"

else:
flg_buy = 1
days = 10
else:
flg_rise = 0
elif (1 == flg_buy):
if (0 >= days):
print "days die"
flg_rise, flg_buy, = 0, 0
if (high_max < high):
buy = high_max + 0.01
flg_buy = 2
print "buy"

days -= 1
elif (2 == flg_buy):
if (high_max <= high):
high_max = high
if (0.9 >= close / buy):
total =  total * close / buy
flg_buy = 0
cnt_win += 1
else:
if (0.9 >= close / high_max):
total = total * close / buy
flg_buy = 0
cnt_lost += 1
else:
flg_rise, flg_buy, = 0, 0
date_old, open_old, high_old, low_old, close_old = date, open, high, low, close
print total, cnt_win, cnt_lost
#print date, open, high, low, close

def main(argv=None):
read_file('csv_test2.csv', quotes)
if (0 == len(quotes)):
return
plot(quotes)

if __name__ == '__main__':
sys.exit(main())