in utils/ryzenai/notification_service.py [0:0]
def retrieve_available_artifacts():
class Artifact:
def __init__(self, name: str):
self.name = name
def __str__(self):
return self.name
def add_path(self, path: str):
self.path = path
_available_artifacts: Dict[str, Artifact] = {}
directories = filter(os.path.isdir, os.listdir())
for directory in directories:
if directory == "reports":
for artifact_name in os.listdir(directory):
_available_artifacts[artifact_name] = Artifact(artifact_name)
_available_artifacts[artifact_name].add_path(os.path.join(directory, artifact_name))
return _available_artifacts