Python写脚本常用模块OS基础用法详解

收集了一些关于OS库的用法,整理归纳一下,方便使用

import os

# 系统操作
print(os.sep)       # 获取当前系统的路径分隔符
print(os.name)      # 获取当前使用的工作平台
print(os.getenv('PATH')) # 获取名为 PATH 的环境变量
print(os.getcwd())    # 获取当前的路径
print(os.environ['PATH']) # 可以返回环境相关的信息 不传参时,以字典的方式返回所有环境变量

# 调用系统命令
os.system(command) # 将linux命令传入这里，就可以执行 Execute the command in a subshell. 

# 目录操作 - 增删改查
dir = "/opt/"
listdir_opt = os.listdir(dir) # 返回指定目录下的所有文件何目录名
print(listdir_opt) 
os.mkdir("/opt/os-make/") # 创建一个目录，只创建一个目录文件
os.mknod("/root/python-test.txt") # 创建一个文件
os.rmdir("/opt/ooo/") # 删除一个空目录，若目录中有文件则无法删除
os.remove("/tmp/test.txt") # 用于删除文件，若是目录，则抛出 IsDirectoryError 异常
os.makedirs("/opt/os-make-again/os-make-again-again") # 可以生成多层递归目录，如果目录全部存在，则创建目录失败
os.removedirs() # 从最下级目录开始，逐级删除指定路径，遇到非空目录即停止
os.chdir("/tmp/") # 改变当前目录，到指定目录
os.rename("/opt/ooo/","/opt/AAA/") # 重命名目录名或者文件名。重命名后的文件已存在，则重命名失败。
"""
os.rename()函数的作用是将文件或路径重命名，一般调用格式为os.rename(src, dst)，即将src指向的文件或路径重命名为dst指定的名称。

注意，如果指定的目标路径在其他目录下，该函数还可实现文件或路径的“剪切并粘贴”功能。但无论直接原地重命名还是“剪切粘贴”，中间路径都必须要存在，否则就会抛出FileNotFoundError异常。如果目标路径已存在，Windows 下会抛出FileExistsError异常；Linux 下，如果目标路径为空且用户权限允许，则会静默覆盖原路径，否则抛出OSError异常，
和上两个函数一样，该函数也有对应的递归版本os.renames()，能够创建缺失的中间路径。

注意，这两种情况下，如果函数执行成功，都会调用os.removedir()函数来递归删除源路径的最下级目录。
"""

# 判断
if os.path.exists("/root"):
 print("/root 目录存在！")

if os.path.isfile("/root"):
 print("/root 文件存在！")

if os.path.isdir("/etc"):
 print("/etc 目录存在！")

if os.path.isabs("/etc"):
 print("/etc 是绝对路径！")

# path模块
"""
os.path中的函数基本上是纯粹的字符串操作。换句话说，传入该模块函数的参数甚至不需要是一个有效路径，该模块也不会试图访问这个路径，而仅仅是按照“路径”的通用格式对字符串进行处理。
"""
path = "/etc/passwd"
filename = os.path.basename(path) # 返回文件名，如果是目录则为空 实际上是传入路径最后一个分隔符之后的子字符串，也就是说，如果最下级目录之后还有一个分隔符，得到的就会是一个空字符串
filedir = os.path.dirname(path) # 返回的是最后一个分隔符前的整个字符串
filesplit = os.path.split(path) # 将传入路径以最后一个分隔符为界，分成两个字符串，并打包成元组的形式返回
"""
类似的
os.path.splitext("ooo.txt")
('ooo', '.txt')
"""
filesize = os.path.getsize(path) # 获取文件的大小 相当于 ls -l 单位为bytes
fileAbsPath = os.path.abspath(path) # 获取文件的绝对路径
filejoin = os.path.join(path,"test.txt") # 拼接新的路径
"""
如果传入路径中存在一个“绝对路径”格式的字符串，且这个字符串不是函数的第一个参数，那么其他在这个参数之前的所有参数都会被丢弃，余下的参数再进行组合。更准确地说，只有最后一个“绝对路径”及其之后的参数才会体现在返回结果中。

例子如下:

os.path.join("just", "do", "/opt/", "it")
结果： /opt/it
os.path.join("just", "do", "/opt/", "python", "dot", "/root", "com")
结果：/root/com
"""

