class CLIClient(object): def __init__(self, server): self.client = IngesterPlatformAPI(server, None) def file_or_json(self, s): """Try to open the file or parse the string as JSON""" obj = None if os.path.exists(s): with open(r, "r") as f: obj = json.load(f) else: obj = json.loads(s) return self.client._marshaller.dict_to_obj(obj) def ping(self): return self.client.ping() def enable(self, ds_id): return self.client.enableDataset(ds_id) def disable(self, ds_id): return self.client.disableDataset(ds_id) def search(self, criteria, limit=20, offset=0): return self.client._marshaller.obj_to_dict( self.client.search(self.file_or_json(criteria), int(offset), int(limit))) def logs(self, ds_id): return self.client._marshaller.obj_to_dict( self.client.getIngesterLogs(ds_id)) def post(self, s): obj = self.file_or_json(s) return (self.client._marshaller.obj_to_dict(self.client.post(obj))) def get(self, *args): if args[0] == "schema": return (self.client.getSchema(args[1])) elif args[0] == "location": return (self.client.getLocation(args[1])) elif args[0] == "dataset": return (self.client.getDataset(args[1]))
class CLIClient(object): def __init__(self, server): self.client = IngesterPlatformAPI(server, None) def file_or_json(self, s): """Try to open the file or parse the string as JSON""" obj = None if os.path.exists(s): with open(r, "r") as f: obj = json.load(f) else: obj = json.loads(s) return self.client._marshaller.dict_to_obj(obj) def ping(self): return self.client.ping() def enable(self, ds_id): return self.client.enableDataset(ds_id) def disable(self, ds_id): return self.client.disableDataset(ds_id) def search(self, criteria, limit=20, offset=0): return self.client._marshaller.obj_to_dict(self.client.search(self.file_or_json(criteria), int(offset), int(limit))) def logs(self, ds_id): return self.client._marshaller.obj_to_dict(self.client.getIngesterLogs(ds_id)) def post(self, s): obj = self.file_or_json(s) return (self.client._marshaller.obj_to_dict(self.client.post(obj))) def get(self, *args): if args[0] == "schema": return (self.client.getSchema(args[1])) elif args[0] == "location": return (self.client.getLocation(args[1])) elif args[0] == "dataset": return (self.client.getDataset(args[1]))