diff options
Diffstat (limited to 'vk/mutuals.py')
| -rw-r--r-- | vk/mutuals.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/vk/mutuals.py b/vk/mutuals.py index 5f4adb4..d291b28 100644 --- a/vk/mutuals.py +++ b/vk/mutuals.py @@ -6,7 +6,7 @@ import abc import argparse from collections import OrderedDict -from enum import Enum +from enum import auto, StrEnum import sys from vk.api import API @@ -56,12 +56,9 @@ class OutputSinkJSON(OutputSinkMutualFriends): self._writer.write(friend_list) -class OutputFormat(Enum): - CSV = "csv" - JSON = "json" - - def __str__(self): - return self.value +class OutputFormat(StrEnum): + CSV = auto() + JSON = auto() @staticmethod def open_file(path=None): |
