scott/bitwarden-ssh-agent
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix encoding & add session param

Browse source
This commit is contained in:
Weidows 2022-10-13 22:23:49 +08:00
parent f4d7dd2bf0
commit c56b7041e4
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bw_add_sshkeys.py
View file

@ -56,12 +56,13 @@ def cli_supports(feature: str) -> bool:
return False return False
def get_session() -> str: def get_session(session: str) -> str:
""" """
Function to return a valid Bitwarden session Function to return a valid Bitwarden session
""" """
# Check for an existing, user-supplied Bitwarden session # Check for an existing, user-supplied Bitwarden session
session = os.environ.get('BW_SESSION', '') if not session:
session = os.environ.get('BW_SESSION', '')
if session: if session:
logging.debug('Existing Bitwarden session found') logging.debug('Existing Bitwarden session found')
return session return session
@ -101,6 +102,7 @@ def get_folders(session: str, foldername: str) -> str:
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
universal_newlines=True, universal_newlines=True,
check=True, check=True,
encoding = "utf-8",
) )
folders = json.loads(proc_folders.stdout) folders = json.loads(proc_folders.stdout)
@ -128,6 +130,7 @@ def folder_items(session: str, folder_id: str) -> List[Dict[str, Any]]:
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
universal_newlines=True, universal_newlines=True,
check=True, check=True,
encoding = "utf-8",
) )
data: List[Dict[str, Any]] = json.loads(proc_items.stdout) data: List[Dict[str, Any]] = json.loads(proc_items.stdout)
@ -270,6 +273,12 @@ if __name__ == '__main__':
default='passphrase', default='passphrase',
help='custom field name where key passphrase is stored', help='custom field name where key passphrase is stored',
) )
parser.add_argument(
'-s',
'--session',
default='',
help='session key of bitwarden',
)
return parser.parse_args() return parser.parse_args()
@ -289,7 +298,7 @@ if __name__ == '__main__':
try: try:
logging.info('Getting Bitwarden session') logging.info('Getting Bitwarden session')
session = get_session() session = get_session(args.session)
logging.debug('Session = %s', session) logging.debug('Session = %s', session)
logging.info('Getting folder list') logging.info('Getting folder list')