def get_selected_id(select_dict, value):
"""
Exits script, if value is invalid. Exits in error case
:param select_dict: Dictionary which describes the data.py
:param value: id or description
:return: id or, in initial case, empty string
"""
if value.isdigit():
key = 'id'
log('res is digit', value)
else:
key = 'description'
try:
res_list = [
v for v in select_dict['values']
if value.lower() in v[key].lower()
]
if len(res_list) > 1:
print("ERROR")
print("%s '%s' is non-unique, found %s canidates:" %
(select_dict['description'], value, len(res_list)))
for res in res_list:
print(" * %s (%s)" % (res['description'], res['id']))
exit_on_rc_error("Please set a unique value for ",
select_dict['description'])
res = res_list[0]['id']
log("Found for %s" % (select_dict['description']), res_list[0])
return res
except IndexError:
exit_on_rc_error("Unknown value", value)
def load_workout_data(workload_id):
"""
Requests the workload for the given ID. Precondition: SSO established
Returns HTML tree
"""
log("load_workload_data", "start")
# just for for testing
# return html.parse(open('workout.html', 'rt')).getroot()
url = "https://app.velohero.com/workouts/edit/{}".format(workload_id)
log("url", str(url))
r = requests.post(url,
headers={
'user-agent': my_user_agent,
},
data={
'sso': sso_id,
})
if r.status_code != 200:
exit_on_rc_error("HTTP error {}. Status code".format(url),
r.status_code)
return html.fromstring(r.text)
def get_simple_input(tree, name):
"""
Returns dictionary with id, value, description
"""
# log('get_simple_input', str(name))
try:
label = tree.xpath("//label[@for='%s']" % name)[0].text
# label inside formatting element, e.g. <span>
if label is None:
label = tree.xpath("//label[@for='%s']" % name)[0][0].text
# log('label', label)
element = tree.xpath("//input[@id='%s']" % name)[0]
# log('element', element)
ret = dict(
id=element.name,
value=element.value,
description=label,
)
# log("get_simple_input ret=", ret)
return ret
except IndexError:
exit_on_rc_error("Element not found", name)
def load_equipments():
"""
Requests the settings for the given user. Precondition: SSO established
Returns HTML tree
"""
log("load_equipments", "start")
# just for for testing
# return html.parse(open('velo_hero_equipments.html', 'rt')).getroot()
url = "https://app.velohero.com/equipment/list"
log("url", str(url))
r = requests.post(url,
headers={
'user-agent': my_user_agent,
},
data={
'sso': sso_id,
})
if r.status_code != 200:
exit_on_rc_error("HTTP error {}. Status code".format(url),
r.status_code)
return html.fromstring(r.text)
def get_select_values(tree, name):
"""
Returns dictionary with id, value, description, for instance
dict=(id='sport_id',
description='Sportart',
values=[(id=1, description='Radsport', data.py-subtext=None, selectecd=True),
id=6, description='Mountainbike', data.py-subtext=None, selected=False),
...
]),
"""
ret = None
try:
element = tree.xpath("//select[@id='%s']" % name)[0]
label_elements = tree.xpath("//label[@for='%s']" % name)
# Proper HTML: label has a for attribute
if len(label_elements) > 0:
label = label_elements[0].text
# Just for 'Material' / equipment_ids:
else:
label = element.getparent().getparent()[0].text
ret = dict(id=element.name, description=label, values=[])
# exclude empty item
for option in [o for o in element if len(o.get('value')) > 0]:
if option.get('selected'):
# log('selected', "True")
selected = True
else:
# log('selected', "False")
selected = False
if option.get('data.py-subtext'):
data_subtext = option.get('data.py-subtext')
else:
data_subtext = None
ret['values'].append(
dict(id=option.get('value'),
description=option.text,
data_subtext=data_subtext,
selected=selected))
except IndexError as e:
exit_on_rc_error("Element '%s' error" % name, e)
return ret
def read_tcx(file_name):
"""
Read TCX file grab some data.py
:param file_name: TCX file name
:return: instance of a Load
"""
if not path.exists(file_name):
exit_on_rc_error("File not found", file_name)
load = Stage(file_name)
tcxparser = TCXParser(file_name)
load.init_by_tcx(tcxparser)
return load
def velohero_do_update(workout_id, args, load):
"""
Update existing workout
:param workout_id Workout ID to update
:param args Args from the vh_update command or, if load set, None.
:param load Load with attributes to update, or if args set, None.
"""
log("do_set", "start")
velohero_check_sso_login()
tree = load_workout_data(workout_id)
# Init all field with old data.py
workout_date = get_simple_input(tree, 'workout_date')['value']
workout_start_time = get_simple_input(tree, 'workout_start_time')['value']
workout_dist_km = get_simple_input(tree, 'workout_dist_km')['value']
workout_asc_m = get_simple_input(tree, 'workout_asc_m')['value']
workout_dur_time = get_simple_input(tree, 'workout_dur_time')['value']
sport_id = get_select_values_value(tree, 'sport_id')
type_id = get_select_values_value(tree, 'type_id')
route_id = get_select_values_value(tree, 'route_id')
workout_dsc_m = get_simple_input(tree, 'workout_dsc_m')['value']
workout_alt_min_m = get_simple_input(tree, 'workout_alt_min_m')['value']
workout_alt_max_m = get_simple_input(tree, 'workout_alt_max_m')['value']
workout_spd_avg_kph = get_simple_input(tree,
'workout_spd_avg_kph')['value']
workout_spd_max_kph = get_simple_input(tree,
'workout_spd_max_kph')['value']
workout_hr_avg_bpm = get_simple_input(tree, 'workout_hr_avg_bpm')['value']
workout_hr_max_bpm = get_simple_input(tree, 'workout_hr_max_bpm')['value']
workout_comment = get_text_area(tree, 'workout_comment')['value']
equipment_ids = get_select_values_values(tree, 'equipment_ids')
log('Actual data.py', ":")
log('workout_date', workout_date)
log('workout_start_time', workout_start_time)
log('workout_dist_km', workout_dist_km)
log('workout_asc_m', workout_asc_m)
log('workout_dur_time', workout_dur_time)
log('sport_id', sport_id)
log('type_id', type_id)
log('route_id', route_id)
log('workout_dsc_m', workout_dsc_m)
log('workout_alt_min_m', workout_alt_min_m)
log('workout_alt_max_m', workout_alt_max_m)
log('workout_spd_avg_kph', workout_spd_avg_kph)
log('workout_spd_max_kph', workout_spd_max_kph)
log('workout_hr_avg_bpm', workout_hr_avg_bpm)
log('workout_hr_max_bpm', workout_hr_max_bpm)
log('equipment_ids', equipment_ids)
log('workout_comment', workout_comment)
# Options
sport_id_dict = get_select_values(tree, 'sport_id')
type_id_dict = get_select_values(tree, 'type_id')
route_id_dict = get_select_values(tree, 'route_id')
equipment_ids_dict = get_select_values(tree, 'equipment_ids')
log('sport_id_dict', sport_id_dict)
log('type_id_dict', type_id_dict)
log('route_id_dict', route_id_dict)
log('equipment_ids_dict', equipment_ids_dict)
# Update fields by vh-update
if args is not None:
if args.workout_date:
workout_date = args.workout_date
if args.workout_start_time:
workout_start_time = args.workout_start_time
if args.workout_dist_km:
workout_dist_km = args.workout_dist_km
if args.workout_asc_m:
workout_asc_m = args.workout_asc_m
if args.workout_dur_time:
workout_dur_time = args.workout_dur_time
if args.sport_id:
sport_id = get_selected_id(sport_id_dict, args.sport_id)
if args.type_id:
type_id = get_selected_id(type_id_dict, args.type_id)
if args.route_id:
route_id = get_selected_id(route_id_dict, args.route_id)
if args.workout_dsc_m:
workout_dsc_m = args.workout_dsc_m
if args.workout_alt_min_m:
workout_alt_min_m = args.workout_alt_min_m
if args.workout_alt_max_m:
workout_alt_max_m = args.workout_alt_max_m
if args.workout_spd_avg_kph:
workout_spd_avg_kph = args.workout_spd_avg_kph
if args.workout_spd_max_kph:
workout_spd_max_kph = args.workout_spd_max_kph
if args.workout_hr_avg_bpm:
workout_hr_avg_bpm = args.workout_hr_avg_bpm
if args.workout_hr_max_bpm:
workout_hr_max_bpm = args.workout_hr_max_bpm
if args.equipment_ids:
equipment_ids = get_selected_ids(equipment_ids_dict,
values=args.equipment_ids)
if args.workout_comment:
workout_comment = args.workout_comment
# Update fields by command transfer
elif load is not None:
if load.new_activity_type:
new_sport_id = map_sport_id(sport_id_dict, load.new_activity_type)
if new_sport_id:
sport_id = new_sport_id
# log('load.training_type', load.training_type)
if load.training_type is not None:
if load.training_type == '':
type_id = '0'
else:
type_id = get_selected_id(type_id_dict, load.training_type)
# Check explicitly for 'not None', otherwise an empty list will be ignored
if load.equipment_names is not None:
equipment_ids = get_selected_ids(equipment_ids_dict,
value_list=load.equipment_names)
# Check explicitly for 'not None', otherwise an empty list will be ignored
if load.route_name is not None:
route_id = get_selected_id(route_id_dict, load.route_name)
# Map description to comment
if load.title is not None:
workout_comment = load.title
log('New data.py', ":")
log('workout_date', workout_date)
log('workout_start_time', workout_start_time)
log('workout_dist_km', workout_dist_km)
log('workout_asc_m', workout_asc_m)
log('workout_dur_time', workout_dur_time)
log('sport_id', sport_id)
log('type_id', type_id)
log('route_id', route_id)
log('workout_dsc_m', workout_dsc_m)
log('workout_alt_min_m', workout_alt_min_m)
log('workout_alt_max_m', workout_alt_max_m)
log('workout_spd_avg_kph', workout_spd_avg_kph)
log('workout_spd_max_kph', workout_spd_max_kph)
log('workout_hr_avg_bpm', workout_hr_avg_bpm)
log('workout_hr_max_bpm', workout_hr_max_bpm)
log('equipment_ids', equipment_ids)
log('workout_comment', workout_comment)
url = "https://app.velohero.com/workouts/edit/{}".format(workout_id)
log("url", str(url))
# Must contains all workout fields
r = requests.post(
url,
headers={
'user-agent': my_user_agent,
},
data={
'sso': sso_id,
'submit': '1',
'workout_date': workout_date,
'workout_start_time': workout_start_time,
'workout_dist_km': workout_dist_km,
'workout_asc_m': workout_asc_m,
'workout_dur_time': workout_dur_time,
'sport_id': sport_id,
'type_id': type_id,
'route_id': route_id,
'workout_dsc_m': workout_dsc_m,
'workout_alt_min_m': workout_alt_min_m,
'workout_alt_max_m': workout_alt_max_m,
'workout_spd_avg_kph': workout_spd_avg_kph,
'workout_spd_max_kph': workout_spd_max_kph,
'workout_hr_avg_bpm': workout_hr_avg_bpm,
'workout_hr_max_bpm': workout_hr_max_bpm,
'workout_comment': workout_comment,
'equipment_ids': equipment_ids,
# 'equipment_ids': ['17481', '3793'],
})
if r.status_code != 200:
exit_on_rc_error("HTTP error for {}".format(url), r.status_code)
# log("response", str(r.text))
log("do_set", "done")