-
-
Save gbaman/b3137e18c739e0cf98539bf4ec4366ad to your computer and use it in GitHub Desktop.
# An example to get the remaining rate limit using the Github GraphQL API. | |
import requests | |
headers = {"Authorization": "Bearer YOUR API KEY"} | |
def run_query(query): # A simple function to use requests.post to make the API call. Note the json= section. | |
request = requests.post('https://api.github.com/graphql', json={'query': query}, headers=headers) | |
if request.status_code == 200: | |
return request.json() | |
else: | |
raise Exception("Query failed to run by returning code of {}. {}".format(request.status_code, query)) | |
# The GraphQL query (with a few aditional bits included) itself defined as a multi-line string. | |
query = """ | |
{ | |
viewer { | |
login | |
} | |
rateLimit { | |
limit | |
cost | |
remaining | |
resetAt | |
} | |
} | |
""" | |
result = run_query(query) # Execute the query | |
remaining_rate_limit = result["data"]["rateLimit"]["remaining"] # Drill down the dictionary | |
print("Remaining rate limit - {}".format(remaining_rate_limit)) |
Easy to read, flexible and still works in 2021, thanks for this!!!!
thanks!
How do I paginate thru results in python code?
How do I paginate thru results in python code?
Hello!!
Hopefully my solution works for you
I did something like this:
def run_query(): # A simple function to use requests.post to make the API call. Note the json= section.
end_cursor = ""
users_list = []
while True:
has_next_page = False
query = """
{
organization(login:"[YOUR_ORG]") {
samlIdentityProvider{
externalIdentities(first:100""" + end_cursor + """){
pageInfo {
startCursor
hasNextPage
endCursor
}
nodes{
samlIdentity{
nameId
username
}
user{
login
id
name
email
}
}
}
}
}
}
"""
request = requests.post('https://api.github.com/graphql', json={'query': query}, headers=headers)
if request.status_code == 200:
result = request.json()
has_next_page = (result["data"]["organization"]
['samlIdentityProvider']
['externalIdentities']['pageInfo']
['hasNextPage'])
new_cursor = (result["data"]["organization"]
['samlIdentityProvider']
['externalIdentities']['pageInfo']
['endCursor'])
result_data = (result["data"]["organization"]
['samlIdentityProvider']
['externalIdentities']['nodes'])
if has_next_page:
end_cursor = ', after:"' + new_cursor + '"'
else:
break
else:
raise Exception("Query failed to run by returning code of {}. {}".format(request.status_code, query))
return
To pass in variables to the query you need to:
- add
variables
parameter in yourrun_query
function.- add
variables
key to the json you that is going to be appended to the graphql endpoint- create the variables dictionary with the parameters you want to pass
my python is a bit rusty but this works
import requests class graphQL: headers = {"Authorization": "token <token_here>"} """docstring for graphQL""" def __init__(self): super(graphQL, self).__init__() def run_query(self, query, variables): # A simple function to use requests.post to make the API call. Note the json= section. request = requests.post('https://api.github.com/graphql', json={'query': query, 'variables': variables}, headers=self.headers) if request.status_code == 200: return request.json() else: raise Exception("Query failed to run by returning code of {}. {}".format(request.status_code, query)) def getClosedIssuesActors(self): listOfNames = [] query = ''' query($owner: String!, $name: String!) { repository(owner: $owner, name: $name){ issues(states: CLOSED, first:10){ edges{ node{ ... on Issue{ timeline(last: 100){ edges{ node{ __typename ... on ClosedEvent{ actor{ login } } } } } } } } } } }''' variables = { "owner": "tatmush", "name": "Saturday-THORN-Dev-Rank" } result = self.run_query(query, variables) #execute query print(result)
don't forget to declare the variables in the query itself.
This approach saved a lot of time for me. thank you for sharing this
query = """ { rateLimit { limit cost remaining resetAt } } """
is not necessary. the ratelimit status is sent in every response.headers
import requests
from datetime import datetime, timezone
import time
github_token = 'ghp_xxxxxxxxxxxxxxxxxxxx'
real_requests_post = requests.post
def wrap_requests_post(*args, **kwargs):
if not 'headers' in kwargs:
kwargs['headers'] = {}
kwargs['headers']['Authorization'] = 'token ' + github_token
response = real_requests_post(*args, **kwargs)
if 'x-ratelimit-used' in response.headers._store:
print("ratelimit status: used %s of %s. next reset in %s minutes" % (
response.headers['X-RateLimit-Used'],
response.headers['X-RateLimit-Limit'],
datetime.fromtimestamp(int(response.headers["X-RateLimit-Reset"]) - time.time(), tz=timezone.utc).strftime("%M:%S")
))
return response
requests.post = wrap_requests_post
query = """
{ viewer { starredRepositories(first: 5) {
nodes { name nameWithOwner sshUrl description }
pageInfo { endCursor hasNextPage }
} } }
"""
response = requests.post('https://api.github.com/graphql', json={'query': query})
data = response.json()
sample output
ratelimit status: used 8 of 5000. next reset in 43:57 minutes
For anyone trying to pass variables for these queries, I suggest looking at string.Template in python:
from string Import Template queryTemplate = Template( """{ viewer { repositories(first: $num) { pageInfo { hasNextPage endCursor } edges { node { name } } } } }""" ) query = queryTemplate.substitute(num=n)
This worked well. thx!
Is there a way to search for all issues across an org? Using REST I would do https://api.github.com/search/issues?q=user:{org_name}&per_page=100&page=1
Any help reformatting this for Graphql would be appreciated.
@els-pnw I was trying to find a way of doing it with graphql explorer but got stuck for this error
"message": "Although you appear to have the correct authorization credentials, the
kubernetes
organization has enabled OAuth App access restrictions, meaning that data access to third-parties is limited. For more information on these restrictions, including how to enable this app, visit https://docs.github.com/articles/restricting-access-to-your-organization-s-data/"
I ended up with something like this
query ($org_name: String!) {
organization(login: $org_name) {
repositories(last: 3) {
nodes {
issues(last: 2) {
nodes {
number
assignees {
edges {
node {
name
}
}
}
author {
login
}
state
}
}
}
}
}
}
Not sure how much it will help you, again the explorer is nice, just play with it, hope you will get your answer.
so cool, exactly what I need, thank you!