# -*- coding: utf-8
# Copyright 2005 by Meelis Mets

from Products.Archetypes.public import BaseSchema, Schema
from Products.Archetypes.public import StringField, LinesField
from Products.Archetypes.public import LinesWidget, TextAreaWidget, IdWidget, StringWidget, SelectionWidget
from Products.Archetypes.public import OrderedBaseFolder, registerType
from Globals import InitializeClass
from Products.CMFCore.utils import getToolByName
from Permissions import *
from types import UnicodeType
from AccessControl import ClassSecurityInfo, Unauthorized
from config import PROJECT_NAME
schema = BaseSchema


class SchoolDataBase(OrderedBaseFolder):
    """ SchoolDataBase product """
    meta_type = "SchoolDataBase"
    archetype_name = "SchoolDataBase"
    global_allow = 0
    filter_content_types = True
    allowed_content_types = ('School',)
    security = ClassSecurityInfo()
    schema = schema
    actions = (
        {
        'id':'view',
        'name':'View',
        'action':'string:${object_url}/schooldatabase_view',
        'permissions': ('View',),
        },
#        {
#        'id':'registration',
#        'name':'Registration',
#        'action':'string:${object_url}/registration_view',
#        'permissions': ('View',),
#        },
        {
        'id':'update',
        'name':'Update',
        'action':'string:${object_url}/update_schooldata',
        'permissions': ('Manage Portal',),
        },
    )
    
    def __init__(self, id):
	self.id = id
        self.schoolList = []
    
