ipsilon/tools/files.py

   1 # Copyright (C) 2014  Simo Sorce <simo@redhat.com>
   2 #
   3 # see file 'COPYING' for use and warranty information
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 import os
  19 import pwd
  20 from string import Template
  21
  22
  23 def fix_user_dirs(path, user=None, mode=0700):
  24     pw = None
  25     if user:
  26         pw = pwd.getpwnam(user)
  27     for t in os.walk(path, topdown=False):
  28         root, files = t[0], t[2]
  29         for name in files:
  30             target = os.path.join(root, name)
  31             if pw:
  32                 os.chown(target, pw.pw_uid, pw.pw_gid)
  33             os.chmod(target, mode & 0666)
  34         if pw:
  35             os.chown(root, pw.pw_uid, pw.pw_gid)
  36         os.chmod(root, mode)
  37
  38
  39 def write_from_template(destfile, template, opts):
  40     with open(template) as f:
  41         t = Template(f.read())
  42     text = t.substitute(**opts)
  43     with open(destfile, 'w+') as f:
  44         f.write(text)