| import sys,os | import sys,os |
| sys.path.insert(0, os.path.join(os.path.dirname(__file__) or '.', '../')) | sys.path.insert(0, os.path.join(os.path.dirname(__file__) or '.', '../')) |
| import scrape | import scrape |
| from bs4 import BeautifulSoup | from bs4 import BeautifulSoup |
| import parsedatetime as pdt | |
| from time import mktime | from time import mktime |
| from datetime import datetime | |
| import feedparser | import feedparser |
| import abc | import abc |
| import unicodedata, re | import unicodedata, re |
| import dateutil | |
| from dateutil.parser import * | |
| from datetime import * | |
| class GenericDisclogScraper(object): | class GenericDisclogScraper(object): |
| __metaclass__ = abc.ABCMeta | __metaclass__ = abc.ABCMeta |
| agencyID = None | agencyID = None |
| disclogURL = None | disclogURL = None |
| def remove_control_chars(self, input): | def remove_control_chars(self, input): |
| return "".join([i for i in input if ord(i) in range(32, 127)]) | return "".join([i for i in input if ord(i) in range(32, 127)]) |
| def getAgencyID(self): | def getAgencyID(self): |
| """ disclosr agency id """ | """ disclosr agency id """ |
| if self.agencyID == None: | if self.agencyID == None: |
| self.agencyID = os.path.basename(sys.argv[0]).replace(".py","") | self.agencyID = os.path.basename(sys.argv[0]).replace(".py","") |
| return self.agencyID | return self.agencyID |
| def getURL(self): | def getURL(self): |
| """ disclog URL""" | """ disclog URL""" |
| if self.disclogURL == None: | if self.disclogURL == None: |
| agency = scrape.agencydb.get(self.getAgencyID()) | agency = scrape.agencydb.get(self.getAgencyID()) |
| self.disclogURL = agency['FOIDocumentsURL'] | self.disclogURL = agency['FOIDocumentsURL'] |
| return self.disclogURL | return self.disclogURL |
| @abc.abstractmethod | @abc.abstractmethod |
| def doScrape(self): | def doScrape(self): |
| """ do the scraping """ | """ do the scraping """ |
| return | return |
| @abc.abstractmethod | @abc.abstractmethod |
| def getDescription(self, content, entry, doc): | def getDescription(self, content, entry, doc): |
| """ get description""" | """ get description""" |
| return | return |
| class GenericRSSDisclogScraper(GenericDisclogScraper): | class GenericRSSDisclogScraper(GenericDisclogScraper): |
| def doScrape(self): | def doScrape(self): |
| foidocsdb = scrape.couch['disclosr-foidocuments'] | foidocsdb = scrape.couch['disclosr-foidocuments'] |
| (url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID()) | (url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID()) |
| feed = feedparser.parse(content) | feed = feedparser.parse(content) |
| for entry in feed.entries: | for entry in feed.entries: |
| #print entry | #print entry |
| print entry.id | print entry.id |
| hash = scrape.mkhash(entry.id) | hash = scrape.mkhash(entry.id) |
| #print hash | #print hash |
| doc = foidocsdb.get(hash) | doc = foidocsdb.get(hash) |
| #print doc | #print doc |
| if doc == None: | if doc == None: |
| print "saving" | print "saving" |
| edate = datetime.fromtimestamp(mktime( entry.published_parsed)).strftime("%Y-%m-%d") | edate = datetime.fromtimestamp(mktime( entry.published_parsed)).strftime("%Y-%m-%d") |
| doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': entry.link, 'docID': entry.id, | doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': entry.link, 'docID': entry.id, |
| "date": edate,"title": entry.title} | "date": edate,"title": entry.title} |
| self.getDescription(entry,entry, doc) | self.getDescription(entry,entry, doc) |
| foidocsdb.save(doc) | foidocsdb.save(doc) |
| else: | else: |
| print "already saved" | print "already saved" |
| def getDescription(self, content, entry, doc): | def getDescription(self, content, entry, doc): |
| """ get description from rss entry""" | """ get description from rss entry""" |
| doc.update({'description': content.summary}) | doc.update({'description': content.summary}) |
| return | return |
| class GenericOAICDisclogScraper(GenericDisclogScraper): | class GenericOAICDisclogScraper(GenericDisclogScraper): |
| __metaclass__ = abc.ABCMeta | __metaclass__ = abc.ABCMeta |
| @abc.abstractmethod | @abc.abstractmethod |
| def getColumns(self,columns): | def getColumns(self,columns): |
| """ rearranges columns if required """ | """ rearranges columns if required """ |
| return | return |
| def getColumnCount(self): | def getColumnCount(self): |
| return 5 | return 5 |
| def getDescription(self, content, entry, doc): | def getDescription(self, content, entry, doc): |
| """ get description from rss entry""" | """ get description from rss entry""" |
| descriptiontxt = "" | descriptiontxt = "" |
| for string in content.stripped_strings: | for string in content.stripped_strings: |
| descriptiontxt = descriptiontxt + " \n" + string | descriptiontxt = descriptiontxt + " \n" + string |
| doc.update({'description': descriptiontxt}) | doc.update({'description': descriptiontxt}) |
| return | return |
| def getTitle(self, content, entry, doc): | def getTitle(self, content, entry, doc): |
| doc.update({'title': content.string}) | doc.update({'title': content.string}) |
| return | return |
| def getTable(self, soup): | def getTable(self, soup): |
| return soup.table | return soup.table |
| def getDate(self, content, entry, doc): | def getDate(self, content, entry, doc): |
| dtresult = cal.parseDateText(content.string) | edate = parse(content.string.strip(), dayfirst=True, fuzzy=True).strftime("%Y-%m-%d") |
| if len(dtresult) == 2: | |
| (dtdate,dtr) = dtresult | |
| edate = ""+str(dtdate[0])+'-'+str(dtdate[1])+'-'+str(dtdate[2]) | |
| else: | |
| edate = datetime.strptime(date.string.strip(), "%d %B %Y").strftime("%Y-%m-%d") | |
| print edate | print edate |
| doc.update({'date': edate}) | doc.update({'date': edate}) |
| return | return |
| def doScrape(self): | def doScrape(self): |
| cal = pdt.Calendar() | |
| foidocsdb = scrape.couch['disclosr-foidocuments'] | foidocsdb = scrape.couch['disclosr-foidocuments'] |
| (url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID()) | (url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID()) |
| if content != None: | if content != None: |
| if mime_type == "text/html" or mime_type == "application/xhtml+xml" or mime_type =="application/xml": | if mime_type == "text/html" or mime_type == "application/xhtml+xml" or mime_type =="application/xml": |
| # http://www.crummy.com/software/BeautifulSoup/documentation.html | # http://www.crummy.com/software/BeautifulSoup/documentation.html |
| soup = BeautifulSoup(content) | soup = BeautifulSoup(content) |
| table = self.getTable(soup) | table = self.getTable(soup) |
| for row in table.find_all('tr'): | for row in table.find_all('tr'): |
| columns = row.find_all('td') | columns = row.find_all('td') |
| if len(columns) == self.getColumnCount(): | if len(columns) == self.getColumnCount(): |
| (id, date, description, title, notes) = self.getColumns(columns) | (id, date, description, title, notes) = self.getColumns(columns) |
| print id.string | print id.string |
| if id.string == None: | if id.string == None: |
| hash = scrape.mkhash(self.remove_control_chars(url+date.string)) | hash = scrape.mkhash(self.remove_control_chars(url+date.string)) |
| else: | else: |
| hash = scrape.mkhash(self.remove_control_chars(url+id.string)) | hash = scrape.mkhash(self.remove_control_chars(url+id.string)) |
| links = [] | links = [] |
| for atag in row.find_all("a"): | for atag in row.find_all("a"): |
| if atag.has_key('href'): | if atag.has_key('href'): |
| links.append(scrape.fullurl(url,atag['href'])) | links.append(scrape.fullurl(url,atag['href'])) |
| doc = foidocsdb.get(hash) | doc = foidocsdb.get(hash) |
| if doc == None: | if doc == None: |
| print "saving" | print "saving" |
| doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': self.getURL(), 'docID': id.string} | doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': self.getURL(), 'docID': id.string} |
| if links != []: | if links != []: |
| doc.update({'links': links}) | doc.update({'links': links}) |
| self.getTitle(title,row, doc) | self.getTitle(title,row, doc) |
| self.getDate(date,row, doc) | self.getDate(date,row, doc) |
| self.getDescription(description,row, doc) | self.getDescription(description,row, doc) |
| if notes != None: | if notes != None: |
| doc.update({ 'notes': notes.string}) | doc.update({ 'notes': notes.string}) |
| foidocsdb.save(doc) | foidocsdb.save(doc) |
| else: | else: |
| print "already saved "+hash | print "already saved "+hash |
| elif len(row.find_all('th')) == self.getColumnCount(): | elif len(row.find_all('th')) == self.getColumnCount(): |
| print "header row" | print "header row" |
| else: | else: |
| print "ERROR number of columns incorrect" | print "ERROR number of columns incorrect" |
| print row | print row |