python/samba/mdb_util.py

   1 # Unix SMB/CIFS implementation.
   2 # mdb util helpers
   3 #
   4 # Copyright (C) Andrew Bartlett <abartlet@samba.org> 2018
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 import samba
  20 import subprocess
  21 import os
  22 from samba.netcmd import CommandError
  23
  24
  25 def mdb_copy(file1, file2):
  26     """Copy mdb file using mdb_copy utility and rename it
  27     """
  28     # Find the location of the mdb_copy tool
  29     dirs = os.getenv('PATH').split(os.pathsep)
  30     found = False
  31     for d in dirs:
  32         toolpath = os.path.join(d, "mdb_copy")
  33         if os.path.exists(toolpath):
  34             found = True
  35             break
  36
  37     if not found:
  38         raise CommandError("mdb_copy not found. "
  39                            "You may need to install the lmdb-utils package")
  40
  41     mdb_copy_cmd = [toolpath, "-n", file1, "%s.copy.mdb" % file1]
  42     status = subprocess.check_call(mdb_copy_cmd, close_fds=True, shell=False)
  43
  44     os.rename("%s.copy.mdb" % file1, file2)