def get_sub_wheel_size()

in api/choice_algorithm.py [0:0]

• 8 lines of code
• 6 McCabe index (conditional complexity)

def get_sub_wheel_size(wheel_name):
    resp = Wheel.query(
              IndexName='name_index',
              KeyConditionExpression=Key('name').eq(wheel_name)
           )
    if len(resp['Items']):  # if a matching wheel is found
        return int(resp['Items'][0]['participant_count']) or 1  # if wheel size is 0, default to 1
    return 1 # default to 1 if no matching wheel is found