From 2f5b47c8b30d1b7d86443391332be2f3805cdafd Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Tue, 8 Feb 2022 20:21:02 -0800 Subject: Cleanup more contextlib.AbstractContextManagers and Literal[False]s. --- text_utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'text_utils.py') diff --git a/text_utils.py b/text_utils.py index 4384a1e..7910990 100644 --- a/text_utils.py +++ b/text_utils.py @@ -3,11 +3,12 @@ """Utilities for dealing with "text".""" +import contextlib import logging import math import sys from collections import defaultdict -from typing import Dict, Generator, List, NamedTuple, Optional, Tuple +from typing import Dict, Generator, List, Literal, NamedTuple, Optional, Tuple from ansi import fg, reset @@ -261,7 +262,7 @@ def wrap_string(text: str, n: int) -> str: return out -class Indenter(object): +class Indenter(contextlib.AbstractContextManager): """ with Indenter(pad_count = 8) as i: i.print('test') @@ -289,10 +290,11 @@ class Indenter(object): self.level += 1 return self - def __exit__(self, exc_type, exc_value, exc_tb): + def __exit__(self, exc_type, exc_value, exc_tb) -> Literal[False]: self.level -= 1 if self.level < -1: self.level = -1 + return False def print(self, *arg, **kwargs): import string_utils -- cgit v1.3