fix: shutdown application on exit

ApeWorX · Nov 13, 2023 · c6b632b · c6b632b
1 parent df0e0d4
commit c6b632b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/silverback/application.py b/silverback/application.py
@@ -1,3 +1,4 @@
+import asyncio
 import atexit
 from datetime import timedelta
 from typing import Callable, Dict, Optional, Union
@@ -56,6 +57,10 @@ def __init__(self, settings: Optional[Settings] = None):
         # NOTE: This allows using connected ape methods e.g. `Contract`
         provider = self.network.__enter__()
 
+        def shutdown():
+            asyncio.run(self.shutdown())
+
+        atexit.register(shutdown)
         atexit.register(self.network.__exit__)
 
         self.signer = settings.get_signer()