# 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

id = 215
group = 'introduction'
number = 6
visible = True

solution = '''\
robot = mindstorms_widgets()
robot.connect_motor( 'left' )
robot.connect_motor( 'right' )
robot.connect_sensor( 'ultrasonic' )
robot.move_steering( 'on', power=80 )
while robot.ultrasonic_sensor_measure( 'distance-cm' ) > 500:
	pass
robot.move_steering( 'on', power=20 )
while robot.ultrasonic_sensor_measure( 'distance-cm' ) > 200:
	pass
robot.move_steering( 'off' )
'''

hint_type = {
    'mW_init': Hint('mW_init'),
    'connectMotorLeft': Hint('connectMotorLeft'),
    'connectMotorRight': Hint('connectMotorRight'),
    'moveSteeringOn': Hint('moveSteeringOn'),
    'moveSteeringOff': Hint('moveSteeringOff'),
    'connectUltrasonicSensor': Hint('connectUltrasonicSensor'),
    'ultrasonicSensorMeasure': Hint('ultrasonicSensorMeasure'),
    'power80': Hint('power80'),
    'power20': Hint('power20'),
    'while': Hint('while')
}

def hint( code ):
    tokens = get_tokens(code)
    lines = code.split('\n')

    # 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, ['connect_sensor', '(']) and 'ultrasonic' in code) :
        return [{'id': 'connectUltrasonicSensor'}]

    if not (has_token_sequence(tokens, ['move_steering', '(']) and 'on' in code):
        return [{'id': 'moveSteeringOn'}]

    if not (has_token_sequence(tokens, ['move_steering', '(']) and 'off' in code):
        return [{'id': 'moveSteeringOff'}]

    #if not (has_token_sequence(tokens, ['ultrasonic_sensor_measure', '(']) and 'distance-cm' in code):
    if not any(('ultrasonic_sensor_measure' in s and '(' in s and 'distance-cm' in s) for s in lines):
        return [{'id': 'ultrasonicSensorMeasure'}]

    if not any(('power' in s and '80' in s and '=' in s) for s in lines):
        return [{'id': 'power80'}]
    
    if not any(('power' in s and '20' in s and '=' in s) for s in lines):
        return [{'id': 'power20'}]
    
    if not has_token_sequence(tokens, ['while']):
        return [{'id': 'while'}]

    return None