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"
