made print statements optional

AADeLucia · Jan 28, 2021 · fb685fd · fb685fd
1 parent 4d8a14c
commit fb685fd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/littlebird/tweet_utils.py b/littlebird/tweet_utils.py
@@ -51,7 +51,8 @@ def __init__(self, filename: str):
     def read_tweets(self,
         skip_deleted: bool = True,
         skip_withheld: bool = True,
-        skip_retweeted_and_quoted: bool = False
+        skip_retweeted_and_quoted: bool = False,
+        print_stats: bool = False
     ) -> Iterable[Any]:
         # Initialize counters
         count_total = 0
@@ -84,7 +85,8 @@ def read_tweets(self,
         # Close file
         self.f.close()
         # Print stats
-        logging.warning(f"""
+        if print_stats:
+            logging.warning(f"""
 Total iterated tweets: {count_total:,}
 Remaining tweets: {count_total - (count_deleted + count_withheld + count_retweeted):,}
 Skipped tweets:

diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setuptools.setup(
     name="littlebird",
-    version="0.0.1",
+    version="0.0.2",
     author="Alexandra DeLucia",
     author_email="aadelcuia@jhu.edu",
     description="Utilities for opening and processing Tweets",