From: Kern Sibbald <kern@sibbald.com>
Date: Mon, 29 Nov 2004 18:16:04 +0000 (+0000)
Subject: - Add python restart command in Console.
X-Git-Tag: Release-7.0.0~9061
X-Git-Url: https://git.sur5r.net/?a=commitdiff_plain;h=eb563660471b32eeffc5db53502bcc7d9e76cdd9;p=bacula%2Fbacula

- Add python restart command in Console.
- Make built-in variables table driven.


git-svn-id: https://bacula.svn.sourceforge.net/svnroot/bacula/trunk@1730 91ce42f0-d328-0410-95d8-f526ca767f89
---

diff --git a/bacula/src/dird/python.c b/bacula/src/dird/python.c
index 1c8488059e..bda706e664 100644
--- a/bacula/src/dird/python.c
+++ b/bacula/src/dird/python.c
@@ -49,41 +49,75 @@ PyMethodDef BaculaMethods[] = {
 };
 
 
+struct s_vars {
+   const char *name;
+   char *fmt;
+};     
+
+static struct s_vars vars[] = {
+   { N_("Job"),        "s"},
+   { N_("Dir"),        "s"},
+   { N_("Level"),      "s"},
+   { N_("Type"),       "s"},
+   { N_("JobId"),      "i"},
+   { N_("Client"),     "s"},
+   { N_("NumVols"),    "i"},
+   { N_("Pool"),       "s"},
+   { N_("Storage"),    "s"},
+   { N_("Catalog"),    "s"},
+   { N_("MediaType"),  "s"},
+   { N_("JobName"),    "s"},
+   
+   { NULL,	       NULL}
+};
+
 /* Return Bacula variables */
 PyObject *bacula_get(PyObject *self, PyObject *args)
 {
    PyObject *CObject;
    JCR *jcr;
    char *item;
+   bool found = false;
+   int i;
+
    if (!PyArg_ParseTuple(args, "Os:get", &CObject, &item)) {
       return NULL;
    }
    jcr = (JCR *)PyCObject_AsVoidPtr(CObject);
-   /* ***FIXME*** put this in a table */
-   if (strcmp(item, "JobId") == 0) {
-      return Py_BuildValue("i", jcr->JobId);
-   } else if (strcmp(item, "Client") == 0) {
-      return Py_BuildValue("s", jcr->client->hdr.name);
-   } else if (strcmp(item, "Pool") == 0) {
-      return Py_BuildValue("s", jcr->pool->hdr.name);
-   } else if (strcmp(item, "Storage") == 0) {
-      return Py_BuildValue("s", jcr->store->hdr.name);
-   } else if (strcmp(item, "Catalog") == 0) {
-      return Py_BuildValue("s", jcr->catalog->hdr.name);
-   } else if (strcmp(item, "MediaType") == 0) {
-      return Py_BuildValue("s", jcr->store->media_type);
-   } else if (strcmp(item, "NumVols") == 0) {
-      return Py_BuildValue("i", jcr->NumVols);
-   } else if (strcmp(item, "DirName") == 0) {
-      return Py_BuildValue("s", my_name);
-   } else if (strcmp(item, "Level") == 0) {
-      return Py_BuildValue("s", job_level_to_str(jcr->JobLevel));
-   } else if (strcmp(item, "Type") == 0) {
-      return Py_BuildValue("s", job_type_to_str(jcr->JobType));
-   } else if (strcmp(item, "Job") == 0) {
-      return Py_BuildValue("s", jcr->job->hdr.name);
-   } else if (strcmp(item, "JobName") == 0) {
-      return Py_BuildValue("s", jcr->Job);
+   for (i=0; vars[i].name; i++) {
+      if (strcmp(vars[i].name, item) == 0) {
+	 found = true;
+	 break;
+      }
+   }
+   if (!found) {
+      return NULL;
+   }
+   switch (i) {
+   case 0:			      /* Job */
+      return Py_BuildValue(vars[i].fmt, jcr->job->hdr.name);
+   case 1:                            /* Director's name */
+      return Py_BuildValue(vars[i].fmt, my_name);
+   case 2:			      /* level */
+      return Py_BuildValue(vars[i].fmt, job_level_to_str(jcr->JobLevel));
+   case 3:			      /* type */
+      return Py_BuildValue(vars[i].fmt, job_type_to_str(jcr->JobType));
+   case 4:			      /* JobId */
+      return Py_BuildValue(vars[i].fmt, jcr->JobId);
+   case 5:			      /* Client */
+      return Py_BuildValue(vars[i].fmt, jcr->client->hdr.name);
+   case 6:			      /* NumVols */
+      return Py_BuildValue(vars[i].fmt, jcr->NumVols);
+   case 7:			      /* Pool */
+      return Py_BuildValue(vars[i].fmt, jcr->pool->hdr.name);
+   case 8:			      /* Storage */
+      return Py_BuildValue(vars[i].fmt, jcr->store->hdr.name);
+   case 9:
+      return Py_BuildValue(vars[i].fmt, jcr->catalog->hdr.name);
+   case 10:			      /* MediaType */
+      return Py_BuildValue(vars[i].fmt, jcr->store->media_type);
+   case 11:			      /* JobName */
+      return Py_BuildValue(vars[i].fmt, jcr->Job);
    }
    return NULL;
 }
@@ -101,6 +135,7 @@ PyObject *bacula_set(PyObject *self, PyObject *args, PyObject *keyw)
       return NULL;
    }
    jcr = (JCR *)PyCObject_AsVoidPtr(CObject);
