取得股票資訊 (Python)
在ptt上看到的,luke93所分享
import httplib
import urllib2
import re
def debug_print( s, msg = None ):
#print "[DEBUG]", msg, s
pass
httplib.HTTPConnection.debuglevel = 1
stock_ids = ( 2324, 8078, 2311, 2330, 2891, 8926 )
opener = urllib2.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
# For finding stock price
iRE_price = re.compile( r".*nowrap><b>([\d.]+)<.*", \
re.I | re.U | re.M | re.S )
# For finding stock name
# pattern: >2330台積電</a><br><a href="/pf/pfsel?stocklist=
e = ".*>\d+" + r'(.+)</a><br><a href="/pf/pfsel\?stocklist=.*'
debug_print( e, "for name" )
iRE_name = re.compile( e, re.I | re.U | re.M | re.S )
for stock_id in stock_ids:
# Get web page content
content = opener.open( 'http://tw.stock.yahoo.com/q/q?s=' + \
str( stock_id ) ).read()
# Print the whole content for debugging
#print content
stock_price = iRE_price.match( content ).groups()[ 0 ]
stock_name = unicode( iRE_name.match( content ).groups()[ 0 ], "BIG5" )
# Print result
print "%d\t%s\t%.2f" % ( int( stock_id ), stock_name, \
float( stock_price ) )
raw_input( "Press any key..." )
如有錯誤 歡迎指正
