From b604a66731ec21a33bd28ad92114c1c4a8b6755c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jonah=20Br=C3=BCchert?= Date: Tue, 16 Apr 2024 18:02:46 +0200 Subject: Add type hints --- hostinfo/prefix.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'hostinfo/prefix.py') diff --git a/hostinfo/prefix.py b/hostinfo/prefix.py index 8271d67..7233c48 100644 --- a/hostinfo/prefix.py +++ b/hostinfo/prefix.py @@ -1,9 +1,12 @@ # -*- coding: utf-8 -*- +class Flags: + oneline: bool = False + nospaces: bool = False class Printer: - flags = list() + flags = Flags() def __init__(self, full_key='', printer=None): if printer is None: @@ -20,7 +23,7 @@ class Printer: def set_label(self, label='', maxlength=0): self.label = self._get_label(label, maxlength) - def _get_label(self, label, maxlength): + def _get_label(self, label: str, maxlength: int): if label == '': return label @@ -33,7 +36,7 @@ class Printer: else: return label.ljust(maxlength+2) - def pprint(self, data): + def pprint(self, data: str): self.output("%s%s" % (self.label, data)) self.has_output = True if not self.empty: -- cgit v1.2.3-1-g7c22