if len(sys.argv) < 2:
print Colors.FAIL + "Please enter a hero name. Eg. 'Faceless Void'" + Colors.ENDC
else:
hero_name = sys.argv[1]
client = MongoClient('localhost', 27017)
print Colors.OKGREEN + "Connected to MongoDB." + Colors.ENDC
db = client.assignment_1
heroes_coll = db.heroes
benchmarks_coll = db.hero_benchmarks
matchups_coll = db.hero_matchups
# time.sleep(5) # hack for the mongoDb database to get running
scrapper = Request()
heroes = scrapper.getHeroes()
get_heroes(db)
selected_hero = get_selected_hero()
if selected_hero != None:
print Colors.OKGREEN + "\nDisplaying Hero Data for " + hero_name + ": " + Colors.ENDC
print pd.DataFrame(selected_hero)
selected_hero_id = selected_hero['id']
get_hero_benchmark()
get_hero_matchups()
