python 遍历所有文件夹和子文件夹_用python遍历目录下文件(不包括子目录) | 求索阁...

#!/usr/bin/env python

#coding=gbk

__author__ = 'lc'

import MySQLdb

import glob

import os

import sys

DB_HOST = "localhost"

DB_USER = "root"

DB_PWD = "root"

DB_DATABASE = "database"

CHARVAR_PATH = "dir"

NOT_EXIST_FILE = "./not_exist_character.txt"

EXPORT_SQL_FILE = "./character_var.sql"

def check_arg():

if len(sys.argv) != 2:

print("参数不正确! 用法: convert_charvar.py 变量目录")

exit(1)

global CHARVAR_PATH

CHARVAR_PATH = sys.argv[1]

def convert_and_dump():

check_arg()

print("正在执行转换, 请稍候......")

try:

fp = None

cursor = None

sql_fp = open(EXPORT_SQL_FILE, "w")

not_exist_fp = open(NOT_EXIST_FILE, "w")

dbConn = MySQLdb.connect(DB_HOST, DB_USER, DB_PWD, DB_DATABASE)

cursor = dbConn.cursor()

for varFile in glob.glob(CHARVAR_PATH + os.sep + "*.txt"):

fileName = os.path.basename(varFile)

charName = fileName.rsplit(".txt")[0]

sql = "select id from tbl_character_info where name = '{}'".format(charName)

resNum = cursor.execute(sql)

if resNum == 0:

print("玩家 {} 在数据库角色表中查询不存在".format(charName))

not_exist_fp.write(charName + "\n")

continue

row = cursor.fetchone()

charid = row[0]

fp = open(varFile, "r")

varLst = fp.readlines()

fp.close()

for var in varLst:

oneLst = var.split("=")

if len(oneLst) != 2:

continue

name = oneLst[0].strip()

value = oneLst[1].strip(" \n\"")

INSERT_PREFIX = "insert into tbl_character_var (charid, name, value) values({}, '{}', '{}');    #角色名:{}"

sql = INSERT_PREFIX.format(charid, name, value, charName)

sql_fp.write(sql + "\n")

except Exception as exp:

print("严重错误: " + str(exp))

else:

print("执行完毕, sql语句保存在" + EXPORT_SQL_FILE + ", 不存在的玩家名保存在" + NOT_EXIST_FILE)

finally:

dbConn.close()

sql_fp.close()

not_exist_fp.close()

if cursor is not None:

cursor.close()

if fp is not None:

fp.close()

if __name__ == "__main__":

convert_and_dump()