mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-22 04:43:00 +00:00
missing/wrong type hints
This commit is contained in:
parent
1c8f016208
commit
e151562ab9
4 changed files with 11 additions and 11 deletions
|
@ -1,7 +1,7 @@
|
||||||
import functools
|
import functools
|
||||||
from ipaddress import IPv4Network
|
from ipaddress import IPv4Network
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional, Dict, List, Any, TextIO
|
from typing import Optional, Dict, List, Any, TextIO, Tuple
|
||||||
|
|
||||||
from pydantic import BaseModel, constr, root_validator, validator
|
from pydantic import BaseModel, constr, root_validator, validator
|
||||||
|
|
||||||
|
@ -55,7 +55,7 @@ class ProjectConfig(BaseModel):
|
||||||
|
|
||||||
@validator("override_storage", pre=True)
|
@validator("override_storage", pre=True)
|
||||||
@classmethod
|
@classmethod
|
||||||
def unify_storage(cls, value):
|
def unify_storage(cls, value) -> Dict[str, Any]:
|
||||||
"""parse different storage notations"""
|
"""parse different storage notations"""
|
||||||
|
|
||||||
if value is None or isinstance(value, dict):
|
if value is None or isinstance(value, dict):
|
||||||
|
@ -135,7 +135,7 @@ class KiwiConfig(BaseModel):
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@functools.lru_cache(maxsize=5)
|
@functools.lru_cache(maxsize=5)
|
||||||
def from_directory(cls, directory: Path):
|
def from_directory(cls, directory: Path) -> "KiwiConfig":
|
||||||
"""parses an actual kiwi.yml from disk (cached)"""
|
"""parses an actual kiwi.yml from disk (cached)"""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -148,7 +148,7 @@ class KiwiConfig(BaseModel):
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@functools.lru_cache(maxsize=1)
|
@functools.lru_cache(maxsize=1)
|
||||||
def from_default(cls):
|
def from_default(cls) -> "KiwiConfig":
|
||||||
"""returns the default config (cached)"""
|
"""returns the default config (cached)"""
|
||||||
|
|
||||||
return cls()
|
return cls()
|
||||||
|
@ -268,7 +268,7 @@ class KiwiConfig(BaseModel):
|
||||||
def unify_environment(cls, value) -> Dict[str, Optional[str]]:
|
def unify_environment(cls, value) -> Dict[str, Optional[str]]:
|
||||||
"""parse different environment notations"""
|
"""parse different environment notations"""
|
||||||
|
|
||||||
def parse_str(var_val: Any) -> (str, Optional[str]):
|
def parse_str(var_val: Any) -> Tuple[str, Optional[str]]:
|
||||||
"""parse a "<variable>=<value>" string"""
|
"""parse a "<variable>=<value>" string"""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -311,7 +311,7 @@ class KiwiConfig(BaseModel):
|
||||||
|
|
||||||
@validator("storage", pre=True)
|
@validator("storage", pre=True)
|
||||||
@classmethod
|
@classmethod
|
||||||
def unify_storage(cls, value):
|
def unify_storage(cls, value) -> Dict[str, Any]:
|
||||||
"""parse different storage notations"""
|
"""parse different storage notations"""
|
||||||
|
|
||||||
if value is None:
|
if value is None:
|
||||||
|
@ -336,7 +336,7 @@ class KiwiConfig(BaseModel):
|
||||||
|
|
||||||
@validator("network", pre=True)
|
@validator("network", pre=True)
|
||||||
@classmethod
|
@classmethod
|
||||||
def unify_network(cls, value):
|
def unify_network(cls, value) -> Dict[str, Any]:
|
||||||
"""parse different network notations"""
|
"""parse different network notations"""
|
||||||
|
|
||||||
if value is None:
|
if value is None:
|
||||||
|
|
|
@ -46,7 +46,7 @@ class Services:
|
||||||
def __bool__(self) -> bool:
|
def __bool__(self) -> bool:
|
||||||
return bool(self.content)
|
return bool(self.content)
|
||||||
|
|
||||||
def filter_existing(self, service_names: List[str]):
|
def filter_existing(self, service_names: List[str]) -> "Services":
|
||||||
return Services([
|
return Services([
|
||||||
service
|
service
|
||||||
for service in self.content
|
for service in self.content
|
||||||
|
|
|
@ -23,7 +23,7 @@ class YAML(ruamel.yaml.YAML):
|
||||||
return stream.getvalue()
|
return stream.getvalue()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _format_kiwi_yml(yml_string: str):
|
def _format_kiwi_yml(yml_string: str) -> str:
|
||||||
# insert newline before every main key
|
# insert newline before every main key
|
||||||
yml_string = re.sub(r'^(\S)', r'\n\1', yml_string, flags=re.MULTILINE)
|
yml_string = re.sub(r'^(\S)', r'\n\1', yml_string, flags=re.MULTILINE)
|
||||||
|
|
||||||
|
|
|
@ -67,9 +67,9 @@ class Rootkit:
|
||||||
f"{IMAGES_DIRECTORY_NAME}"
|
f"{IMAGES_DIRECTORY_NAME}"
|
||||||
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||||||
|
|
||||||
def run(self, process_args, **kwargs):
|
def run(self, process_args, **kwargs) -> Optional[subprocess.CompletedProcess]:
|
||||||
self.__build_image()
|
self.__build_image()
|
||||||
DOCKER_EXE.run([
|
return DOCKER_EXE.run([
|
||||||
'run', '--rm',
|
'run', '--rm',
|
||||||
'-v', '/:/mnt',
|
'-v', '/:/mnt',
|
||||||
'-u', 'root',
|
'-u', 'root',
|
||||||
|
|
Loading…
Reference in a new issue