Reorganized

twint.py

@@ -20,38 +20,7 @@ def error():
 	print("[-] {}: {}".format(error, message))
 	sys.exit(0)
 
-def main():
-	ap = argparse.ArgumentParser(prog="tw.py", usage="python3 %(prog)s [options]", description="tw.py - An Advanced Twitter Scraping Tool")
-	ap.add_argument("-u", "--username", help="User's Tweets you want to scrape.")
-	ap.add_argument("-s", "--search", help="Search for Tweets containing this word or phrase.")
-	ap.add_argument("-g", "--geo", help="Search for geocoded tweets.")
-	ap.add_argument("-l", "--lang", help="Serch for Tweets in a specific language")
-	ap.add_argument("-o", "--output", help="Save output to a file.")
-	ap.add_argument("-es", "--elasticsearch", help="Index to Elasticsearch")
-	ap.add_argument("-t", "--timedelta", help="Time intervall for every request")
-	ap.add_argument("--year", help="Filter Tweets before specified year.")
-	ap.add_argument("--since", help="Filter Tweets sent since date (Example: 2017-12-27).")
-	ap.add_argument("--until", help="Filter Tweets sent until date (Example: 2017-12-27).")
-	ap.add_argument("--fruit", help="Display 'low-hanging-fruit' Tweets.", action="store_true")
-	ap.add_argument("--tweets", help="Display Tweets only.", action="store_true")
-	ap.add_argument("--verified", help="Display Tweets only from verified users (Use with -s).", action="store_true")
-	ap.add_argument("--users", help="Display users only (Use with -s).", action="store_true")
-	ap.add_argument("--csv", help="Write as .csv file.", action="store_true")
-	ap.add_argument("--json", help="Write as .json file.", action="store_true")
-	ap.add_argument("--hashtags", help="Output hashtags in seperate column.", action="store_true")
-	ap.add_argument("--userid", help="Twitter user id")
-	ap.add_argument("--limit", help="Number of Tweets to pull (Increments of 20).")
-	ap.add_argument("--count", help="Display number Tweets scraped at the end of session.", action="store_true")
-	ap.add_argument("--stats", help="Show number of replies, retweets, and likes", action="store_true")
-	ap.add_argument("--database", help="Store tweets in the database")
-	ap.add_argument("--to", help="Search Tweets to a user")
-	ap.add_argument("--all", help="Search all Tweets associated with a user")
-	ap.add_argument("--followers", help="Scrape a person's followers", action="store_true")
-	ap.add_argument("--following", help="Scrape who a person follows.", action="store_true")
-	ap.add_argument("--favorites", help="Scrape Tweets a user has liked.", action="store_true")
-	ap.add_argument("--debug", help="Debug mode", action="store_true")
-	args = ap.parse_args()
-	
+def check(args):
 	if args.username is not None:
 		if args.users:
 			error("Contradicting Args", "Please use --users in combination with -s.")
@@ -64,6 +33,7 @@ def main():
 	if args.csv and args.output is None:
 		error("Error", "Please specify an output file (Example: -o file.csv).")
 
+def initialize(args):
 	c = twint.Config()
 	c.Username = args.username
 	c.User_id = args.userid
@@ -90,6 +60,45 @@ def main():
 	c.To = args.to
 	c.All = args.all
 	c.Debug = args.debug
+	return c
+
+def options():
+	ap = argparse.ArgumentParser(prog="tw.py", usage="python3 %(prog)s [options]", description="tw.py - An Advanced Twitter Scraping Tool")
+	ap.add_argument("-u", "--username", help="User's Tweets you want to scrape.")
+	ap.add_argument("-s", "--search", help="Search for Tweets containing this word or phrase.")
+	ap.add_argument("-g", "--geo", help="Search for geocoded tweets.")
+	ap.add_argument("-l", "--lang", help="Serch for Tweets in a specific language")
+	ap.add_argument("-o", "--output", help="Save output to a file.")
+	ap.add_argument("-es", "--elasticsearch", help="Index to Elasticsearch")
+	ap.add_argument("-t", "--timedelta", help="Time intervall for every request")
+	ap.add_argument("--year", help="Filter Tweets before specified year.")
+	ap.add_argument("--since", help="Filter Tweets sent since date (Example: 2017-12-27).")
+	ap.add_argument("--until", help="Filter Tweets sent until date (Example: 2017-12-27).")
+	ap.add_argument("--fruit", help="Display 'low-hanging-fruit' Tweets.", action="store_true")
+	ap.add_argument("--tweets", help="Display Tweets only.", action="store_true")
+	ap.add_argument("--verified", help="Display Tweets only from verified users (Use with -s).", action="store_true")
+	ap.add_argument("--users", help="Display users only (Use with -s).", action="store_true")
+	ap.add_argument("--csv", help="Write as .csv file.", action="store_true")
+	ap.add_argument("--json", help="Write as .json file.", action="store_true")
+	ap.add_argument("--hashtags", help="Output hashtags in seperate column.", action="store_true")
+	ap.add_argument("--userid", help="Twitter user id")
+	ap.add_argument("--limit", help="Number of Tweets to pull (Increments of 20).")
+	ap.add_argument("--count", help="Display number Tweets scraped at the end of session.", action="store_true")
+	ap.add_argument("--stats", help="Show number of replies, retweets, and likes", action="store_true")
+	ap.add_argument("--database", help="Store tweets in the database")
+	ap.add_argument("--to", help="Search Tweets to a user")
+	ap.add_argument("--all", help="Search all Tweets associated with a user")
+	ap.add_argument("--followers", help="Scrape a person's followers", action="store_true")
+	ap.add_argument("--following", help="Scrape who a person follows.", action="store_true")
+	ap.add_argument("--favorites", help="Scrape Tweets a user has liked.", action="store_true")
+	ap.add_argument("--debug", help="Debug mode", action="store_true")
+	args = ap.parse_args()
+	return args
+
+def main():
+	args = options()
+	check(args)
+	c = initialize(args)
 
 	if args.favorites:
 		twint.Favorites(c)