116 lines
2.8 KiB
Python
116 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from collections import namedtuple
|
|
import subprocess as sp
|
|
import pprint
|
|
import json
|
|
import logging
|
|
import tempfile
|
|
import os
|
|
import csv
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
|
|
OP_BIN = 'op'
|
|
ZIP_BIN = 'zip'
|
|
|
|
Record = namedtuple('Record', [
|
|
'title',
|
|
'username',
|
|
'password',
|
|
'url',
|
|
])
|
|
|
|
Vault = namedtuple('Vault', ['title', 'records'])
|
|
|
|
DUMMY_RECORD_SET = [
|
|
Record(
|
|
title='title' + str(i),
|
|
username='username' + str(i),
|
|
password='password' + str(i),
|
|
url='url' + str(i),
|
|
)
|
|
for i in range(10)
|
|
]
|
|
|
|
def main():
|
|
# dir_path = tempfile.mkdtemp(suffix='password-export')
|
|
dir_path = './tmp'
|
|
vaults = [Vault(title='test', records=DUMMY_RECORD_SET)]
|
|
for vault in vaults:
|
|
save_vault(dir_path, vault)
|
|
|
|
|
|
def process_vault(vault):
|
|
vault_uuid = vault['uuid']
|
|
vault_name = vault['name']
|
|
logging.info('Vault "{}": Fetch items'.format(vault_name))
|
|
items = retrieve_items(vault_uuid)
|
|
records = []
|
|
logging.info('Vault "{}": {}/{}'.format(vault_name, 0, len(items)))
|
|
for index, item in enumerate(items):
|
|
item_data = retrieve_item(item['uuid'])
|
|
extracted = extract_item_fields(item_data)
|
|
records.append(extracted)
|
|
logging.info('Vault "{}": {}/{}'.format(vault_name, index + 1, len(items)))
|
|
return Vault(title=vault_name, records=records)
|
|
|
|
|
|
def save_vault(base_dir, vault):
|
|
file_name = os.path.join(base_dir, vault.title.lower())
|
|
with open(file_name, 'w') as csvfile:
|
|
record_writer = csv.writer(csvfile)
|
|
for record in vault.records:
|
|
record_writer.writerow([
|
|
record.title,
|
|
record.username,
|
|
record.password,
|
|
record.url
|
|
])
|
|
|
|
|
|
def retrieve_vaults():
|
|
data = catch_op_json([OP_BIN, 'list', 'vaults'])
|
|
return data
|
|
|
|
def retrieve_items(vault_uuid):
|
|
data = catch_op_json([OP_BIN, 'list', 'items', '--vault=' + vault_uuid])
|
|
return data
|
|
|
|
|
|
def retrieve_item(item_uuid):
|
|
data = catch_op_json([OP_BIN, 'get', 'item', item_uuid])
|
|
return data
|
|
|
|
|
|
def extract_item_fields(item_data):
|
|
details = item_data.get('details', {})
|
|
fields = details.get('fields', [])
|
|
overview = item_data.get('overview', {})
|
|
username = value_from_fields(fields, 'username')
|
|
password = value_from_fields(fields, 'password') or details.get('password', '')
|
|
return Record(
|
|
title=overview.get('title', ''),
|
|
username=username,
|
|
password=password,
|
|
url=overview.get('url', ''),
|
|
)
|
|
|
|
|
|
def value_from_fields(item_fields, key):
|
|
for f in item_fields:
|
|
if f.get('designation', '') == key:
|
|
return f.get('value', '')
|
|
return ''
|
|
|
|
|
|
def catch_op_json(args):
|
|
output = sp.run(args, check=True, stdout=sp.PIPE).stdout
|
|
return json.loads(output.decode('utf-8'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|