blob: d07e9810dbfc1c5dc17749dd7f8ffd185a0a9098 [file] [log] [blame]
# -*- mode: python -*-
# Find virtme's resources
# Copyright © 2014-2019 Andy Lutomirski
# Licensed under the GPLv2, which is available in the virtme distribution
# as a file called LICENSE with SHA-256 hash:
# 8177f97513213526df2cf6184d8ff986c675afb514d4e68a404010521b880643
"""Helpers to find virtme's guest tools and host scripts."""
import os
import shutil
import pkg_resources
import subprocess
def find_guest_tools():
"""Return the path of the guest tools installed with the running virtme.
if pkg_resources.resource_isdir(__name__, 'guest'):
return pkg_resources.resource_filename(__name__, 'guest')
# No luck. This is somewhat surprising.
return None
def find_script(name) -> str:
# If we're running out of a source checkout, we can find scripts through
# the 'virtme/scripts' symlink.
fn = pkg_resources.resource_filename(__name__, 'scripts/%s' % name)
if os.path.isfile(fn):
return fn
# Otherwise assume we're actually installed and in PATH.
guess = shutil.which(name)
if guess is not None:
return guess
# No luck. This is somewhat surprising.
raise Exception('could not find script %s' % name)
def run_script(name, **kwargs) -> None:
fn = find_script(name)
subprocess.check_call(executable=fn, args=[fn], **kwargs)