From 4d7351ef070176283d7626ae2501a4ad7acc093e Mon Sep 17 00:00:00 2001
From: Petr Viktorin <pviktori@redhat.com>
Date: Thu, 15 May 2014 15:42:48 +0200
Subject: ipalib.cli: Add filename argument to ipa console

This allows writing simple IPA scripts using the shebang
    #! /usr/bin/ipa console

https://fedorahosted.org/freeipa/ticket/4351

Reviewed-By: Nathaniel McCallum <npmccallum@redhat.com>
---
 ipalib/cli.py | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

(limited to 'ipalib/cli.py')

diff --git a/ipalib/cli.py b/ipalib/cli.py
index f03db9c61..ea47c7bb0 100644
--- a/ipalib/cli.py
+++ b/ipalib/cli.py
@@ -30,6 +30,7 @@ import fcntl
 import termios
 import struct
 import base64
+import traceback
 try:
     #pylint: disable=F0401
     import default_encoding_utf8
@@ -875,15 +876,33 @@ class show_mappings(frontend.Command):
 
 
 class console(frontend.Command):
-    """Start the IPA interactive Python console."""
+    """Start the IPA interactive Python console, or run a script.
 
+    An IPA API object is initialized and made available
+    in the `api` global variable.
+    """
+
+    takes_args = ('filename?',)
     has_output = tuple()
 
-    def run(self, **options):
-        code.interact(
-            '(Custom IPA interactive Python console)',
-            local=dict(api=self.api)
-        )
+    def run(self, filename=None, **options):
+        local = dict(api=self.api)
+        if filename:
+            try:
+                script = open(filename)
+            except IOError as e:
+                exit("%s: %s" % (e.filename, e.strerror))
+            try:
+                with script:
+                    exec(script, globals(), local)
+            except:
+                traceback.print_exc()
+                exit(1)
+        else:
+            code.interact(
+                '(Custom IPA interactive Python console)',
+                local=local
+            )
 
 
 class show_api(frontend.Command):
-- 
cgit