def strip_s3_key_prefix()

in backend/model_garden/utils.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

def strip_s3_key_prefix(prefix: str, key: str, delimiter: str = '/') -> str:
  """Return the key without the prefix and without the leading delimiter.

  >>> strip_s3_key_prefix('foo', 'foo/bar/baz')
  'bar/baz'
  """
  if not key:
    return key

  if not key.startswith(prefix):
    return key

  suffix = key[len(prefix):]
  if suffix.startswith(delimiter):
    return suffix[len(delimiter):]

  return suffix