#!/usr/bin/env python

# Tool for verifying sanity of units in Profile.xls / fitparse/profile.py

import os

import sys

import xlrd # Dev requirement for parsing Excel spreadsheet

from fitparse.profile import MESSAGE_TYPES

from fitparse.utils import scrub_method_name

def do_profile_xls():

workbook = xlrd.open_workbook(sys.argv[1])

sheet = workbook.sheet_by_name('Messages')

all_unit_values = []

for unit_value in sheet.col_values(8): # Extract unit column values

unit_value = unit_value.strip()

if unit_value:

# Deal with comma separated components

unit_values = [v.strip() for v in unit_value.split(',')]

all_unit_values.extend(unit_values)

print('In Profile.xls:')

for unit_value in sorted(set(all_unit_values)):

print(' * %s' % unit_value)

def do_fitparse_profile():

unit_values = []

for message_type in MESSAGE_TYPES.values():

for field in message_type.fields.values():

unit_values.append(field.units)

if field.components:

for component in field.components:

unit_values.append(component.units)

if field.subfields:

for subfield in field.subfields:

unit_values.append(subfield.units)

if subfield.components:

for component in subfield.components:

unit_values.append(component.units)

unit_values = filter(None, unit_values)

print('In fitparse/profile.py:')

for unit_value in sorted(set(unit_values)):

print(' * {} [{}]'.format(

unit_value,

scrub_method_name('process_units_%s' % unit_value, convert_units=True)

))

if __name__ == '__main__':

if len(sys.argv) < 2:

print("Usage: {0} Profile.xls".format(os.path.basename(__file__)))

sys.exit(0)

do_profile_xls()

print()

do_fitparse_profile()