def output_add(form_add):
action = '{action} {controller}'.format(
action=TRANSLATIONS['add']['title'],
controller=TRANSLATIONS['output']['title'])
error = []
dep_unmet, _ = return_dependencies(form_add.output_type.data.split(',')[0])
if dep_unmet:
list_unmet_deps = []
for each_dep in dep_unmet:
list_unmet_deps.append(each_dep[0])
error.append(
"The {dev} device you're trying to add has unmet dependencies: "
"{dep}".format(dev=form_add.output_type.data,
dep=', '.join(list_unmet_deps)))
if len(form_add.output_type.data.split(',')) < 2:
error.append("Must select an Output type")
if not is_int(form_add.output_quantity.data, check_range=[1, 20]):
error.append("{error}. {accepted_values}: 1-20".format(
error=gettext("Invalid quantity"),
accepted_values=gettext("Acceptable values")))
if not error:
for _ in range(0, form_add.output_quantity.data):
try:
output_type = form_add.output_type.data.split(',')[0]
interface = form_add.output_type.data.split(',')[1]
new_output = Output()
new_output.name = str(OUTPUT_INFO[output_type]['name'])
new_output.output_type = output_type
new_output.interface = interface
if output_type in [
'wired', 'wireless_rpi_rf', 'command', 'python'
]:
new_output.measurement = 'duration_time'
new_output.unit = 's'
elif output_type in OUTPUTS_PWM:
new_output.measurement = 'duty_cycle'
new_output.unit = 'percent'
new_output.channel = 0
if output_type == 'wired':
new_output.state_startup = '0'
new_output.state_shutdown = '0'
elif output_type == 'wireless_rpi_rf':
new_output.pin = None
new_output.protocol = 1
new_output.pulse_length = 189
new_output.on_command = '22559'
new_output.off_command = '22558'
elif output_type == 'command':
new_output.on_command = '/home/pi/script_on.sh'
new_output.off_command = '/home/pi/script_off.sh'
elif output_type == 'command_pwm':
new_output.pwm_command = '/home/pi/script_pwm.sh ((duty_cycle))'
elif output_type == 'pwm':
new_output.pwm_hertz = 22000
new_output.pwm_library = 'pigpio_any'
elif output_type == 'python':
new_output.on_command = """
timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
write_string = "{ts}: ID: {id}: ON\\n".format(id=output_id, ts=timestamp)
with open("/home/pi/Mycodo/OutputTest.txt", "a") as myfile:
myfile.write(write_string)"""
new_output.off_command = """
timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
write_string = "{ts}: ID: {id}: OFF\\n".format(id=output_id, ts=timestamp)
with open("/home/pi/Mycodo/OutputTest.txt", "a") as myfile:
myfile.write(write_string)"""
elif output_type == 'python_pwm':
new_output.pwm_command = """
timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
write_string = "{ts}: ID: {id}: Duty Cycle: ((duty_cycle)) %\\n".format(
id=output_id, ts=timestamp)
with open("/home/pi/Mycodo/OutputTest.txt", "a") as myfile:
myfile.write(write_string)"""
elif output_type == 'atlas_ezo_pmp':
new_output.output_mode = 'fastest_flow_rate'
new_output.flow_rate = 10
if interface == 'FTDI':
new_output.location = '/dev/ttyUSB0'
elif interface == 'I2C':
new_output.location = '0x67'
new_output.i2c_bus = 1
elif interface == 'UART':
new_output.location = '/dev/ttyAMA0'
new_output.baud_rate = 9600
if not error:
new_output.save()
display_order = csv_to_list_of_str(
DisplayOrder.query.first().output)
DisplayOrder.query.first().output = add_display_order(
display_order, new_output.unique_id)
db.session.commit()
# Add device measurements
for measurement, measure_data in OUTPUT_INFO[
new_output.output_type]['measure'].items():
for unit, unit_data in measure_data.items():
for channel, _ in unit_data.items():
new_measurement = DeviceMeasurements()
new_measurement.device_id = new_output.unique_id
new_measurement.name = ''
new_measurement.is_enabled = True
new_measurement.measurement = measurement
new_measurement.unit = unit
new_measurement.channel = channel
new_measurement.save()
manipulate_output('Add', new_output.unique_id)
except sqlalchemy.exc.OperationalError as except_msg:
error.append(except_msg)
except sqlalchemy.exc.IntegrityError as except_msg:
error.append(except_msg)
flash_success_errors(error, action, url_for('routes_page.page_output'))
if dep_unmet:
return 1
from mycodo.config import OUTPUT_INFO
output_unique_id = {}
# Go through each output to get output unique_id
with session_scope(MYCODO_DB_PATH) as output_sess:
for each_output in output_sess.query(Output).all():
output_unique_id[each_output.unique_id] = []
# Create measurements in device_measurements table
for measurement, measure_data in OUTPUT_INFO[each_output.output_type]['measure'].items():
for unit, unit_data in measure_data.items():
for channel, channel_data in unit_data.items():
new_measurement = DeviceMeasurements()
new_measurement.device_id = each_output.unique_id
new_measurement.name = ''
new_measurement.is_enabled = True
new_measurement.measurement = measurement
new_measurement.unit = unit
new_measurement.channel = channel
output_sess.add(new_measurement)
output_sess.commit()
output_unique_id[each_output.unique_id].append(
new_measurement.unique_id)
# Update all outputs in Dashboard elements to new unique_ids
for each_dash in output_sess.query(Dashboard).all():
each_dash.output_ids = each_dash.output_ids.replace(',output', '')
output_sess.commit()
for each_output_id, list_device_ids in output_unique_id.items():
