1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import sys,os sys.path.insert(0, os.path.join(os.path.dirname(__file__) or '.', '../')) import genericScrapers import dateutil from dateutil.parser import * from datetime import * import scrape from bs4 import BeautifulSoup class ScraperImplementation(genericScrapers.GenericOAICDisclogScraper): def __init__(self): super(ScraperImplementation, self).__init__() def getTable(self, soup): return soup.find(id='content') def getDescription(self,content, entry,doc): link = None links = [] description = "" for atag in entry.find_all('a'): if atag.has_attr('href'): link = scrape.fullurl(self.getURL(), atag['href']) (url, mime_type, htcontent) = scrape.fetchURL(scrape.docsdb, link, "foidocuments", self.getAgencyID(), False) if htcontent != None: if mime_type == "text/html" or mime_type == "application/xhtml+xml" or mime_type =="application/xml": soup = BeautifulSoup(htcontent) row = soup.find(id="foidetails") if row == None: row = soup.find(id="content").table if row == None: row = soup.find(id="content") description = ''.join(row.stripped_strings) for atag in row.find_all("a"): if atag.has_attr('href'): links.append(scrape.fullurl(link, atag['href'])) if links != []: doc.update({'links': links}) if description != "": doc.update({ 'description': description}) def getColumnCount(self): return 3 def getColumns(self, columns): (id, title, date) = columns return (id, date, title, title, None) if __name__ == '__main__': print 'Subclass:', issubclass(ScraperImplementation, genericScrapers.GenericOAICDisclogScraper) print 'Instance:', isinstance(ScraperImplementation(), genericScrapers.GenericOAICDisclogScraper) ScraperImplementation().doScrape() |