Allow DataSelectors in PropsDB getindex

legend-exp · Oct 2, 2023 · 16f49a7 · 16f49a7
1 parent 1a6802c
commit 16f49a7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/props_db.jl b/src/props_db.jl
@@ -207,10 +207,18 @@ end
 (@nospecialize(pd::PropsDB{Nothing}))(filekey::FileKey) = pd(ValiditySelection(filekey))
 
 
+function Base.getindex(@nospecialize(pd::PropsDB), a, b, cs...)
+    getindex(getindex(pd, a), b, cs...)
+end
+
 function Base.getindex(@nospecialize(pd::PropsDB), s::Symbol)
     _get_md_property(pd, s)
 end
 
+function Base.getindex(@nospecialize(pd::PropsDB), s::DataSelector)
+    getindex(pd, Symbol(string(s)))
+end
+
 function Base.getindex(@nospecialize(pd::PropsDB), S::AbstractArray{<:Symbol})
     _get_md_property.(Ref(pd), S)
 end