Feat: add listing fluxes
This commit is contained in:
parent
2a387a1bc8
commit
8882317a47
@ -3,6 +3,7 @@ from datetime import datetime
|
|||||||
import csv
|
import csv
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
from plesna.libs.string_tools import StringToolsError, extract_values_from_pattern
|
||||||
from plesna.storage.metadata_repository.metadata_repository import (
|
from plesna.storage.metadata_repository.metadata_repository import (
|
||||||
ExecutionLog,
|
ExecutionLog,
|
||||||
MetaDataRepository,
|
MetaDataRepository,
|
||||||
@ -34,7 +35,15 @@ class FSMetaDataRepository(MetaDataRepository):
|
|||||||
|
|
||||||
def fluxes(self) -> list[str]:
|
def fluxes(self) -> list[str]:
|
||||||
"""List fluxes's ids"""
|
"""List fluxes's ids"""
|
||||||
raise NotImplementedError
|
fluxes = []
|
||||||
|
for filepath in self._basepath.iterdir():
|
||||||
|
try:
|
||||||
|
founded = extract_values_from_pattern(self.FILEMODEL["execution"], filepath.name)
|
||||||
|
except StringToolsError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
fluxes.append(founded["flux_id"])
|
||||||
|
return fluxes
|
||||||
|
|
||||||
def add_flux(self, flux_id: str) -> str:
|
def add_flux(self, flux_id: str) -> str:
|
||||||
"""Get the flux metadata"""
|
"""Get the flux metadata"""
|
||||||
|
@ -114,3 +114,10 @@ def test_register_and_get_last_log(metadata_repository):
|
|||||||
datetime=datetime(2024, 3, 15, 14, 30),
|
datetime=datetime(2024, 3, 15, 14, 30),
|
||||||
output=FluxMetaData(data={"truc": "chose"}),
|
output=FluxMetaData(data={"truc": "chose"}),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_register_and_list_fluxes(metadata_repository):
|
||||||
|
flux_ids = ["my_flux", "flux2", "blahblah"]
|
||||||
|
for f in flux_ids:
|
||||||
|
metadata_repository.add_flux(f)
|
||||||
|
assert metadata_repository.fluxes() == flux_ids
|
||||||
|
Loading…
Reference in New Issue
Block a user