How do I cleanup the widget after it was deleted?

[roundedrectangle]

In on_mount i call a worker function (with work decorator). It creates an instance of an aiohttp.Session object. It is required to call await session.close() on it to properly delete it. Where would I put this code in Textual? I need the session to be alive while the widget is active since there are buttons in my UI which utilize it when pressed. I cannot recreate the session every time either because I need cookies to persist, and re-creating the session while storing cookies in a separate container would be impractical.

Here is an example:

from textual import work
from textual.app import App
from textual.screen import Screen

from textual.widgets import Static, Footer

import aiohttp

class MyScreen(Screen):
    def on_mount(self):
        self.init()
    @work
    async def init(self):
        self.session = aiohttp.ClientSession()
    
    def compose(self):
        yield Static('Now try quitting with ctrl+q and see the "Unclosed client session" warning')

class MyApp(App):
    BINDINGS = [('a','push_screen("screen1")', "Open screen1")]
    SCREENS = {
        'screen1': MyScreen,
    }

    def compose(self):
        yield Static("Press a")
        yield Footer()


if __name__ == "__main__":
    MyApp().run()

[roundedrectangle]

Ok, nevermind. I fixed this for switching screens by using on_screen_resume and on_screen_suspend events, and for quitting I simply did this in the on_quit function:

1. check if the current screen is the one which needs to be cleaned up, and if no just exit and do nothing else
2. otherwise, run a special worker on the screen instance which in the end calls the app.exit function
3. if the user presses ctrl+q again while worker is running, app cancels the worker and exits immediately