earthengine-api/python/ee/dictionary.py
Max Shawabkeh 2403df3657 v0.1.51.
2015-04-22 16:36:30 -07:00

63 lines
1.6 KiB
Python

#!/usr/bin/env python
"""A wrapper for dictionaries."""
import apifunction
import computedobject
import ee_exception
# Using lowercase function naming to match the JavaScript names.
# pylint: disable=g-bad-name
class Dictionary(computedobject.ComputedObject):
"""An object to represent dictionaries."""
_initialized = False
def __init__(self, arg):
"""Construct a dictionary wrapper.
This constuctor accepts the following args:
1) A bare dictionary.
2) A ComputedObject returning a dictionary.
Args:
arg: The dictionary to wrap.
"""
self.initialize()
if isinstance(arg, dict):
super(Dictionary, self).__init__(None, None)
self._dictionary = arg
elif isinstance(arg, computedobject.ComputedObject):
super(Dictionary, self).__init__(arg.func, arg.args, arg.varName)
self._dictionary = None
else:
raise ee_exception.EEException(
'Invalid argument specified for ee.Dictionary(): %s' % arg)
@classmethod
def initialize(cls):
"""Imports API functions to this class."""
if not cls._initialized:
apifunction.ApiFunction.importApi(cls, 'Dictionary', 'Dictionary')
cls._initialized = True
@classmethod
def reset(cls):
"""Removes imported API functions from this class."""
apifunction.ApiFunction.clearApi(cls)
cls._initialized = False
@staticmethod
def name():
return 'Dictionary'
def encode(self, opt_encoder=None):
if self._dictionary is not None:
return opt_encoder(self._dictionary)
else:
return super(Dictionary, self).encode(opt_encoder)