# 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 = 207
group = 'introduction'
number = 12
visible = True

solution = '''\
import time

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

start = time.time()
while time.time()-start < 10:
    if robot.color_sensor_measure('reflected_light_intensity') < 50:
        L, R = 0, 30
    else:
        L, R = 30, 0
    robot.move_tank( 'on', lr_power=[L,R])
robot.move_tank( 'off' )
'''

hint_type = {
    'mW_init': Hint('mW_init'),
    'connectMotorLeft': Hint('connectMotorLeft'),
    'connectMotorRight': Hint('connectMotorRight'),
    'moveTankOn': Hint('moveTankOn'),
    'lrPower': Hint('lr_power'),
    'moveTankOff': Hint('moveTankOff'),
    'connectColorSensor': Hint('connectColorSensor'),
    'colorSensorMeasureRLI': Hint('colorSensorMeasureRLI'),
    'while': Hint('while'),
    'time': Hint('time'),
    'if': Hint('if')
}

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

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

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

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

    if not (has_token_sequence(tokens, ['color_sensor_measure', '(']) and 'reflected_light_intensity' in code):
        return [{'id': 'colorSensorMeasureRLI'}]
    
    if not has_token_sequence(tokens, ['while']):
        return [{'id': 'while'}]

    if not has_token_sequence(tokens, ['time']):
        return [{'id': 'time'}]
    
    if not has_token_sequence(tokens, ['if']):
        return [{'id': 'if'}]
    
    return None