Format code
This commit is contained in:
parent
ec2c0a3277
commit
eeddce803f
18
1export.py
18
1export.py
@ -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'))
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user