First working version

client/once.py

@@ -0,0 +1,68 @@
+'''
+Simple command to share one-time files
+'''
+
+import os
+import json
+import time
+
+import click
+import requests
+from pygments import highlight, lexers, formatters
+
+
+ONCE_API_URL = os.getenv('ONCE_API_URL')
+
+
+def highlight_json(obj):
+    formatted_json = json.dumps(obj, sort_keys=True, indent=4)
+    return highlight(formatted_json, lexers.JsonLexer(), formatters.TerminalFormatter())
+
+
+def echo_obj(obj):
+    click.echo(highlight_json(obj))
+
+
+def api_req(method: str, url: str, verbose: bool = False, **kwargs):
+    method = method.lower()
+    if method not in ['get', 'post']:
+        raise ValueError(f'Unsupported HTTP method "{method}"')
+
+    actual_url = f'{ONCE_API_URL}{url}'
+
+    if verbose:
+        print(f'{method.upper()} {actual_url}')
+
+    response = getattr(requests, method)(actual_url, **kwargs)
+
+    if verbose:
+        print(f'Server response status: {response.status_code}')
+        echo_obj(response.json())
+
+    return response
+
+
+@click.command('share')
+@click.argument('file', type=click.File(mode='rb'), required=True)
+@click.option('--verbose', '-v', is_flag=True, default=False, help='Enables verbose output.')
+def share(file: click.File, verbose: bool):
+    entry = api_req('GET', '/',
+        params={'f': os.path.basename(file.name)},
+        verbose=verbose).json()
+
+    once_url = entry['once_url']
+    upload_data = entry['presigned_post']
+    files = {'file': file}
+
+    upload_started = time.time()
+    response = requests.post(upload_data['url'],
+        data=upload_data['fields'],
+        files=files)
+
+    upload_time = time.time() - upload_started
+    print(f"File uploaded in {upload_time}s")    
+    print(f"File can be downloaded once at: {once_url}")
+
+
+if __name__ == '__main__':
+    share()

client/requirements.txt

@@ -0,0 +1,3 @@
+click
+pygments
+requests