3 # sends purge volume action=all to bacula using bconsole.
4 # list of storages is read from stdin
6 # Copyright 2013,2017 Peter Palfrader
8 # Permission is hereby granted, free of charge, to any person obtaining
9 # a copy of this software and associated documentation files (the
10 # "Software"), to deal in the Software without restriction, including
11 # without limitation the rights to use, copy, modify, merge, publish,
12 # distribute, sublicense, and/or sell copies of the Software, and to
13 # permit persons to whom the Software is furnished to do so, subject to
14 # the following conditions:
16 # The above copyright notice and this permission notice shall be
17 # included in all copies or substantial portions of the Software.
19 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
20 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
21 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
22 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
23 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
24 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
25 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
31 import psycopg2.extras
33 parser = argparse.ArgumentParser()
34 parser.add_argument('-d', '--db-connect-string', metavar='connect-string', dest='db',
35 help='Database connect string')
36 parser.add_argument('-D', '--db-connect-string-file', metavar='FILE', dest='dbfile',
37 default='/etc/dsa/bacula-reader-database',
38 help='File to read database connect string from (/etc/dsa/bacula-reader-database)')
39 parser.add_argument("-v", "--verbose", dest="verbose",
40 default=False, action="store_true",
41 help="Be more verbose.")
42 parser.add_argument("-n", "--nodo", dest="nodo",
43 default=False, action="store_true",
44 help="Print to cat rather than bconsole.")
45 args = parser.parse_args()
47 if args.db is not None:
49 elif args.dbfile is not None:
50 args.db = open(args.dbfile).read().rstrip()
52 print >>sys.stderr, 'Need one of -d or -D.'
56 conn = psycopg2.connect(args.db)
57 cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
58 cursor.execute("""SELECT DISTINCT mediatype FROM media""", {})
59 storages = set(r['mediatype'] for r in cursor.fetchall())
63 c = 'truncate allpools storage=%s'%(s,)
71 print("Will run: %s"%(c,))
73 p = subprocess.Popen(['/usr/sbin/bconsole'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
74 (out, err) = p.communicate(c.encode())
76 raise Exception("bconsole failed. stdout:\n%s\nstderr:%s\n"%(out, err))
80 sys.stdout.buffer.write(out)
84 print("bconsole said on stderr:\n", file=sys.stderr)
85 sys.stderr.buffer.write(out)
86 print("", file=sys.stderr)