#    def manage_afterAdd(self,item,container):
#        """sets permissions"""
#        self.manage_permission('Manage Portal', ('Manager','School') ,1)

    def getSchools(self):
        """school objects"""
        return self.objectValues("School")

    def getSchoolObj(self,id='all'):
        if id=='all':
            objects=[]
            for name in self.getSchools():
                objects.append(getattr(self,name.id))
            return objects
        else:
            return getattr(self,id)

    def getSchoolTitle(self,id):
        """returns title"""
        return getattr(self,id).title

    def getMembersName(self,member):
        """returns members fullname"""
        pm = getToolByName(self, 'portal_membership')
        member = pm.getMemberById(str(member))
        return member.getProperty('fullname')

    def getSchoolLocations(self):
        """returns list of school locations"""
        data = self.getFolderContents()
        schoolLocation = []
        for school in data:
            if school.location in schoolLocation:
                pass
            else:
                schoolLocation.append(school.location)
        return schoolLocation

    def getSchoolsByLocationAndBySubjects(self, loc, sub):
        """returns list of schools by locations"""
        locations = self.getSchoolLocations()
        subjects = [0,0,0]
        if loc==None:
            loc='0'
        c = 0
        for s in sub:
            if sub[c]!=None:
                subjects[c]=sub[c]
            c = c + 1
        location = locations[int(loc)]
        data = self.getFolderContents()
        schoolList = []
        for school in data:
            school = getattr(self,school.id)
            if school.location == location:
                if school.subjects[int(subjects[0])]['nr']>0 and school.subjects[int(subjects[1])]['nr']>0 and school.subjects[int(subjects[2])]['nr']>0:
                    nr = school.subjects[int(subjects[0])]['nr']
                    if school.subjects[int(subjects[1])]['nr'] < nr:
                        nr = school.subjects[int(subjects[1])]['nr']
                        if school.subjects[int(subjects[2])]['nr'] < nr:
                            nr = school.subjects[int(subjects[2])]['nr']
                    schoolList.append({'name':school.title, 'id':school.id, 'nr':nr})
        return schoolList
    
    security.declareProtected('Manage Portal', 'updateSchoolList')
    def updateSchoolList(self):
        """ get data from db as list of schools"""
        import urllib2
        id_list = []
        count=0
        size='50'
        for value in range(1, 901, 50):
            if count>16:
                size='34'
            count=count+1
            url='https://eh-jas.hm.ee/avalik/oas/Otsing.uix?event=goto&source=tabel&value='+str(value)+'&size='+size
            f = urllib2.urlopen(url)
            html = f.read()
            list = html.split('<span class="xg">')
            list.pop(0)
            for item in list:
                name=''
                id=''
                reg=''
                splitter=item.split('</span>')
                reg=splitter[0]
                if reg not in ['','0']:
                    if len(splitter)>1:
                        splitter=splitter[1].split('<a href="')
                        if len(splitter)>1:
                            splitter=splitter[1].split('">')
                            id=splitter[0].split('?')
                            id=id[1].split('&')
                            id='?'+id[0]
                            if len(splitter)>1:
                                splitter=splitter[1].split('<')
                                name=splitter[0]
                                print name
                    if name!='' and id!='':
                        id_list.append({'name':name,'id':id,'reg':reg})
        self.schoolList = id_list
        self._p_changed = 1
        
    security.declareProtected('View', 'getSchoolList')
    def getSchoolList(self):
        """returns schoolList"""
        return self.schoolList

    security.declareProtected('Manage Portal', 'updateSchool')
    def updateSchool(self, REQUEST):
        """ get data from db as data of schools"""
        import urllib2
        new_list = ['null']
        key = REQUEST.get('otsingKey')
        url='https://eh-jas.hm.ee/avalik/oas/muuda2.uix?otsingKey='+key+'&event=valiOtsing'
        f = urllib2.urlopen(url)
        html = f.read()
        html = self.charReplace(html)
        html = html.replace('<span class="xd">','<split_begin>')
        html = html.replace('<td width="250" align="right" class="xd">','<split_begin>')
        html = html.replace('</td><td width="12">','<split_end>')
        html = html.replace('</label></span>','<split_end>')
        html = html.replace('<split_end><split_end>','<split_end>')
        html = html.replace('<span class="xg">','<split_begin_2>')   
        html = html.replace('<a href="','<split_begin_2>')
        html = html.replace('</span></td>','<split_end_2>')
        html = html.replace('" target="_blank">','<split_end_2>')
        data = html.split('<split_begin>')
        if len(data)>1:
            data.pop(0)
            data.pop(0)
        new_data = []
        for d in data:
            d=d.split('<split_end>')
            da=d[0].split('">')
            if len(da)>1:
                da=da[1]
            else:
                da=da[0]
            db=d[1].split('<split_begin_2>')
            if len(db)>1:
                db=db[1].split('<split_end_2>')
                db=db[0]
            else:
                db=''
            if da!='':
                new_data.append({'name':da,'value':db})
        schoolInfo = new_data
        #school abjects
        typestool = getToolByName(self, 'portal_types')
        puf_type = getattr(typestool, 'School', None)
        if not puf_type:
            return 0
        puf_type.global_allow = 1
        id = schoolInfo[0]['value']
        if hasattr(self, id):
            #this school object is already in database
            pass
        else:
            id = self.invokeFactory("School", id=id,
                 title=schoolInfo[1]['value'],
                 schooltype=schoolInfo[4]['value'],
                 address=schoolInfo[12]['value'],
                 location=schoolInfo[11]['value'].replace(',',''),
                 email=schoolInfo[9]['value'],
                 www=schoolInfo[10]['value'],
                 phone=schoolInfo[7]['value'],)
        puf_type.global_allow = 0
        return REQUEST.RESPONSE.redirect(self.absolute_url()+'/update_schooldata')

    security.declareProtected('Manage Portal', 'isInDB')
    def isInDB(self, id):
        """if school is in db"""
        if hasattr(self, id):
            return '#CCFFCC'
        return '#FFCCCC'

    def charReplace(self,string):
        string=string.replace('&auml;','\xc3\xa4') #ä
        string=string.replace('&Auml;','\xc3\x84') #Ä 
        string=string.replace('&ouml;','\xc3\xb6') #ö
        string=string.replace('&Ouml;','\xc3\xb6') #Ö 
        string=string.replace('&otilde;','\xc3\xb5') #õ
        string=string.replace('&Otilde;','\xC3\x95') #Õ 
        string=string.replace('&uuml;','\xc3\xbc') #ü
        string=string.replace('&Uuml;','\xc3\x9c') #Ü
        string=string.replace('&#382;','z') #z~
        string=string.replace('&#353;','s') #s~
        return string

