import sys,os
sys.path.insert(0, os.path.join(os.path.dirname(__file__) or '.', '../'))
import genericScrapers
import scrape
from bs4 import BeautifulSoup
import codecs
#http://www.doughellmann.com/PyMOTW/abc/
class NewScraperImplementation(genericScrapers.GenericOAICDisclogScraper):
        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":
                                        # http://www.crummy.com/software/BeautifulSoup/documentation.html
                                                soup = BeautifulSoup(htcontent)
                                                for text in soup.find(class_ = "mainContent").stripped_strings:
                                                    description = description + text.encode('ascii', 'ignore')

                                                for atag in soup.find(id="SortingTable").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 2
	def getTable(self,soup):
		return soup.find(id = "TwoColumnSorting")
	def getColumns(self,columns):
		( title, date) = columns
		return (title, date, title, title, None)
class OldScraperImplementation(genericScrapers.GenericOAICDisclogScraper):
        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":
                                        # http://www.crummy.com/software/BeautifulSoup/documentation.html
                                                soup = BeautifulSoup(htcontent)
                                                for text in soup.find(id="content-item").stripped_strings:
                                                    description = description + text + " \n"
                                                for atag in soup.find(id="content-item").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})

		if links != []:
                 	doc.update({'links': links})
                if description != "":
                        doc.update({ 'description': description})

	def getColumnCount(self):
		return 2
	def getTable(self,soup):
		return soup.find(class_ = "doc-list")
	def getColumns(self,columns):
		(date, title) = columns
		return (title, date, title, title, None)

if __name__ == '__main__':
    print 'Subclass:', issubclass(NewScraperImplementation, genericScrapers.GenericOAICDisclogScraper)
    print 'Instance:', isinstance(NewScraperImplementation(), genericScrapers.GenericOAICDisclogScraper)
    NewScraperImplementation().doScrape()
    print 'Subclass:', issubclass(OldScraperImplementation, genericScrapers.GenericOAICDisclogScraper)
    print 'Instance:', isinstance(OldScraperImplementation(), genericScrapers.GenericOAICDisclogScraper)
    osi = OldScraperImplementation()
    osi.disclogURL = "http://archive.treasury.gov.au/content/foi_publications.asp?year=-1&abstract=0&classification=&=&titl=Disclosure+Log+-+Documents+Released+Under+FOI"
    osi.doScrape()