summaryrefslogtreecommitdiff
path: root/docker/utils/decorators.py
blob: 8d07f936e2a6940688e1dd7e8dafef4de6035136 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
import docker.errors


def check_resource(f):
    def wrapped(self, resource_id=None, *args, **kwargs):
        if resource_id is None and (
            kwargs.get('container') is None and kwargs.get('image') is None
        ):
            raise docker.errors.NullResource(
                'image or container param is None'
            )
        return f(self, resource_id, *args, **kwargs)
    return wrapped