shell编程实战之rm命令太危险，使用自定义命令del替换

by Posted on

不知道大家对在服务器上使用rm命令有什么感觉，反正每次我使用的时候，都会好好确认，即使这样，按下回车的时候，心里还是会有点慌慌的感觉。因为和windows系统不一样，使用win系统默认删除文件，该文件就会被放入到回收站中，想要恢复直接从回收站找到文件并恢复即可。但是在linux中，用rm删除文件，那就是真的删除掉了，是没有回收站这个概念的。

真因为rm命令的危险性，所以想要不自己写一个shell脚本，用来删除文件。自己定义的这个命令删除文件时，会将原文件的位置移动“回收站”目录，然后再将原文件给删除掉。如果误删了，就可以从回收站中找到该文件并恢复。

首先，第一步，我们需要先创建回收站目录，如果已经有了此目录从不用重新创建

# 回收站目录
DEL_BAK_DIR="/root/.delbak"

if [ ! -d "$DEL_BAK_DIR" ];then
    mkdir "$DEL_BAK_DIR" || exit 1
fi

然后我们来获取用户输入的参数，循环遍历有所参数，调用删除函数来删除。当然，我们首先需要检查用户是否输入了参数，如果用户没有输入参数那么就直接退出程序，不往下走了。

# 获取参数
if (( $# < 1 ));then
    echo "Usage 
# 获取参数
if (( $# < 1 ));then
echo "Usage ${0} file1 file2 ..."
exit 2
fi
# 删除文件
for file in $*
do
delete $file
done
 file1 file2 ..."
    exit 2
fi

# 删除文件
for file in $*
do
    delete $file
done

最后，我们需要编写delete这个函数了。这个函数应该有哪些功能呢？首先，它肯定能对待删除的文件名做检查，如果不是文件或目录，它就不继续往下走了。另外，每次将删除文件放到回收站目录前，应该对回收站目录所在分区剩余空间做检查，如果空间不足也不往下走。

下面是文件是否存在的检查

if [ ! -e $1 ];then
    echo "file $1 is not exists"
    exit 3
fi

而关于分区空间的检查则稍微复杂些。首先，我们要获取到回收站目录所在分区剩余空间，然后获取到待删除文件的大小，拿这两个数据做对比。

# 分区剩余空间是否充足校验
part_free=$(df -k $DEL_BAK_DIR | grep -iv filesystem | awk '{print $4}')
file_size=$(du -ks $1 | awk '{print $1}')    

if (( part_free > file_size))
then
    # 备份且删除源文件
    ……
else
    echo "The disk size is not enough"
    exit 4
fi

到这步，所有的准备工作都做好了，就差最后的备份源文件，然后删除源文件了。下面给出完整的代码：

#!/bin/bash
# 使用自定义命令del来代替rm命令

export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:~/bin
export LANG=en

# 回收站目录
DEL_BAK_DIR="/root/.delbak"

function delete ()
{
    # 文件名是否存在校验
    if [ ! -e $1 ];then
        echo "file $1 is not exists"
        exit 3
    fi

    # 分区剩余空间是否充足校验
    part_free=$(df -k $DEL_BAK_DIR | grep -iv filesystem | awk '{print $4}')
    file_size=$(du -ks $1 | awk '{print $1}')    

    if (( part_free > file_size))
    then
        # 备份且删除源文件
        now=$(date +%Y%m%d%H%M)

        if [ ! -d "$DEL_BAK_DIR/.$now" ];then
            mkdir "$DEL_BAK_DIR/.$now" || exit 5
        fi

        mv -f $1 $DEL_BAK_DIR/.$now/$(basename $1)
    else
        echo "The disk size is not enough"
        exit 4
    fi
}

if [ ! -d "$DEL_BAK_DIR" ];then
    mkdir "$DEL_BAK_DIR" || exit 1
fi

# 获取参数
if (( $# < 1 ));then
    echo "Usage 
#!/bin/bash
# 使用自定义命令del来代替rm命令
export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:~/bin
export LANG=en
# 回收站目录
DEL_BAK_DIR="/root/.delbak"
function delete ()
{
# 文件名是否存在校验
if [ ! -e $1 ];then
echo "file $1 is not exists"
exit 3
fi
# 分区剩余空间是否充足校验
part_free=$(df -k $DEL_BAK_DIR | grep -iv filesystem | awk '{print $4}')
file_size=$(du -ks $1 | awk '{print $1}')    
if (( part_free > file_size))
then
# 备份且删除源文件
now=$(date +%Y%m%d%H%M)
if [ ! -d "$DEL_BAK_DIR/.$now" ];then
mkdir "$DEL_BAK_DIR/.$now" || exit 5
fi
mv -f $1 $DEL_BAK_DIR/.$now/$(basename $1)
else
echo "The disk size is not enough"
exit 4
fi
}
if [ ! -d "$DEL_BAK_DIR" ];then
mkdir "$DEL_BAK_DIR" || exit 1
fi
# 获取参数
if (( $# < 1 ));then
echo "Usage ${0} file1 file2 ..."
exit 2
fi
# 删除文件
for file in $*
do
delete $file
done
 file1 file2 ..."
    exit 2
fi

# 删除文件
for file in $*
do
    delete $file
done

关于该脚本本人已经测试过，大家可以拿去放心用。该脚本支持通配符哦，比如想删除所有以.html结尾的文件，只需输入del *.html即可。

Published by 1024@phper

热爱编程，热爱生活