de0e662f51
Fix up 2to3 complaints. I've decided to do with wrapping list comprehension inside list() to get the generators to run in several places instead of list(map( or list(filter( which seem less readable to me.
109 lines
2.7 KiB
Python
109 lines
2.7 KiB
Python
#
|
|
# sysutils.py
|
|
#
|
|
# Copyright (C) 2009-2015 Red Hat, Inc.
|
|
#
|
|
# 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, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# Red Hat Author(s): Martin Gracik <mgracik@redhat.com>
|
|
#
|
|
|
|
__all__ = ["joinpaths", "touch", "replace", "chown_", "chmod_", "remove",
|
|
"linktree"]
|
|
|
|
import sys
|
|
import os
|
|
import re
|
|
import fileinput
|
|
import pwd
|
|
import grp
|
|
import glob
|
|
import shutil
|
|
|
|
from pylorax.executils import runcmd
|
|
|
|
def joinpaths(*args, **kwargs):
|
|
path = os.path.sep.join(args)
|
|
|
|
if kwargs.get("follow_symlinks"):
|
|
return os.path.realpath(path)
|
|
else:
|
|
return path
|
|
|
|
|
|
def touch(fname):
|
|
# python closes the file when it goes out of scope
|
|
open(fname, "w").write("")
|
|
|
|
|
|
def replace(fname, find, sub):
|
|
fin = fileinput.input(fname, inplace=1)
|
|
pattern = re.compile(find)
|
|
|
|
for line in fin:
|
|
line = pattern.sub(sub, line)
|
|
sys.stdout.write(line)
|
|
|
|
fin.close()
|
|
|
|
|
|
def chown_(path, user=None, group=None, recursive=False):
|
|
uid = gid = -1
|
|
|
|
if user is not None:
|
|
uid = pwd.getpwnam(user)[2]
|
|
if group is not None:
|
|
gid = grp.getgrnam(group)[2]
|
|
|
|
for fname in glob.iglob(path):
|
|
os.chown(fname, uid, gid)
|
|
|
|
if recursive and os.path.isdir(fname):
|
|
for nested in os.listdir(fname):
|
|
nested = joinpaths(fname, nested)
|
|
chown_(nested, user, group, recursive)
|
|
|
|
|
|
def chmod_(path, mode, recursive=False):
|
|
for fname in glob.iglob(path):
|
|
os.chmod(fname, mode)
|
|
|
|
if recursive and os.path.isdir(fname):
|
|
for nested in os.listdir(fname):
|
|
nested = joinpaths(fname, nested)
|
|
chmod_(nested, mode, recursive)
|
|
|
|
|
|
def cpfile(src, dst):
|
|
shutil.copy2(src, dst)
|
|
if os.path.isdir(dst):
|
|
dst = joinpaths(dst, os.path.basename(src))
|
|
|
|
return dst
|
|
|
|
def mvfile(src, dst):
|
|
if os.path.isdir(dst):
|
|
dst = joinpaths(dst, os.path.basename(src))
|
|
os.rename(src, dst)
|
|
return dst
|
|
|
|
def remove(target):
|
|
if os.path.isdir(target) and not os.path.islink(target):
|
|
shutil.rmtree(target)
|
|
else:
|
|
os.unlink(target)
|
|
|
|
def linktree(src, dst):
|
|
runcmd(["/bin/cp", "-alx", src, dst])
|