73 lines
3.1 KiB
Python
73 lines
3.1 KiB
Python
from flask_table import Table, Col, LinkCol, BoolCol, DatetimeCol, NestedTableCol
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import SelectMultipleField, StringField, BooleanField, SelectField, validators, HiddenField
|
|
from wtforms_alchemy import ModelForm
|
|
|
|
from pb.models import RequiredDocument, Investigator, StudyDetails
|
|
|
|
|
|
class StudyForm(FlaskForm):
|
|
STUDYID = HiddenField()
|
|
TITLE = StringField('Title', [validators.required()])
|
|
NETBADGEID = StringField('User UVA Computing Id', [validators.required()])
|
|
requirements = SelectMultipleField("Requirements",
|
|
render_kw={'class': 'multi'},
|
|
choices=[(rd.AUXDOCID, rd.AUXDOC) for rd in RequiredDocument.all()])
|
|
HSRNUMBER = StringField('HSR Number')
|
|
Q_COMPLETE = BooleanField('Complete in Protocol Builder?', default='checked',
|
|
false_values=(False, 'false', 0, '0'))
|
|
|
|
class InvestigatorForm(FlaskForm):
|
|
NETBADGEID = StringField('UVA Id')
|
|
INVESTIGATORTYPE = SelectField("InvestigatorType", choices=[(i.INVESTIGATORTYPE, i.INVESTIGATORTYPEFULL) for i in Investigator.all_types()])
|
|
|
|
class StudyDetailsForm(ModelForm, FlaskForm):
|
|
class Meta:
|
|
model = StudyDetails
|
|
|
|
class RequirementsTable(Table):
|
|
AUXDOCID = Col('Code')
|
|
AUXDOC = Col('Name')
|
|
|
|
class InvestigatorsTable(Table):
|
|
NETBADGEID = Col('UVA Id')
|
|
INVESTIGATORTYPE = Col('Type')
|
|
delete = LinkCol(
|
|
'delete', 'del_investigator', url_kwargs=dict(inv_id='id'),
|
|
anchor_attrs={'class': 'btn btn-icon btn-warn', 'title': 'Delete Investigator'},
|
|
th_html_attrs={'class': 'mat-icon text-center', 'title': 'Delete Investigator'}
|
|
)
|
|
|
|
|
|
class StudyTable(Table):
|
|
def sort_url(self, col_id, reverse=False):
|
|
pass
|
|
edit = LinkCol(
|
|
'edit', 'edit_study', url_kwargs=dict(study_id='STUDYID'),
|
|
anchor_attrs={'class': 'btn btn-icon btn-primary', 'title': 'Edit Study'},
|
|
th_html_attrs={'class': 'mat-icon text-center', 'title': 'Edit Study'}
|
|
)
|
|
delete = LinkCol(
|
|
'delete', 'del_study', url_kwargs=dict(study_id='STUDYID'),
|
|
anchor_attrs={'class': 'btn btn-icon btn-warn', 'title': 'Delete Study'},
|
|
th_html_attrs={'class': 'mat-icon text-center', 'title': 'Delete Study'}
|
|
)
|
|
details = LinkCol(
|
|
'ballot', 'study_details', url_kwargs=dict(study_id='STUDYID'),
|
|
anchor_attrs={'class': 'btn btn-icon btn-default', 'title': 'Edit Questions'},
|
|
th_html_attrs={'class': 'mat-icon text-center', 'title': 'Edit Questions'}
|
|
)
|
|
add_inv = LinkCol(
|
|
'person_add', 'new_investigator', url_kwargs=dict(study_id='STUDYID'),
|
|
anchor_attrs={'class': 'btn btn-icon btn-accent', 'title': 'Add Investigator'},
|
|
th_html_attrs={'class': 'mat-icon text-center', 'title': 'Add Investigator'}
|
|
)
|
|
STUDYID = Col('Study Id')
|
|
TITLE = Col('Title')
|
|
NETBADGEID = Col('User')
|
|
DATE_MODIFIED = DatetimeCol('Last Update', "medium")
|
|
Q_COMPLETE = BoolCol('Complete?')
|
|
requirements = NestedTableCol('Requirements', RequirementsTable)
|
|
investigators = NestedTableCol('Investigators', InvestigatorsTable)
|
|
|