-
Notifications
You must be signed in to change notification settings - Fork 0
/
github_connector.py
62 lines (45 loc) · 2.12 KB
/
github_connector.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
#%%writefile /content/src/github_connector.py
import streamlit as st
from streamlit.connections import ExperimentalBaseConnection
import requests
class GitHubConnection(ExperimentalBaseConnection):
BASE_URL = "https://api.github.com"
def __init__(self, connection_name):
super().__init__(connection_name)
self.token = st.secrets["github"]["token"]
def _connect(self):
return self
def _make_request(self, endpoint):
url = f"{self.BASE_URL}{endpoint}"
headers = {
"Authorization": f"token {self.token}",
"Accept": "application/vnd.github.v3+json"
}
response = requests.get(url, headers=headers)
if response.status_code == 200:
return response.json()
else:
st.error(f"Error {response.status_code}: {response.text}")
return []
def get_user_profile(self, username):
return self._make_request(f"/users/{username}")
def get_user_activity(self, username):
return self._make_request(f"/users/{username}/events/public")
def get_user_repositories(self, username):
return self._make_request(f"/users/{username}/repos")
def get_user_issues(self, username):
return self._make_request(f"/search/issues?q=author:{username}")
def get_user_pull_requests(self, username):
return self._make_request(f"/search/issues?q=author:{username} type:pr")
def get_user_starred_repos(self, username):
return self._make_request(f"/users/{username}/starred")
def get_user_followers(self, username):
return self._make_request(f"/users/{username}/followers")
def get_user_following(self, username):
return self._make_request(f"/users/{username}/following")
def get_user_gists(self, username):
return self._make_request(f"/users/{username}/gists")
def get_user_organizations(self, username):
return self._make_request(f"/users/{username}/orgs")
def get_repo_languages(self, owner, repo):
return self._make_request(f"/repos/{owner}/{repo}/languages")