Add basic type hints

This commit is contained in:
Einar Forselv 2019-04-17 03:01:12 +02:00
parent d809496eb2
commit 2c9c418c6e
1 changed files with 14 additions and 9 deletions

View File

@ -98,26 +98,27 @@ class Mount:
self._container = container self._container = container
@property @property
def container(self): def container(self) -> Container:
return self._container return self._container
@property @property
def type(self): def type(self) -> str:
"""bind/volume"""
return self._data.get('Type') return self._data.get('Type')
@property @property
def name(self): def name(self) -> str:
return self._data.get('Name') return self._data.get('Name')
@property @property
def source(self): def source(self) -> str:
return self._data.get('Source') return self._data.get('Source')
@property @property
def destination(self): def destination(self) -> str:
return self._data.get('Destination') return self._data.get('Destination')
def mount_string(self): def mount_string(self) -> str:
if self.type == VOLUME_TYPE_VOLUME: if self.type == VOLUME_TYPE_VOLUME:
return "- {}:{}:ro".format(self.name.split('_')[-1], self.destination) return "- {}:{}:ro".format(self.name.split('_')[-1], self.destination)
elif self.type == VOLUME_TYPE_BIND: elif self.type == VOLUME_TYPE_BIND:
@ -125,10 +126,10 @@ class Mount:
else: else:
raise ValueError("Uknown volume type: {}".format(self.type)) raise ValueError("Uknown volume type: {}".format(self.type))
def __repr__(self): def __repr__(self) -> str:
return str(self) return str(self)
def __str__(self): def __str__(self) -> str:
return str(self._data) return str(self._data)
def __hash__(self): def __hash__(self):
@ -147,6 +148,7 @@ class RunningContainers:
all_containers = utils.list_containers() all_containers = utils.list_containers()
self.containers = [] self.containers = []
self.this_container = None self.this_container = None
self.backup_process_container = None
# Find the container we are running in. # Find the container we are running in.
# If we don't have this information we cannot continue # If we don't have this information we cannot continue
@ -195,7 +197,10 @@ class RunningContainers:
# """Host mapped volumes""" # """Host mapped volumes"""
# return set(mnt for mnt in self.gen_volumes(VOLUME_TYPE_BIND)) # return set(mnt for mnt in self.gen_volumes(VOLUME_TYPE_BIND))
def get_service(self, name): def backup_process_running(self) -> bool:
return self.backup_process_container is not None
def get_service(self, name) -> Container:
for container in self.containers: for container in self.containers:
if container.service_name == name: if container.service_name == name:
return container return container