def setUpClass(cls):
token = environ.get('KB_AUTH_TOKEN', None)
config_file = environ.get('KB_DEPLOYMENT_CONFIG', None)
cls.cfg = {}
config = ConfigParser()
config.read(config_file)
for nameval in config.items('kb_mash'):
cls.cfg[nameval[0]] = nameval[1]
# Getting username from Auth profile for token
# authServiceUrl = cls.cfg['auth-service-url']
# auth_client = _KBaseAuth(authServiceUrl)
# user_id = auth_client.get_user(token)
# WARNING: don't call any logging methods on the context object,
# it'll result in a NoneType error
cls.ctx = MethodContext(None)
cls.ctx.update({
'token':
token,
# 'user_id': user_id,
'provenance': [{
'service': 'kb_mash',
'method': 'please_never_use_it_in_production',
'method_params': []
}],
'authenticated':
1
})
cls.wsURL = cls.cfg['workspace-url']
cls.wsClient = workspaceService(cls.wsURL)
# suffix = int(time.time() * 1000)
# wsName = "test_kb_mash_" + str(suffix)
# cls.ws_info = cls.wsClient.create_workspace({'workspace': wsName})
cls.serviceImpl = kb_mash(cls.cfg)
cls.scratch = cls.cfg['scratch']
cls.callback_url = os.environ['SDK_CALLBACK_URL']
def get_config():
if not get_config_file():
return None
retconfig = {}
config = ConfigParser()
config.read(get_config_file())
for nameval in config.items(get_service_name() or 'kb_mash'):
retconfig[nameval[0]] = nameval[1]
return retconfig
config = get_config()
from kb_mash.kb_mashImpl import kb_mash # noqa @IgnorePep8
impl_kb_mash = kb_mash(config)
class JSONObjectEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, set):
return list(obj)
if isinstance(obj, frozenset):
return list(obj)
if hasattr(obj, 'toJSONable'):
return obj.toJSONable()
return json.JSONEncoder.default(self, obj)
class JSONRPCServiceCustom(JSONRPCService):
def get_config():
if not get_config_file():
return None
retconfig = {}
config = ConfigParser()
config.read(get_config_file())
for nameval in config.items(get_service_name() or 'kb_mash'):
retconfig[nameval[0]] = nameval[1]
return retconfig
config = get_config()
from kb_mash.kb_mashImpl import kb_mash # noqa @IgnorePep8
impl_kb_mash = kb_mash(config)
class JSONObjectEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, set):
return list(obj)
if isinstance(obj, frozenset):
return list(obj)
if hasattr(obj, 'toJSONable'):
return obj.toJSONable()
return json.JSONEncoder.default(self, obj)
class JSONRPCServiceCustom(JSONRPCService):
def call(self, ctx, jsondata):