Skip to content
Snippets Groups Projects

Core sampler: don't leave a DB connection out there flapping in the breeze

Merged Janet Goldstein requested to merge core-sampler-close-hanging-conn into main
1 file
+ 10
5
Compare changes
  • Side-by-side
  • Inline
@@ -188,11 +188,16 @@ def main():
action="store",
)
ns = parser.parse_args()
if ns.project_code:
CoreSampler(MDDBConnector()).sample_project(ns.project_code[0])
elif ns.sdm_name:
CoreSampler(MDDBConnector()).sample_eb(ns.sdm_name[0])
conn = MDDBConnector()
sampler = CoreSampler(conn)
try:
if ns.project_code:
sampler.sample_project(ns.project_code[0])
elif ns.sdm_name:
sampler.sample_eb(ns.sdm_name[0])
finally:
conn.close()
return 0
Loading