Python操作文件读写

import csv

from pdfminer.converter import PDFPageAggregator
from pdfminer.layout import LAParams
from pdfminer.pdfparser import PDFParser, PDFDocument
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter


class DealFile(object):
    """
        读取csv文件
        @:param path: 文件路径
        @:return result: 以列表形式将文件内容返回
    """

    @staticmethod
    def read_csv(path):
        result = []
        with open(path, 'r') as f:
            reader = csv.reader(f)
            for row in reader:
                result.append(row)
        return result

    """
        向csv文件中写入内容
        @:param path: 文件路径
        @:param data: 以列表形式将内容传入
    """

    @staticmethod
    def write_csv(path, data):
        with open(path, 'w') as f:
            writer = csv.writer(f)
            for row_data in data:
                writer.writerow(row_data)

    """
        读取pdf文件
        @:param path: 文件路径
        @:return result: 以列表形式将文件内容返回
    """

    @staticmethod
    def read_pdf(path):
        result = []
        # 创建文档对象
        fp = open(path, "rb")

        # 创建一个与文档关联的解释器
        parser = PDFParser(fp)

        # 创建一个PDF的文档对象
        doc = PDFDocument()

        # 连接解释器和文档对象
        parser.set_document(doc)
        doc.set_parser(parser)

        # 初始化文档
        doc.initialize("")

        # 创建PDF资源管理器
        resource = PDFResourceManager()

        # 参数分析器
        laparam = LAParams()

        # 聚合器
        device = PDFPageAggregator(resource, laparams=laparam)

        # 页面解释器
        interpreter = PDFPageInterpreter(resource, device)

        # 使用文档对象得到页面的集合
        for page in doc.get_pages():
            # 使用页面解释器来读取
            interpreter.process_page(page)

            # 使用聚合器获得内容
            layout = device.get_result()

            for out in layout:
                if hasattr(out, "get_text"):
                    result.append(out.get_text)
        return result

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值