import csv
import os
import sys
import gnupg
## Usage
# Run pass2csv /path/to/password-storage
## options
# field names
userfield = "user"
urlfield = "url"
def traverse(path):
for root, dirs, files in os.walk(path):
if '.git' in dirs:
dirs.remove('.git')
for name in files:
yield os.path.join(root, name)
def parse(basepath, path, data):
name = os.path.splitext(os.path.basename(path))[0]
group = os.path.dirname(os.path.os.path.relpath(path, basepath))
split_data = data.split('\n')
password = split_data[0]
matching_user = [s for s in split_data if userfield+": " in s]
user = None
url = None
if matching_user:
for x in matching_user:
user_split = x.split(userfield+": ")
if len(user_split) == 2:
user = user_split[1]
else:
user = None
matching_url = [s for s in split_data if urlfield+": " in s]
if matching_url:
for x in matching_url:
url_split = x.split(urlfield+": ")
if len(url_split) == 2:
url = url_split[1]
else:
url = None
if url == "None":
if user == "None":
return [group, name, password]
else:
return [group, name, password, user]
else:
if user == "None":
return [group, name, password, url]
else:
return [group, name, password, url, user]
def main(path):
gpg = gnupg.GPG()
gpg.encoding = 'utf-8'
csv_data = []
for file_path in traverse(path):
if os.path.splitext(file_path)[1] == '.gpg':
with open(file_path, 'rb') as f:
data = str(gpg.decrypt_file(f))
csv_data.append(parse(path, file_path, data))
with open('pass.csv', 'w', newline='') as csv_file:
writer = csv.writer(csv_file, delimiter=',')
writer.writerows(csv_data)
if __name__ == '__main__':
path = os.path.abspath(sys.argv[1])
main(path)
