python实现将html表格转换成CSV文件的方法

发布于 2015-11-18 16:53:39 | 155 次阅读 | 评论: 0 | 来源: 网友投递

Python编程语言Python 是一种面向对象、解释型计算机程序设计语言，由Guido van Rossum于1989年底发明，第一个公开发行版发行于1991年。Python语法简洁而清晰，具有丰富和强大的类库。它常被昵称为胶水语言，它能够把用其他语言制作的各种模块(尤其是C/C++)很轻松地联结在一起。

这篇文章主要介绍了python实现将html表格转换成CSV文件的方法,涉及Python操作csv文件的相关技巧,需要的朋友可以参考下

本文实例讲述了python实现将html表格转换成CSV文件的方法。分享给大家供大家参考。具体如下：

使用方法：python html2csv.py *.html

这段代码使用了 HTMLParser 模块

#!/usr/bin/python

# -*- coding: iso-8859-1 -*-

# Hello, this program is written in Python - http://python.org

programname = 'html2csv - version 2002-09-20 - http://sebsauvage.net'

import sys, getopt, os.path, glob, HTMLParser, re

try: import psyco ; psyco.jit() # If present, use psyco to accelerate the program

except: pass

def usage(progname):

''' Display program usage. '''

progname = os.path.split(progname)[1]

if os.path.splitext(progname)[1] in ['.py','.pyc']: progname = 'python '+progname

return '''%s

A coarse HTML tables to CSV (Comma-Separated Values) converter.

Syntax : %s source.html

Arguments : source.html is the HTML file you want to convert to CSV.

By default, the file will be converted to csv with the same

name and the csv extension (source.html -> source.csv)

You can use * and ?.

Examples : %s mypage.html

: %s *.html

This program is public domain.

Author : Sebastien SAUVAGE

http://sebsauvage.net

''' % (programname, progname, progname, progname)

class html2csv(HTMLParser.HTMLParser):

''' A basic parser which converts HTML tables into CSV.

Feed HTML with feed(). Get CSV with getCSV(). (See example below.)

All tables in HTML will be converted to CSV (in the order they occur

in the HTML file).

You can process very large HTML files by feeding this class with chunks

of html while getting chunks of CSV by calling getCSV().

Should handle badly formated html (missing

, , ,

extraneous , ...).

This parser uses HTMLParser from the HTMLParser module,

not HTMLParser from the htmllib module.

Example: parser = html2csv()

parser.feed( open('mypage.html','rb').read() )

open('mytables.csv','w+b').write( parser.getCSV() )

This class is public domain.

Author: Sébastien SAUVAGE

http://sebsauvage.net

Versions:

2002-09-19 : - First version

2002-09-20 : - now uses HTMLParser.HTMLParser instead of htmllib.HTMLParser.

- now parses command-line.

To do:

- handle

 tags

- convert html entities (&name; and ref;) to Ascii.

'''

def __init__(self):

HTMLParser.HTMLParser.__init__(self)

self.CSV = '' # The CSV data

self.CSVrow = '' # The current CSV row beeing constructed from HTML

self.inTD = 0 # Used to track if we are inside or outside a

... tag.

self.inTR = 0 # Used to track if we are inside or outside a

... tag.

self.re_multiplespaces = re.compile('s+') # regular expression used to remove spaces in excess

self.rowCount = 0 # CSV output line counter.

def handle_starttag(self, tag, attrs):

if tag == 'tr': self.start_tr()

elif tag == 'td': self.start_td()

def handle_endtag(self, tag):

if tag == 'tr': self.end_tr()

elif tag == 'td': self.end_td()

def start_tr(self):

if self.inTR: self.end_tr() #

implies

self.inTR = 1

def end_tr(self):

if self.inTD: self.end_td() # implies

self.inTR = 0

if len(self.CSVrow) > 0:

self.CSV += self.CSVrow[:-1]

self.CSVrow = ''

self.CSV += 'n'

self.rowCount += 1

def start_td(self):

if not self.inTR: self.start_tr() #

implies

self.CSVrow += '"'

self.inTD = 1

def end_td(self):

if self.inTD:

self.CSVrow += '",'

self.inTD = 0

def handle_data(self, data):

if self.inTD:

self.CSVrow += self.re_multiplespaces.sub(' ',data.replace('t',' ').replace('n','').replace('r','').replace('"','""'))

def getCSV(self,purge=False):

''' Get output CSV.

If purge is true, getCSV() will return all remaining data,

even if

or are not properly closed.

(You would typically call getCSV with purge=True when you do not have

any more HTML to feed and you suspect dirty HTML (unclosed tags). '''

if purge and self.inTR: self.end_tr() # This will also end_td and append last CSV row to output CSV.

dataout = self.CSV[:]

self.CSV = ''

return dataout

if __name__ == "__main__":

try: # Put getopt in place for future usage.

opts, args = getopt.getopt(sys.argv[1:],None)

except getopt.GetoptError:

print usage(sys.argv[0]) # print help information and exit:

sys.exit(2)

if len(args) == 0:

print usage(sys.argv[0]) # print help information and exit:

sys.exit(2)

print programname

html_files = glob.glob(args[0])

for htmlfilename in html_files:

outputfilename = os.path.splitext(htmlfilename)[0]+'.csv'

parser = html2csv()

print 'Reading %s, writing %s...' % (htmlfilename, outputfilename)

try:

htmlfile = open(htmlfilename, 'rb')

csvfile = open( outputfilename, 'w+b')

data = htmlfile.read(8192)

while data:

parser.feed( data )

csvfile.write( parser.getCSV() )

sys.stdout.write('%d CSV rows written.r' % parser.rowCount)

data = htmlfile.read(8192)

csvfile.write( parser.getCSV(True) )

csvfile.close()

htmlfile.close()

except:

print 'Error converting %s ' % htmlfilename

try: htmlfile.close()

except: pass

try: csvfile.close()

except: pass

print 'All done. '

相关阅读:

python实现将html表格转换成CSV文件的方法

python实现将pvr格式转换成pvr.ccz的方法

Python实现将绝对URL替换成相对URL的方法

python通过pil模块将raw图片转换成png图片的方法

python使用PythonMagick将jpg图片转换成ico图片的方法

python实现支持目录FTP上传下载文件的方法

Python实现压缩与解压gzip大文件的方法

python实现将英文单词表示的数字转换成阿拉伯数字的方法

python将图片文件转换成base64编码的方法

python将文本转换成图片输出的方法

python实现将元祖转换成数组的方法

Python实现简单HTML表格解析的方法