def device_management():
return render_template(
'device_management.html',
names=pretty_names,
fields=device_public_properties,
devices=Device.serialize(),
add_device_form=AddDevice(request.form)
)
def view(view_type):
add_device_form = AddDevice(request.form)
add_link_form = AddLink(request.form)
all_devices = Device.choices()
add_link_form.source.choices = all_devices
add_link_form.destination.choices = all_devices
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.devices.choices = all_devices
scheduling_form.pools.choices = Pool.choices()
labels = {'device': 'name', 'link': 'name'}
if 'view_options' in request.form:
# update labels
labels = {
'device': request.form['device_label'],
'link': request.form['link_label']
}
# for the sake of better performances, the view defaults to markercluster
# if there are more than 2000 devices
view = 'leaflet' if len(Device.query.all()) < 2000 else 'markercluster'
if 'view' in request.form:
view = request.form['view']
# name to id
name_to_id = {
device.name: id for id, device in enumerate(Device.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_device_form=add_device_form,
add_link_form=add_link_form,
device_fields=device_public_properties,
link_fields=link_public_properties,
labels=labels,
names=pretty_names,
device_subtypes=device_subtypes,
link_colors=link_subtype_to_color,
name_to_id=name_to_id,
devices=Device.serialize(),
links=Link.serialize()
)
def objects_download():
devices = Device.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
for tab, header in cls_to_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 device in devices:
for k, v in device.items():
if k is not 'id':
try:
ws[device['type']].write(
ws[device['type']].row_index,
column,
v,
style1
)
column = column + 1
except Exception:
continue
column = 0
ws[device['type']].row_index = ws[device['type']].row_index + 1
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
def view(view_type):
add_link_form = AddLink(request.form)
all_devices = Device.choices()
add_link_form.source.choices = all_devices
add_link_form.destination.choices = all_devices
labels = {'device': 'name', 'link': 'name'}
if 'view_options' in request.form:
labels = {
'device': request.form['device_label'],
'link': request.form['link_label']
}
if len(Device.query.all()) < 50:
view = 'glearth'
elif len(Device.query.all()) < 2000:
view = 'leaflet'
else:
view = 'markercluster'
if 'view' in request.form:
view = request.form['view']
# name to id
name_to_id = {
device.name: id
for id, device in enumerate(Device.query.all())
}
return render_template(f'{view_type}_view.html',
pools=Pool.query.all(),
parameters=Parameters.query.one().serialized,
view=view,
view_options_form=ViewOptionsForm(request.form),
google_earth_form=GoogleEarthForm(request.form),
add_device_form=AddDevice(request.form),
add_link_form=add_link_form,
device_fields=device_public_properties,
link_fields=link_public_properties,
labels=labels,
names=pretty_names,
device_subtypes=device_subtypes,
link_colors=link_subtype_to_color,
name_to_id=name_to_id,
devices=Device.serialize(),
links=Link.serialize())