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

attachments now will be downloaded to a temporary directory

Browse source
This commit is contained in:
Joao Jacome 2020-03-07 20:00:11 +00:00
parent a814bc8eaf
commit e1d4980819
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
ssh.py
View file

@ -2,6 +2,7 @@ import subprocess
import os import os
import sys import sys
import json import json
import tempfile
from urllib.parse import urlparse, urlencode from urllib.parse import urlparse, urlencode
from urllib.request import urlopen, Request from urllib.request import urlopen, Request
from urllib.error import HTTPError from urllib.error import HTTPError
@ -46,6 +47,9 @@ except Exception as e:
sys.exit(1) sys.exit(1)
keys = [] keys = []
directory = tempfile.TemporaryDirectory()
temporary_file_name = os.path.join(directory.name + './private_key')
try: try:
for item in items: for item in items:
private_key_file = [k['value'] for k in item['fields'] if k['name'] == 'private' and k['type'] == 0][0] private_key_file = [k['value'] for k in item['fields'] if k['name'] == 'private' and k['type'] == 0][0]
@ -53,9 +57,9 @@ try:
private_key_id = [k['id'] for k in item['attachments'] if k['fileName'] == private_key_file][0] private_key_id = [k['id'] for k in item['attachments'] if k['fileName'] == private_key_file][0]
# would be nice if there was an option to retrieve the attachment file directly to the stdout # would be nice if there was an option to retrieve the attachment file directly to the stdout
subprocess.check_output(['bw', 'get', 'attachment', private_key_id, '--itemid', item['id'], '--output', './private_key'] + session) subprocess.check_output(['bw', 'get', 'attachment', private_key_id, '--itemid', item['id'], '--output', temporary_file_name] + session)
private_key = open('private_key', 'r').read() private_key = open(temporary_file_name, 'r').read()
os.remove('./private_key') os.remove(temporary_file_name)
keys.append({'private_key': private_key}) keys.append({'private_key': private_key})
except: except:
print('Something happened.') print('Something happened.')