这是我修改的一些代码,可以在 pyscholar 中使用。
#!/usr/bin/python
#author: Bryan Bishop <kanzure@gmail.com>
#date: 2010-03-03
#purpose: given a link on the command line to sciencedirect.com, download the associated PDF and put it in "sciencedirect.pdf" or something
import os
import re
import pycurl
#from BeautifulSoup import BeautifulSoup
from lxml import etree
import lxml.html
from StringIO import StringIO
from string import join, split
user_agent = "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.1.5) Gecko/20091123 Iceweasel/3.5.5 (like Firefox/3.5.5; Debian-3.5.5-1)"
def interscience(url):
'''downloads the PDF from sciencedirect given a link to an article'''
url = str(url)
buffer = StringIO()
curl = pycurl.Curl()
curl.setopt(curl.URL, url)
curl.setopt(curl.WRITEFUNCTION, buffer.write)
curl.setopt(curl.VERBOSE, 0)
curl.setopt(curl.USERAGENT, user_agent)
curl.setopt(curl.TIMEOUT, 20)
curl.perform()
curl.close()
buffer = buffer.getvalue().strip()
html = lxml.html.parse(StringIO(buffer))
pdf_href = []
for item in html.getroot().iter('a'):
if (('id' in item.attrib) and ('href' in item.attrib) and item.attrib['id']=='pdfLink'):
pdf_href.append(item.attrib['href'])
pdf_href = pdf_href[0]
#now let's get the article title
title_div = html.find("head/title")
paper_title = title_div.text
paper_title = paper_title.replace("\n", "")
if paper_title[-1] == " ": paper_title = paper_title[:-1]
re.sub('[^a-zA-Z0-9_\-.() ]+', '', paper_title)
paper_title = paper_title.strip()
paper_title = re.sub(' ','_',paper_title)
#now fetch the document for the user
command = "wget --user-agent=\"pyscholar/blah\" --output-document=\"%s.pdf\" \"%s\"" % (paper_title, pdf_href)
os.system(command)
print "\n\n"
interscience("http://www.sciencedirect.com/science/article/pii/S0163638307000628")