Format code

This commit is contained in:
Anton Vakhrushev 2019-10-24 12:35:13 +03:00
parent ec2c0a3277
commit eeddce803f

View File

@ -5,26 +5,22 @@ import argparse
import csv import csv
import json import json
import os import os
import shutil
import subprocess as sp import subprocess as sp
import tempfile import tempfile
from collections import namedtuple from collections import namedtuple
OP_BIN = 'op' OP_BIN = 'op'
GPG_BIN = 'gpg' GPG_BIN = 'gpg'
TAR_BIN = 'tar' TAR_BIN = 'tar'
Record = namedtuple('Record', [ Record = namedtuple('Record', [
'title', 'title',
'username', 'username',
'password', 'password',
'url', 'url',
]) ])
Vault = namedtuple('Vault', ['title', 'records']) Vault = namedtuple('Vault', ['title', 'records'])
DUMMY_RECORD_SET = [ DUMMY_RECORD_SET = [
Record( Record(
@ -87,8 +83,8 @@ def process_vault(vault):
def save_vault(base_dir, vault): def save_vault(base_dir, vault):
file_name = os.path.join(base_dir, vault.title.lower()) file_name = os.path.join(base_dir, vault.title.lower())
with open(file_name, 'w') as csvfile: with open(file_name, 'w') as csv_file:
record_writer = csv.writer(csvfile) record_writer = csv.writer(csv_file)
for record in vault.records: for record in vault.records:
record_writer.writerow([ record_writer.writerow([
record.title, record.title,
@ -108,8 +104,8 @@ def export_encrypted(dir_path, file_name, password):
tar_name = os.path.join(tar_dir_path, os.path.basename(file_name) + '.tar') tar_name = os.path.join(tar_dir_path, os.path.basename(file_name) + '.tar')
print('Store temp tar in ' + tar_name) print('Store temp tar in ' + tar_name)
sp.run([TAR_BIN, '-cf', tar_name, '-C', dir_path, '.']) sp.run([TAR_BIN, '-cf', tar_name, '-C', dir_path, '.'])
sp.run([GPG_BIN, '--symmetric', '--batch', '--yes', '--passphrase', password, sp.run([GPG_BIN, '--symmetric', '--batch', '--yes', '--passphrase', password,
'--output', gpg_name, tar_name]) '--output', gpg_name, tar_name])
def retrieve_vaults(): def retrieve_vaults():
@ -118,7 +114,7 @@ def retrieve_vaults():
return data return data
def retrieve_items(vault_uuid): def retrieve_items(vault_uuid):
data = catch_op_json([OP_BIN, 'list', 'items', '--vault=' + vault_uuid]) data = catch_op_json([OP_BIN, 'list', 'items', '--vault=' + vault_uuid])
return data return data
@ -150,7 +146,7 @@ def value_from_fields(item_fields, key):
def catch_op_json(args): def catch_op_json(args):
output = sp.run(args, check=True, stdout=sp.PIPE).stdout output = sp.run(args, check=True, stdout=sp.PIPE).stdout
return json.loads(output.decode('utf-8')) return json.loads(output.decode('utf-8'))