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
 
class GenericDisclogScraper(object):
        __metaclass__ = abc.ABCMeta
        agencyID = None
        disclogURL = None
        def getAgencyID(self):
                """ disclosr agency id """
                if self.agencyID == None:
                        self.agencyID = os.path.basename(sys.argv[0]).replace(".py","")
                return self.agencyID
 
        def getURL(self):
                """ disclog URL"""
                if self.disclogURL == None:
                        agency = scrape.agencydb.get(self.getAgencyID())
                        self.disclogURL = agency['FOIDocumentsURL']
                return self.disclogURL
 
        @abc.abstractmethod
        def doScrape(self):
                """ do the scraping """
                return
 
 
 
class GenericRSSDisclogScraper(GenericDisclogScraper):
        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(GenericDisclogScraper):
        __metaclass__ = abc.ABCMeta
        @abc.abstractmethod
        def getColumns(self,columns):
                """ rearranges columns if required """
                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:
                        if mime_type == "text/html" or mime_type == "application/xhtml+xml" or mime_type =="application/xml":
                        # http://www.crummy.com/software/BeautifulSoup/documentation.html
                                soup = BeautifulSoup(content)
                                for row in soup.table.find_all('tr'):
                                        columns = row.find_all('td')
                                        if len(columns) == 5:
                                                (id, date, description, title, notes) = self.getColumns(columns)
                                                print id.string
                                                hash = scrape.mkhash(url+id.string)
                                                links = []
                                                for atag in row.find_all("a"):
                                                        if atag.has_key('href'):
                                                                links.append(scrape.fullurl(url,atag['href']))
                                                doc = foidocsdb.get(hash)
                                                descriptiontxt = ""
                                                for string in description.stripped_strings:
                                                        descriptiontxt = descriptiontxt + " \n" + string
                                                        
                                                if doc == None:
                                                        print "saving"
                                                        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)
                                                else:
                                                        print "already saved"
                                        
                                        elif len(row.find_all('th')) == 5:
                                                print "header row"
                                        
                                        else:
                                                print "ERROR number of columns incorrect"
                                                print row