added method to represent tweet as json

tweet_model/tweet.py

 # -*- coding: utf-8 -*-
 """Main module."""
 
+import json
+import pysnooper
+
+from typing import List, Dict
+
 
 class Tweet():
     """
@@ -366,5 +371,30 @@ class Tweet():
     def set_trtext(self, trtext):
         self.trtext = trtext
 
+    def get_tweet_fields_subset(self, fields: List[str]) -> Dict:
+        """
+        Keep just the specified fields and return a dict
+        with just the information specified
+        """
+
+        tweet_subset = {}
+        for field in fields:
+            try:
+                tweet_subset[field] = self[field]
+            except AttributeError:
+                pass
+        return tweet_subset
+
+    def as_json(self) -> Dict:
+        """
+        Return the Tweet object in a JSON-like representation (nested dicts)
+        """
+
+        json_tweet = {}
+        for key, value in self.__dict__.items():
+            if value is not None:
+                json_tweet[key] = value
+        return json_tweet
+
     def __getitem__(self, key):
         return getattr(self, key)

tweet_model/utils.py

@@ -92,24 +92,6 @@ def get_tweets_from_csv(csv_file):
     return tweets
 
 
-def get_tweet_fields_subset(
-        tweet: Tweet,
-        fields: List[str]
-        ) -> Dict:
-    """
-    Given a Tweet objects, keep just the specified fields and return a dict
-    with just the information specified
-    """
-
-    tweet_subset = {}
-    for field in fields:
-        try:
-            tweet_subset[field] = tweet[field]
-        except AttributeError:
-            pass
-    return tweet_subset
-
-
 def get_tweet_collection_fields_subset(
         tweet_collection: Union[List[Tweet], Generator[Tweet, None, None]],
         fields: List[str]
@@ -119,7 +101,7 @@ def get_tweet_collection_fields_subset(
     return a generator of dicts with just the information specified
     """
     for tweet in tweet_collection:
-        yield get_tweet_fields_subset(tweet, fields)
+        yield tweet.get_tweet_fields_subset(fields)
 
 
 class NotValidTweetError(Exception):