easyengine/ee/core/fileutils.py

"""EasyEngine file utils core classes."""
import shutil
import os
import sys
import glob
import shutil
import fileinput
from ee.core.logging import Log


class EEFileUtils():
    """Method to operate on files"""
    def __init__():
        pass

    def remove(self, filelist):
        for file in filelist:
            if os.path.isfile(file):
                self.app.log.info("Removing "+os.path.basename(file)+" ...")
                os.remove(file)
                self.app.log.debug('file Removed')
                self.app.log.info("Done")
            if os.path.isdir(file):
                try:
                    print("Removing "+os.path.basename(file)+"...")
                    shutil.rmtree(file)
                    self.app.log.info("Done")
                except shutil.Error as e:
                    Log.error(self, 'Unable to Remove file {err}'
                              .format(err=str(e.reason)))
                    sys.exit(1)

    def create_symlink(self, paths, errormsg=''):
        src = paths[0]
        dst = paths[1]
        if not os.path.islink(dst):
            try:
                os.symlink(src, dst)
            except Exception as e:
                self.app.log.error("Unable to create symbolic link ...\n {0}"
                                   " {1}".format(e.errno, e.strerror))
                sys.exit(1)
        else:
            self.app.log.debug("Destination: {0} exists".format(dst))

    def remove_symlink(self, filepath):
        try:
            os.unlink(filepath)
        except Exception as e:
            Log.error(self, "Unable to reomove symbolic link ...\n {0} {1}"
                      .format(e.errno, e.strerror))
            sys.exit(1)

    def copyfile(self, src, dest):
        try:
            shutil.copy2(src, dest)
        except shutil.Error as e:
            print('Error: {0}'.format(e))
        except IOError as e:
            print('Error: {e}'.format(e.strerror))

    def searchreplace(self, fnm, sstr, rstr):
        try:
            for line in fileinput.input(fnm, inplace=True):
                print(line.replace(sstr, rstr), end='')
            fileinput.close()
        except Exception as e:
            print('Error : {0}'.format(e))

    def mvfile(self, src, dst):
        try:
            shutil.move(src, dst)
        except shutil.Error as e:
            Log.error(self, 'Unable to move file {err}'
                      .format(err=str(e.reason)))
            sys.exit(1)

    def chdir(self, path):
        try:
            os.chdir(path)
        except OSError as e:
            Log.error(self, 'Unable to Change Directory {err}'
                      .format(err=e.strerror))
            sys.exit(1)

    def chown(self, path, user, group, recursive=False):
        try:
            if recursive:
                for root, dirs, files in os.walk(path):
                    for d in dirs:
                        shutil.chown(os.path.join(root, d), user=user,
                                     group=group)
                    for f in files:
                        shutil.chown(os.path.join(root, f), user=user,
                                     group=group)
            else:
                shutil.chown(path, user=user, group=group)
        except shutil.Error as e:
            Log.error(self, "Unable to change owner : {0} ".format(e))
            sys.exit(1)
        except Exception as e:
            Log.error(self, "Unable to change owner {0}".format(e))
            sys.exit(1)

    def chmod(self, path, perm, recursive=False):
        try:
            if recursive:
                for root, dirs, files in os.walk(path):
                    for d in dirs:
                        os.chmod(os.path.join(root, d), perm)
                    for f in files:
                        os.chmod(os.path.join(root, f), perm)
            else:
                os.chmod(path, perm)
        except OSError as e:
            Log.error(self, "Unable to change owner {0}".format(e.strerror))
            sys.exit(1)

    def mkdir(self, path):
        try:
            os.makedirs(path)
        except OSError as e:
            Log.error(self, "Unable to create directory {0}"
                      .format(e.strerror))
            sys.exit(1)

    def isexist(self, path):
        try:
            if os.path.exists(path):
                return (True)
            else:
                return (False)
        except OSError as e:
            Log.error(self, "Unable to check path {0}"
                      .format(e.strerror))
            sys.exit(1)

    def grep(self, fnm, sstr):
        try:
            for line in open(fnm):
                if sstr in line:
                    return line
        except OSError as e:
            Log.error(self, "Unable to Search string {0}"
                      .format(e.strerror))
            sys.exit(1)