Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
抓大盤日成交量
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 抓大盤日成交量
# By Toomore
from datetime import datetime, timedelta
import urllib2, logging, csv, re
class ggm(object):
def __init__(self,year,savef):
starttime = 0
daterange = datetime.today()
f = open('%s.csv' % savef,'wt')
writer = csv.writer(f)
while 1:
if daterange.year <= year:
break
else:
daterange = datetime.today() - timedelta(days = 30 * starttime)
for i in self.fetch_data(daterange):
if self.ckinv(i):
ti = []
for ii in range(len(i)):
i[ii] = i[ii].replace(',','')
i[ii] = i[ii].replace(' ','')
ti.append(i[ii])
print ti
writer.writerow(tuple(ti))
starttime += 1
f.close()
def fetch_data(self, nowdatetime):
""" Fetch data from twse.com.tw
return list.
"""
url = "http://www.twse.com.tw/ch/trading/exchange/FMTQIK/FMTQIK2.php?STK_NO=&myear=%(year)d&mmon=%(mon)02d&type=csv" % {'year': nowdatetime.year, 'mon': nowdatetime.month}
logging.info(url)
cc = urllib2.urlopen(url)
#print cc.info().headers
csv_read = csv.reader(cc)
return csv_read
def ckinv(self,oo):
""" check the value is date or not """
pattern = re.compile(r"[0-9]{2}/[0-9]{2}/[0-9]{2}")
b = re.search(pattern, oo[0])
try:
b.group()
return True
except:
return False
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment