From 060db6c242024314a6a728ef77672cfcf897a073 Mon Sep 17 00:00:00 2001 From: kthouky Date: Wed, 9 Aug 2023 15:10:52 +0200 Subject: [PATCH] fix: enum --- pythonseer/types/format.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pythonseer/types/format.py b/pythonseer/types/format.py index 238dee6..dd97c08 100644 --- a/pythonseer/types/format.py +++ b/pythonseer/types/format.py @@ -6,11 +6,11 @@ class FormatType(Enum): LIST = auto() CSV = auto() - @classmethod - def get_query(cls, query_join=''): - if cls == cls.FULL: + def get_query(self, query_join=''): + print(self) + if self == FormatType.FULL: return "" - elif cls == cls.LIST: + elif self == FormatType.LIST: return f"{query_join}domains=true" - elif cls == cls.CSV: + elif self == FormatType.CSV: return f"{query_join}csv=true"