-
Notifications
You must be signed in to change notification settings - Fork 112
/
fund-rank.py
271 lines (226 loc) · 8.04 KB
/
fund-rank.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
# !/usr/bin/python
# -*- coding: utf-8 -*-
import time
import datetime
import glob
import urllib2
import json
import sys
import re
# 使用方法
def usage():
print 'fund-rank.py usage:'
print '\tpython fund.py start-date end-date fund-code=none\n'
print '\tdate format ****-**-**'
print '\t\tstart-date must before end-date'
print '\tfund-code default none'
print '\t\tif not input, get top 20 funds from all more than 6400 funds'
print '\t\telse get that fund\'s rate of rise\n'
print '\teg:\tpython fund-rank.py 2017-03-01 2017-03-25'
print '\teg:\tpython fund-rank.py 2017-03-01 2017-03-25 377240'
# 获取某一基金在某一日的累计净值数据
def get_jingzhi(strfundcode, strdate):
try:
url = 'http://fund.eastmoney.com/f10/F10DataApi.aspx?type=lsjz&code=' + \
strfundcode + '&page=1&per=20&sdate=' + strdate + '&edate=' + strdate
#print url + '\n'
response = urllib2.urlopen(url)
except urllib2.HTTPError, e:
print e
urllib_error_tag = True
except StandardError, e:
print e
urllib_error_tag = True
else:
urllib_error_tag = False
if urllib_error_tag == True:
return '-1'
json_fund_value = response.read().decode('utf-8')
#print json_fund_value
tr_re = re.compile(r'<tr>(.*?)</tr>')
item_re = re.compile(r'''<td>(\d{4}-\d{2}-\d{2})</td><td.*?>(.*?)</td><td.*?>(.*?)</td><td.*?>(.*?)</td><td.*?>(.*?)</td><td.*?>(.*?)</td><td.*?></td>''', re.X)
# 获取不到 返回-1
jingzhi = '-1'
for line in tr_re.findall(json_fund_value):
#print line + '\n'
match = item_re.match(line)
if match:
entry = match.groups()
date = datetime.datetime.strptime(entry[0], '%Y-%m-%d')
#jingzhi = entry[2]
# result.append([date, float(entry[1]), entry[3]])
jingzhi1 = entry[1]
jingzhi2 = entry[2]
#print jingzhi2
if jingzhi2.strip() == '':
# 040028
# 净值日期 每万份收益 7日年化收益率（%） 申购状态 赎回状态 分红送配
# 2017-01-06 1.4414 暂停申购 暂停赎回
# 2017-01-05 1.4369 暂停申购 暂停赎回
jingzhi = '-1'
elif jingzhi2.find('%') > -1:
# 040003
# 净值日期 每万份收益 7日年化收益率（%） 申购状态 赎回状态 分红送配
# 2017-03-27 1.1149 3.9450% 限制大额申购 开放赎回
# 2017-03-26* 2.2240 3.8970% 限制大额申购 开放赎回
jingzhi = '-1'
elif float(jingzhi1) > float(jingzhi2):
# 502015
# 净值日期 单位净值 累计净值 日增长率 申购状态 赎回状态 分红送配
# 2017-03-27 0.6980 0.3785 -2.24% 场内买入 场内卖出
# 2017-03-24 0.7140 0.3945 5.15% 场内买入 场内卖出
jingzhi = entry[1]
else:
#
# 净值日期 单位净值 累计净值 日增长率 申购状态 赎回状态 分红送配
# 2017-03-28 1.7720 1.7720 -0.23% 开放申购 开放赎回
# 2017-03-27 1.7761 1.7761 -0.43% 开放申购 开放赎回
jingzhi = entry[2]
return jingzhi
def main(argv):
# 获取前20个
gettopnum = 50
# 1、参数处理
#print sys.argv
if len(sys.argv) != 3 and len(sys.argv) != 4:
usage()
sys.exit(1)
# 1.1 起始时间
strsdate = sys.argv[1]
stredate = sys.argv[2]
# 今日零时
strtoday = datetime.datetime.strftime(datetime.datetime.now(), '%Y-%m-%d')
tdatetime = datetime.datetime.strptime(strtoday, '%Y-%m-%d')
#print tdatetime
# 开始时间 如果是周六 周日 调整到周五
#print strsdate
sdatetime = datetime.datetime.strptime(strsdate, '%Y-%m-%d')
sdatetime.isoweekday()
if sdatetime.isoweekday() == 7:
sdatetime = sdatetime + datetime.timedelta(days=-2)
elif sdatetime.isoweekday() == 6:
sdatetime = sdatetime + datetime.timedelta(days=-1)
strsdate = datetime.datetime.strftime(sdatetime, '%Y-%m-%d')
#print strsdate
# 结束时间 如果是周六 周日 调整到周五
#print stredate
edatetime = datetime.datetime.strptime(stredate, '%Y-%m-%d')
edatetime.isoweekday()
if edatetime.isoweekday() == 7:
edatetime = edatetime + datetime.timedelta(days=-2)
elif edatetime.isoweekday() == 6:
edatetime = edatetime + datetime.timedelta(days=-1)
stredate = datetime.datetime.strftime(edatetime, '%Y-%m-%d')
#print stredate
# 判断时间段 今日净值要下午才出 一律不处理
if edatetime <= sdatetime or tdatetime <= sdatetime or tdatetime <= edatetime:
print 'date input error!\n'
usage()
sys.exit(1)
# 2 对单一基金进行处理
if len(sys.argv) == 4:
strfundcode = sys.argv[3]
jingzhimin = get_jingzhi(strfundcode, strsdate)
jingzhimax = get_jingzhi(strfundcode, stredate)
if jingzhimin == '-1' or jingzhimax == '-1' or jingzhimin.strip() == '' or jingzhimax.strip() == '':
print 'maybe date input error!\n'
usage()
sys.exit(1)
jingzhidif = float(jingzhimax) - float(jingzhimin)
jingzhirise = float('%.2f' %(jingzhidif * 100 / float(jingzhimin)))
print 'fund:' + strfundcode + '\n'
print strsdate + '\t' + stredate + '\t净增长' + '\t' + '增长率'
print jingzhimin + '\t\t' + jingzhimax + '\t\t' + str(jingzhidif) + '\t' + str(jingzhirise) + '%'
sys.exit(0)
# 3、获取基金列表
# 如果存在文件 fundlist-*.txt 文件，则读取该文件
# 如果该文件不存在 url获取列表 然后存文件
fundlist_files = glob.glob('fundlist-*.txt')
if (len(fundlist_files) > 0) :
# print fundlist_files[0]
# 读取文件内容
file_object = open(fundlist_files[0], 'r')
try:
all_funds_txt = file_object.read()
#print all_funds_txt
finally:
file_object.close()
else:
# 从天天基金网获取所有基金
response_all_funds = urllib2.urlopen('http://fund.eastmoney.com/js/fundcode_search.js')
all_funds_txt = response_all_funds.read()
#存文件
file_object = open('fundlist-' + strtoday + '.txt', 'w')
try:
file_object.write(all_funds_txt)
#print all_funds_txt
finally:
file_object.close()
#处理数据 将其转化为list
all_funds_txt = all_funds_txt[all_funds_txt.find('=')+2:all_funds_txt.rfind(';')]
all_funds_list = json.loads(all_funds_txt.decode('utf-8'))
print 'start:'
print datetime.datetime.now()
print 'funds sum:' + str(len(all_funds_list))
# 4 循环处理每个基金
for fund in all_funds_list:
print 'process fund:\t' + fund[0].encode('gb18030') + '\t' + fund[2].encode('gb18030')
strfundcode = fund[0]
# 获取净值
jingzhimin = get_jingzhi(strfundcode, strsdate)
jingzhimax = get_jingzhi(strfundcode, stredate)
if jingzhimin == '-1' or jingzhimax == '-1' or jingzhimin.strip() == '' or jingzhimax.strip() == '':
# 有的获取不到 如 000002 华夏成长(后端) 那就都取0吧 NND
# 040028 华安月月鑫短期理财债券A 有可能为空
jingzhimin = '0'
jingzhimax = '0'
jingzhidif = 0
jingzhirise = 0
'''
print 'maybe date input error!\n'
usage()
sys.exit(1)
'''
elif jingzhimin.find('%') > -1 or jingzhimax.find('%') > -1:
# 还有这种 NND 000037 广发理财7天债券A
# 净值日期 每万份收益 7日年化收益率（%） 最近运作期年化收益率 申购状态 赎回状态 分红送配
# 2016-03-01 0.7740 2.7010% 2.7010% 限制大额申购 开放赎回
jingzhidif = 0
jingzhirise = 0
else:
# 计算增长率
jingzhidif = float('%.4f' %(float(jingzhimax) - float(jingzhimin)))
jingzhirise = float('%.2f' %(jingzhidif * 100 / float(jingzhimin)))
# 将数据加入到list中
fund.append(jingzhimin)
fund.append(jingzhimax)
fund.append(jingzhidif)
fund.append(jingzhirise)
# 是否需要控制速度
#time.sleep(1)
# 5 排序 写文件 打印结果
fileobject = open('result_' + strsdate + '_' + stredate + '.txt', 'w')
# 根据增长率逆排序
all_funds_list.sort(key=lambda fund: fund[8], reverse=True)
strhead = '排序' + '\t' + '编码' + '\t\t' + '名称' + '\t\t' + '类型' + '\t\t' + \
strsdate + '\t' + stredate + '\t' + '净增长' + '\t' + '增长率' + '\n'
print strhead
fileobject.write(strhead)
# 打印
for index in range(len(all_funds_list)):
#print all_funds_list[index]
strcontent = str(index+1) + '\t' + all_funds_list[index][0].encode('gb18030') + '\t' + all_funds_list[index][2].encode('gb18030') + \
'\t\t' + all_funds_list[index][3].encode('gb18030') + '\t\t' + all_funds_list[index][5].encode('gb18030') + '\t\t' + \
all_funds_list[index][6].encode('gb18030') + '\t\t' + str(all_funds_list[index][7]) + '\t' + str(all_funds_list[index][8]) + '%\n'
print strcontent
fileobject.write(strcontent)
if index >= gettopnum:
break;
fileobject.close()
print 'end:'
print datetime.datetime.now()
sys.exit(0)
if __name__ == "__main__":
reload(sys)
sys.setdefaultencoding('utf-8')
main(sys.argv)