#!/usr/bin/python3


import os
import shutil
import subprocess
import sys

def get_vmlinuz_path():
    return os.path.join('mnt', os.readlink('mnt/vmlinuz').lstrip('/'))

def get_initrd_path():
    return os.path.join('mnt', os.readlink('mnt/initrd.img').lstrip('/'))

if not os.path.exists('mnt'):
    os.mkdir('mnt')

subprocess.run([
    'guestmount',
    '-a', 'unstable-armhf.qcow2',
    '-m', '/dev/sda1',
    '--ro',
    'mnt'])
print(get_vmlinuz_path())
print(get_initrd_path())

shutil.copyfile(get_vmlinuz_path(), 'vmlinuz', follow_symlinks=True)
shutil.copyfile(get_initrd_path(), 'initrd.img', follow_symlinks=True)

subprocess.run(['fusermount', '-u', 'mnt'])