PythonLibs(12) - sys, os, glob

sys是Python的一个「标准库」，也就是官方出的「模块」，是「System」的简写，封装了一些系统的信息和接口。

1 sys

1.1 sys.argv

「argv」是「argument variable」参数变量的简写，一般在 命令行调用 的时候由 系统 传递给 程序 。在.py脚本里可用于获取 脚本调用传参。

# sys.argv 是一个List列表:
# argv[0] 一般是“被调用脚本文件名或全路径”(与操作系统有关);
# argv[1]  是传入系统命令参数;
python using sys.args.py “whoami”  # sys.argv[1]是“whoami”这个参数

1.2 sys._getframe()

输出代码位置信息，方便调试-定位。

#coding=utf-8
import sys
import os

# sys._getframe() 调试-定位-好帮手
# print(sys._getframe().f_code.co_filename)  # 当前文件名， 与__file__作用等价
# print(sys._getframe(0).f_code.co_name)     # 当前函数名, 如果没有被调用，则返回<module>
# print(sys._getframe(1).f_code.co_name)     # 上层调用函数名，如果未有上层调用报错-ValueError: call stack is not deep enough
# print(sys._getframe(0).f_lineno)           # 当前函数的行号 
# print(sys._getframe(1).f_lineno)           # 上层调用函数的行号，如果未有上层调用报错-ValueError: call stack is not deep enough

def get_location_str(normal_mode=True):
    """
    @brief: normal_mode [文件名:行号] 方便VsCode文件搜索[command + p]跳转
            detail_mode [文件名:行号] [函数名] 
    """
    if normal_mode:
        location_str = "[{0}:{1}]".format(os.path.basename(sys._getframe().f_code.co_filename), 
                                                           sys._getframe(1).f_lineno)
    else:
        location_str = "[{0}:{1}] [{2}]".format(os.path.basename(sys._getframe().f_code.co_filename), 
                                                                 sys._getframe(1).f_lineno,
                                                                 sys._getframe(1).f_code.co_name)
    return location_str

if __name__ == "__main__":

    print("{} test0".format(get_location_str()))
    print("{} test1".format(get_location_str(False)))

执行结果

[sys_test.py:28] test0
[sys_test.py:29] [<module>] test1

2.os 模块

文件/目录的常用管理操作
在终端或者文件浏览器中可以执行的常规操作。例如：创建，重命名，删除，改变路径，查看目录内容等。通过Python实现上述功能，需要导入os模块。

2.1 改变工作目录os.chdir()

使用脚本运行，默认的工作目录为脚本所在的目录中。可以使用chdir 改变工作目录

import os
print(os.getcwd())
# linux 命令行使用pwd来输出当前路径
os.chdir("/Users/chenyingying")
print(os.getcwd())

输出

/Users/chenyingying/PycharmProjects/DataMining_python
/Users/chenyingying

2.2 文件重命名os.rename()

os.rename(源文件名.扩展名，目标文件名.扩展名)

2.3 删除文件os.remove()

os.remove(文件名.扩展名)

demo:删除文件而保留目录结构

import os
g=os.walk('/home/abc/Desktop/cyy_test')
for path,dir,files in g:  #路径path 文件夹中的文件名files和文件夹名dir（都是list）
	print(path,dir,files)
	while files!=[]:
		for file in files:
			print(file)
			os.remove(os.path.join(path,file))
			files.remove(file)
print('end')

由顶至下的遍历所有文件夹

2.4 删除空路径os.removedirs()

删除一个空文件夹，如果文件夹里有文件，或者文件夹是删除不成功的。

os.removedirs(’./cyy_test/subtest123’)

删除一个非空路径（如果文件夹里面有文件也会一起被删掉）

shutil.rmtree(’./cyy_test/subtest2’)

2.5 目录列表os.listdir()

os.listdir(目录名)
os.listdir(“.”) # 查看当前目录的目录列表

2.6 创建目录os.mkdir()

os.mkdir(“目录名”)

2.7 删除目录os.rmdir()

os.rmdir(”目录名“)

2.8 判断某个目录存在否,如果不在,创建这个目录

import os
log_path = os.getcwd() + '/Logs/' #当前文件夹下的Logs文件夹
if not os.path.exists(log_path) :  #判断文件夹或文件
   os.makedirs(log_path)  #创建

# Note 不能使用通配符
print(os.path.exists("cyy_test*"))      
print(os.path.exists("cyy_test1.py"))

参考资料:https://blog.csdn.net/xielifu/article/details/81013536

2.9 os.path.split(“xxx/xx/xxxx”)

将路径分割出来,一次分割一个级别,每次分割得到两个元素的tuple.

>>> import os 
>>> a=os.getcwd()
>>> a
'/home/pp/Desktop/testsplit/aaa/bbb'

>>> a=os.getcwd()+"/1.txt"
>>> a
'/home/pp/Desktop/testsplit/aaa/bbb/1.txt'

>>> b=os.path.split(a)
>>> b
('/home/pp/Desktop/testsplit/aaa/bbb', '1.txt')

>>> c=os.path.split(b[0])
>>> c
('/home/pp/Desktop/testsplit/aaa', 'bbb')

2.10 os.path.isdir()

目录名：要查看的目录的名字。确定目录列表中的字符串为一个目录，而不是一个文件。返回的True或者False。

os.path.isdir(“xxx”)

先判断路径是否存在，不存在就创建该路径

if not os.path.isdir(tarDir): 
	os.makedirs(tarDir)

3 glob 模块

glob是python自带的一个操作文件的模块，可用于查找 指定路径 中 匹配的（支持模糊匹配） 文件, 返回【路径】列表。

”*”匹配0个或多个字符；
”?”匹配单个字符；
”[]”匹配指定范围内的字符，如：[0-9]匹配数字。

(base) pp@pp-System-Product-Name:~/Desktop/test_glob$ tree
.
├── a
│   ├── 12.txt
│   ├── 1.txt
│   └── 2.txt
└── b
    ├── 12.txt
    ├── 1.txt
    └── 2.txt
2 directories, 6 files

# 1. 使用glob查找 返回符合模糊匹配规则的 文件路径列表
import glob
>>> glob.glob("/home/pp/Desktop/test_glob/a/1*")  # *用于匹配任意字符
['/home/pp/Desktop/test_glob/a/12.txt', '/home/pp/Desktop/test_glob/a/1.txt']

# 2 os.listdir() 只是列出给定路径中的 文件或目录名
>>> import os
>>> os.listdir("/home/pp/Desktop/test_glob/a")
['2.txt', '12.txt', '1.txt']
>>> os.listdir("/home/pp/Desktop/test_glob")
['a', 'b']

参考资料：https://blog.csdn.net/gufenchen/article/details/90723418