from invoke import task

cmd = '''
HOME=/root
export HOME
eval `keychain --quiet --eval id_ecdsa`
'''

@task
def init_workspace(ctx):
    ctx.run(cmd + 'pdm run -p .. setup_pi.py --init', pty=True)

@task
def setup_pi(ctx, hostname, serial):
    ctx.run(cmd + f'pdm run -p .. setup_pi.py {hostname} {serial}', pty=True)