Skip to content

Commit 598f5f1

Browse files
committed
Add click to call data gen script.
1 parent e091d61 commit 598f5f1

1 file changed

Lines changed: 17 additions & 8 deletions

File tree

query.py

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os
22
import requests
33
import json
4+
import click
45

56
token = os.environ["GRAPH_API_KEY"]
67
endpoint = 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()

0 commit comments

Comments
 (0)