Add json output flag (#139)

This commit is contained in:
Dave G
2022-07-05 00:11:00 -04:00
committed by GitHub
parent 3df07dc65e
commit 260ef74d4e
7 changed files with 148 additions and 57 deletions

View File

@@ -1,5 +1,6 @@
# pylint: disable=old-style-class
import os
import json
from colorama import Fore, Style
from .scores import get_vote_color
@@ -11,6 +12,17 @@ class Post:
self.user = user
class PostEncoder(json.JSONEncoder):
def default(self, o):
if isinstance(o, Post):
return dict(
body=o.body,
score=o.score,
user=o.user,
)
return json.JSONEncoder.default(self, o)
def get_terminal_width():
_, columns = os.popen("stty size", "r").read().split()
return int(columns)