生成一个python 资源管理类 要求 遍历文件夹下目录 可选择类型(txt和png)读取 pnt图片用cv2加载 按文件夹名字分类 使用时根据分类名返回list
时间: 2024-03-17 16:42:05 浏览: 145
Java_PNT:“ Escuelita Somos PNT”迷你项目的存储库
好的,我可以帮你生成这样一个Python资源管理类。以下是代码示例:
```python
import os
import cv2
class ResourceLoader:
def __init__(self, folder_path):
self.folder_path = folder_path
self.resources = {}
self.load_resources()
def load_resources(self):
for root, dirs, files in os.walk(self.folder_path):
for file in files:
file_path = os.path.join(root, file)
if file.endswith('.txt'):
self.load_text_resource(file_path)
elif file.endswith('.png'):
self.load_image_resource(file_path)
def load_text_resource(self, file_path):
with open(file_path, 'r') as file:
content = file.read()
folder_name = os.path.basename(os.path.dirname(file_path))
if folder_name not in self.resources:
self.resources[folder_name] = {}
self.resources[folder_name][file_path] = content
def load_image_resource(self, file_path):
img = cv2.imread(file_path)
folder_name = os.path.basename(os.path.dirname(file_path))
if folder_name not in self.resources:
self.resources[folder_name] = {}
self.resources[folder_name][file_path] = img
def get_resources_by_folder(self, folder_name):
return self.resources.get(folder_name, {}).values()
```
使用方法:
```python
resource_loader = ResourceLoader('path/to/folder')
resources = resource_loader.get_resources_by_folder('folder_name')
```
其中,`path/to/folder` 是你要遍历的文件夹路径,`folder_name` 是你要获取资源的文件夹名称。在上面的代码中,我们使用了 `os.walk` 函数来遍历文件夹下的所有文件和子文件夹。对于每个文件,我们根据文件类型调用不同的加载函数。加载的资源会按照文件夹名称进行分类,并存储在 `self.resources` 字典中。最后,我们提供了一个 `get_resources_by_folder` 方法,可以根据文件夹名称返回该文件夹下的所有资源列表。
阅读全文