From ba908cdd493d049a1089a27d3f5f1d7739a9d3e7 Mon Sep 17 00:00:00 2001
From: Melanie Walsh <melanie.walsh@cornell.edu>
Date: Thu, 12 Mar 2020 16:04:50 -0400
Subject: [PATCH] add option for number of emojis

---
 utils/emojis.py | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/utils/emojis.py b/utils/emojis.py
index dd26163e..69fcb186 100755
--- a/utils/emojis.py
+++ b/utils/emojis.py
@@ -4,14 +4,30 @@
 import json
 import fileinput
 import collections
+import optparse
 
 import emoji
 
+opt_parser = optparse.OptionParser()
+
+opt_parser.add_option(
+    "-n",
+    "--number",
+    dest="number",
+    type="int",
+    default= 10
+)
+options, args = opt_parser.parse_args()
+tweets = args
+
+number_of_emojis = options.number
+tweets = tweets.pop()
+
 counts = collections.Counter()
 
 EMOJI_RE = emoji.get_emoji_regexp()
 
-for line in fileinput.input():
+for line in open(tweets):
     tweet = json.loads(line)
     if 'full_text' in tweet:
         text = tweet['full_text']
@@ -20,5 +36,5 @@
     for char in EMOJI_RE.findall(text):
         counts[char] += 1
 
-for char, count in counts.most_common():
+for char, count in counts.most_common(number_of_emojis):
     print("%s %5i" % (char, count))