kubernator 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of kubernator might be problematic. Click here for more details.

kubernator/__init__.py CHANGED
@@ -16,12 +16,12 @@
16
16
  # limitations under the License.
17
17
  #
18
18
 
19
- __version__ = "1.0.4"
19
+ __version__ = "1.0.6"
20
20
 
21
21
 
22
22
  def _main():
23
23
  from kubernator import app
24
- return app.main()
24
+ return app.main() or 0
25
25
 
26
26
 
27
27
  def main():
kubernator/api.py CHANGED
@@ -152,8 +152,12 @@ def _download_remote_file(url, file_name, cache: dict):
152
152
  return dict(r.headers)
153
153
 
154
154
 
155
+ def get_app_cache_dir():
156
+ return Path(user_config_dir("kubernator"))
157
+
158
+
155
159
  def get_cache_dir(category: str, sub_category: str = None):
156
- config_dir = Path(user_config_dir("kubernator")) / category
160
+ config_dir = get_app_cache_dir() / category
157
161
  if sub_category:
158
162
  config_dir = config_dir / sub_category
159
163
  if not config_dir.exists():
@@ -224,6 +228,19 @@ def validator_with_defaults(validator_class):
224
228
  return validators.extend(validator_class, {"properties": set_defaults})
225
229
 
226
230
 
231
+ def install_python_k8s_client(run, package_major, logger_stdout, logger_stderr):
232
+ cache_dir = get_cache_dir("python")
233
+ package_major_dir = cache_dir / str(package_major)
234
+
235
+ if not package_major_dir.exists():
236
+ package_major_dir.mkdir(parents=True, exist_ok=True)
237
+
238
+ run(["pip", "install", "--no-deps", "--no-cache-dir", "--no-input", "--target", str(package_major_dir),
239
+ f"kubernetes~={package_major}.0"], logger_stdout, logger_stderr).wait()
240
+
241
+ return package_major_dir
242
+
243
+
227
244
  class _PropertyList(MutableSequence):
228
245
 
229
246
  def __init__(self, seq, read_parent, name):
@@ -620,6 +637,7 @@ def clone_url_str(url):
620
637
 
621
638
 
622
639
  def prepend_os_path(path):
640
+ path = str(path)
623
641
  paths = os.environ["PATH"].split(os.pathsep)
624
642
  if path not in paths:
625
643
  paths.insert(0, path)
kubernator/app.py CHANGED
@@ -26,11 +26,13 @@ import urllib.parse
26
26
  from collections import deque
27
27
  from collections.abc import MutableMapping, Callable
28
28
  from pathlib import Path
29
+ from shutil import rmtree
29
30
  from typing import Optional, Union
30
31
 
31
32
  import kubernator
32
33
  from kubernator.api import (KubernatorPlugin, Globs, scan_dir, PropertyDict, config_as_dict, config_parent,
33
- download_remote_file, load_remote_file, Repository, StripNL, jp)
34
+ download_remote_file, load_remote_file, Repository, StripNL, jp, get_app_cache_dir,
35
+ install_python_k8s_client)
34
36
  from kubernator.proc import run, run_capturing_out
35
37
 
36
38
  TRACE = 5
@@ -55,10 +57,16 @@ logger = logging.getLogger("kubernator")
55
57
 
56
58
 
57
59
  def define_arg_parse():
