Toolshed Repo creation command

planemo/commands/cmd_shed_repo_create.py

@@ -0,0 +1,77 @@
+"""
+"""
+import click
+
+from planemo.cli import pass_context
+from planemo import options
+from planemo import shed
+from planemo.io import error
+from planemo.io import info
+from planemo.io import shell
+import json
+
+# TODO: Implement alternative tool per repo upload strategy.
+# TODO: Use git commit hash and origin to generated commit message.
+@click.command("shed_repo_create")
+@options.optional_project_arg(exists=True)
+@options.shed_owner_option()
+@options.shed_name_option()
+@options.shed_target_option()
+@click.option(
+    '--shed_key',
+    help="API key for Tool Shed access (required unless e-mail/pass "
+         "specified)."
+)
+@click.option(
+    '--shed_email',
+    help="E-mail for Tool Shed auth (required unless shed_key is "
+         "specified)."
+)
+@click.option(
+    '--shed_password',
+    help="Password for Tool Shed auth (required unless shed_key is "
+         "specified)."
+)
+@pass_context
+def cli(ctx, path, **kwds):
+    """Create a repository in the toolshed from a .shed.yml file
+    """
+    tsi = shed.tool_shed_client(ctx, **kwds)
+    repo_id = __find_repository(ctx, tsi, path, **kwds)
+    if repo_id is None:
+        __create_repository(ctx, tsi, path, **kwds)
+        info("Repository created")
+
+
+def __find_repository(ctx, tsi, path, **kwds):
+    """More advanced error handling for finding a repository by ID
+    """
+    try:
+        repo_id = shed.find_repository_id(ctx, tsi, path, **kwds)
+        return repo_id
+    except Exception as e:
+        error("Could not update %s" % path)
+        try:
+            error(e.read())
+        except AttributeError:
+            # I've seen a case where the error couldn't be read, so now
+            # wrapped in try/except
+            error("Could not find repository in toolshed")
+    return None
+
+
+def __create_repository(ctx, tsi, path, **kwds):
+    """Wrapper for creating the endpoint if it doesn't exist
+    """
+    try:
+        repo = shed.create_repository(ctx, tsi, path, **kwds)
+        return repo['id']
+    # Have to catch missing snyopsis/bioblend exceptions
+    except Exception as e:
+        # TODO: galaxyproject/bioblend#126
+        try:
+            upstream_error = json.loads(e.read())
+            error(upstream_error['err_msg'])
+        except Exception as e2:
+            error(str(e))
+        return None