1

我的程序接受用户输入并通过特定网页进行搜索。此外，我希望它去点击一个特定的链接，然后下载那里的文件。

例子 ：

  1. 网页：http ://www.rcsb.org/pdb/home/home.do
  2. 搜索词：“1AW0”
  3. 在网站上搜索这个词后，它会将您带到： http ://www.rcsb.org/pdb/explore/explore.do?structureId=1AW0

我希望程序位于网页的右侧并从DOWNLOAD FILES选项下载 pdb 文件

我已经设法使用 mechanize 模块编写了一个程序来自动搜索单词但是找不到我可以点击链接的方法

我的代码：

import urllib2
import re
import mechanize

br = mechanize.Browser()
br.open("http://www.rcsb.org/pdb/home/home.do")
## name of the form that holds the search text area 
br.select_form("headerQueryForm")

## "q" name of the teaxtarea in the html script
br["q"] = str("1AW0")
response = br.submit()
print response.read()

任何帮助或任何建议都会有所帮助。

顺便说一句，我是 Python 的中级程序员，我正在尝试学习 Jython 模块以尝试完成这项工作。

提前致谢

4

1 回答 1

1

以下是我的做法：

'''
Created on Dec 9, 2012

@author: Daniel Ng
'''

import urllib

def fetch_structure(structureid, filetype='pdb'):
  download_url = 'http://www.rcsb.org/pdb/download/downloadFile.do?fileFormat=%s&compression=NO&structureId=%s'
  filetypes = ['pdb','cif','xml']
  if (filetype not in filetypes):
    print "Invalid filetype...", filetype
  else:
    try:
      urllib.urlretrieve(download_url % (filetype,structureid), '%s.%s' % (structureid,filetype))
    except Exception, e:
      print "Download failed...", e
    else:
      print "Saved to", '%s.%s' % (structureid,filetype)

if __name__ == "__main__":
  fetch_structure('1AW0')
  fetch_structure('1AW0', filetype='xml')
  fetch_structure('1AW0', filetype='png')

它提供了这个输出：

Saved to 1AW0.pdb
Saved to 1AW0.xml
Invalid filetype... png

连同 2 个文件1AW0.pdb一起1AW0.xml保存到脚本目录（对于此示例）。

http://docs.python.org/2/library/urllib.html#urllib.urlretrieve

于 2012-12-09T10:18:37.333 回答