Python替换文件内容(文件处理)

发布时间:2019-08-27 08:02:16编辑:auto阅读(1641)

    #需求

    用Python来替换掉文件里面的内容


    #脚本内容如下

    #!/usr/bin/env python
    #_*_ coding:utf-8 _*_
    import sys,os
    
    #小于四个位置变量则退出脚本
    if len(sys.argv) <4:
        print "useage: ./file_replace.py old_text new_text filename"
        sys.exit()
        
    #设置三个位置变量
    old_text,new_text = sys.argv[1],sys.argv[2]
    file_name = sys.argv[3]
    
    #设置原文件可读,新文件可写
    f = file(file_name,'rb')
    new_file = file('.%s.bak' % file_name,'wb')
    
    #把替换内容写入新文件
    for line in f.xreadlines():
        new_file.write(line.replace(old_text,new_text))
     
    #关闭文件   
    f.close()
    new_file.close()
    
    #--bak判断用户需不需要先备份再修改文件
    if '--bak' in sys.argv:
    
        #加上--bak就备份原文件
        os.rename(file_name, '%s.bak' %file_name)   #原文件备份为xx.bak
        os.rename('.%s.bak' %file_name , file_name) #替换的新文件改为xx
        
    else:
    
        #不加--bak替换后的新文件直接复盖原文件
        os.rename('.%s.bak' %file_name , file_name)


    #查看要替换的文件

    [root@localhost opt]# cat passwd 
    ntp:x:38:38::/etc/ntp:/sbin/nologin
    mysql:x:27:27:MySQL Server:/var/lib/mysql:/bin/bash
    wsyht:x:500:500::/home/wsyht:/bin/bash


    #执行脚本不备份原有文件

    [root@localhost opt]# python replace.py : ! passwd
    [root@localhost opt]# cat passwd
    qemu!x!107!107!qemu user!/!/sbin/nologin
    radvd!x!75!75!radvd user!/!/sbin/nologin
    wsyht!x!500!500!!/home/wsyht!/bin/bash


    #执行脚本备份原有文件

    [root@localhost opt]# python replace.py ! : passwd --bak  
    [root@localhost opt]# cat passwd
    qemu:x:107:107:qemu user:/:/sbin/nologin
    radvd:x:75:75:radvd user:/:/sbin/nologin
    wsyht:x:500:500::/home/wsyht:/bin/bash
    [root@localhost opt]# cat passwd.bak 
    qemu!x!107!107!qemu user!/!/sbin/nologin
    radvd!x!75!75!radvd user!/!/sbin/nologin
    wsyht!x!500!500!!/home/wsyht!/bin/bash


关键字