AHK extenstion example with state changes

my_extension.py

from ahk import AHK
from ahk.extensions import Extension

extension_script = '''\
variables := {}

set_variable(varname, value) {
    global variables
    variables[(varname)] := value
    return FormatNoValueResponse()
}

get_variable(varname) {
    global variables
    val := variables[(varname)]
    return FormatResponse("ahk.message.StringResponseMessage", val)
}

'''

extension = Extension(script_text=extension_script)

@extension.register
def get_variable(ahk: AHK, varname: str) -> str:
    return ahk.function_call('get_variable', [varname])

@extension.register
def set_variable(ahk: AHK, varname: str, value: str) -> None:
    ahk.function_call('set_variable', [varname, value])

ahk = AHK(extensions=[extension], version='v1')

ahk.set_variable(varname='foo', value='bar')

print(ahk.get_variable('foo'))  # 'bar'