《零基础入门学习Python》第30讲笔记：文件系统：介绍一个高大上的东西

TommyTimfy

模块介绍
1.        OS：Operation System
Python是跨平台的语言，OS使我们无需担心平台

查询：扩展阅读

>>> import os
>>> os.getcwd()
'D:\\软件\\Python'
>>> os. chdir('E:\\')
>>> os.getcwd
<built-in function getcwd>
>>> os.getcwd()
'E:\\'

‘.’指代当前目录，‘..’指代上一级目录（os.pardir）

>>> os.curdir
'.'
>>> os.listdir(os.curdir)  # 列举当前目录文件名，也可以os.listdir(‘.’)

os.path若干用法

相对路径 / 绝对路径

练习：
0.

# 输入文件名/开始搜索的路径，搜索该文件名是否存在。
# 若遇到文件夹，则进入文件夹继续搜索

import os

def search_file(start_dir, target):
    os.chdir(start_dir)
    
    for each_file in os.listdir(os.curdir):
        if each_file == target:  # 找到目标
            print(os.getcwd() + os.sep + each_file)  # os.sep使程序更标准
            # 当前目录 + 路径分隔符 + 文件名
        if os.path.isdir(each_file):
            search_file(each_file, target)  # 递归调用
            os.chdir(os.pardir)             # 记得返回上一层目录！
            
start_dir = input('请输入待查找的初始目录：')
target = input('请输入待查找的目标文件：')
search_file(start_dir, target)

1.

# 输入开始搜索路径，查找该路径下所有的指定格式文件（mp4,avi,rmvb）
# 并创建一个文件（VedioList.txt）存放找到的文件的路径

import os

def search_file(start_dir, target):
    os.chdir(start_dir)
    
    for each_file in os.listdir(os.curdir):
        ext = os.path.splitext(each_file)[1] # 获取后缀
        if ext in target :
            vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep) # 使用os.sep让程序更标准
        if os.path.isdir(each_file) :
            search_file(each_file, target) # 递归调用
            os.chdir(os.pardir) # 递归调用后切记返回上一层目录

start_dir = input('请输入待查找的初始目录：')
program_dir = os.getcwd()

target = ['.mp4', '.avi', '.rmvb']
vedio_list = []  # 存储对应格式文件路径

search_file(start_dir, target)

f = open(program_dir + os.sep + 'vedioList.txt', 'w')
f.writelines(vedio_list)
f.close()

2.

# 用户输入关键字，查找当前文件夹内（如果当前文件夹包含文件夹，则进入文件夹继续搜索）
# 所有含有该关键字的文本文件（.txt后缀），要求显示该文件所在位置以及关键字在文件中的
# 具体位置（第几行，第几个字符）

import os

def print_pos(key_dict):
    keys = key_dict.keys()
    keys = sorted(keys) # 由于字典是无序的，我们这里对行数进行排序
    for each_key in keys:
        print('关键字出现在第 %s 行，第 %s 个位置。' % (each_key, str(key_dict[each_key])))


def pos_in_line(line, key):
    pos = []
    begin = line.find(key)
    while begin != -1:
        pos.append(begin + 1) # 用户的角度是从1开始数
        begin = line.find(key, begin+1) # 从下一个位置继续查找

    return pos


def search_in_file(file_name, key):
    f = open(file_name)
    count = 0 # 记录行数
    key_dict = dict() # 字典，用户存放key所在具体行数对应具体位置
    
    for each_line in f:
        count += 1
        if key in each_line:
            pos = pos_in_line(each_line, key) # key在每行对应的位置
            key_dict[count] = pos
    
    f.close()
    return key_dict


def search_files(key, detail):    
    all_files = os.walk(os.getcwd())  # 遍历path，返回(path, [包含目录], [包含文件])
    txt_files = []

    for i in all_files:
        for each_file in i[2]:
            if os.path.splitext(each_file)[1] == '.txt': # 根据后缀判断是否文本文件
                each_file = os.path.join(i[0], each_file)
                txt_files.append(each_file)

    for each_txt_file in txt_files:
        key_dict = search_in_file(each_txt_file, key)
        if key_dict:  # 字典内有内容
            print('================================================================')
            print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
            if detail in ['YES', 'Yes', 'yes']:
                print_pos(key_dict)


key = input('请将该代码放于待查找的文件夹内，请输入关键字：')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：' % key)
search_files(key, detail)

3.

# 编写一个程序，统计当前目录下每个文件类型的文件数
import os


all_files = os.listdir(os.curdir)  # 存储当前目录内的文件名
type_dict = dict()  # 字典

for each_file in all_files:
    if os.path.isdir(each_file):  # 指定路径是否存在且是一个目录(即文件夹！)
        type_dict.setdefault('文件夹', 0)  # 默认规则插入内容
        type_dict['文件夹'] += 1
    else:
        ext = os.path.splitext(each_file)[1]  # 分离文件名、扩展名，只返回后者
        type_dict.setdefault(ext, 0)
        type_dict[ext] += 1

for each_type in type_dict.keys():   # 遍历字典中的键
    print('该文件夹下共有类型为【%s】的文件%d个' % \
          (each_type, type_dict[each_type]))

4.

# 计算当前文件夹下所有文件的大小
import os

all_files = os.listdir(os.curdir)  # 存储当前目录内的文件名
file_dict = dict()  # 字典

for each_file in all_files:
    if os.path.isfile(each_file):  
        # 指定路径是否存在且是一个文件
        file_size = os.path.getsize(each_file)  # 获取大小
        file_dict[each_file] = file_size        # 存储

for each in file_dict.items():
    print('%s【%d Bytes】' % (each[0], each[1]))