《零基础入门学习Python》第29讲笔记：一个任务

TommyTimfy

# 接受用户输入并保存为新的文件

def file_write(file_name):
    f = open(file_name,'w')
    print("请输入内容【单独输入\':w\'将保存退出】：")

    while True:
        write_some = input()
        if write_some != ':w':
            f.write('%s\n' % write_some)
        else:
            break

    f.close()

temp = input('请输入文件名：')
file_name = str(temp)+'.txt'
file_write(file_name)

# 对两个文件找不同

def file_compare(file1, file2):
    f1 = open(file1)
    f2 = open(file2)
    count = 0 # 统计行数
    differ = [] # 统计不一样的数量

    for line1 in f1:
        line2 = f2.readline()  # 一次读一行
        count += 1
        if line1 != line2:
            differ.append(count)

    f1.close()
    f2.close()
    return differ

file1 = input('请输入需要比较的头一个文件名：')
file2 = input('请输入需要比较的另一个文件名：')

differ = file_compare(file1, file2)

if len(differ) == 0:
    print('两个文件完全一样！')
else:
    print('两个文件共有【%d】处不同：' % len(differ))
    for each in differ:
        print('第 %d 行不一样' % each)

# 输出用户要求的前N行内容

def file_view(file_name, line_num):
    print('\n文件%s的前%s的内容如下：\n' % (file_name, line_num))
    f = open(file_name)
    for i in range(int(line_num)):   # 打印前N行
        print(f.readline(), end= '')

    f.close()

file_name = input(r'请输入要打开的文件（C:\\test.txt）：')
line_num = input('请输入需要显示该文件前几行：')
file_view(file_name, line_num)

# 输出用户要求的指定行的内容

def file_view(file_name, line_num):
    if line_num.strip() == ':':   # 删去左右的空格，判断：若只有‘:’
        begin = '1'
        end = '-1'
        
    (begin, end) = line_num.split(':')  # 获得前后俩数字

    if begin == '':
        begin = '1'
    if end == '':
        end = '-1'

    if begin == '1' and end == '-1':
        prompt = '的全文'
    elif begin == '1':
        prompt = '从开始到%s' % end
    elif end == '-1':
        prompt = '从%s到结束' % begin
    else:
        prompt = '从第%s行到第%s行' % (begin, end)

    print('\n文件%s%s的内容如下：\n' % (file_name, prompt))

    begin = int(begin) - 1
    end = int(end)
    lines = end - begin

    f = open(file_name)  
    
    for i in range(begin):  # 用于消耗掉begin之前的内容
        f.readline()

    if lines < 0:
        print(f.read())
    else:
        for j in range(lines):
            print(f.readline(), end='')
    
    f.close()

file_name = input(r'请输入要打开的文件（C:\\test.txt）：')
line_num = input('请输入需要显示的行数【格式如 13:21 或 :21 或 21: 或 : 】：')
file_view(file_name, line_num)

# 替换文本的内容

def file_replace(file_name, rep_word, new_word):
    f_read = open(file_name)

    content = []    # 替换后内容
    count = 0       # 替换个数

    for eachline in f_read:
        if rep_word in eachline:
            count += eachline.count(rep_word)  # count感觉应该用这个，获取须替换的词汇数目，此处应累加！
            eachline = eachline.replace(rep_word, new_word)  # 替换操作
        content.append(eachline)       #  记录替换后的内容

    decide = input('\n文件 %s 中共有%s个【%s】\n您确定要把所有的【%s】替换为【%s】吗？\n【YES/NO】：' \
                   % (file_name, count, rep_word, rep_word, new_word))

    if decide in ['YES', 'Yes', 'yes', 'y', 'Y']:
        f_write = open(file_name, 'w')   # 命名技巧
        f_write.writelines(content)
        f_write.close()

    f_read.close()


file_name = input('请输入文件名：')
rep_word = input('请输入需要替换的单词或字符：')
new_word = input('请输入新的单词或字符：')
file_replace(file_name, rep_word, new_word)