From a16238edda4626622c2f023fbe1888388b05b73d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= Date: Wed, 17 Sep 2025 09:05:44 +0200 Subject: [PATCH] [util] add __enter__ & __exit__ methods to NullResponse (#8227) --- gallery_dl/util.py | 6 ++++++ test/test_util.py | 3 +++ 2 files changed, 9 insertions(+) diff --git a/gallery_dl/util.py b/gallery_dl/util.py index 5da89444..49c1ba8e 100644 --- a/gallery_dl/util.py +++ b/gallery_dl/util.py @@ -641,6 +641,12 @@ class NullResponse(): self.url = url self.reason = str(reason) + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_value, traceback): + pass + def __str__(self): return "900 " + self.reason diff --git a/test/test_util.py b/test/test_util.py index 4a767690..bfaab01b 100644 --- a/test/test_util.py +++ b/test/test_util.py @@ -1041,6 +1041,9 @@ value = 123 self.assertEqual(response.links.get("next"), None) self.assertEqual(response.close(), None) + with response as ctx: + self.assertIs(response, ctx) + class TestExtractor(): category = "test_category"