aboutsummaryrefslogtreecommitdiffstats
path: root/libbe
diff options
context:
space:
mode:
Diffstat (limited to 'libbe')
-rw-r--r--libbe/arch.py17
-rw-r--r--libbe/config.py44
2 files changed, 58 insertions, 3 deletions
diff --git a/libbe/arch.py b/libbe/arch.py
index 6cc3dbb..b1b88fd 100644
--- a/libbe/arch.py
+++ b/libbe/arch.py
@@ -1,15 +1,26 @@
from popen2 import Popen4
import os
+import config
+client = config.get_val("arch_client")
+if client is None:
+ client = "tla"
+ config.set_val("arch_client", client)
def invoke(args):
q=Popen4(args)
output = q.fromchild.read()
status = q.wait()
if os.WIFEXITED(status):
- return (os.WEXITSTATUS(status))
+ return os.WEXITSTATUS(status)
+ raise Exception("Command failed")
+
+def invoke_client(*args, **kwargs):
+ status = invoke((client,) + args)
+ if status not in (0,):
+ raise Exception("Command failed")
def add_id(filename):
- return invoke(("tla", "add-id", filename))
+ invoke_client("add-id", filename)
def delete_id(filename):
- return invoke(("tla", "delete-id", filename))
+ invoke_client("delete-id", filename)
diff --git a/libbe/config.py b/libbe/config.py
new file mode 100644
index 0000000..bc05e9a
--- /dev/null
+++ b/libbe/config.py
@@ -0,0 +1,44 @@
+import ConfigParser
+import os.path
+def path():
+ """Return the path to the per-user config file"""
+ return os.path.expanduser("~/.bugs_everywhere")
+
+def set_val(name, value, section="DEFAULT"):
+ """Set a value in the per-user config file
+
+ :param name: The name of the value to set
+ :param value: The new value to set (or None to delete the value)
+ :param section: The section to store the name/value in
+ """
+ config = ConfigParser.ConfigParser()
+ config.read(path())
+ if value is not None:
+ config.set(section, name, value)
+ else:
+ config.remove_option(section, name)
+ config.write(file(path(), "wb"))
+ pass
+
+def get_val(name, section="DEFAULT"):
+ """
+ Get a value from the per-user config file
+
+ :param name: The name of the value to get
+ :section: The section that the name is in
+ :return: The value, or None
+ >>> get_val("junk") is None
+ True
+ >>> set_val("junk", "random")
+ >>> get_val("junk")
+ 'random'
+ >>> set_val("junk", None)
+ >>> get_val("junk") is None
+ True
+ """
+ config = ConfigParser.ConfigParser()
+ config.read(path())
+ try:
+ return config.get(section, name)
+ except ConfigParser.NoOptionError:
+ return None