Added support for uploading to only public repositories.

jackton1 · jackton1 · commit 93014c0396c6 · 2021-07-30T06:06:31.000-04:00
diff --git a/github_deploy/commands/_utils.py b/github_deploy/commands/_utils.py
@@ -1,2 +1,6 @@
 def get_repo(*, org, project):
     return "{org}/{project}".format(project=project, org=org)
+
+
+def can_upload(*, repo, include_private):
+    return True if include_private and repo['private'] == True else not repo['private']
diff --git a/github_deploy/commands/upload.py b/github_deploy/commands/upload.py
@@ -8,7 +8,7 @@
 import certifi
 
 from github_deploy.commands._constants import BASE_URL, REPOS_URL
-from github_deploy.commands._utils import get_repo
+from github_deploy.commands._utils import get_repo, can_upload
 
 
 async def get(*, session, url, headers=None, skip_missing=False):
@@ -199,7 +199,13 @@ async def list_repos(*, session, org, token):
     help="Overwrite existing files.",
     default=False,
 )
-async def main(org, token, source, dest, overwrite):
+@click.option(
+    "--private/--no-private",
+    prompt=click.style("Should we Include private repositories", bold=True),
+    help="Upload files to private repositories.",
+    default=True,
+)
+async def main(org, token, source, dest, overwrite, private):
     """Upload a file to all repositories owned by an organization/user."""
     # create instance of Semaphore: max concurrent requests.
     semaphore = asyncio.Semaphore(1000)
@@ -209,13 +215,14 @@ async def main(org, token, source, dest, overwrite):
     async with aiohttp.ClientSession() as session:
         response = await list_repos(org=org, token=token, session=session)
         repos = [
-            get_repo(org=org, project=v["name"])
-            for v in response["items"]
-            if not v["archived"]
+            get_repo(org=org, project=r["name"])
+            for r in response["items"]
+            if not r["archived"] and can_upload(repo=r, include_private=private)
         ]
+        repo_type = 'public and private' if private else 'public'
         click.echo(
             click.style(
-                "Found '{}' repositories non archived repositories:".format(len(repos)),
+                "Found '{}' repositories non archived {} repositories:".format(len(repos), repo_type),
                 fg="green",
             )
         )
diff --git a/github_deploy/main.py b/github_deploy/main.py
@@ -5,15 +5,15 @@
 
 
 class GithubDeploy(click.MultiCommand):
-    
+
     def list_commands(self, ctx):
         rv = []
         for filename in os.listdir(plugin_folder):
             if filename.endswith('.py') and not filename.startswith('__init__') and not filename.startswith('_'):
                 rv.append(filename[:-3])
         rv.sort()
         return rv
-    
+
     def get_command(self, ctx, name):
         ns = {}
         fn = os.path.join(plugin_folder, name + '.py')
