近来尝试使用python来写poc,每次在创建一个python文件时,头部信息总是需要自己去手动添加,添加了几次之后,发现这项工作太过于枯燥,于是想写一个脚本来自动生成这些信息。思路是:使用shell脚本来创建.py文件,将该shell脚本设置为系统命令,然后在该linux虚拟机中,任何一个目录下通过调用改脚本加上文件名即可成功创建,具体操作如下:
1、在/usr/local/sbin/目录下创建vimpy.sh文件,shell脚本如下:
:<<!
作者:张乐
创建时间:2018.5.28 9:20
用途:新建python脚本并且初始化开头结尾
!
#shell脚本中$1代表程序后的第一个参数,$0表示该脚本的文件名,脚本是否输入要创建的文件名
if [ ! "$1" ]
then
echo '请输入要新建的文件名称'
exit 1
fi
#如果该文件已经存在,则直接用vim打开,不存在则创建文件
if [ ! -f "$1" ]; then
touch "$1"
else
vim "$1"
fi
#写入注释
echo "#-*- coding:utf-8 -*-">>"$1"
echo "">>"$1"
echo "'''">>"$1"
echo "作者:张乐">>"$1"
time=$(date "+%Y-%m-%d %H:%M:%S")
#echo ${time}
echo "创建时间&