registerType(SchoolDataBase, PROJECT_NAME)
    
class School(OrderedBaseFolder):
    """ School """
    meta_type = "School"
    archetype_name = "School"
    exclude_from_nav = True
    global_allow = 0
    filter_content_types = True
    allowed_content_types = ()
    security = ClassSecurityInfo()
    schema = schema + Schema((
    
    StringField('title',
    widget=StringWidget(
        label="School name",
        description="",
        label_msgid='label_school_name',
        description_msgid='description_school_name',
        i18n_domain="eportfolio"),
        ),
    StringField('schooltype',
    widget=StringWidget(
        label="School type",
        description="",
        label_msgid='label_school_type',
        description_msgid='description_school_type',
        i18n_domain="eportfolio"),
        ),
    StringField('address',
    widget=StringWidget(
        label="Address",
        description="",
        label_msgid='label_school_address',
        description_msgid='description_school_address',
        i18n_domain="eportfolio"),
        ),
    StringField('location',
    widget=StringWidget(
        label="Location",
        description="",
        label_msgid='label_school_location',
        description_msgid='description_school_location',
        i18n_domain="eportfolio"),
        ),
    StringField('headmaster',
    widget=StringWidget(
        label="Headmaster",
        description="",
        label_msgid='label_school_headmaster',
        description_msgid='description_school_headmaster',
        i18n_domain="eportfolio"),
        ),
    StringField('email',
    widget=StringWidget(
        label="E-mail",
        description="",
        label_msgid='label_school_email',
        description_msgid='description_school_email',
        i18n_domain="eportfolio"),
        ),
    StringField('www',
    widget=StringWidget(
        label="WWW",
        description="",
        label_msgid='label_school_www',
        description_msgid='description_school_www',
        i18n_domain="eportfolio"),
        ),
    StringField('phone',
    widget=StringWidget(
        label="Phone",
        description="",
        label_msgid='label_school_phone',
        description_msgid='description_school_phone',
        i18n_domain="eportfolio"),
        ),
    StringField('studinglanguage',
    widget=StringWidget(
        label="Language",
        description="",
        label_msgid='label_school_language',
        description_msgid='description_school_language',
        i18n_domain="eportfolio"),
        ),
    StringField('nrofstudents',
    widget=StringWidget(
        label="Number of students",
        description="",
        label_msgid='label_school_nrofstudents',
        description_msgid='description_school_nrofstudents',
        i18n_domain="eportfolio"),
        ),
))
    
    actions = (
        {
        'id':'view',
        'name':'View',
        'action':'string:${object_url}/base_view',
        'permissions': ('View',),
        },
        {
        'id':'owner',
        'name':'Owner',
        'action':'string:${object_url}/school_owner',
        'permissions': ('Manage Portal',),
        },
        {
        'id':'subjects',
        'name':'Subjects',
        'action':'string:${object_url}/school_subjects',
        'permissions': ('Modify portal content',),
        },
    )


    def __init__(self, id):
	self.id = id
        self.subjects = []
        self.time = []
        self.practice = []
    
#    def manage_afterAdd(self,item,container):
#        """sets permissions"""
#        self.manage_permission('Manage Portal', ('Manager','School') ,1)
#        self.manage_permission('View', ('Manager','School') ,1)

    def getSubjects(self):
        """get subjects"""
        return self.subjects

    def getNumbers(self,max,min=0):
        """get numbers"""
        numbers = []
        for nr in range(min,max+min,1):
            nr = str(nr)
            if len(nr)<2:
                nr='0'+nr
            numbers.append(nr)
        return numbers

registerType(School, PROJECT_NAME)