def main_menu(self): geo=None ids=[] print "-----------------Menu Instagram-------------------" main_menu={1:"Buscar",2:"Cargar Red Desde Json"} user_input=Dock.display_menu(main_menu) if main_menu[user_input]=="Buscar": insta_results,insta_friends_json,geo= self.buscar() elif main_menu[user_input]=="Cargar Red Desde Json": data,proyecto= self.load() if geo: self.process(insta_results,insta_friends_json) else: self.process(data,proyecto)
def main_menu(self): geo=None ids=[] print "-----------------Menu Twitter-------------------" main_menu={1:"Buscar",2:"Cargar Red Desde Json"} user_input=Dock.display_menu(main_menu) if main_menu[user_input]=="Buscar": data,proyecto,geo= self.buscar() elif main_menu[user_input]=="Cargar Red Desde Json": data,proyecto= self.load() if geo: self.process(data,proyecto,geo) else: self.process(data,proyecto)
def cargar_json(files): count=0 for dirname, dirnames, filenames in os.walk(path+'/json'): for subdirname in dirnames: print os.path.join(dirname, subdirname) for filename in filenames: if filename != ".DS_Store" and "users_" not in filename: count+=1 files[count]=os.path.join(dirname, filename) proyecto=raw_input("\nComo se llama tu proyecto?") print "escoger todos los archivos que quiera cargar separados por coma" user_input=Dock.display_menu(files,output=2) print user_input input_split=user_input.split(",") paths=[] for item in input_split: paths.append(files[int(item)]) loaded_data=Dock.combine_json(paths)
except twitter.TwitterError as e: print "solo se puede conseguir informacion de usuarios si se carga un json" graph=create_graph(loaded_data) Dock.json_csv_xlwt(loaded_data,"tweet",proyecto,csv_path) Dock.json_csv_xlwt(count_json,"freq_count",proyecto,csv_path) if geo != "": gexf_filename='search_results_'+proyecto+'_'+geo+'.gexf' save_to_gexf(graph,gexf_path+'/'+gexf_filename) else: gexf_filename='search_results_'+proyecto+'.gexf' save_to_gexf(graph,gexf_path+'/'+gexf_filename) if __name__ == '__main__': Dock.create_folders(os.path.dirname(sys.argv[0])) main_menu={1:"Twitter",2:"Instagram"} user_input=Dock.display_menu(main_menu) if main_menu[user_input] is 'Twitter': tweet_menu=Twitter_menu() tweet_menu.main_menu() elif main_menu[user_input] is 'Instagram': insta_menu=Instagram_menu() insta_menu.main_menu() print "done"