58
- parser = argparse.ArgumentParser(description="Kubernetes Provisioning Tool",
60
+ parser = argparse.ArgumentParser(prog="kubernator",
61
+ description="Kubernetes Provisioning Tool",
59
62
  formatter_class=argparse.ArgumentDefaultsHelpFormatter)
60
- parser.add_argument("--version", action="version", version=kubernator.__version__,
61
- help="print version and exit")
63
+ g = parser.add_mutually_exclusive_group()
64
+ g.add_argument("--version", action="version", version=kubernator.__version__,
65
+ help="print version and exit")
66
+ g.add_argument("--clear-cache", action="store_true",
67
+ help="clear cache and exit")
68
+ g.add_argument("--pre-cache-k8s-client", action="extend", nargs="+", type=int,
69
+ help="download specified K8S client library minor(!) version(s) and exit")
62
70
  parser.add_argument("--log-format", choices=["human", "json"], default="human",
63
71
  help="whether to log for human or machine consumption")
64
72
  parser.add_argument("--log-file", type=argparse.FileType("w"), default=None,
@@ -158,47 +166,50 @@ class App(KubernatorPlugin):
158
166
 
159
167
  self.register_plugin(self)
160
168
 
161
- while True:
162
- cwd = self.next()
163
- if not cwd:
164
- logger.debug("No paths left to traverse")
165
- break
169
+ try:
170
+ while True:
171
+ cwd = self.next()
172
+ if not cwd:
173
+ logger.debug("No paths left to traverse")
174
+ break
166
175
 
167
- context = self.context
168
-
169
- logger.debug("Inspecting directory %s", self._display_path(cwd))
170
- self._run_handlers(KubernatorPlugin.handle_before_dir, False, context, None, cwd)
176
+ context = self.context
171
177
 
172
- if (ktor_py := (cwd / ".kubernator.py")).exists():
173
- self._run_handlers(KubernatorPlugin.handle_before_script, False, context, None, cwd)
178
+ logger.debug("Inspecting directory %s", self._display_path(cwd))
179
+ self._run_handlers(KubernatorPlugin.handle_before_dir, False, context, None, cwd)
174
180
 
175
- for h in self.context._plugins:
176
- h.set_context(context)
181
+ if (ktor_py := (cwd / ".kubernator.py")).exists():
182
+ self._run_handlers(KubernatorPlugin.handle_before_script, False, context, None, cwd)
177
183
 
178
- self._exec_ktor(ktor_py)
184
+ for h in self.context._plugins:
185
+ h.set_context(context)
179
186
 
180
- for h in self.context._plugins:
181
- h.set_context(None)
187
+ self._exec_ktor(ktor_py)
182
188
 
183
- self._run_handlers(KubernatorPlugin.handle_after_script, True, context, None, cwd)
189
+ for h in self.context._plugins:
190
+ h.set_context(None)
184
191
 
185
- self._run_handlers(KubernatorPlugin.handle_after_dir, True, context, None, cwd)
192
+ self._run_handlers(KubernatorPlugin.handle_after_script, True, context, None, cwd)
186
193
 
187
- self.context = self._top_dir_context
188
- context = self.context
194
+ self._run_handlers(KubernatorPlugin.handle_after_dir, True, context, None, cwd)
189
195
 
190
- self._run_handlers(KubernatorPlugin.handle_apply, True, context, None)
196
+ self.context = self._top_dir_context
197
+ context = self.context
191
198
 
192
- self._run_handlers(KubernatorPlugin.handle_verify, True, context, None)
199
+ self._run_handlers(KubernatorPlugin.handle_apply, True, context, None)
193
200
 
194
- self._run_handlers(KubernatorPlugin.handle_shutdown, True, context, None)
201
+ self._run_handlers(KubernatorPlugin.handle_verify, True, context, None)
202
+ finally:
203
+ self.context = self._top_dir_context
204
+ context = self.context
205
+ self._run_handlers(KubernatorPlugin.handle_shutdown, True, context, None)
195
206
 
196
207
  def discover_plugins(self):
197
208
  importlib.invalidate_caches()
198
209
  search_path = Path(kubernator.__path__[0], "plugins")
199
210
  [importlib.import_module(name)
200
211
  for finder, name, is_pkg in
201
- pkgutil.iter_modules([search_path], "kubernator.plugins.")]
212
+ pkgutil.iter_modules([str(search_path)], "kubernator.plugins.")]
202
213
 
203
214
  for plugin in KubernatorPlugin.__subclasses__():
204
215
  if plugin._name in self._plugin_types:
@@ -440,11 +451,36 @@ class App(KubernatorPlugin):
440
451
  return "Kubernator"
441
452
 
442
453
 
454
+ def clear_cache():
455
+ cache_dir = get_app_cache_dir()
456
+ logger.info("Clearing application cache at %s", cache_dir)
457
+ if cache_dir.exists():
458
+ rmtree(cache_dir)
459
+
460
+
461
+ def pre_cache_k8s_clients(*versions):
462
+ proc_logger = logger.getChild("proc")
463
+ stdout_logger = StripNL(proc_logger.info)
464
+ stderr_logger = StripNL(proc_logger.warning)
465
+
466
+ for v in versions:
467
+ logger.info("Caching K8S client library ~=v%s.0...", v)
468
+ install_python_k8s_client(run, v, stdout_logger, stderr_logger)
469
+
470
+
443
471
  def main():
444
472
  args = define_arg_parse().parse_args()
445
473
  init_logging(args.verbose, args.log_format, args.log_file)
446
474
 
447
475
  try:
476
+ if args.clear_cache:
477
+ clear_cache()
478
+ return
479
+
480
+ if args.pre_cache_k8s_client:
481
+ pre_cache_k8s_clients(*args.pre_cache_k8s_client)
482
+ return
483
+
448
484
  with App(args) as app:
449
485
  app.run()
450
486
  except SystemExit as e:
@@ -54,7 +54,7 @@ class AwsCliPlugin(KubernatorPlugin):
54
54
  def set_context(self, context):
55
55
  self.context = context
56
56
 
57
- def cmd(self, *args, output="json", region=None):
57
+ def stanza(self, *args, output="json", region=None):
58
58
  context = self.context
59
59
  stanza = [context.awscli.aws_file, "--output", output]
60
60
  if region:
@@ -101,7 +101,7 @@ class AwsCliPlugin(KubernatorPlugin):
101
101
  aws_file.chmod(0o500)
102
102
 
103
103
  self.aws_dir = awscli_cache_dir
104
- prepend_os_path(str(self.aws_dir))
104
+ prepend_os_path(self.aws_dir)
105
105
 
106
106
  self.aws_file = str(aws_file)
107
107
 
@@ -112,7 +112,7 @@ class AwsCliPlugin(KubernatorPlugin):
112
112
 
113
113
  context.globals.awscli = dict(version=version,
114
114
  aws_file=self.aws_file,
115
- cmd=self.cmd
115
+ stanza=self.stanza
116
116
  )
117
117
 
118
118
  def __repr__(self):
kubernator/plugins/eks.py CHANGED
@@ -64,7 +64,7 @@ class EksPlugin(KubernatorPlugin):
64
64
  def handle_init(self):
65
65
  context = self.context
66
66
 
67
- self.context.app.run(context.awscli.cmd(
67
+ self.context.app.run(context.awscli.stanza(
68
68
  "eks", "update-kubeconfig", "--name", self.name, "--kubeconfig", context.eks.kubeconfig,
69
69
  region=self.region),
70
70
  stdout_logger,
@@ -106,7 +106,7 @@ class HelmPlugin(KubernatorPlugin):
106
106
  def set_context(self, context):
107
107
  self.context = context
108
108
 
109
- def helm_stanza(self):
109
+ def stanza(self):
110
110
  context = self.context
111
111
  stanza = [context.helm.helm_file, f"--kubeconfig={context.kubeconfig.kubeconfig}"]
112
112
  if logger.getEffectiveLevel() < logging.INFO:
@@ -132,7 +132,7 @@ class HelmPlugin(KubernatorPlugin):
132
132
  copy(Path(self.helm_dir.name)/f"{get_golang_os()}-{get_golang_machine()}"/"helm", helm_file)
133
133
 
134
134
  os.chmod(helm_file, 0o500)
135
- prepend_os_path(str(self.helm_dir))
135
+ prepend_os_path(self.helm_dir.name)
136
136
  else:
137
137
  # Use current version
138
138
  helm_file = which("helm")
@@ -145,13 +145,13 @@ class HelmPlugin(KubernatorPlugin):
145
145
  default_excludes=Globs([".*"], True),
146
146
  namespace_transformer=True,
147
147
  helm_file=helm_file,
148
- helm_stanza=self.helm_stanza,
148
+ stanza=self.stanza,
149
149
  add_helm_template=self.add_helm_template,
150
150
  add_helm=self.add_helm,
151
151
  )
152
152
 
153
153
  def handle_init(self):
154
- version = self.context.app.run_capturing_out(self.helm_stanza() + ["version", "--template", "{{.Version}}"],
154
+ version = self.context.app.run_capturing_out(self.stanza() + ["version", "--template", "{{.Version}}"],
155
155
  logger.error)
156
156
  logger.info("Found Helm version %s", version)
157
157
 
@@ -204,10 +204,10 @@ class HelmPlugin(KubernatorPlugin):
204
204
  logger.debug("Repository %s mapping to %s", repository, repository_hash)
205
205
  if repository_hash not in self.repositories:
206
206
  logger.info("Adding and updating repository %s mapping to %s", repository, repository_hash)
207
- self.context.app.run(self.helm_stanza() + ["repo", "add", repository_hash, repository],
207
+ self.context.app.run(self.stanza() + ["repo", "add", repository_hash, repository],
208
208
  stdout_logger,
209
209
  stderr_logger).wait()
210
- self.context.app.run(self.helm_stanza() + ["repo", "update"],
210
+ self.context.app.run(self.stanza() + ["repo", "update"],
211
211
  stdout_logger,
212
212
  stderr_logger).wait()
213
213
  self.repositories.add(repository_hash)
@@ -233,7 +233,7 @@ class HelmPlugin(KubernatorPlugin):
233
233
 
234
234
  stdin = write_stdin
235
235
 
236
- resources = self.context.app.run_capturing_out(self.helm_stanza() +
236
+ resources = self.context.app.run_capturing_out(self.stanza() +
237
237
  ["template",
238
238
  name,
239
239
  f"{repository_hash}/{chart}",
@@ -77,7 +77,7 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
77
77
  istio_tar.extractall(self.istioctl_dir.name)
78
78
 
79
79
  os.chmod(istioctl_file, 0o500)
80
- prepend_os_path(str(self.istioctl_dir))
80
+ prepend_os_path(self.istioctl_dir.name)
81
81
  else:
82
82
  # Use current version
83
83
  istioctl_file = which("istioctl")
@@ -90,19 +90,19 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
90
90
  default_includes=Globs(["*.istio.yaml", "*.istio.yml"], True),
91
91
  default_excludes=Globs([".*"], True),
92
92
  istioctl_file=istioctl_file,
93
- istioctl_stanza=self.istioctl_stanza,
93
+ stanza=self.stanza,
94
94
  test=self.test_istioctl
95
95
  )
96
96
 
97
97
  def test_istioctl(self):
98
98
  context = self.context
99
- version_out: str = context.app.run_capturing_out(context.istio.istioctl_stanza() + ["version", "-o", "json"],
99
+ version_out: str = context.app.run_capturing_out(context.istio.stanza() + ["version", "-o", "json"],
100
100
  stderr_logger)
101
101
 
102
102
  version_out_js = json.loads(version_out)
103
103
  version = version_out_js["clientVersion"]["version"]
104
104
  logger.info("Using istioctl %r version %r with stanza %r",
105
- self.context.istio.istioctl_file, version, context.istio.istioctl_stanza())
105
+ self.context.istio.istioctl_file, version, context.istio.stanza())
106
106
 
107
107
  logger.info("Found Istio client version %s", version)
108
108
 
@@ -111,7 +111,7 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
111
111
  def set_context(self, context):
112
112
  self.context = context
113
113
 
114
- def istioctl_stanza(self):
114
+ def stanza(self):
115
115
  context = self.context.istio
116
116
  return [context.istioctl_file, f"--kubeconfig={self.context.kubeconfig.kubeconfig}"]
117
117
 
@@ -145,7 +145,8 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
145
145
  # This plugin only deals with Istio Operator, so only load that stuff
146
146
  self.resource_definitions_schema = load_remote_file(logger,
147
147
  f"https://raw.githubusercontent.com/kubernetes/kubernetes/"
148
- f"{self.context.k8s.server_version}/api/openapi-spec/swagger.json",
148
+ f"{self.context.k8s.server_git_version}"
149
+ f"/api/openapi-spec/swagger.json",
149
150
  FileType.JSON)
150
151
  self._populate_resource_definitions()
151
152
  self.add_remote_crds(f"{url_prefix}/crd-operator.yaml", FileType.YAML)
@@ -188,9 +189,9 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
188
189
 
189
190
  if context.app.args.command == "apply":
190
191
  logger.info("Running Istio precheck")
191
- context.app.run(context.istio.istioctl_stanza() + ["x", "precheck"],
192
+ context.app.run(context.istio.stanza() + ["x", "precheck"],
192
193
  stdout_logger, stderr_logger).wait()
193
- context.app.run(context.istio.istioctl_stanza() + ["validate", "-f", operators_file.name],
194
+ context.app.run(context.istio.stanza() + ["validate", "-f", operators_file.name],
194
195
  stdout_logger, stderr_logger).wait()
195
196
 
196
197
  self._operator_init(operators_file, True)
@@ -229,7 +230,7 @@ class IstioPlugin(KubernatorPlugin, K8SResourcePluginMixin):
229
230
  raise
230
231
 
231
232
  logger.info("Running Istio operator init%s", status_details)
232
- istio_operator_init = context.istio.istioctl_stanza() + ["operator", "init", "-f", operators_file.name]
233
+ istio_operator_init = context.istio.stanza() + ["operator", "init", "-f", operators_file.name]
233
234
  context.app.run(istio_operator_init + (["--dry-run"] if dry_run else []),
234
235
  stdout_logger,
235
236
  stderr_logger).wait()
kubernator/plugins/k8s.py CHANGED
@@ -24,19 +24,30 @@ import sys
24
24
  import types
25
25
  from collections.abc import Mapping
26
26
  from functools import partial
27
+ from importlib.metadata import version as pkg_version
27
28
  from pathlib import Path
28
29
  from typing import Iterable, Callable, Sequence
29
30
 
30
31
  import jsonpatch
31
32
  import yaml
32
33
 
33
- from kubernator.api import (KubernatorPlugin, Globs, scan_dir, load_file, FileType, load_remote_file)
34
+ from kubernator.api import (KubernatorPlugin,
35
+ Globs,
36
+ scan_dir,
37
+ load_file,
38
+ FileType,
39
+ load_remote_file,
40
+ StripNL,
41
+ install_python_k8s_client)
34
42
  from kubernator.plugins.k8s_api import (K8SResourcePluginMixin,
35
43
  K8SResource,
36
44
  K8SResourcePatchType,
37
45
  K8SPropagationPolicy)
38
46
 
39
47
  logger = logging.getLogger("kubernator.k8s")
48
+ proc_logger = logger.getChild("proc")
49
+ stdout_logger = StripNL(proc_logger.info)
50
+ stderr_logger = StripNL(proc_logger.warning)
40
51
 
41
52
 
42
53
  def final_resource_validator(resources: Sequence[K8SResource],
@@ -62,6 +73,8 @@ class KubernetesPlugin(KubernatorPlugin, K8SResourcePluginMixin):
62
73
  super().__init__()
63
74
  self.context = None
64
75
 
76
+ self.embedded_pkg_version = self._get_kubernetes_client_version()
77
+
65
78
  self._transformers = []
66
79
  self._validators = []
67
80
 
@@ -109,30 +122,64 @@ class KubernetesPlugin(KubernatorPlugin, K8SResourcePluginMixin):
109
122
  def _kubeconfig_changed(self):
110
123
  self.setup_client()
111
124
 
125
+ def _get_kubernetes_client_version(self):
126
+ return pkg_version("kubernetes").split(".")
127
+
112
128
  def setup_client(self):
129
+ k8s = self.context.k8s
130
+ if "server_version" not in k8s:
131
+ self._setup_client()
132
+
133
+ server_minor = k8s.server_version[1]
134
+ pkg_major = self.embedded_pkg_version[0]
135
+ if server_minor != pkg_major:
136
+ logger.info("Bundled Kubernetes client version %s doesn't match server version %s",
137
+ ".".join(self.embedded_pkg_version), ".".join(k8s.server_version))
138
+ pkg_dir = install_python_k8s_client(self.context.app.run, server_minor, stdout_logger, stderr_logger)
139
+
140
+ modules_to_delete = []
141
+ for k, v in sys.modules.items():
142
+ if k == "kubernetes" or k.startswith("kubernetes."):
143
+ modules_to_delete.append(k)
144
+ for k in modules_to_delete:
145
+ del sys.modules[k]
146
+
147
+ logger.info("Adding sys.path reference to %s", pkg_dir)
148
+ sys.path.insert(0, str(pkg_dir))
149
+ self.embedded_pkg_version = self._get_kubernetes_client_version()
150
+ logger.info("Switching to Kubernetes client version %s", ".".join(self.embedded_pkg_version))
151
+ self._setup_client()
152
+ else:
153
+ logger.info("Bundled Kubernetes client version %s matches server version %s",
154
+ ".".join(self.embedded_pkg_version), ".".join(k8s.server_version))
155
+
156
+ logger.debug("Reading Kubernetes OpenAPI spec for %s", k8s.server_git_version)
157
+
158
+ k8s_def = load_remote_file(logger, f"https://raw.githubusercontent.com/kubernetes/kubernetes/"
159
+ f"{k8s.server_git_version}/api/openapi-spec/swagger.json",
160
+ FileType.JSON)
161
+ self.resource_definitions_schema = k8s_def
162
+
163
+ self._populate_resource_definitions()
164
+
165
+ def _setup_client(self):
113
166
  from kubernetes import client
114
167
 
115
168
  context = self.context
169
+ k8s = context.k8s
116
170
 
117
- context.k8s.client = self._setup_k8s_client()
118
- version = client.VersionApi(context.k8s.client).get_code()
171
+ k8s.client = self._setup_k8s_client()
172
+ version = client.VersionApi(k8s.client).get_code()
119
173
  if "-eks-" in version.git_version:
120
174
  git_version = version.git_version.split("-")[0]
121
175
  else:
122
176
  git_version = version.git_version
123
177
 
124
- context.k8s.server_version = git_version
125
-
126
- logger.info("Found Kubernetes %s on %s", version.git_version, context.k8s.client.configuration.host)
178
+ k8s.server_version = git_version[1:].split(".")
179
+ k8s.server_git_version = git_version
127
180
 
128
- logger.debug("Reading Kubernetes OpenAPI spec for version %s", git_version)
129
-
130
- k8s_def = load_remote_file(logger, f"https://raw.githubusercontent.com/kubernetes/kubernetes/"
131
- f"{git_version}/api/openapi-spec/swagger.json",
132
- FileType.JSON)
133
- self.resource_definitions_schema = k8s_def
134
-
135
- self._populate_resource_definitions()
181
+ logger.info("Found Kubernetes %s on %s", k8s.server_git_version, k8s.client.configuration.host)
182
+ K8SResource._k8s_client_version = tuple(map(int, pkg_version("kubernetes").split(".")))
136
183
 
137
184
  def handle_before_dir(self, cwd: Path):
138
185
  context = self.context
@@ -337,6 +337,8 @@ class K8SResourceKey(namedtuple("K8SResourceKey", ["group", "kind", "name", "nam
337
337
 
338
338
 
339
339
  class K8SResource:
340
+ _k8s_client_version = None
341
+
340
342
  def __init__(self, manifest: dict, rdef: K8SResourceDef, source: Union[str, Path] = None):
341
343
  self.key = self.get_manifest_key(manifest)
342
344
 
@@ -413,7 +415,7 @@ class K8SResource:
413
415
  rdef = self.rdef
414
416
  kwargs = {"name": self.name,
415
417
  "body": json_patch
416
- if patch_type != K8SResourcePatchType.SERVER_SIDE_PATCH
418
+ if patch_type != K8SResourcePatchType.SERVER_SIDE_PATCH or self._k8s_client_version[0] > 24
417
419
  else json.dumps(json_patch),
418
420
  "_preload_content": False,
419
421
  "field_manager": "kubernator",
@@ -49,7 +49,7 @@ class KubectlPlugin(KubernatorPlugin):
49
49
  def set_context(self, context):
50
50
  self.context = context
51
51
 
52
- def kubectl_stanza(self):
52
+ def stanza(self):
53
53
  context = self.context.kubectl
54
54
  return [context.kubectl_file, f"--kubeconfig={context.kubeconfig}"]
55
55
 
@@ -73,7 +73,7 @@ class KubectlPlugin(KubernatorPlugin):
73
73
  kubectl_file = str(Path(self.kubectl_dir.name) / "kubectl")
74
74
  copy(kubectl_file_dl, kubectl_file)
75
75
  os.chmod(kubectl_file, 0o500)
76
- prepend_os_path(str(self.kubectl_dir))
76
+ prepend_os_path(self.kubectl_dir.name)
77
77
  else:
78
78
  # Use current version
79
79
  kubectl_file = which("kubectl")
@@ -85,14 +85,14 @@ class KubectlPlugin(KubernatorPlugin):
85
85
  context.globals.kubectl = dict(version=version,
86
86
  kubeconfig=kubeconfig,
87
87
  kubectl_file=kubectl_file,
88
- kubectl_stanza=self.kubectl_stanza,
88
+ stanza=self.stanza,
89
89
  test=self.test_kubectl
90
90
  )
91
91
 
92
92
  context.globals.kubectl.version = context.kubectl.test()
93
93
 
94
94
  def test_kubectl(self):
95
- version_out: str = self.context.app.run_capturing_out(self.kubectl_stanza() +
95
+ version_out: str = self.context.app.run_capturing_out(self.stanza() +
96
96
  ["version", "--client=true", "-o", "json"],
97
97
  stderr_logger)
98
98
 
@@ -100,7 +100,7 @@ class KubectlPlugin(KubernatorPlugin):
100
100
  kubectl_version = version_out_js["clientVersion"]["gitVersion"][1:]
101
101
 
102
102
  logger.info("Using kubectl %r version %r with stanza %r",
103
- self.context.kubectl.kubectl_file, kubectl_version, self.kubectl_stanza())
103
+ self.context.kubectl.kubectl_file, kubectl_version, self.stanza())
104
104
 
105
105
  return kubectl_version
106
106
 
@@ -18,11 +18,16 @@
18
18
 
19
19
  import logging
20
20
  import os
21
+ import tempfile
22
+ from pathlib import Path
21
23
 
22
24
  from kubernator.api import (KubernatorPlugin,
23
25
  StripNL,
24
26
  get_golang_os,
25
- get_golang_machine
27
+ get_golang_machine,
28
+ prepend_os_path,
29
+ get_cache_dir,
30
+ CalledProcessError
26
31
  )
27
32
 
28
33
  logger = logging.getLogger("kubernator.minikube")
@@ -38,6 +43,7 @@ class MinikubePlugin(KubernatorPlugin):
38
43
 
39
44
  def __init__(self):
40
45
  self.context = None
46
+ self.minikube_dir = None
41
47
  self.minikube_home_dir = None
42
48
  self.kubeconfig_dir = None
43
49
 
@@ -64,38 +70,132 @@ class MinikubePlugin(KubernatorPlugin):
64
70
  map(lambda line: line.split()[1][11:].split("."),
65
71
  versions.splitlines(False))))))[-1])))
66
72
 
67
- def cmd(self, ):
68
- pass
73
+ def cmd(self, *extra_args):
74
+ stanza, env = self._stanza(list(extra_args))
75
+ return self.context.app.run(stanza, stdout_logger, stderr_logger, env=env).wait()
69
76
 
70
- def register(self, minikube_version=None, k8s_version=None):
77
+ def cmd_out(self, *extra_args):
78
+ stanza, env = self._stanza(list(extra_args))
79
+ return self.context.app.run_capturing_out(stanza, stderr_logger, env=env)
80
+
81
+ def _stanza(self, extra_args):
82
+ context = self.context
83
+ minikube = context.minikube
84
+ stanza = [context.minikube.minikube_file, "-p", f"minikube-{minikube.profile}"] + extra_args
85
+ env = dict(os.environ)
86
+ env["MINIKUBE_HOME"] = str(self.minikube_home_dir)
87
+ env["KUBECONFIG"] = str(minikube.kubeconfig)
88
+ return stanza, env
89
+
90
+ def register(self, minikube_version=None, profile="default", k8s_version=None,
91
+ keep_running=False, start_fresh=False,
92
+ nodes=1, driver="docker", cpus="no-limit", extra_args=None):
71
93
  context = self.context
72
94
 
73
- context.app.register_plugin("kubectl")
95
+ context.app.register_plugin("kubeconfig")
96
+
97
+ if not k8s_version:
98
+ msg = "No Kubernetes version is specified for Minikube"
99
+ logger.critical(msg)
100
+ raise RuntimeError(msg)
74
101
 
75
102
  if not minikube_version:
76
103
  minikube_version = self.get_latest_minikube_version()
77
104
  logger.info("No minikube version is specified, latest is %s", minikube_version)
78
105
 
79
- minikube_file = context.app.download_remote_file(logger,
80
- f"https://github.com/kubernetes/minikube/releases/download/"
81
- f"v{minikube_version}/"
82
- f"minikube-{get_golang_os()}-{get_golang_machine()}", "bin")
83
- os.chmod(minikube_file, 0o500)
106
+ minikube_dl_file, _ = context.app.download_remote_file(logger,
107
+ f"https://github.com/kubernetes/minikube/releases"
108
+ f"/download/v{minikube_version}/"
109
+ f"minikube-{get_golang_os()}-{get_golang_machine()}",
110
+ "bin")
111
+
112
+ os.chmod(minikube_dl_file, 0o500)
113
+ self.minikube_dir = tempfile.TemporaryDirectory()
114
+ minikube_file = Path(self.minikube_dir.name) / "minikube"
115
+ minikube_file.symlink_to(minikube_dl_file)
116
+ prepend_os_path(self.minikube_dir.name)
84
117
  version_out: str = self.context.app.run_capturing_out([minikube_file, "version", "--short"],
85
- stderr_logger)
118
+ stderr_logger).strip()
86
119
  version = version_out[1:]
87
- logger.info("Found minikube %s in %r", version, minikube_file)
120
+ logger.info("Found minikube %s in %s", version, minikube_file)
121
+
122
+ profile_dir = get_cache_dir("minikube", profile)
123
+ self.minikube_home_dir = profile_dir / "home"
124
+ self.minikube_home_dir.mkdir(parents=True, exist_ok=True)
125
+ self.kubeconfig_dir = profile_dir / ".kube"
126
+ self.kubeconfig_dir.mkdir(parents=True, exist_ok=True)
88
127
 
89
128
  context.globals.minikube = dict(version=version,
90
- minikube_file=minikube_file,
91
- cmd=self.cmd
129
+ minikube_file=str(minikube_file),
130
+ profile=profile,
131
+ k8s_version=k8s_version,
132
+ start_fresh=start_fresh,
133
+ keep_running=keep_running,
134
+ nodes=nodes,
135
+ driver=driver,
136
+ cpus=cpus,
137
+ extra_args=extra_args or [],
138
+ kubeconfig=str(self.kubeconfig_dir / "config"),
139
+ cmd=self.cmd,
140
+ cmd_out=self.cmd_out
92
141
  )
93
-
94
- def handle_init(self):
95
- pass
142
+ context.kubeconfig.kubeconfig = context.minikube.kubeconfig
143
+
144
+ logger.info("Minikube Home is %s", self.minikube_home_dir)
145
+ logger.info("Minikube Kubeconfig is %s", context.minikube.kubeconfig)
146
+
147
+ def minikube_is_running(self):
148
+ try:
149
+ out = self.cmd_out("status", "-o", "json")
150
+ logger.info("Minikube profile %r is running: %s", self.context.minikube.profile,
151
+ out.strip())
152
+ return True
153
+ except CalledProcessError as e:
154
+ logger.info("Minikube profile %r is not running: %s", self.context.minikube.profile,
155
+ e.output.strip())
156
+ return False
157
+
158
+ def minikube_start(self):
159
+ minikube = self.context.minikube
160
+ if not self.minikube_is_running():
161
+ logger.info("Starting minikube profile %r...", minikube.profile)
162
+ self.cmd("start",
163
+ "--driver", str(minikube.driver),
164
+ "--nodes", str(minikube.nodes),
165
+ "--cpus", str(minikube.cpus),
166
+ "--kubernetes-version", str(minikube.k8s_version),
167
+ "--wait", "apiserver")
168
+ else:
169
+ logger.warning("Minikube profile %r is already running!", minikube.profile)
170
+
171
+ logger.info("Updating minikube profile %r context", minikube.profile)
172
+ self.cmd("update-context")
173
+
174
+ def minikube_stop(self):
175
+ minikube = self.context.minikube
176
+ if self.minikube_is_running():
177
+ logger.info("Shutting down minikube profile %r...", minikube.profile)
178
+ self.cmd("stop", "-o", "json")
179
+
180
+ def minikube_delete(self):
181
+ minikube = self.context.minikube
182
+ self.minikube_stop()
183
+ logger.warning("Deleting minikube profile %r!", minikube.profile)
184
+ self.cmd("delete")
185
+
186
+ def handle_start(self):
187
+ minikube = self.context.minikube
188
+ if minikube.start_fresh:
189
+ self.minikube_delete()
190
+
191
+ self.minikube_start()
96
192
 
97
193
  def handle_shutdown(self):
98
- pass
194
+ minikube = self.context.minikube
195
+ if not minikube.keep_running:
196
+ self.minikube_stop()
197
+ else:
198
+ logger.warning("Will keep minikube profile %s running!", minikube.profile)
99
199
 
100
200
  def __repr__(self):
101
- return "AWS CLI Plugin"
201
+ return "Minikube Plugin"
@@ -51,7 +51,7 @@ class TerraformPlugin(KubernatorPlugin):
51
51
  def set_context(self, context):
52
52
  self.context = context
53
53
 
54
- def tf_stanza(self):
54
+ def stanza(self):
55
55
  return [self.tf_file]
56
56
 
57
57
  def register(self, version=None):
@@ -71,7 +71,7 @@ class TerraformPlugin(KubernatorPlugin):
71
71
  tf_zip.extractall(self.tf_dir.name)
72
72
 
73
73
  os.chmod(tf_file, 0o500)
74
- prepend_os_path(str(self.tf_dir))
74
+ prepend_os_path(self.tf_dir.name)
75
75
  else:
76
76
  # Use current version
77
77
  tf_file = which("terraform")
@@ -95,7 +95,7 @@ class TerraformPlugin(KubernatorPlugin):
95
95
 
96
96
  context.globals.terraform = dict(version=version,
97
97
  tf_file=self.tf_file,
98
- tf_stanza=self.tf_stanza,
98
+ stanza=self.stanza,
99
99
  )
100
100
 
101
101
  logger.info("Found Terraform version %s at %s", version, self.tf_file)
@@ -127,10 +127,10 @@ class TerraformPlugin(KubernatorPlugin):
127
127
  if not tf_detected:
128
128
  return
129
129
 
130
- context.app.run(self.tf_stanza() + ["init", "-reconfigure", "-input=false", "-upgrade=false"],
130
+ context.app.run(self.stanza() + ["init", "-reconfigure", "-input=false", "-upgrade=false"],
131
131
  stdout_logger, stderr_logger, cwd=cwd).wait()
132
132
 
133
- output = json.loads(context.app.run_capturing_out(self.tf_stanza() + ["output", "-json"],
133
+ output = json.loads(context.app.run_capturing_out(self.stanza() + ["output", "-json"],
134
134
  stderr_logger, cwd=cwd))
135
135
  if not output:
136
136
  raise RuntimeError("Terraform output produced no values. Please check if Terraform is functioning.")
@@ -49,7 +49,7 @@ class TerragruntPlugin(KubernatorPlugin):
49
49
  def set_context(self, context):
50
50
  self.context = context
51
51
 
52
- def tg_stanza(self):
52
+ def stanza(self):
53
53
  return [self.tg_file]
54
54
 
55
55
  def register(self, version=None):
@@ -66,7 +66,7 @@ class TerragruntPlugin(KubernatorPlugin):
66
66
  tg_file = Path(self.tg_dir.name) / "terragrunt"
67
67
  copy(tg_file_cache, tg_file)
68
68
  os.chmod(tg_file, 0o500)
69
- prepend_os_path(str(self.tg_dir))
69
+ prepend_os_path(self.tg_dir.name)
70
70
  else:
71
71
  # Use current version
72
72
  tg_file = which("terragrunt")
@@ -81,7 +81,7 @@ class TerragruntPlugin(KubernatorPlugin):
81
81
  version = version_out.split(" ")[-1][1:].strip()
82
82
  context.globals.terragrunt = dict(version=version,
83
83
  tg_file=self.tg_file,
84
- tg_stanza=self.tg_stanza,
84
+ stanza=self.stanza,
85
85
  )
86
86
 
87
87
  logger.info("Found Terragrunt version %s at %s", version, self.tg_file)
@@ -113,10 +113,10 @@ class TerragruntPlugin(KubernatorPlugin):
113
113
  if not tg_detected:
114
114
  return
115
115
 
116
- context.app.run(self.tg_stanza() + ["run-all", "init", "-reconfigure", "-input=false", "-upgrade=false"],
116
+ context.app.run(self.stanza() + ["run-all", "init", "-reconfigure", "-input=false", "-upgrade=false"],
117
117
  stdout_logger, stderr_logger, cwd=cwd).wait()
118
118
 
119
- output_json = context.app.run_capturing_out(self.tg_stanza() + ["run-all", "output", "-json"],
119
+ output_json = context.app.run_capturing_out(self.stanza() + ["run-all", "output", "-json"],
120
120
  stderr_logger, cwd=cwd)
121
121
 
122
122
  json_decoder = json.JSONDecoder()
kubernator/proc.py CHANGED
@@ -70,13 +70,16 @@ class ProcessRunner:
70
70
  safe_args=None, universal_newlines=True, **kwargs):
71
71
  self._safe_args = safe_args or args
72
72
  logger.trace("Starting %r", self._safe_args)
73
+
74
+ if "env" not in kwargs:
75
+ kwargs["env"] = os.environ
76
+
73
77
  self._proc = Popen(args,
74
78
  stdout=PIPE if isinstance(stdout, Callable) else (stdout if stdout is not None else DEVNULL),
75
79
  stderr=PIPE if isinstance(stderr, Callable) else (stderr if stderr is not None else DEVNULL),
76
80
  stdin=PIPE if isinstance(stdin, (Callable, bytes, str)) else
77
81
  (stdin if stdin is not None else DEVNULL),
78
82
  universal_newlines=universal_newlines,
79
- env=os.environ if "env" not in kwargs else kwargs["env"],
80
83
  **kwargs)
81
84
 
82
85
  self._stdin_writer = (spawn(partial(stream_writer_text if universal_newlines else stream_writer_buf,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: kubernator
3
- Version: 1.0.4
3
+ Version: 1.0.6
4
4
  Summary: Kubernator is the a pluggable framework for K8S provisioning
5
5
  Home-page: https://github.com/karellen/kubernator
6
6
  Author: Express Systems USA, Inc.
@@ -0,0 +1,27 @@
1
+ kubernator/LICENSE,sha256=wKKdOCMTCPQRV5gDkVLAsXX8qSnRJ5owk7yWPO1KZNo,11387
2
+ kubernator/__init__.py,sha256=PTgWFGl8toikIoJ1DhGqzDU4n_T6ysuFFY_78v2Ho3w,914
3
+ kubernator/__main__.py,sha256=f0S60wgpLu--1UlOhzfWail-xt8zyIuODodX98_yPN0,707
4
+ kubernator/api.py,sha256=nu6dHXAayk1M4Mc5lni4GiQ6kKsrjWmaUM4BwmotR5E,24168
5
+ kubernator/app.py,sha256=kGS8jVcf-nCY1mP6MdqMR27XNcEMb5Uxek9FEW4GWLg,19472
6
+ kubernator/proc.py,sha256=8YlgbppyHic_51fVTPD7OP8x5yuRuL8j1kThR8MJjNI,5119
7
+ kubernator/plugins/__init__.py,sha256=h9TLYK8UFEi53ipZSZsTBjp0ljKhisWsgPpt_PkWrO8,670
8
+ kubernator/plugins/awscli.py,sha256=S6X7-qFiaZ7NDVDl2Jg0t-ih9KAJ45cUjjzd5Qe93ZM,4252
9
+ kubernator/plugins/eks.py,sha256=xe7vyPHNwuP8gEYDSzPyBkm-RkAtP64wCOqs9U5I7xI,2273
10
+ kubernator/plugins/helm.py,sha256=CHuK97TTLf6j6lTL1FSB51xyQ_Jqf-lAVZId_NHhQfQ,10469
11
+ kubernator/plugins/istio.py,sha256=ixY-6H_hH55rBBcWJqq6SCkyHrVIu4iJdKm18okxdfk,9777
12
+ kubernator/plugins/k8s.py,sha256=klbAU-DIDNzY-aXfbhxQKHwGhStTmJ-k7ydr1ckUqr8,20248
13
+ kubernator/plugins/k8s_api.py,sha256=xsDQii32khCUOc4hMlNc0zHLQK73VTb4kdnug8ukmy4,26000
14
+ kubernator/plugins/kops.py,sha256=QsrQJUF6wGJo2QRVqP92pG5vmOTYQIc25PD_DWCzrAA,9635
15
+ kubernator/plugins/kubeconfig.py,sha256=uwtHmF2I6LiTPrC3M88G5SfYxDWtuh0MqcMfrHHoNRA,2178
16
+ kubernator/plugins/kubectl.py,sha256=4Z01zDCni0B60F4L2ImuA3Ob8HMD1YK-DD9FXfZ1Z6Y,3861
17
+ kubernator/plugins/minikube.py,sha256=cJYnjGcQdNNrTAisDzF5kJXuAeYbRHUBc-XCuxBwEY8,8386
18
+ kubernator/plugins/template.py,sha256=o6UqLw1wK91gCfN7q-Wwc3P5iycAeELUryXh-egA2vQ,8070
19
+ kubernator/plugins/terraform.py,sha256=bw-0KUBYB1bKEeBTfSYD5A1PGHtuGwOVLV86z6zznkc,5231
20
+ kubernator/plugins/terragrunt.py,sha256=UbV_Hm2wO8lPpkRvE9PGCbtpcWVjDJedAv-SHw7KngE,4996
21
+ kubernator-1.0.6.dist-info/METADATA,sha256=SBKFOc51gi4deyJyGzk2l6Sg4oMOPYCWsHYRMXZKiHs,10300
22
+ kubernator-1.0.6.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
23
+ kubernator-1.0.6.dist-info/entry_points.txt,sha256=IWDtHzyTleRqDSuVRXEr5GImrI7z_kh21t5DWZ8ldcQ,47
24
+ kubernator-1.0.6.dist-info/namespace_packages.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
25
+ kubernator-1.0.6.dist-info/top_level.txt,sha256=_z1CxWeKMI55ckf2vC8HqjbCn_E2Y_P5RdrhE_QWcIs,11
26
+ kubernator-1.0.6.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
27
+ kubernator-1.0.6.dist-info/RECORD,,
@@ -1,27 +0,0 @@
1
- kubernator/LICENSE,sha256=wKKdOCMTCPQRV5gDkVLAsXX8qSnRJ5owk7yWPO1KZNo,11387
2
- kubernator/__init__.py,sha256=vrdZZ51ljtDBriYt-0tQJ9VzAEoxoaNtXoB6kHOzeis,909
3
- kubernator/__main__.py,sha256=f0S60wgpLu--1UlOhzfWail-xt8zyIuODodX98_yPN0,707
4
- kubernator/api.py,sha256=ugpHtUMJypwMJ9E3aZvmiPhQu0DNosCGJPsak2NtLnU,23582
5
- kubernator/app.py,sha256=Un6BvgD_L8DUhB4rZiRDoWGVE1TKDcMcBR0UKSYWwgI,18087
6
- kubernator/proc.py,sha256=MNJwcpSwY_j-WfxEtI3CIhC59rID47GQwUMOFiIQEMk,5131
7
- kubernator/plugins/__init__.py,sha256=h9TLYK8UFEi53ipZSZsTBjp0ljKhisWsgPpt_PkWrO8,670
8
- kubernator/plugins/awscli.py,sha256=TvxKHTtu_oxS0O4jTpOBqxfnn9a9J--mqTYS9MNLOMg,4248
9
- kubernator/plugins/eks.py,sha256=PbbnopFxy7sF66GLAq_BazjUKSPQvKsS_Z4vqUDJ7pA,2270
10
- kubernator/plugins/helm.py,sha256=9WkeYCjrLWzUN5e_MZdH_L_sQXiXtyLZ10sW47CpRLU,10504
11
- kubernator/plugins/istio.py,sha256=WnfRuClbl4Z5tYRMJjVwPIHn66BXHxV7QVBAt6RMMig,9781
12
- kubernator/plugins/k8s.py,sha256=EBb92zf7pV1EgOmjVvhnrwBRHihUBxxZVF4MH-Mjwjk,18129
13
- kubernator/plugins/k8s_api.py,sha256=SDxU86Uthy1fcTotAdRWSv13BMoyO0znfcP_7aQ0yIM,25932
14
- kubernator/plugins/kops.py,sha256=QsrQJUF6wGJo2QRVqP92pG5vmOTYQIc25PD_DWCzrAA,9635
15
- kubernator/plugins/kubeconfig.py,sha256=uwtHmF2I6LiTPrC3M88G5SfYxDWtuh0MqcMfrHHoNRA,2178
16
- kubernator/plugins/kubectl.py,sha256=l3H3hjZsf_QTU-UnM5WqaNQkzz6NzTErQy6wlzSWsQU,3901
17
- kubernator/plugins/minikube.py,sha256=crXSXVf7O4JLS1d5-2tjKOFEBurYFgzM18x9dEmbpc8,3760
18
- kubernator/plugins/template.py,sha256=o6UqLw1wK91gCfN7q-Wwc3P5iycAeELUryXh-egA2vQ,8070
19
- kubernator/plugins/terraform.py,sha256=2fP776nTWH06sfBcyOlJTaAMZGEnHPy8eZsDs0JPYMo,5246
20
- kubernator/plugins/terragrunt.py,sha256=FEAqohdvOB2rAXHOR-SyS3A_GuHhvrLuE3P02qlVDSI,5011
21
- kubernator-1.0.4.dist-info/METADATA,sha256=-jhH3tNI3oqHTZGYQ7xdlJQVY2dYCjb15uUD6vqiHNM,10300
22
- kubernator-1.0.4.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
23
- kubernator-1.0.4.dist-info/entry_points.txt,sha256=IWDtHzyTleRqDSuVRXEr5GImrI7z_kh21t5DWZ8ldcQ,47
24
- kubernator-1.0.4.dist-info/namespace_packages.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
25
- kubernator-1.0.4.dist-info/top_level.txt,sha256=_z1CxWeKMI55ckf2vC8HqjbCn_E2Y_P5RdrhE_QWcIs,11
26
- kubernator-1.0.4.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
27
- kubernator-1.0.4.dist-info/RECORD,,