Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions remo/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ class BaseAPI:
def __init__(self, server, email, password):
self.server = server
self.token = None
self._public_url = ''
self._login(email, password)

def _login(self, email, password):
Expand All @@ -64,6 +65,13 @@ def _auth_header(self):
raise Exception('Not authenticated')
return {'Authorization': 'Token {}'.format(self.token)}

def set_public_url(self, public_url: str):
self._public_url = public_url

def public_url(self, endpoint, *args, **kwargs):
url = self._public_url if self._public_url else self.server
return self._build_url(url, endpoint, *args, **kwargs)

def url(self, endpoint, *args, **kwargs):
return self._build_url(self.server, endpoint, *args, **kwargs)

Expand Down
5 changes: 4 additions & 1 deletion remo/sdk.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,9 @@ def __init__(self, server: str, email: str, password: str, viewer: str = 'browse
self.viewer = None
self.set_viewer(viewer)

def set_public_url(self, public_url: str):
self.api.set_public_url(public_url)

def set_viewer(self, viewer: str):
"""
Allows to choose one of available viewers
Expand Down Expand Up @@ -780,7 +783,7 @@ def view_annotation_stats(self, annotation_set_id: int):
return self._view(frontend.annotation_set_insights.format(annotation_set_id))

def _view(self, url, *args, **kwargs):
return self.viewer.browse(self.api.url(url, *args, **kwargs))
return self.viewer.browse(self.api.public_url(url, *args, **kwargs))

def generate_annotations_from_folders(self, path_to_data_folder: str):
"""
Expand Down
2 changes: 1 addition & 1 deletion remo/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = '0.1.4'
__version__ = '0.1.5'