File: nssdb.py

package info (click to toggle)
pglistener 4
  • links: PTS
  • area: main
  • in suites: buster, stretch
  • size: 224 kB
  • ctags: 132
  • sloc: python: 594; perl: 364; sh: 120; sql: 89; makefile: 2
file content (40 lines) | stat: -rw-r--r-- 1,466 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# Postgres Listener, writes out flatfiles when SQL NOTIFYs are triggered
# Copyright (C) 2005 Rob Bradford <rob@robster.org.uk>
# Copyright (C) 2007-8 Robert McQueen <robert.mcqueen@bluelinux.co.uk>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# 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.

from pglistener import PgListener

import bsddb.db as db
import os

class NssDb(PgListener):
  def do_write(self,result,target):
    tmp = "%s.tmp" % (target, )
    _db = db.DB()
    _db.open(tmp, None, db.DB_BTREE, db.DB_CREATE|db.DB_TRUNCATE, mode=0644)
    db_keys = self.options['db-keys']

    for index, row in enumerate(result):
      entry = self.do_format(row) + '\0'

      for field in db_keys.keys():
        _db.put(db_keys[field] % row[field], entry)
      _db.put("0%d" % index, entry)

    _db.sync()
    _db.close()
    os.rename(tmp, target)