处理文本文件什么的，python最好用了

原创于 2015-05-25 15:01:35 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #mysqldump #数据恢复

本文介绍了一个使用Python编写的简单脚本，该脚本用于解决MySQLWorkbench无法直接导入mysqldump备份文件的问题。通过在每个备份文件头部添加指定的数据库使用命令，实现了备份文件的有效恢复。

mysql workbench用来恢复数据最好用了，可惜无法恢复mysqldump备份的文件。问题在开头的“use。。。”语句。花了一天时间入门python写了个小东西，好用的不得了

#coding:utf-8
import os
fileDir='e:/mysql_backup/01'
newFileDir='e:/mysql_backup/add_use_xh_erp'
#删除目标目录下文件
if os.path.exists(newFileDir):
    nameList=os.listdir(newFileDir)
    for fileName in nameList:
        print fileName+",deleted"
        os.remove(newFileDir+"/"+fileName)
else:#目录不存在，新建一个
    os.makedirs(newFileDir)

nameList=os.listdir(fileDir)
for fileName in nameList:
    fileOpen=open(fileDir+'/'+fileName)
    try:
        all_text = fileOpen.read()
    finally:
        fileOpen.close()
    fileNew=open(newFileDir+'/'+fileName,'a')
    fileNew.write("USE `xh_erp`; \r\n") 
    fileNew.write(all_text)
    print(fileName+", add 'use xh_erp' ok")