diff options
-rwxr-xr-x | exmachina.py | 9 | ||||
-rwxr-xr-x | test.py | 4 |
2 files changed, 13 insertions, 0 deletions
diff --git a/exmachina.py b/exmachina.py index a72ac55..229a465 100755 --- a/exmachina.py +++ b/exmachina.py @@ -141,6 +141,9 @@ class ExMachinaClient(): self.initd.start = self.conn.call.initd_start self.initd.stop = self.conn.call.initd_stop self.initd.restart = self.conn.call.initd_restart + + def close(self): + self.sock.close() def run_server(socket_path="/tmp/exmachina.sock"): # TODO: check for root permissions, warn if not root @@ -174,6 +177,10 @@ def main(): default=False, help="Show more debugging statements", action="store_true") + parser.add_option("-q", "--quiet", + default=False, + help="Show fewer informational statements", + action="store_true") (options, args) = parser.parse_args() @@ -188,6 +195,8 @@ def main(): if options.verbose: log.setLevel(logging.DEBUG) + elif options.quiet: + log.setLevel(logging.ERROR) else: log.setLevel(logging.INFO) @@ -30,11 +30,15 @@ def main(): print c.call.initd_status("bluetooth") print "hostname: %s" % c.call.augeas_get("/files/etc/hostname/*") print "localhost: %s" % c.call.augeas_get("/files/etc/hosts/1/canonical") + sock.close() print "========= Testing user client library" client = ExMachinaClient() print client.augeas.match("/files/etc/*") print client.initd.restart("bluetooth") + print client.initd.status("bluetooth") + print client.initd.status("greentooth") + client.close() if __name__ == '__main__': main() |