搜索所有股票的相关信息并保存

草稿算完成了，但是代码显然价值很低，未进行任何优化，就是图个快速写完，所以看官别骂
下一步考虑在这个代码的基础上做优化，估计首先应该是解决代码'味道'的问题，然后是多线程的问题。其实有点想能否将这个代码修改成golang的，不过不知道golang是否有selenium的包包。

#coding=utf-8
from selenium import webdriver
import time
import os
import re
import sys  
from selenium.common.exceptions import NoSuchElementException
reload(sys)  
sys.setdefaultencoding('utf8')   

def showlink(linklist):
    for x in linklist:
        print x.get_attribute('href')
    print "股票数:%d"%(len(linklist))

def getinfo(mydriver,gourl):
    linetext=""
    mydriver.get(gourl)
    try:
        gupiaoming=mydriver.find_element_by_xpath(title).text
        gupiaocode=mydriver.find_element_by_xpath(code).text
        hexinshuju=mydriver.find_element_by_class_name('pad5')
        shujuhang=hexinshuju.find_elements_by_tag_name('tr')
        for i in range(len(shujuhang)-2):
            shujulie=shujuhang[i].find_elements_by_tag_name('td')
            tmpshuju=myre.split(shujulie[0].text)
            linetext=linetext+"~"+tmpshuju[1]
        shuju=myre.split(shujuhang[8].text)

        linetext=linetext+"~"+shuju[1]
        tmpshuju=myre.split(shujuhang[9].text)
        linetext=linetext+"~"+tmpshuju[1]
        linetext="%s~%s%s"%(gupiaoming,gupiaocode,linetext)
        print "数据写入",linetext
        myfile.write("%s\n"%(linetext))
    except  NoSuchElementException,e:
        print "不是股票"
######################

br=webdriver.Firefox()
brsub=webdriver.Firefox()
baseurl="http://quote.eastmoney.com/"
indexurl="stocklist.html"
gourl="%s%s"%(baseurl,indexurl)
br.get(gourl)

shxpath="/html/body/div[9]/div[2]/div/ul[1]"
szxpath="/html/body/div[9]/div[2]/div/ul[2]"

shgupiao=br.find_element_by_xpath(shxpath)
szgupiao=br.find_element_by_xpath(szxpath)

shgupiaolist=shgupiao.find_elements_by_tag_name('a')
szgupiaolist=szgupiao.find_elements_by_tag_name('a')


title='//*[@id="name"]'
code='//*[@id="code"]'
hexinshujuxpath="/html/body/div[14]/div[1]/div[4]/div[1]"
restr="：".decode('utf8')
myre=re.compile(restr,re.I|re.M|re.S)


#mylist=['http://quote.eastmoney.com/sh603678.html','http://quote.eastmoney.com/sh603686.html']
print "####    获得上海股票数据               ############"
print "####    数据保存到shgupiaodata.text   ############"
filename='shgupiaodata.txt'
myfile = open(filename, 'w')
for i in shgupiaolist:
    myurl=i.get_attribute('href')
    print myurl
    if myurl!=None:
        getinfo(brsub,myurl)
myfile.close()


print "####    获得深圳股票数据               ############"
print "####    数据保存到szgupiaodata.text   ############"
filename='szgupiaodata.txt'
myfile = open(filename, 'w')
for i in szgupiaolist:
    myurl=i.get_attribute('href')
    print myurl
    if myurl!=None:
        getinfo(brsub,myurl)
myfile.close()

print "数据下载完成"
br.quit()
brsub.quit()