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 subprocess
  20 import os
  21 from samba.netcmd import CommandError
  22
  23
  24 def mdb_copy(file1, file2):
  25     """Copy mdb file using mdb_copy utility and rename it
  26     """
  27     # Find the location of the mdb_copy tool
  28     dirs = os.getenv('PATH').split(os.pathsep)
  29     found = False
  30     for d in dirs:
  31         toolpath = os.path.join(d, "mdb_copy")
  32         if os.path.exists(toolpath):
  33             found = True
  34             break
  35
  36     if not found:
  37         raise CommandError("mdb_copy not found. "
  38                            "You may need to install the lmdb-utils package")
  39
  40     mdb_copy_cmd = [toolpath, "-n", file1, "%s.copy.mdb" % file1]
  41     status = subprocess.check_call(mdb_copy_cmd, close_fds=True, shell=False)
  42
  43     os.rename("%s.copy.mdb" % file1, file2)