adding get_url operator

Author: brimoor

plugins/io/README.md

@@ -158,3 +158,25 @@ dataset_or_view.draw_labels(
 where the operator's form allows you to configure the output directory on disk,
 the label field(s) to render, and any other optional arguments for
 `draw_labels()`.
+
+### get_url (Enterprise-only)
+
+[FiftyOne Enterprise](https://docs.voxel51.com/enterprise/index.html) users can
+use this operator to retrieve signed URLs to read/write/delete assets in cloud
+storage.
+
+This operator is essentially a wrapper around the
+`fiftyone.core.storage.get_url()` method:
+
+```py
+import fiftyone.core.storage as fos
+
+path = "s3://..."
+method = "GET"  # GET|PUT|DELETE
+hours = 24
+
+url = fos.get_url(path, method=method, hours=hours)
+```
+
+where the operator's form allows you to select the path, HTTP verb, and desired
+expiration time of the URL.

plugins/io/__init__.py

@@ -2795,6 +2795,94 @@ def _draw_labels_inputs(ctx, inputs):
     return True
 
 
+class GetURL(foo.Operator):
+    @property
+    def config(self):
+        return foo.OperatorConfig(
+            name="get_url",
+            label="Get URL",
+            light_icon="/assets/icon-light.svg",
+            dark_icon="/assets/icon-dark.svg",
+            allow_delegated_execution=False,
+            allow_immediate_execution=True,
+            default_choice_to_delegated=False,
+            dynamic=True,
+        )
+
+    def resolve_input(self, ctx):
+        inputs = types.Object()
+
+        _get_url_inputs(ctx, inputs)
+
+        return types.Property(inputs, view=types.View(label="Get URL"))
+
+    def execute(self, ctx):
+        filepath = _parse_path(ctx, "filepath")
+        hours = ctx.params.get("hours", 24)
+        method = ctx.params.get("method", "GET")
+
+        # pylint: disable=no-member
+        url = fos.get_url(filepath, hours=hours, method=method)
+
+        return {"url": url}
+
+    def resolve_output(self, ctx):
+        outputs = types.Object()
+
+        # @todo if method=GET, render a download button
+        outputs.str("url", label="URL", view=types.MarkdownView())
+
+        return types.Property(
+            outputs, view=types.View(label="Get URL samples")
+        )
+
+
+def _get_url_inputs(ctx, inputs):
+    file_explorer = types.FileExplorerView(button_label="Choose a file...")
+    prop = inputs.file(
+        "filepath",
+        required=True,
+        label="File",
+        description="Choose a file for which to generate a signed URL",
+        view=file_explorer,
+    )
+    filepath = _parse_path(ctx, "filepath")
+
+    inputs.int(
+        "hours",
+        default=24,
+        label="Hours",
+        description="The number of hours that the URL should be valid",
+    )
+    hours = ctx.params.get("hours", 24)
+
+    method_choices = types.Choices()
+    method_choices.add_choice("GET", label="GET")
+    method_choices.add_choice("PUT", label="PUT")
+    method_choices.add_choice("DELETE", label="DELETE")
+
+    inputs.enum(
+        "method",
+        method_choices.values(),
+        required=True,
+        default="GET",
+        label="Method",
+        description="The HTTP verb (GET, PUT, DELETE) to authorize",
+        view=method_choices,
+    )
+    method = ctx.params.get("method", "GET")
+
+    try:
+        # pylint: disable=no-member
+        url = fos.get_url(filepath, hours=hours, method=method)
+        view = types.Notice(label=f"[{url}]({url})")
+    except Exception as e:
+        view = types.Error(label=str(e))
+
+    prop = inputs.view("result", view)
+    prop.invalid = True
+
+
 def _parse_path(ctx, key):
     value = ctx.params.get(key, None)
     return value.get("absolute_path", None) if value else None
@@ -2833,3 +2921,5 @@ def register(p):
     p.register(MergeLabels)
     p.register(ExportSamples)
     p.register(DrawLabels)
+    if hasattr(foc, "TEAMS_VERSION"):
+        p.register(GetURL)

plugins/io/fiftyone.yml

@@ -11,3 +11,4 @@ operators:
   - merge_labels
   - export_samples
   - draw_labels
+  - get_url