问题描述
我正在尝试熟悉抽象类,但不知道如何附加到类中的空列表。这是我正在工作的目录的样子:
my_dir
|__data
| |__colors
| | |__light_colors.csv
| | |__dark_colors.csv
| | |__cold_colors.json
| |__ages
|__code
|__data_reader.py
|__config.py
在我的配置文件中:
import os
REPO_ROOT_DIRECTORY = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
colors_data_dirECTORY = os.path.join(REPO_ROOT_DIRECTORY,"data")
在我的 data_reader.py 文件中,我有以下代码,它是一个抽象类。
from abc import ABC,abstractmethod
from config import colors_data_dirECTORY
import os
class PathReader(ABC):
@abstractmethod
def read_data_type(self): pass
class colorReader(PathReader):
def __init__(self):
self.path_to_types = os.path.join(colors_data_dirECTORY,'colors')
colors= []
def read_data_type(self):
for files in os.listdir(self.path_to_types):
colors.append(os.path.join(self.path_to_types,files))
g= colorReader()
print(g.read_data_type())
然而,似乎列表 colors= [] 是空的,理想情况下我想在另一个抽象类中使用这个列表,所以,我可以检查存储在列表中的文件的扩展名,所以这就是代码的方式data_reader.py 继续。
class ExtensionReader(ABC):
@abstractmethod
def file_extension_reader(self): pass
class csvFileExtension(ExtensionReader):
def __init__(self,file_type: PathReader):
self.file_type = file_type
def file_extension_reader(self):
for files in self.file_type:
if files.split(".")[1] == 'csv':
print(files)
但这是我不知道如何使用颜色 =[] 列表来检查以 .csv 结尾的文件的扩展名的地方
解决方法
这与抽象类无关。您忘记将 colors
变量设为 self
的属性:
from abc import ABC,abstractmethod
from config import colors_DATA_DIRECTORY
import os
class PathReader(ABC):
@abstractmethod
def read_data_type(self): pass
class colorReader(PathReader):
def __init__(self):
self.path_to_types = os.path.join(colors_DATA_DIRECTORY,'colors')
self.colors= []
def read_data_type(self):
for files in os.listdir(self.path_to_types):
self.colors.append(os.path.join(self.path_to_types,files))
g= colorReader()
print(g.read_data_type())