-
Notifications
You must be signed in to change notification settings - Fork 15
/
ego-network-2levels-fast.py
76 lines (66 loc) · 2.11 KB
/
ego-network-2levels-fast.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# -*- coding: utf-8 -*-
#
# Ego-network analysis of followers in GitHub, Degree 2
# Fast version (only account name)
#
# Author: Massimo Menichinelli
# Homepage: http://www.openp2pdesign.org
# License: GPL v.3
#
# Requisite:
# install pyGithub with pip install PyGithub
# install NetworkX with pip install networkx
#
# PyGitHub documentation can be found here:
# https://github.com/jacquev6/PyGithub
#
from github import Github
import networkx as nx
import getpass
import os
# Clear screen
os.system('cls' if os.name=='nt' else 'clear')
print "Social Network Analisys of your GitHub network"
print ""
username = raw_input("Enter your username: ")
password = getpass.getpass("Enter yor password: ")
user = raw_input("Enter the username to mine: ")
print ""
g = Github( username, password )
graph = nx.DiGraph()
graph.add_node(user,label=g.get_user(user).name)
print "Looking for the followers of",user,"..."
for f in g.get_user(user).get_followers():
print " -", f.login
graph.add_node(f.login,label=f.login)
graph.add_edge(f.login,user)
print " And his/her followers:"
for i in f.get_followers():
print " --", i.login
graph.add_node(i.login,label=i.login)
graph.add_edge(i.login,f.login)
print " And the users she/he's following:"
for i in f.get_following():
print " --", i.login
graph.add_node(i.login,label=i.login)
graph.add_edge(f.login,i.login)
print "-----"
print "Looking for the users",user,"is following..."
for f in g.get_user(user).get_following():
print " -", f.login
graph.add_node(f.login,label=f.login)
graph.add_edge(user,f.login)
print " And his/her followers:"
for i in f.get_followers():
print " --", i.login
graph.add_node(i.login,label=i.login)
graph.add_edge(i.login,f.login)
print " And the users she/he's following:"
for i in f.get_following():
print " --", i.login
graph.add_node(i.login,label=i.login)
graph.add_edge(f.login,i.login)
print "-----"
print "Saving the network..."
nx.write_gexf(graph, user+"_ego-network_2_levels.gexf")
print "Done."