repos -r option for specific repos, closes #42

Author: simonw

README.md

@@ -91,6 +91,12 @@ You can pass more than one username to fetch for multiple users or organizations
 
     $ github-to-sqlite repos github.db simonw dogsheep
 
+## Fetching specific repositories
+
+You can use `-r` with the `repos` command one or more times to fetch just specific repositories.
+
+    $ github-to-sqlite repos github.db -r simonw/datasette -r dogsheep/github-to-sqlite
+
 ## Fetching repos that have been starred by a user
 
 The `starred` command fetches the repos that have been starred by a user.

github_to_sqlite/cli.py

@@ -172,24 +172,35 @@ def stargazers(db_path, repos, auth):
     default="auth.json",
     help="Path to auth.json token file",
 )
+@click.option(
+    "-r",
+    "--repo",
+    multiple=True,
+    help="Just fetch these repos",
+)
 @click.option(
     "--load",
     type=click.Path(file_okay=True, dir_okay=False, allow_dash=True, exists=True),
     help="Load issues JSON from this file instead of the API",
 )
-def repos(db_path, usernames, auth, load):
+def repos(db_path, usernames, auth, repo, load):
     "Save repos owened by the specified (or authenticated) username or organization"
     db = sqlite_utils.Database(db_path)
     token = load_token(auth)
     if load:
-        for repo in json.load(open(load)):
-            utils.save_repo(db, repo)
+        for loaded_repo in json.load(open(load)):
+            utils.save_repo(db, loaded_repo)
     else:
-        if not usernames:
-            usernames = [None]
-        for username in usernames:
-            for repo in utils.fetch_all_repos(username, token):
-                utils.save_repo(db, repo)
+        if repo:
+            # Just these repos
+            for full_name in repo:
+                utils.save_repo(db, utils.fetch_repo(full_name, token))
+        else:
+            if not usernames:
+                usernames = [None]
+            for username in usernames:
+                for repo in utils.fetch_all_repos(username, token):
+                    utils.save_repo(db, repo)
     utils.ensure_db_shape(db)