# coding=utf-8

from python.util import has_token_sequence, string_almost_equal, \
    string_contains_number, get_tokens, get_numbers, get_exception_desc
from server.hints import Hint, HintSequence

id = 213
group = 'introduction'
number = 4
visible = True

solution = '''\
from ev3dev import *
from mindstorms_widgets import mindstorms_widgets

robot = mindstorms_widgets()
robot.connect_motor( 'left' )
robot.connect_motor( 'right' )

turntime = .95
robot.move_steering( 'on_for_seconds', direction=0, power=20, seconds=2 )
robot.move_tank( 'on_for_seconds', lr_power=[20,-20], seconds=turntime )
robot.move_steering( 'on_for_seconds', direction=0, power=20, seconds=2 )
robot.move_tank( 'on_for_seconds', lr_power=[20,-20], seconds=turntime )
robot.move_steering( 'on_for_seconds', direction=0, power=20, seconds=2 )
robot.move_tank( 'on_for_seconds', lr_power=[20,-20], seconds=turntime )
robot.move_steering( 'on_for_seconds', direction=0, power=20, seconds=2 )
robot.move_tank( 'on_for_seconds', lr_power=[20,-20], seconds=turntime )
'''

hint_type = {
    'mW_init': Hint('mW_init'),
    'connectMotorLeft': Hint('connectMotorLeft'),
    'connectMotorRight': Hint('connectMotorRight'),
    'moveSteering': Hint('moveSteering'),
    'onForSeconds': Hint('onForSeconds'),
    'direction': Hint('direction'),
    'seconds': Hint('seconds'),
    'moveTank': Hint('moveTank'),
    'lrPower': Hint('lrPower')
}

def hint( code):
    tokens = get_tokens(code)

    # if code does not include mindstorms_widgets(), a student gets a hint that the robot should be somehow represented in the program
    if not has_token_sequence(tokens, ['mindstorms_widgets', '(',')']):
        return [{'id': 'mW_init'}]

    # if code does not include connect_motor statement, a student needs to learn about how to connect the motors
    if not has_token_sequence(tokens, ['connect_motor']) and not has_token_sequence(tokens, ['left']):
        return [{'id': 'connectMotorLeft'}]

    # if code does not include connect_motor statement, a student needs to learn about how to connect the motors
    if not has_token_sequence(tokens, ['connect_motor']) and not has_token_sequence(tokens, ['right']):
        return [{'id': 'connectMotorRight'}]

    if not has_token_sequence(tokens, ['move_steering']):
        return [{'id': 'moveSteering'}]

    if not 'on_for_seconds' in code:
        return [{'id': 'onForSeconds'}]
    
    if not 'direction' in code:
        return [{'id': 'direction'}]
    
    if not 'seconds' in code:
        return [{'id': 'seconds'}]
    
    if not has_token_sequence(tokens, ['move_tank']):
        return [{'id': 'moveTank'}]

    if not 'lr_power' in code:
        return [{'id': 'lrPower'}]

    return None