File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11import os
22import requests
33import json
4+ import click
45
56token = os .environ ["GRAPH_API_KEY" ]
67endpoint = r"https://api.github.com/graphql"
@@ -180,21 +181,29 @@ def dump(self, outfile):
180181 json .dump (self .raw_data , outf )
181182
182183
183- if __name__ == "__main__" :
184- repo = "networkx"
184+ @click .command ()
185+ @click .argument ('repo_owner' )
186+ @click .argument ('repo_name' )
187+ def main (repo_owner , repo_name ):
188+ """Download and save issue and pr data for `repo_owner`/`repo_name`."""
185189 issues = GithubGrabber (
186190 'query_examples/issue_activity_since_date.gql' ,
187191 'issues' ,
188- repo_owner = repo ,
189- repo_name = repo ,
192+ repo_owner = repo_owner ,
193+ repo_name = repo_name ,
190194 )
191195 issues .get ()
192- issues .dump (f"_data/{ repo } _issues.json" )
196+ issues .dump (f"_data/{ repo_name } _issues.json" )
193197 prs = GithubGrabber (
194198 'query_examples/pr_data_query.gql' ,
195199 'pullRequests' ,
196- repo_owner = repo ,
197- repo_name = repo ,
200+ repo_owner = repo_owner ,
201+ repo_name = repo_name ,
198202 )
199203 prs .get ()
200- prs .dump (f"_data/{ repo } _prs.json" )
204+ prs .dump (f"_data/{ repo_name } _prs.json" )
205+
206+
207+
208+ if __name__ == "__main__" :
209+ main ()
You can’t perform that action at this time.
0 commit comments