aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/vk/mutuals.py
diff options
context:
space:
mode:
Diffstat (limited to 'vk/mutuals.py')
-rw-r--r--vk/mutuals.py11
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):