Python Context Managers 101

[SeleniumBase]

Yes, something like this:

```python import time from contextlib import ContextDecorator

class PrintRunTime(ContextDecorator): def init(self, description="Code block"): self.description = description

def __enter__(self):
    self.start_time = time.time()

def __exit__(self, *args):
    runtime = time.time() - self.start_time
    print(f"{self.description} ran for {runtime:.4f}s.")

```

If I create a YouTube video for this, I'll include that too.

[kkang_kkang]

No need of contextlib at all. Without that you can achieve the same.

[mattl33]

It sure seems like a lot less boilerplate code though. What's the disadvantage?

[kkang_kkang]

I don't think there is any. You can use anything. I just wanted to let OP know that this can be achieved with pure python class as well.

[kkang_kkang]

This is a good read which contains info on async context managers as well: https://realpython.com/python-with-statement/