프로그래밍/파이썬

파이썬 엑셀 입력하기

레곤플라이 2009. 6. 15. 14:08

pyXLWriter를 통해서 엑셀에 손쉽게 데이터를 넣을 수 있다.

pyXLWriter의 설치는 setup.py install 명령으로 할 수 있다.

아래의 프로그램은 python 파일이 존재하는 디렉토리에서
하위의 파일 리스트들을 검색하여
파일 리스트를 엑셀로 저장하는 프로그램이다.

#-*- coding: cp949 -*-

import os
import pyXLWriter as xl

def search(dirname):
 filelist = os.listdir(dirname)
 for f in filelist:
  next = os.path.join(dirname, f)
  if os.path.isdir(next):
   search(next)
   
  else:
   doFileWork(next)
   
def doFileWork(filename):
 global id
 print os.path.dirname(filename)+'\\'+os.path.basename(filename)

 ext=os.path.splitext(filename)[-1].upper()
 if ext == ".JPG" or ext == ".GIF":
  return

 worksheet.write([id, 0],  os.path.dirname(filename)+'\\'+os.path.basename(filename))
 id=id+1


global worksheet
workbook  = xl.Writer(r"c:\korea.xls")
worksheet = workbook.add_worksheet()
id=1
search("./")

workbook.close()