|
|
from django import forms
|
|
|
import django.utils.html
|
|
|
import django.utils.safestring
|
|
|
import django.template.defaulttags
|
|
|
|
|
|
# third party imports
|
|
|
import numpy
|
|
|
|
|
|
# Madrigal imports
|
|
|
import madrigal.metadata
|
|
|
import madrigal.ui.web
|
|
|
|
|
|
# madrigal imports
|
|
|
import madrigal._derive
|
|
|
import madrigal.metadata
|
|
|
import madrigal.ui.web
|
|
|
import madrigal.cedar
|
|
|
import madrigal.isprint
|
|
|
import madweb.forms
|
|
|
|
|
|
import datetime, time
|
|
|
|
|
|
def getSelection(keyword, args, kwargs):
|
|
|
"""getSelection returns '0' if keyword not a key in either args[0] or kwargs,
|
|
|
otherwise the value
|
|
|
|
|
|
args, kwargs - arguments as passed into SingleExpDefaultForm __init__
|
|
|
"""
|
|
|
if len(args) == 0 and len(list(kwargs.keys())) == 0:
|
|
|
return('0') # default case when no data passed in
|
|
|
elif len(args) > 0:
|
|
|
# args[0] is data dict argument to bind data
|
|
|
if keyword in args[0]:
|
|
|
return(args[0][keyword])
|
|
|
else:
|
|
|
return('0')
|
|
|
elif keyword in kwargs:
|
|
|
return(kwargs[keyword])
|
|
|
elif 'data' in kwargs:
|
|
|
if keyword in kwargs['data']:
|
|
|
return(kwargs['data'][keyword])
|
|
|
else:
|
|
|
return('0')
|
|
|
else:
|
|
|
return('0')
|
|
|
|
|
|
def getExperimentList(args, kwargs, madWeb, header='Select experiment: '):
|
|
|
|
|
|
instrumentsId= int(getSelection('instruments', args, kwargs))
|
|
|
|
|
|
kinstList = [int(instrumentsId)]
|
|
|
startDate = datetime.datetime(1950,1,1)
|
|
|
startDT = datetime.datetime(startDate.year, startDate.month, startDate.day, 0, 0, 0)
|
|
|
now = datetime.datetime.now()
|
|
|
endDate = datetime.datetime(now.year, 12, 31, 23, 59, 59)
|
|
|
endDT = datetime.datetime(endDate.year, endDate.month, endDate.day, 23, 59, 59)
|
|
|
experiments = madWeb.getExperimentList(kinstList,startDT, endDT, True)
|
|
|
expListin = [('0', header),]
|
|
|
for exp in experiments:
|
|
|
expListin.append((exp[0], exp[2]))
|
|
|
|
|
|
# Using set
|
|
|
seen = set()
|
|
|
|
|
|
# using list comprehension
|
|
|
expList = [(a, b) for a, b in expListin
|
|
|
if not (b in seen or seen.add(b))]
|
|
|
|
|
|
return(expList)
|
|
|
|
|
|
class UpdataForm(forms.Form):
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
super(UpdataForm, self).__init__(*args, **kwargs)
|
|
|
madDB = madrigal.metadata.MadrigalDB()
|
|
|
madInstData = madrigal.metadata.MadrigalInstrumentData(madDB)
|
|
|
instruments = madInstData.getInstruments(0, True)
|
|
|
instList = [('0', "Select Instrument"), ]
|
|
|
for kinst, instDesc, siteID in instruments:
|
|
|
instList.append((str(kinst), instDesc))
|
|
|
|
|
|
instrumentSelection = getSelection('instruments', args, kwargs)
|
|
|
self.fields['instruments'] = django.forms.ChoiceField(widget = django.forms.Select(attrs={"onChange":'populateExp(this)'}),
|
|
|
choices=instList,
|
|
|
initial=instrumentSelection,
|
|
|
label='Instrument:')
|
|
|
|
|
|
madWebObj = madrigal.ui.web.MadrigalWeb(madDB)
|
|
|
experimentSelection = getSelection('experiments', args, kwargs)
|
|
|
self.fields['experiments'] = django.forms.ChoiceField(choices=getExperimentList(args, kwargs, madWebObj),
|
|
|
initial=experimentSelection,
|
|
|
required=False, label='Experiment:')
|
|
|
|
|
|
description = forms.CharField(widget=forms.Textarea(attrs={'cols': 40,'rows': 3, 'style': 'resize:none'}), label='Description')
|
|
|
type = forms.ChoiceField(choices=[('0', 'Public'),('1', 'Private')], initial=0,widget=forms.RadioSelect(attrs={'class': 'custom-radio'}))
|
|
|
file = forms.FileField(label='Select Files', widget=forms.ClearableFileInput(attrs={'multiple': True}))
|
|
|
|
|
|
class ExpForm(forms.Form):
|
|
|
"""SingleExpInstForm is a Form class for the instrument select field in the Single Experiment interface.
|
|
|
Use this because its faster to create than the full SingleExpDefaultForm
|
|
|
"""
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
super(ExpForm, self).__init__(*args, **kwargs)
|
|
|
madDB = madrigal.metadata.MadrigalDB()
|
|
|
madWebObj = madrigal.ui.web.MadrigalWeb(madDB)
|
|
|
experimentSelection = getSelection('experiments', args, kwargs)
|
|
|
self.fields['experiments'] = django.forms.ChoiceField(choices=getExperimentList(args, kwargs, madWebObj),
|
|
|
initial=experimentSelection,
|
|
|
required=False, label='Experiment')
|
|
|
|
