工作中使用python编写自动化脚本,提前两小时下班不是梦
大家好,十点多才下班不好意思,虽然基本没人给我留言说喜欢学习啥,但是通过我最近发文的阅读量来看,大家对shell脚本那期貌似感兴趣的很多。于是,我觉得大家可能会更喜欢使用python脚本做一些平常的自动化操作。(我都学会分析读者心理了,如果猜对了大家记得关注点赞啊,最重要的事说三遍,关注关注+关注啊)。
于是就有了这期,因为其实shell脚本存在很多难以调试,无法应对大规模的自动化运维工作,语法陈旧,模块化程度低等问题。
目前基本上所有的大公司都主推的是使用python 做工程自动化编译,自动化部署等工作了,甚至就直接写到公司的编码规范中了,shell脚本往后会使用的越来越少,而模块化程度高,相对而言,功能更加强大的python 是目前编写自动化脚本的不二语言。
这期的话,主要想给大家介绍下,使用Python 做自动化脚本编写时,常见的一些基础命令,最后,我们会附上一个简单的示例,一个小脚本,来归拢下我们今天所学。
废话不多说,我们开始吧。
首先呢,使用python做自动化脚本编写,那我们经常可能遇到的需求或者是可以自动化的基础动作有哪些呢?无外乎是文件的增删改查和文件夹的增删改查操作,还有一些类似获取文件夹内部所有信息,筛选符合条件的文件等这类操作会比较多。
于是我们今天就先介绍这些操作的基本命令和使用效果。
我们先从最简单的开始:
os.getcwd() // 此命令的作用是获取执行python文件时的当前路径
我先新建一个test.py的python脚本,内容如下:
import os
print(os.getcwd())
然后我执行如下操作后结果:
这里大家要分清楚,getcwd是获取当前执行脚本时,所在的路径,而并不是指test.py这个源文件保存的路径。
然后介绍的第二个命令是:
os.listdir(dir) // 入参dir是个文件夹路径的字符串
上面命令的作用是,获取到所有dir文件夹下的文件和子文件夹的名称,以list的字符串的形式返回。(如果大家不熟悉python基本数据结构和语法,需要我讲的话,留言啊,随时开讲)
我们在test.py脚本中写下如下的代码:
import os
# print(os.getcwd()) // 之前的命令暂时注释掉
print(os.listdir("E:/")) // 显示E盘下面的所有文件和文件夹名称
额,格式有点不好看,来给大家变换一下:
import os
name_list = os.listdir("E:/")
for name in name_list:
print(name)
结果如下:
可以看到,基本完全一致,除了有个隐藏文件夹之外。。
额,由于今天已经很晚了,就用一个简单的自动化脚本,完成的功能如下:
- 过滤出来在某个给定文件夹下所有的文件路径和文件,然后进行分类打印。
这个需求比较简单,代码如下:
import os
import sys
if len(sys.argv) != 2:
print("error input params")
exit(1)
dir_path = sys.argv[1] // sys.argv[1] 获取python 脚本入参
for name in os.listdir(dir_path): // 上面学过的命令,忘了的话再回头看看
fullname = os.path.join(dir_path, name) // 把字符串dir_path和name连接起来
if os.path.isdir(fullname): // os.path.isdir判断是否是路径
print("{} is a directory".format(fullname))
else:
print("{} is a file".format(fullname))
都有注释,如果大家对python基础感兴趣的话,留言给我哟,关注我,我可以分N多期,给大家从零开始介绍如何使用python。
最后贴一下执行结果:
最后的最后,求关注,求留言。。每天分析大家的喜好也是挺费脑子的。。。关注我直接留言,你想学的这里都有。。。