ProteinStructureStore do syntax

MurrellGroup · Oct 11, 2024 · a19bae0 · a19bae0
1 parent 5693494
commit a19bae0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/store/store.jl b/src/store/store.jl
@@ -92,6 +92,15 @@ end
 Base.show(io::IO, store::ProteinStructureStore) = print(io, "$ProteinStructureStore(\"$(store.filename)\", $(store.mode))")
 Base.show(io::IO, ::MIME"text/plain", store::ProteinStructureStore) = print(io, summary(store))
 
+function ProteinStructureStore(f::Function, args...)
+    store = ProteinStructureStore(args...)
+    try
+        return f(store)
+    finally
+        close(store)
+    end
+end
+
 """
     serialize(filename::AbstractString, structures::AbstractVector{<:ProteinStructure})