"""Forms for cedar application

$Id: forms.py 7033 2019-10-03 20:15:55Z brideout $
"""

# standard python imports
import os.path
import datetime

# django imports
import django.forms
import django.utils.html
import django.utils.safestring

# Madrigal imports
import madrigal.metadata
import madrigal.ui.userData



class AdminLogForm(django.forms.Form):
    """AdminLogForm is a Form class for the get log admin interface
    """
    def __init__(self, *args, **kwargs):
        super(AdminLogForm, self).__init__(*args, **kwargs)
        
        selectedSite = None
        kinstChoices = [('0', 'Select a site first'),]
        if len(args) > 0:
            try:
                if 'site' in args[0]:
                    selectedSite = int(args[0]['site'])
            except:
                pass
        
        madDB = madrigal.metadata.MadrigalDB()
        madSiteObj = madrigal.metadata.MadrigalSite(madDB)
        madInstObj = madrigal.metadata.MadrigalInstrument(madDB)
        self.madUserObj = madrigal.ui.userData.MadrigalUserData(madDB)
        
        madExpObj = madrigal.metadata.MadrigalExperiment(madDB, os.path.join(madDB.getMadroot(), 'metadata/expTabAll.txt'))
        siteDict = {} # temp dict with key = siteId, value = list of kinst where there is data at that site
        for i in range(madExpObj.getExpCount()):
            siteId = madExpObj.getExpSiteIdByPosition(i)
            kinst = madExpObj.getKinstByPosition(i)
            try:
                if kinst in siteDict[siteId]:
                    continue
            except KeyError:
                pass
            expUrl = madExpObj.getExpUrlByPosition(i)
            if madDB.isTestExperiment(expUrl):
                continue
            if siteId in siteDict:
                siteDict[siteId].append(kinst)
            else:
                siteDict[siteId] = [kinst]
                
        # make siteList, which is sorted list of (siteid, kinst, instName) and siteChoices
        # which is (siteID, siteName)
        self.siteList = []
        self.siteChoices = [("0", "Select your site")]
        sites = list(siteDict.keys())
        sites.sort()
        for site in sites:
            if site not in (10,): #skip CEDAR site
                self.siteChoices.append((site, madSiteObj.getSiteName(site)))
            kinstList = siteDict[site]
            kinstList.sort()
            for kinst in kinstList:
                self.siteList.append((site, kinst, madInstObj.getInstrumentName(kinst)))
                if not selectedSite is None:
                    if selectedSite == site:
                        kinstChoices.append((kinst, madInstObj.getInstrumentName(kinst)))
            
        # now add to self.siteList and self.siteChoices from hostSite.txt file
        filename = os.path.join(madDB.getMadroot(), 'metadata/hostSite.txt')
        f = open(filename)
        
        lines = f.readlines()
        for line in lines:
            items = line.strip().split(',')
            if len(items) < 3:
                continue
            siteId = int(items[0])
            # check this is unique siteId
            if siteId in siteDict:
                raise IOError('File hostSite.txt has siteId %i from siteTab.txt' % (siteId))
            kinstList = []
            for item in items[2:]:
                kinst = int(item)
                kinstList.append(kinst)
                self.siteList.append((siteId, kinst, madInstObj.getInstrumentName(kinst)))
                if not selectedSite is None:
                    if selectedSite == siteId:
                        kinstChoices.append((kinst, madInstObj.getInstrumentName(kinst)))
            self.siteChoices.append((str(siteId), items[1]))
            
        self.fields['site'] = django.forms.ChoiceField(widget = django.forms.Select(attrs={"onChange":'selectSite(this.form)'}),
                                                       choices=self.siteChoices, 
                                                       initial='0')
        
        self.fields['password'] = django.forms.CharField(widget = django.forms.PasswordInput())
        
        self.fields['kinst'] = django.forms.CharField(widget=django.forms.SelectMultiple(choices=kinstChoices))
        
        # time selection
        self.fields['startDate'] = django.forms.DateField(input_formats=['%Y-%m-%d'],
                                                          initial=datetime.datetime(2011,1,1,0,0,0))
        now = datetime.datetime.now()
        self.fields['endDate'] = django.forms.DateField(input_formats=['%Y-%m-%d'],
                                                        initial=datetime.datetime(now.year,12,31,0,0,0))
        
        
    def clean(self):
        
        """clean verifies password
        """
        siteId = int(self.cleaned_data['site'])
        siteName = None
        for id, name in self.siteChoices:
            if int(id) == siteId:
                siteName = name
                break
        if siteName is None:
            raise django.forms.ValidationError('Illegal site choosen')
        if not self.madUserObj.verifyUser(siteName, self.cleaned_data['password']):
            raise django.forms.ValidationError('Incorrect password')