Add bootstrap css
[disclosr.git] / documents / genericScrapers.py
blob:a/documents/genericScrapers.py -> blob:b/documents/genericScrapers.py
--- a/documents/genericScrapers.py
+++ b/documents/genericScrapers.py
@@ -1,10 +1,50 @@
 import sys,os
 sys.path.insert(0, os.path.join(os.path.dirname(__file__) or '.', '../'))
 import scrape
+from bs4 import BeautifulSoup
+import parsedatetime as pdt
+from time import mktime
+from datetime import datetime
+import feedparser
+import abc
 
-from bs4 import BeautifulSoup
-import abc
-import dateutil.parser
+class GenericRSSDisclogScraper(object):
+        __metaclass__ = abc.ABCMeta
+        @abc.abstractmethod
+        def getAgencyID(self):
+                """ disclosr agency id """
+                return
+
+        @abc.abstractmethod
+        def getURL(self):
+                """ disclog URL"""
+                return
+
+        def getDescription(self, entry, doc):
+                """ get description from rss entry"""
+                doc['description'] = entry.summary
+		return
+
+       	def doScrape(self):
+               	foidocsdb = scrape.couch['disclosr-foidocuments']
+                (url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID())
+		feed = feedparser.parse(content)		
+		for entry in feed.entries:
+			#print entry
+			print entry.id
+			hash = scrape.mkhash(entry.id)
+			#print hash
+		  	doc = foidocsdb.get(hash)
+			#print doc
+			if doc == None:
+                        	print "saving"
+				edate = datetime.fromtimestamp(mktime( entry.published_parsed)).strftime("%Y-%m-%d")
+                                doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': entry.link, 'docID': entry.id,
+                                "date": edate,"title": entry.title}
+				self.getDescription(entry, doc)
+                                foidocsdb.save(doc)
+                        else:
+                        	print "already saved"			
 
 class GenericOAICDisclogScraper(object):
 	__metaclass__ = abc.ABCMeta
@@ -24,6 +64,7 @@
 		return
 
 	def doScrape(self):
+		cal = pdt.Calendar()
 		foidocsdb = scrape.couch['disclosr-foidocuments']
 		(url,mime_type,content) = scrape.fetchURL(scrape.docsdb, self.getURL(), "foidocuments", self.getAgencyID())
 		if content != None:
@@ -43,11 +84,17 @@
 						doc = foidocsdb.get(hash)
 						descriptiontxt = ""
 						for string in description.stripped_strings:
-							descriptiontxt = descriptiontxt + string
+							descriptiontxt = descriptiontxt + " \n" + string
 							
 						if doc == None:
 							print "saving"
-                                                        edate = dateutil.parser.parse(date.string).date().strftime("%Y-%m-%d")
+                                                        dtresult = cal.parseDateText(date.string)
+							if len(dtresult) == 2:
+								(dtdate,dtr) = dtresult
+								print dtdate
+                                                        	edate = ""+str(dtdate[0])+'-'+str(dtdate[1])+'-'+str(dtdate[2])
+							else:
+								edate = ""
 							doc = {'_id': hash, 'agencyID': self.getAgencyID(), 'url': self.getURL(), "links": links, 'docID': id.string,
 			 				 "date": edate, "description": descriptiontxt,"title": title.string,"notes": notes.string}
 							foidocsdb.save(doc)