+
    if (msg) {
       Jmsg(jcr, M_INFO, 0, "%s", msg);
    }
diff --git a/bacula/src/dird/ua_cmds.c b/bacula/src/dird/ua_cmds.c
index 2090b43a99..57f99e5ac2 100644
--- a/bacula/src/dird/ua_cmds.c
+++ b/bacula/src/dird/ua_cmds.c
@@ -87,6 +87,7 @@ static int release_cmd(UAContext *ua, const char *cmd);
 static int update_cmd(UAContext *ua, const char *cmd);
 static int wait_cmd(UAContext *ua, const char *cmd);
 static int setip_cmd(UAContext *ua, const char *cmd);
+static int python_cmd(UAContext *ua, const char *cmd);
 static void do_job_delete(UAContext *ua, JobId_t JobId);
 static void delete_job_id_range(UAContext *ua, char *tok);
 
@@ -113,6 +114,7 @@ static struct cmdstruct commands[] = {
  { N_("mount"),      mount_cmd,     _("mount <storage-name>")},
  { N_("prune"),      prunecmd,      _("prune expired records from catalog")},
  { N_("purge"),      purgecmd,      _("purge records from catalog")},
+ { N_("python"),     python_cmd,    _("python control commands")},
  { N_("quit"),       quit_cmd,      _("quit")},
  { N_("query"),      querycmd,      _("query catalog")},
  { N_("restore"),    restore_cmd,   _("restore files")},
@@ -559,6 +561,26 @@ static int create_cmd(UAContext *ua, const char *cmd)
 }
 
 
+extern DIRRES *director;
+
+/*
+ * Python control command
+ *  python restart (restarts interpreter)
+ */
+static int python_cmd(UAContext *ua, const char *cmd)
+{
+   if (strcasecmp(ua->argk[1], _("restart")) == 0) {
+      term_python_interpreter();
+      init_python_interpreter(director->hdr.name, director->scripts_directory ?
+         director->scripts_directory : ".");
+      bsendmsg(ua, _("Python interpreter restarted.\n"));
+   } else {
+      bsendmsg(ua, _("Nothing done.\n"));
+   }
+   return 1;
+}
+
+
 /*
  * Set a new address in a Client resource. We do this only
  *  if the Console name is the same as the Client name