| #http://packages.python.org/CouchDB/client.html | |
| import couchdb | |
| import urllib2 | |
| from BeautifulSoup import BeautifulSoup | |
| import re | |
| import hashlib | |
| #http://diveintopython.org/http_web_services/etags.html | |
| class NotModifiedHandler(urllib2.BaseHandler): | |
| def http_error_304(self, req, fp, code, message, headers): | |
| addinfourl = urllib2.addinfourl(fp, headers, req.get_full_url()) | |
| addinfourl.code = code | |
| return addinfourl | |
| def scrapeAndStore(docsdb, url, depth, agencyID): | |
| hash = hashlib.md5(url).hexdigest() | |
| req = urllib2.Request(url) | |
| print "Fetching %s", url | |
| doc = docsdb['hash'] | |
| #if there is a previous version stored in couchdb, load caching helper tags | |
| if doc.has_key('etag'): | |
| req.add_header("If-None-Match", doc['etag']) | |
| if doc.has_key('last_modified'): | |
| req.add_header("If-Modified-Since", doc['last_modified']) | |
| opener = urllib2.build_opener(NotModifiedHandler()) | |
| url_handle = opener.open(req) | |
| headers = url_handle.info() # the addinfourls have the .info() too | |
| doc['etag'] = headers.getheader("ETag") | |
| doc['last_modified'] = headers.getheader("Last-Modified") | |
| doc['web_server'] = headers.getheader("Server") | |
| doc['file_size'] = headers.getheader("Content-Length") | |
| doc['mime_type'] = headers.getheader("Content-Type") | |
| if hasattr(url_handle, 'code'): | |
| if url_handle.code == 304: | |
| print "the web page has not been modified" | |
| else: | |
| #do scraping | |
| html = url_handle.read() | |
| # http://www.crummy.com/software/BeautifulSoup/documentation.html | |
| soup = BeautifulSoup(html) | |
| links = soup.findAll('a') # soup.findAll('a', id=re.compile("^p-")) | |
| for link in links: | |
| if link.has_key("href"): | |
| print link['href'] | |
| #for each unique link | |
| #if html mimetype | |
| # go down X levels, | |
| # diff with last stored attachment, store in document | |
| #if not | |
| # remember to save parentURL and title (link text that lead to document) | |
| #store as attachment epoch-filename | |
| else: | |
| print "error %s in downloading %s", url_handle.code, URL | |
| #record/alert error to error database | |
| couch = couchdb.Server('http://192.168.1.148:5984/') | |
| # select database | |
| agencydb = couch['disclosr-agencies'] | |
| docsdb = couch['disclosr-documents'] | |
| for row in agencydb.view('app/getScrapeRequired'): #not recently scraped agencies view? | |
| agency = agencydb.get(row.id) | |
| print agency['name'] | |
| scrapeAndStore(docsdb, agency['website'],1,agency['_id']) | |