+/*
+ Bacula® - The Network Backup Solution
+
+ Copyright (C) 2004-2008 Free Software Foundation Europe e.V.
+
+ The main author of Bacula is Kern Sibbald, with contributions from
+ many others, a complete list can be found in the file AUTHORS.
+ This program is Free Software; you can redistribute it and/or
+ modify it under the terms of version two of the GNU General Public
+ License as published by the Free Software Foundation and included
+ in the file LICENSE.
+
+ This program is distributed in the hope that it will be useful, but
+ WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ 02110-1301, USA.
+
+ Bacula® is a registered trademark of Kern Sibbald.
+ The licensor of Bacula is the Free Software Foundation Europe
+ (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
+ Switzerland, email:ftf@fsfeurope.org.
+*/
/*
*
* Bacula interface to Python for the Director
* Version $Id$
*
*/
-/*
- Copyright (C) 2004-2005 Kern Sibbald
-
- This program is free software; you can redistribute it and/or
- modify it under the terms of the GNU General Public License
- version 2 as ammended with additional clauses defined in the
- file LICENSE in the main source directory.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- the file LICENSE for additional details.
-
- */
#include "bacula.h"
#include "dird.h"
#undef _POSIX_C_SOURCE
#include <Python.h>
-extern char *configfile;
-extern JCR *get_jcr_from_PyObject(PyObject *self);
-extern PyObject *find_method(PyObject *eventsObject, PyObject *method,
- const char *name);
+#include <lib/pythonlib.h>
+extern struct s_jl joblevels[];
static PyObject *set_job_events(PyObject *self, PyObject *arg);
static PyObject *job_run(PyObject *self, PyObject *arg);
static PyObject *job_write(PyObject *self, PyObject *arg);
static PyObject *job_cancel(PyObject *self, PyObject *arg);
+static PyObject *job_does_vol_exist(PyObject *self, PyObject *arg);
PyMethodDef JobMethods[] = {
{"set_events", set_job_events, METH_VARARGS, "Set Job events"},
{"run", job_run, METH_VARARGS, "Run a Job"},
{"write", job_write, METH_VARARGS, "Write to output"},
{"cancel", job_cancel, METH_VARARGS, "Cancel a Job"},
+ {"DoesVolumeExist", job_does_vol_exist, METH_VARARGS, "Does Volume Exist"},
{NULL, NULL, 0, NULL} /* last item */
};
-
struct s_vars {
const char *name;
- char *fmt;
+ const char *fmt;
};
/* Read-only variables */
static struct s_vars getvars[] = {
- { N_("Job"), "s"},
- { N_("DirName"), "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"},
- { N_("JobStatus"), "s"},
- { N_("Priority"), "i"},
- { N_("Version"), "(ss)"},
- { N_("ConfigFile"), "s"},
- { N_("WorkingDir"), "s"},
- { N_("CatalogRes"), "(sssssi)"},
+ { "Job", "s"},
+ { "Level", "s"},
+ { "Type", "s"},
+ { "JobId", "i"},
+ { "Client", "s"},
+ { "NumVols", "i"},
+ { "Pool", "s"},
+ { "Storage", "s"},
+ { "Catalog", "s"},
+ { "MediaType", "s"},
+ { "JobName", "s"},
+ { "JobStatus", "s"},
+ { "Priority", "i"},
+ { "VolumeName", "s"},
+ { "CatalogRes", "(sssssis)"},
+ { "JobErrors", "i"},
+ { "JobFiles", "i"},
+ { "SDJobFiles", "i"},
+ { "SDErrors", "i"},
+ { "FDJobStatus","s"},
+ { "SDJobStatus","s"},
{ NULL, NULL}
};
/* Writable variables */
static struct s_vars setvars[] = {
- { N_("JobReport"), "s"},
- { N_("VolumeName"), "s"},
- { N_("Priority"), "i"},
+ { "JobReport", "s"},
+ { "VolumeName", "s"},
+ { "Priority", "i"},
+ { "JobLevel", "s"},
{ NULL, NULL}
};
Dmsg0(100, "In job_getattr.\n");
jcr = get_jcr_from_PyObject(self);
if (!jcr) {
- bstrncpy(errmsg, "Job pointer not found.", sizeof(errmsg));
+ bstrncpy(errmsg, _("Job pointer not found."), sizeof(errmsg));
goto bail_out;
}
for (i=0; getvars[i].name; i++) {
}
switch (i) {
case 0: /* Job */
- return Py_BuildValue(getvars[i].fmt, jcr->job->hdr.name);
- case 1: /* Director's name */
- return Py_BuildValue(getvars[i].fmt, my_name);
- case 2: /* level */
- return Py_BuildValue(getvars[i].fmt, job_level_to_str(jcr->JobLevel));
- case 3: /* type */
- return Py_BuildValue(getvars[i].fmt, job_type_to_str(jcr->JobType));
- case 4: /* JobId */
- return Py_BuildValue(getvars[i].fmt, jcr->JobId);
- case 5: /* Client */
- return Py_BuildValue(getvars[i].fmt, jcr->client->hdr.name);
- case 6: /* NumVols */
- return Py_BuildValue(getvars[i].fmt, jcr->NumVols);
- case 7: /* Pool */
- return Py_BuildValue(getvars[i].fmt, jcr->pool->hdr.name);
- case 8: /* Storage */
- return Py_BuildValue(getvars[i].fmt, jcr->store->hdr.name);
- case 9:
- return Py_BuildValue(getvars[i].fmt, jcr->catalog->hdr.name);
- case 10: /* MediaType */
- return Py_BuildValue(getvars[i].fmt, jcr->store->media_type);
- case 11: /* JobName */
- return Py_BuildValue(getvars[i].fmt, jcr->Job);
- case 12: /* JobStatus */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->job->hdr.name);
+ case 1: /* level */
+ return Py_BuildValue((char *)getvars[i].fmt, job_level_to_str(jcr->get_JobLevel()));
+ case 2: /* type */
+ return Py_BuildValue((char *)getvars[i].fmt, job_type_to_str(jcr->get_JobType()));
+ case 3: /* JobId */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->JobId);
+ case 4: /* Client */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->client->hdr.name);
+ case 5: /* NumVols */
+ POOL_DBR pr;
+ memset(&pr, 0, sizeof(pr));
+ bstrncpy(pr.Name, jcr->pool->hdr.name, sizeof(pr.Name));
+ if (db_get_pool_record(jcr, jcr->db, &pr)) {
+ jcr->NumVols = pr.NumVols;
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->NumVols);
+ } else {
+ bsnprintf(errmsg, sizeof(errmsg), _("Pool record not found."));
+ goto bail_out;
+ }
+ case 6: /* Pool */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->pool->name());
+ case 7: /* Storage */
+ if (jcr->wstore) {
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->wstore->name());
+ } else if (jcr->rstore) {
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->rstore->name());
+ } else {
+ goto bail_out;
+ }
+ case 8:
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->catalog->name());
+ case 9: /* MediaType */
+ if (jcr->wstore) {
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->wstore->media_type);
+ } else if (jcr->rstore) {
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->rstore->media_type);
+ } else {
+ goto bail_out;
+ }
+ case 10: /* JobName */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->Job);
+ case 11: /* JobStatus */
buf[1] = 0;
buf[0] = jcr->JobStatus;
- return Py_BuildValue(getvars[i].fmt, buf);
- case 13: /* Priority */
- return Py_BuildValue(getvars[i].fmt, jcr->JobPriority);
- case 14: /* Version */
- return Py_BuildValue(getvars[i].fmt, VERSION, BDATE);
- case 15: /* Config Dir */
- return Py_BuildValue(getvars[i].fmt, configfile);
- case 16: /* Working Dir */
- return Py_BuildValue(getvars[i].fmt, director->working_directory);
- case 17: /* CatalogRes */
- return Py_BuildValue(getvars[i].fmt,
+ return Py_BuildValue((char *)getvars[i].fmt, buf);
+ case 12: /* Priority */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->JobPriority);
+ case 13:
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->VolumeName);
+ case 14: /* CatalogRes */
+ return Py_BuildValue((char *)getvars[i].fmt,
jcr->catalog->db_name, jcr->catalog->db_address,
jcr->catalog->db_user, jcr->catalog->db_password,
- jcr->catalog->db_socket, jcr->catalog->db_port);
-
+ jcr->catalog->db_socket, jcr->catalog->db_port,
+ db_get_type());
+ case 15: /* JobErrors */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->JobErrors);
+ case 16: /* JobFiles */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->JobFiles);
+ case 17: /* SDJobFiles */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->SDJobFiles);
+ case 18: /* SDErrors */
+ return Py_BuildValue((char *)getvars[i].fmt, jcr->SDErrors);
+ case 19: /* FDJobStatus */
+ buf[1] = 0;
+ buf[0] = jcr->FDJobStatus;
+ return Py_BuildValue((char *)getvars[i].fmt, buf);
+ case 29: /* SDJobStatus */
+ buf[1] = 0;
+ buf[0] = jcr->SDJobStatus;
+ return Py_BuildValue((char *)getvars[i].fmt, buf);
}
- bsnprintf(errmsg, sizeof(errmsg), "Attribute %s not found.", attrname);
+ bsnprintf(errmsg, sizeof(errmsg), _("Attribute %s not found."), attrname);
bail_out:
PyErr_SetString(PyExc_AttributeError, errmsg);
return NULL;
if (setvars[i].fmt != NULL) {
switch (setvars[i].fmt[0]) {
case 's':
- if (!PyArg_Parse(value, setvars[i].fmt, &strval)) {
- PyErr_SetString(PyExc_TypeError, "Read-only attribute");
+ if (!PyArg_Parse(value, (char *)setvars[i].fmt, &strval)) {
+ PyErr_SetString(PyExc_TypeError, _("Read-only attribute"));
return -1;
}
break;
case 'i':
- if (!PyArg_Parse(value, setvars[i].fmt, &intval)) {
- PyErr_SetString(PyExc_TypeError, "Read-only attribute");
+ if (!PyArg_Parse(value, (char *)setvars[i].fmt, &intval)) {
+ PyErr_SetString(PyExc_TypeError, _("Read-only attribute"));
return -1;
}
break;
break;
case 2: /* Priority */
Dmsg1(000, "Set priority=%d\n", intval);
- return 0;
+ if (intval >= 1 && intval <= 100) {
+ jcr->JobPriority = intval;
+ } else {
+ PyErr_SetString(PyExc_ValueError, _("Priority must be 1-100"));
+ return -1;
+ }
+ case 3: /* Job Level */
+ if (strcmp("JobInit", jcr->event) != 0) {
+ PyErr_SetString(PyExc_RuntimeError, _("Job Level can be set only during JobInit"));
+ return -1;
+ }
+ for (i=0; joblevels[i].level_name; i++) {
+ if (strcmp(strval, joblevels[i].level_name) == 0) {
+ if (joblevels[i].job_type == jcr->get_JobType()) {
+ jcr->set_JobLevel(joblevels[i].level);
+ jcr->jr.JobLevel = jcr->get_JobLevel();
+ return 0;
+ }
+ }
+ }
+ PyErr_SetString(PyExc_ValueError, _("Bad JobLevel string"));
+ return -1;
}
bail_out:
PyErr_SetString(PyExc_AttributeError, attrname);
return -1;
}
-
+/*
+ * Set pointer to instantiated events class
+ */
static PyObject *set_job_events(PyObject *self, PyObject *arg)
{
PyObject *eObject;
return Py_None;
}
-/* Run a Bacula command */
+/* Run a Bacula job */
static PyObject *job_run(PyObject *self, PyObject *arg)
{
JCR *jcr;
return NULL;
}
/* Release lock due to recursion */
- PyEval_ReleaseLock();
+// PyEval_ReleaseLock();
jcr = get_jcr_from_PyObject(self);
UAContext *ua = new_ua_context(jcr);
ua->batch = true;
parse_ua_args(ua); /* parse command */
stat = run_cmd(ua, ua->cmd);
free_ua_context(ua);
- PyEval_AcquireLock();
+// PyEval_AcquireLock();
return PyInt_FromLong((long)stat);
}
return NULL;
}
if (text) {
- Jmsg(NULL, M_INFO, 0, "%s", text);
+ JCR *jcr = get_jcr_from_PyObject(self);
+ Jmsg(jcr, M_INFO, 0, "%s", text);
+ }
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+static PyObject *job_does_vol_exist(PyObject *self, PyObject *args)
+{
+ char *VolName = NULL;
+
+ if (!PyArg_ParseTuple(args, "s:does_volume_exist", &VolName)) {
+ Dmsg0(000, "Parse tuple error in job_does_vol_exist\n");
+ return NULL;
+ }
+ if (VolName) {
+ MEDIA_DBR mr;
+ int ok;
+ JCR *jcr = get_jcr_from_PyObject(self);
+ memset(&mr, 0, sizeof(mr));
+ bstrncpy(mr.VolumeName, VolName, sizeof(mr.VolumeName));
+ ok = db_get_media_record(jcr, jcr->db, &mr);
+ return Py_BuildValue("i", ok);
}
Py_INCREF(Py_None);
return Py_None;
}
+
static PyObject *job_cancel(PyObject *self, PyObject *args)
{
JobId_t JobId = 0;
}
foreach_jcr(jcr) {
if (jcr->JobId == 0) {
- free_jcr(jcr);
continue;
}
if (jcr->JobId == JobId) {
break;
}
}
+ /* endeach_jcr(jcr) not needed because freed below */
+
if (!found) {
/* ***FIXME*** raise exception */
return NULL;
}
- PyEval_ReleaseLock();
+// PyEval_ReleaseLock();
UAContext *ua = new_ua_context(jcr);
ua->batch = true;
if (!cancel_job(ua, jcr)) {
}
free_ua_context(ua);
free_jcr(jcr);
- PyEval_AcquireLock();
+// PyEval_AcquireLock();
Py_INCREF(Py_None);
return Py_None;
}
return 0;
}
- PyEval_AcquireLock();
+ lock_python();
+// PyEval_AcquireLock();
method = find_method(events, method, event);
if (!method) {
}
bstrncpy(jcr->event, event, sizeof(jcr->event));
- result = PyObject_CallFunction(method, "O", Job);
+ result = PyObject_CallFunction(method, (char *)"O", Job);
jcr->event[0] = 0; /* no event in progress */
if (result == NULL) {
if (PyErr_Occurred()) {
Py_XDECREF(result);
bail_out:
- PyEval_ReleaseLock();
+ unlock_python();
+// PyEval_ReleaseLock();
return stat;
}