diff options
| author | Egor Tensin <egor@tensin.name> | 2026-10-09 11:35:11 +0500 |
|---|---|---|
| committer | Egor Tensin <egor@tensin.name> | 2026-10-09 11:35:11 +0500 |
| commit | e4af28c4a398cd87378d3833b1d25db0f291ac9e (patch) | |
| tree | 5cbd3dbe7c08a11da0d99c1bc7409bc1bea4c63d /vk/mutuals.py | |
| parent | test: remove matplotlib workaround (diff) | |
| download | vk-scripts-e4af28c4a398cd87378d3833b1d25db0f291ac9e.tar.gz vk-scripts-e4af28c4a398cd87378d3833b1d25db0f291ac9e.zip | |
upgrade to StrEnum
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): |