JustScreaMy/pyinfra
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixes

Browse source
This commit is contained in:
Jakub Kropáček 2024-07-24 00:09:16 +02:00
parent c3ba2b4a27
commit a8b15a8a75
1 changed files with 20 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
scripts/bw2secrets
View file

@ -1,27 +1,19 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse import argparse
import enum
import getpass import getpass
import shutil import shutil
import subprocess as sp import subprocess as sp
from pathlib import Path from pathlib import Path
from typing import Literal
import jinja2 import jinja2
bitwarden_session = None bitwarden_session = None
TemplateEnvType = Literal["password", "username"]
class TemplateEnvType(enum.StrEnum):
def fetch_secret(bw_path: Path, secret_id: str, object_type: str = "password") -> str: USERNAME = "username"
global bitwarden_session PASSWORD = "password"
res = sp.run(
[bw_path, "get", object_type, secret_id, "--session", bitwarden_session],
capture_output=True,
text=True,
)
res.check_returncode()
return res.stdout
class TemplateEnv: class TemplateEnv:
@ -29,19 +21,29 @@ class TemplateEnv:
env_type: TemplateEnvType env_type: TemplateEnvType
cached_items: dict[str, str] cached_items: dict[str, str]
def _fetch_secret(self, secret_id: str) -> str:
global bitwarden_session
res = sp.run(
[
self.bw_path, "get", self.env_type.value,
secret_id, "--session", bitwarden_session,
],
capture_output=True,
text=True,
)
res.check_returncode()
return res.stdout
def __init__(self, _type: TemplateEnvType, bw_path: Path): def __init__(self, _type: TemplateEnvType, bw_path: Path):
self.env_type = _type self.env_type = _type
self.bw_path = bw_path self.bw_path = bw_path
self.cached_items = dict() self.cached_items = dict()
def __getitem__(self, item): def __getitem__(self, item):
print(f"{self.env_type} {self.cached_items}")
if cached_item := self.cached_items.get(item): if cached_item := self.cached_items.get(item):
return cached_item return cached_item
self.cached_items[item] = fetch_secret( self.cached_items[item] = self._fetch_secret(
self.bw_path,
item, item,
self.env_type,
) )
return self.cached_items[item] return self.cached_items[item]
@ -101,8 +103,8 @@ def compile_file(file_path: Path, bw_path: Path):
jinja_env = jinja2.Environment( jinja_env = jinja2.Environment(
loader=jinja2.FileSystemLoader(file_path.parent), loader=jinja2.FileSystemLoader(file_path.parent),
) )
username = TemplateEnv("username", bw_path) username = TemplateEnv(TemplateEnvType.USERNAME, bw_path)
password = TemplateEnv("password", bw_path) password = TemplateEnv(TemplateEnvType.PASSWORD, bw_path)
template = jinja_env.get_template(file_path.name) template = jinja_env.get_template(file_path.name)
rendered_template = template.render( rendered_template = template.render(