summaryrefslogtreecommitdiff
path: root/src/util.py
blob: 146d45038bbd18641ccea9234017ee6e3b76bdd7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python

import subprocess
import atexit
import os
import signal
import dbus

def spawnbus(config = None):
    command = [ "dbus-daemon", "--session", "--nofork", "--print-address" ]
    if config:
        command.append("--config-file=config")

    process = subprocess.Popen (command,
        stdout = subprocess.PIPE,
    )
    atexit.register (lambda p: os.kill (p.pid, signal.SIGKILL), process)

    address = process.stdout.readline().rstrip()
    os.environ ["DBUS_SESSION_BUS_ADDRESS"] = address
    print "Temporary Session bus: %s" % address


def override_env (variable, var):
    os.environ[variable] = var
    print "Setting %s to %s" % (variable, var)

def prepend_env_path (variable, prefix):
    current = os.getenv(variable)
    if current != None:
        p = prefix + ":" + current
    else:
        p = prefix
    override_env (variable, p)

def setup_data_dir (path):
    # Define default if not set as per XDG spec
    if os.getenv("XDG_DATA_DIRS") == None:
        os.environ["XDG_DATA_DIRS"] = " /usr/local/share/:/usr/share/"
    prepend_env_path ("XDG_DATA_DIRS", path)

def setup_run_dir (path):
    # Setup all the various XDG paths
    override_env ("XDG_CONFIG_HOME", os.path.join (path, ".config"))
    override_env ("XDG_CACHE_HOME", os.path.join (path, ".cache"))
    override_env ("XDG_DATA_HOME", os.path.join (path, ".local"))
    override_env ("MC_ACCOUNT_DIR",
        os.path.join (path, ".config", "mission-control", "accounts"))