gst-uninstalled.py

   1 #!/usr/bin/env python3
   2
   3 import os
   4 import argparse
   5 import subprocess
   6
   7
   8 SCRIPTDIR = os.path.dirname(__file__)
   9
  10
  11 def prepend_env_var(env, var, value):
  12     env[var] = os.pathsep + value + os.pathsep + env.get(var, "")
  13     env[var] = env[var].replace(os.pathsep + os.pathsep, os.pathsep).strip(os.pathsep)
  14
  15 def set_prompt_var(options, env):
  16     ps1 = env.get("PS1")
  17     if ps1:
  18         env["PS1"] = "[gst-%s] %s" % (options.gst_version, ps1)
  19
  20     prompt = env.get("PROMPT")
  21     if prompt:
  22         env["PROMPT"] = "[gst-%s] %s" % (options.gst_version, prompt)
  23
  24
  25 def get_subprocess_env(options):
  26     env = os.environ.copy()
  27
  28     PATH = env.get("PATH", "")
  29     subprojects_path = os.path.join(options.builddir, "subprojects")
  30     for proj in os.listdir(subprojects_path):
  31         projpath = os.path.join(subprojects_path, proj)
  32         if not os.path.exists(projpath):
  33             print("Subproject %s does not exist in %s.,\n"
  34                   " Make sure to build everything properly "
  35                   "and try again." % (proj, projpath))
  36             exit(1)
  37
  38         envvars_file = os.path.join(projpath, os.path.basename(projpath) + "-uninstalled-envvars.py")
  39         if os.path.exists(envvars_file):
  40             envvars_env = {"envvars": {}}
  41             with open(envvars_file) as f:
  42                 code = compile(f.read(), envvars_file, 'exec')
  43                 exec(code, None, envvars_env)
  44             for var, value in envvars_env["envvars"].items():
  45                 if var.startswith("+"):
  46                     prepend_env_var(env, var, value.strip("+"))
  47                 else:
  48                     env[var] = value
  49
  50         toolsdir = os.path.join(projpath, "tools")
  51         if os.path.exists(toolsdir):
  52             prepend_env_var(env, "PATH", toolsdir)
  53
  54         prepend_env_var(env, "GST_PLUGIN_PATH", projpath)
  55
  56     env["GST_VALIDATE_SCENARIOS_PATH"] = os.path.normpath(
  57         "%s/subprojects/gst-devtools/validate/data/scenarios" % SCRIPTDIR)
  58     env["GST_VALIDATE_PLUGIN_PATH"] = os.path.normpath(
  59         "%s/subprojects/gst-devtools/validate/plugins" % options.builddir)
  60     prepend_env_var(env, "PATH", os.path.normpath(
  61         "%s/subprojects/gst-devtools/validate/tools" % options.builddir))
  62     env["PATH"] += os.pathsep + PATH
  63     env["GST_VERSION"] = options.gst_version
  64     env["GST_PLUGIN_SYSTEM_PATH"] = ""
  65     env["GST_PLUGIN_SCANNER"] = os.path.normpath(
  66         "%s/subprojects/gstreamer/libs/gst/helpers/gst-plugin-scanner" % options.builddir)
  67     env["GST_PTP_HELPER"] = os.path.normpath(
  68         "%s/subprojects/gstreamer/libs/gst/helpers/gst-ptp-helper" % options.builddir)
  69     env["GST_REGISTRY"] = os.path.normpath(options.builddir + "/registry.dat")
  70
  71     set_prompt_var(options, env)
  72
  73     return env
  74
  75
  76 if __name__ == "__main__":
  77     parser = argparse.ArgumentParser(prog="gstreamer-uninstalled")
  78
  79     parser.add_argument("--builddir",
  80                         default=os.path.join(SCRIPTDIR, "build"),
  81                         help="The meson build directory")
  82     parser.add_argument("--gst-version", default="master",
  83                         help="The meson build directory")
  84     options = parser.parse_args()
  85
  86     if not os.path.exists(options.builddir):
  87         print("GStreamer not built in %s\n\nBuild it and try again" %
  88               options.builddir)
  89         exit(1)
  90
  91     shell_args = [os.environ.get("SHELL", os.path.realpath("/bin/sh"))]
  92     if shell_args[0] == "/bin/bash":
  93         shell_args.append("--noprofile")
  94
  95     try:
  96         exit(subprocess.run(shell_args, env=get_subprocess_env(options)).returncode)
  97     except subprocess.CalledProcessError as e:
  98         exit(e.returncode)