print(filename+"\n"+filedir+"\n"+str(filesize)+"\n"+fileAbsPath+"\n"+filejoin+"\n")

以上模块和函数的用法已经做了一些说明,理解起来也不难,下面再介绍一个函数os.walk()

import os

for item in os.walk("/opt/test-walk/"):
  print(item)

程序输出结果

('/opt/test-walk/', ['a', 'b', 'c'], [])
('/opt/test-walk/a', [], ['a.txt'])
('/opt/test-walk/b', ['b2'], [])
('/opt/test-walk/b/b2', [], ['b.txt'])
('/opt/test-walk/c', [], [])

目录结构

[root@open-1 python_scripts]# tree /opt/test-walk/
/opt/test-walk/
├── a
│  └── a.txt
├── b
│  └── b2
│    └── b.txt
└── c

4 directories, 2 files

由上面的结果可以大致明白os.walk()函数的作用:这个函数需要传入一个路径作为参数，函数的作用是在该路径为根节点的目录树中游走，对树中的每个目录生成一个由(dirpath, dirnames, filenames)三项组成的三元组。其中，dirpath是一个指示这个目录路径的字符串，dirnames是一个dirpath下子目录名（除去 . 和 ..）组成的列表，filenames则是由dirpath下所有非目录的文件名组成的列表。简单来说,就是把目标路径下的所有目录和文件都列出来,结合tree命令的结果,就更好理解这个函数的作用了.

 ----------------------练习------------------------

在当前目录新建目录img, 里面包含多个文件, 文件名各不相同(X4G5.png)

将当前img目录所有以.png结尾的后缀名改为.jpg

import random
import string
import os

def gen_code(len=4):
  # 随机生成4位验证码
  li = random.sample(string.ascii_letters+string.digits,len)
  return ''.join(li)
def create_file():
  # 随机生成100个验证码
  li = {gen_code() for i in range(100)}
  os.mkdir('img')
  for name in li:
    os.mknod('img/' + name + '.png')

create_file()

def modify_suffix(dirname,old_suffix,new_suffix):
  """
  :param dirname:操作的目录
  :param old_suffix: 之前的后缀名
  :param new_suffix: 新的后缀名
  :return:
  """
  # 1.判断查找的目录是否存在，如果不存在，报错
  if os.path.exists(dirname):
    # 2.找出所有以old_suffix(.png)结尾的文件
    pngfile = [filename for filename in os.listdir(dirname)
          if filename.endswith(old_suffix)]
    # 3.将后缀名和文件名分开，留下文件名
    basefiles = [os.path.splitext(filename)[0]
           for filename in pngfile]
    # 4.重命名文件
    for filename in basefiles:
      oldname = os.path.join(dirname,filename+old_suffix)
      newname = os.path.join(dirname,filename+new_suffix)
      os.rename(oldname,newname)
      print('%s命名为%s成功' %(oldname,newname))
  else:
    print('%s不存在,不能操作...' %(dirname))

modify_suffix('redhat','.jpg','.png')

-----------------练习-----------------------

利用time.time()方法，我们可以计算两个时间点之间的时间间隔

但是有些时候我们想要得到/etc/group文件的a/c/m的时间
对应的年月日这些信息
并保存再文件date.txt文件中

import os
import time

time1 = os.path.getctime('/etc/shadow')   #时间戳时间
print(time1)
tuple_time = time.localtime(time1)
print(tuple_time)
year = tuple_time.tm_year
month = tuple_time.tm_mon
day = tuple_time.tm_mday

with open('date.txt','a') as f:
  f.write('%d %d %d' %(year,month,day))
  f.write('\n')

到此这篇关于Python写脚本常用模块OS基础用法详解的文章就介绍到这了,更多相关Python 模块OS使用内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！