import os import pdfplumber # 文本保存目录 txts_dir = "./Users/hh/Desktop/python/fileshunan" for path in os.listdir(files_dir): if os.path.isfile(os.path.join(files_dir, path)): file = os.path.join(files_dir, path) print(file) if file.endswith(".pdf"): with pdfplumber.open(file) as pdf: txt_file = open(txts_dir + path[0:-4] + ".txt", mode='w', encoding='utf-8') for page in pdf.pages: txt_file.write(page.extract_text()) elif file.endswith(".html"): soup = BeautifulSoup(open(file, encoding='utf-8')) with open(txts_dir + path[0:-5] + ".txt", mode='w', encoding='utf-8') as txt_file: txt_file.write(soup.get_text()) #将文件保存为文本
时间: 2024-04-13 15:30:00 浏览: 302
根据你提供的代码,你似乎想要将保存在文件夹中的 PDF 和 HTML 文件转换为文本文件。你使用了 `pdfplumber` 库来处理 PDF 文件,并使用 BeautifulSoup 库处理 HTML 文件。
首先,请确保你已经导入了以下库:
```python
import os
import pdfplumber
from bs4 import BeautifulSoup
```
然后,你可以使用以下代码将文件转换为文本:
```python
# 文本保存目录
txts_dir = "/Users/hh/Desktop/python/fileshunan/"
for path in os.listdir(files_dir):
if os.path.isfile(os.path.join(files_dir, path)):
file = os.path.join(files_dir, path)
print(file)
if file.endswith(".pdf"):
with pdfplumber.open(file) as pdf:
txt_file = open(txts_dir + path[0:-4] + ".txt", mode='w', encoding='utf-8')
for page in pdf.pages:
txt_file.write(page.extract_text())
txt_file.close()
elif file.endswith(".html"):
soup = BeautifulSoup(open(file, encoding='utf-8'), 'lxml')
with open(txts_dir + path[0:-5] + ".txt", mode='w', encoding='utf-8') as txt_file:
txt_file.write(soup.get_text())
```
在上述代码中,我们首先定义了保存文本文件的目录路径 `txts_dir`。
然后,我们使用 `os.listdir()` 函数遍历指定文件夹中的文件。对于每个文件,我们使用 `os.path.isfile()` 函数来判断是否为文件,并使用 `os.path.join()` 函数获取完整路径。
如果文件是以 `.pdf` 结尾的 PDF 文件,我们使用 `pdfplumber.open()` 函数打开 PDF 文件,并使用循环遍历每个页面。我们使用 `page.extract_text()` 方法提取页面的文本内容,并将其写入到以原文件名为基础的新文本文件中。
如果文件是以 `.html` 结尾的 HTML 文件,我们使用 BeautifulSoup 打开文件并解析 HTML 内容。然后,我们使用 `soup.get_text()` 方法获取 HTML 的纯文本内容,并将其写入到以原文件名为基础的新文本文件中。
请确保你已经设置好了 `txts_dir` 变量,它指定了保存文本文件的目录路径。
请注意,以上代码仅供参考,并需要根据你的实际文件路径和需求进行适当的修改。另外,记得导入所需的库。
如果你有任何进一步的问题,请随时提问。
阅读全文