Skip to content

Commit

Permalink
Merge pull request #105 from pastas/improve_empty_library
Browse files Browse the repository at this point in the history
improve empty library
  • Loading branch information
dbrakenhoff authored Oct 24, 2023
2 parents 2b66f79 + 8d90e16 commit fee2fd4
Showing 1 changed file with 18 additions and 7 deletions.
25 changes: 18 additions & 7 deletions pastastore/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -934,13 +934,24 @@ def empty_library(
)
if ui.lower() != "y":
return
names = self._parse_names(None, libname)
for name in (
tqdm(names, desc=f"Deleting items from {libname}") if progressbar else names
):
self._del_item(libname, name)
self._clear_cache(libname)
print(f"Emptied library {libname} in {self.name}: " f"{self.__class__}")

if libname == "models":
# also delete linked modelnames linked to oseries
libs = ["models", "oseries_models"]
else:
libs = [libname]

# delete items and clear caches
for libname in libs:
names = self._parse_names(None, libname)
for name in (
tqdm(names, desc=f"Deleting items from {libname}")
if progressbar
else names
):
self._del_item(libname, name)
self._clear_cache(libname)
print(f"Emptied library {libname} in {self.name}: " f"{self.__class__}")

def _iter_series(self, libname: str, names: Optional[List[str]] = None):
"""Internal method iterate over time series in library.
Expand Down

0 comments on commit fee2fd4

Please sign in to comment.