def objects_management():
add_node_form = AddNode(request.form)
add_link_form = AddLink(request.form)
all_nodes = Node.choices()
add_link_form.source.choices = add_link_form.destination.choices = all_nodes
if request.method == 'POST':
file = request.files['file']
if allowed_file(secure_filename(file.filename), {'xls', 'xlsx'}):
book = open_workbook(file_contents=file.read())
for obj_type, cls in object_class.items():
try:
sheet = book.sheet_by_name(obj_type)
# if the sheet cannot be found, there's nothing to import
except XLRDError:
continue
properties = sheet.row_values(0)
for row_index in range(1, sheet.nrows):
kwargs = dict(zip(properties, sheet.row_values(row_index)))
kwargs['type'] = obj_type
object_factory(**kwargs)
db.session.commit()
return render_template(
'object_management.html',
names=pretty_names,
node_fields=node_public_properties,
nodes=Node.serialize(),
link_fields=link_public_properties,
links=Link.serialize(),
add_node_form=add_node_form,
add_link_form=add_link_form
)
def view(view_type):
add_node_form = AddNode(request.form)
add_link_form = AddLink(request.form)
all_nodes = Node.choices()
add_link_form.source.choices = all_nodes
add_link_form.destination.choices = all_nodes
view_options_form = ViewOptionsForm(request.form)
google_earth_form = GoogleEarthForm(request.form)
scheduling_form = SchedulingForm(request.form)
scheduling_form.job.choices = Job.choices()
scheduling_form.nodes.choices = all_nodes
scheduling_form.pools.choices = Pool.choices()
labels = {'node': 'name', 'link': 'name'}
if 'view_options' in request.form:
# update labels
labels = {
'node': request.form['node_label'],
'link': request.form['link_label']
}
# for the sake of better performances, the view defaults to markercluster
# if there are more than 2000 nodes
view = 'leaflet' if len(Node.query.all()) < 2000 else 'markercluster'
if 'view' in request.form:
view = request.form['view']
# name to id
name_to_id = {node.name: id for id, node in enumerate(Node.query.all())}
return render_template(
f'{view_type}_view.html',
pools=Pool.query.all(),
parameters=Parameters.query.one().serialized,
view=view,
scheduling_form=scheduling_form,
view_options_form=view_options_form,
google_earth_form=google_earth_form,
add_node_form=add_node_form,
add_link_form=add_link_form,
node_fields=node_public_properties,
link_fields=link_public_properties,
labels=labels,
names=pretty_names,
subtypes=list(node_class),
name_to_id=name_to_id,
nodes=Node.serialize(),
links=Link.serialize()
)
def pool_management():
pool_object_form = PoolObjectsForm(request.form)
pool_object_form.nodes.choices = Node.choices()
pool_object_form.links.choices = Link.choices()
return render_template('pool_management.html',
form=AddPoolForm(request.form),
pool_object_form=pool_object_form,
names=pretty_names,
fields=pool_public_properties,
pools=Pool.serialize())
def task_management():
scheduling_form = SchedulingForm(request.form)
scheduling_form.nodes.choices = Node.choices()
scheduling_form.pools.choices = Pool.choices()
scheduling_form.job.choices = Job.choices()
return render_template('task_management.html',
fields=task_public_properties,
tasks=Task.serialize(),
compare_form=CompareForm(request.form),
scheduling_form=scheduling_form)
def scripts():
scheduling_form = SchedulingForm(request.form)
scheduling_form.nodes.choices = Node.choices()
scheduling_form.pools.choices = Pool.choices()
scheduling_form.job.choices = Job.choices()
return render_template(
'script_management.html',
fields=script_public_properties,
type_to_form={t: s(request.form)
for t, s in type_to_form.items()},
names=pretty_names,
scheduling_form=scheduling_form,
scripts=Script.serialize())
def workflow_editor(workflow_id=None):
workflow_editor_form = WorkflowEditorForm(request.form)
workflow_editor_form.workflow.choices = Workflow.choices()
workflow = get_obj(Workflow,
id=workflow_id).serialized if workflow_id else None
scheduling_form = SchedulingForm(request.form)
scheduling_form.scripts.choices = Script.choices()
scheduling_form.nodes.choices = Node.choices()
scheduling_form.pools.choices = Pool.choices()
print(workflow)
return render_template('workflow_editor.html',
workflow_editor_form=workflow_editor_form,
scheduling_form=scheduling_form,
compare_form=CompareForm(request.form),
names=pretty_names,
workflow=workflow)
def workflow_editor(workflow_id=None):
add_existing_task_form = AddExistingTaskForm(request.form)
workflow_editor_form = WorkflowEditorForm(request.form)
workflow_editor_form.workflow.choices = Workflow.choices()
workflow = get_obj(Workflow, id=workflow_id)
scheduling_form = SchedulingForm(request.form)
scheduling_form.job.choices = Job.choices()
scheduling_form.nodes.choices = Node.choices()
scheduling_form.pools.choices = Pool.choices()
add_existing_task_form.task.choices = Task.choices()
return render_template(
'workflow_editor.html',
add_existing_task_form=add_existing_task_form,
workflow_editor_form=workflow_editor_form,
scheduling_form=scheduling_form,
compare_form=CompareForm(request.form),
names=pretty_names,
workflow=workflow.serialized if workflow_id else None)
def objects_download():
nodes = Node.serialize()
ws = {}
wb = xlwt.Workbook()
style0 = xlwt.easyxf(
'font: name Times New Roman, color-index black, bold on',
num_format_str='#,##0.00')
style1 = xlwt.easyxf(num_format_str='#,##0.00')
header_index = 0
# Write tabs and headers
for tab, header in type_to_public_properties.items():
column = 0
ws[tab] = wb.add_sheet(tab)
ws[tab].row_index = 1
for entry in header:
ws[tab].write(header_index, column, entry, style0)
column = column + 1
column = 0
for node in nodes:
for k, v in node.items():
if k is not 'id':
try:
ws[node['type']].write(ws[node['type']].row_index, column,
v, style1)
column = column + 1
except Exception:
continue
column = 0
ws[node['type']].row_index = ws[node['type']].row_index + 1
# Done writing rows
obj_file = Path.cwd() / 'projects' / 'objects.xls'
wb.save(str(obj_file))
sfd = send_file(filename_or_fp=str(obj_file),
as_attachment=True,
attachment_filename='objects.xls')
return sfd