136 lines
4.0 KiB
Python
Executable File
136 lines
4.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
|
|
class Record:
|
|
def __init__(self, name, url, username, password, extra, grouping, fav):
|
|
self.name_val = name
|
|
self.url = url
|
|
self.username = username
|
|
self.password = password
|
|
self.extra = extra
|
|
self.grouping = grouping
|
|
self.fav = fav
|
|
|
|
@property
|
|
def name(self):
|
|
s = "lastpass/"
|
|
if self.grouping:
|
|
s += f"{self.grouping}/"
|
|
if self.name_val:
|
|
s += self.name_val
|
|
s = s.replace(" ", "_").replace("'", "")
|
|
return s
|
|
|
|
def to_str(self):
|
|
s = f"{self.password}\n---\n"
|
|
if self.grouping:
|
|
s += f"{self.grouping} / "
|
|
if self.name_val:
|
|
s += f"{self.name_val}\n"
|
|
if self.username:
|
|
s += f"username: {self.username}\n"
|
|
if self.password:
|
|
s += f"password: {self.password}\n"
|
|
if self.url and self.url != "http://sn":
|
|
s += f"url: {self.url}\n"
|
|
if self.extra:
|
|
s += f"{self.extra}\n"
|
|
return s
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Import LastPass CSV export into pass")
|
|
parser.add_argument(
|
|
"-f", "--force", action="store_true", help="Overwrite existing records"
|
|
)
|
|
parser.add_argument(
|
|
"-d",
|
|
"--default",
|
|
metavar="GROUP",
|
|
default="",
|
|
help="Place uncategorised records into GROUP",
|
|
)
|
|
parser.add_argument("filename", help="Path to LastPass CSV file")
|
|
args = parser.parse_args()
|
|
|
|
print(f"Reading '{args.filename}'...")
|
|
|
|
entries = []
|
|
current_entry = []
|
|
entry_pattern = re.compile(r"^(http|ftp|ssh)")
|
|
|
|
try:
|
|
with open(args.filename, "r") as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if entry_pattern.match(line):
|
|
if current_entry:
|
|
entries.append("\n".join(current_entry))
|
|
current_entry = []
|
|
current_entry.append(line)
|
|
if current_entry:
|
|
entries.append("\n".join(current_entry))
|
|
except FileNotFoundError:
|
|
print(f"Couldn't find {args.filename}!")
|
|
return 1
|
|
|
|
print(f"{len(entries)} records found!")
|
|
|
|
records = []
|
|
for entry in entries:
|
|
parts = entry.split(",")
|
|
url = parts[0]
|
|
username = parts[1] if len(parts) > 1 else ""
|
|
password = parts[2] if len(parts) > 2 else ""
|
|
fav = parts[-1] if len(parts) > 6 else ""
|
|
grouping = parts[-2] if len(parts) > 5 else args.default
|
|
name = parts[-3] if len(parts) > 4 else ""
|
|
extra = ",".join(parts[3:-4])[1:-1] if len(parts) > 7 else ""
|
|
|
|
records.append(Record(name, url, username, password, extra, grouping, fav))
|
|
|
|
print(f"Records parsed: {len(records)}")
|
|
|
|
successful = 0
|
|
errors = []
|
|
for record in records:
|
|
output_path = f"{record.name}.gpg"
|
|
if os.path.exists(output_path) and not args.force:
|
|
print(f"skipped {record.name}: already exists")
|
|
continue
|
|
|
|
print(f"Creating record {record.name}...", end="")
|
|
try:
|
|
proc = subprocess.Popen(
|
|
["pass", "insert", "-m", record.name], stdin=subprocess.PIPE, text=True
|
|
)
|
|
proc.communicate(input=record.to_str())
|
|
if proc.returncode == 0:
|
|
print(" done!")
|
|
successful += 1
|
|
else:
|
|
print(" error!")
|
|
errors.append(record)
|
|
except Exception as e:
|
|
print(f" error! ({str(e)})")
|
|
errors.append(record)
|
|
|
|
print(f"{successful} records successfully imported!")
|
|
|
|
if errors:
|
|
print(f"There were {len(errors)} errors:")
|
|
error_names = [e.name for e in errors]
|
|
print(", ".join(error_names) + ".")
|
|
print(
|
|
"These probably occurred because an identically-named record already existed, or because there were multiple entries with the same name in the csv file."
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|