#!/usr/bin/env python
import fire
import grpc

from cri.api_pb2 import RuntimeServiceStub
from cri.api_pb2 import ImageServiceStub


class Runtime(RuntimeServiceStub):

    def __init__(self, channel):
        RuntimeServiceStub.__init__(self, grpc.insecure_channel(channel))


class Image(ImageServiceStub):

    def __init__(self, channel):
        ImageServiceStub.__init__(self, grpc.insecure_channel(channel))


if __name__ == "__main__":
    fire.Fire({
        'runtime': Runtime,
        'image': Image,
    })
