import os

import yaml
import json

import urllib.request, urllib.error, urllib.parse

from lxml import etree
import lxml.html
import lxml.html.soupparser




class membersParliamentCrawler(object):
    
    def __init__(self, configFile):
        
        with open(configFile, "r") as stream:
            try:
                self.config = yaml.safe_load(stream)
            except yaml.YAMLError as exc:
                print(exc)
  
    
    # input list of countries in form of ['nicaragua', 'honduras', .. , 'mexico']

    def downloadMemberListPagesOfCountries(self, listOfCountries):
        
        # download only html pages of the countries specified in input
        
        for country in listOfCountries:
            for key in self.config:
                if key in listOfCountries:
                    try:
                        memberList = self.config.get(key).get('memberList')
                    except Exception as e:
                        print("There is a problem with the entry memberList in the config.yaml - the original error message is:", e)
                    try:
                        memberListLink = memberList.get('link')
                    except Exception as e:
                        print("No memberListLink defined in config.yaml - the original error message is:", e)
                    
                    # download the html page of the List of Members                    

                    response = urllib.request.urlopen(memberListLink)
                    webContent = response.read().decode('UTF-8')

                    # save interim results to files
                    
                    f = open('pages/' + key +'MemberList.html', 'w+')
                    f.write(webContent)
                    f.close

                    
    def parseMemberListData2dictionary(self, listOfCountries):
        
        for country in listOfCountries:
            
            try:
                                
                #use soupparser to handle broken html
                
                tree = lxml.html.soupparser.parse('pages/' + country + 'MemberList.html')

#                for e in tree.iter():
#
#                    print(e.tag)
#                
#                 for e in tree.xpath('//html//body//form//table//tr//td//table//tr'):
#                     
#                     #print(etree.tostring(e).decode())
                
                dictionaryMemberList = {}
                
                countryConf = self.config.get(country)
                countryDomain = countryConf.get('domain')
                countryConfMemberList = countryConf.get('memberList')
                countryConfMemberListParent = countryConfMemberList.get('parent')
                countryConfMemberListChildName = countryConfMemberList.get('child-name')
                countryConfMemberListChildLink = countryConfMemberList.get('child-link')
                
                for n in range(len(tree.xpath(countryConfMemberListParent))):
                    
                    name = tree.xpath(countryConfMemberListParent + '[' + str(n) + ']' + countryConfMemberListChildName)
                    link = tree.xpath(countryConfMemberListParent + '[' + str(n) + ']' + countryConfMemberListChildLink)
                    
                    if len(name) > 0:
                    
                        dictionaryMemberList[n] = {}
                        dictionaryMemberList[n]['name'] = name[0]
                        
                        if countryDomain in link[0]:
                        
                            dictionaryMemberList[n]['link'] = link[0]
                        
                        if countryDomain not in link[0]:
                            
                            dictionaryMemberList[n]['link'] = countryDomain + link[0]
                            
            except Exception as e:
                
                print('parsing the html did not work. Possibly you first have to downloadMemberListPagesOfCountries(). The original error message is:', e)
            
            # save interim results to files
            
            f = open('output/' + country +'MemberList.txt', 'w+')
            f.write(str(dictionaryMemberList))
            f.close

    def downloadMemberDataHtmls(self, listOfCountries):
        
        for country in listOfCountries:
            
            f = open('output/' + country +'MemberList.txt')
            text = f.read()
            
            dictionaryMemberList = eval(text)
            
            
            for memberid in dictionaryMemberList:
                
                
                memberLink = dictionaryMemberList[memberid]['link']
                
                # download the html page of the Member                    

                response = urllib.request.urlopen(memberLink)
                webContent = response.read().decode('UTF-8')

                # save interim results to files
                
                filename = 'pages/' + country + '/' + str(memberid) +'.html'
                
                os.makedirs(os.path.dirname(filename), exist_ok=True)
                f = open( filename, 'w+')
                f.write(webContent)
                f.close
                
            
    def parseMemberData2dictionary(self, listOfCountries):
        
        for country in listOfCountries:
            
            print('started to parse data of member of ' + country + ' ..')
            
            f = open('output/' + country +'MemberList.txt')
            text = f.read()
            
            dictionaryMemberList = eval(text)
            
            
            countryConf = self.config.get(country)
            countryDomain = countryConf.get('domain')
            countryConfMember = countryConf.get('member')
            countryConfMemberInfo1 = countryConfMember.get('info-1')
            countryConfMemberInfo1Parent = countryConfMemberInfo1.get('parent')
            countryConfMemberInfo1ChildPoliticalParty = countryConfMemberInfo1.get('child-politicalParty')
        
            for memberid in dictionaryMemberList:
                
                print('started to parse data of member with name ' + dictionaryMemberList[memberid]['name'] + ' ..')
                
                filename = 'pages/' + country + '/' + str(memberid) +'.html'
                
                tree = lxml.html.soupparser.parse(filename)
                
                politicalParty = tree.xpath(countryConfMemberInfo1Parent + countryConfMemberInfo1ChildPoliticalParty)
                
                print('oi', politicalParty)
                
                if len(politicalParty) > 0:
                    
                    dictionaryMemberList[memberid]['political party'] = politicalParty[0]
                
                
                
            f = open('output/' + country +'MemberList.txt', 'w+')
            f.write(str(dictionaryMemberList))
            f.close