More sanitization options

pass2csv.py

@@ -92,7 +92,7 @@ bwFieldSet = OrderedDict({
 
 header = [key for key, value in bwFieldSet.items()]
 
-def write(file, entries, get_fields, get_lines):
+def write(file, entries, get_fields, get_lines, notes, ignore, usernames, no_degroup):
     get_field_names = set(x[0] for x in get_fields)
     get_line_names = set(x[0] for x in get_lines)
     field_names = get_field_names | get_line_names
@@ -100,15 +100,31 @@ def write(file, entries, get_fields, get_lines):
     stderr(f"\nWriting data to {file.name}\n")
     csvw.writerow(header)
     for entry in entries:
-        if entry['title'] == 'ChromePasswords':
+        if entry['title'] in ignore:
             continue
         columns = []
         fieldIndex = -1
+        if entry['group'] not in no_degroup:
+            for upat in usernames:
+                m = upat.search(entry['title'])
+                if m:
+                    # username match, clean up the entry
+                    if entry['fields'].get('username') in ['', None]:
+                        entry['fields']['username'] = m[0]
+                    else:
+                        entry['fields']['titleuser'] = m[0]
+                    oldTitle = entry['title']
+                    entry['title'] = upat.sub(str(entry['group']), str(entry['title'])).rstrip(":")
+                    entry['fields']['passEntry'] = '{}/{}'.format(entry['group'], oldTitle)
+                    entry['group'] = ''
         for bwf, pf in bwFieldSet.items():
             if pf is None:
                 columns.append('')
             elif pf == 'type':
-                columns.append('login')
+                if entry['title'] in notes:
+                    columns.append('note')
+                else:
+                    columns.append('login')
             elif pf == '__fields':
                 columns.append('')
                 fieldIndex = len(columns) - 1
@@ -133,7 +149,7 @@ def write(file, entries, get_fields, get_lines):
 
 
 def main(store_path, outfile, grouping_base, gpgbinary, use_agent, encodings,
-         exclude, get_fields, get_lines):
+         exclude, get_fields, get_lines, notes, ignore, usernames, no_degroup):
     entries = []
     failures = []
     path = pathlib.Path(store_path)
@@ -181,7 +197,7 @@ def main(store_path, outfile, grouping_base, gpgbinary, use_agent, encodings,
     if not entries:
         stderr("\nNothing to write.")
         sys.exit(1)
-    write(outfile, entries, get_fields, get_lines)
+    write(outfile, entries, get_fields, get_lines, notes, ignore, usernames, no_degroup)
 
 
 def parse_args(args=None):
@@ -286,6 +302,58 @@ def parse_args(args=None):
         version='%(prog)s ' + __version__
     )
 
+    parser.add_argument(
+        '-n', '--note',
+        metavar='title',
+        action='append',
+        type=str,
+        default=[],
+        help=(
+            "a title of entry that should be added as a note, "
+            "can be specified multiple times"
+        ),
+        dest='notes'
+    )
+
+    parser.add_argument(
+        '-i', '--ignore',
+        metavar='title',
+        action='append',
+        type=str,
+        default=[],
+        help=(
+            "a title of entry that should be ignored, "
+            "can be specified multiple times"
+        ),
+        dest='ignore'
+    )
+
+    parser.add_argument(
+        '-u', '--username',
+        metavar='title',
+        action='append',
+        type=str,
+        default=[],
+        help=(
+            "a regex matching a username in title, "
+            "can be specified multiple times. any match will be replaced with the group name"
+        ),
+        dest='usernames'
+    )
+
+    parser.add_argument(
+        '-D', '--no-degroup',
+        metavar='title',
+        action='append',
+        type=str,
+        default=[],
+        help=(
+            "a list of groups not to perform username degrouping on, "
+            "can be specified multiple times."
+        ),
+        dest='no_degroup'
+    )
+
     return parser.parse_args(args)
 
 
@@ -324,6 +392,13 @@ def cli():
             failed = True
         get_lines.append((name, regexp))
 
+    username_patterns = []
+    for pattern in parsed.usernames:
+        regexp = compile_regexp(pattern)
+        if not regexp:
+            failed = True
+        username_patterns.append(regexp)
+
     if failed:
         sys.exit(1)
 
@@ -346,7 +421,11 @@ def cli():
         'encodings': encodings,
         'exclude': exclude_patterns,
         'get_fields': get_fields,
-        'get_lines': get_lines
+        'get_lines': get_lines,
+        'notes': parsed.notes,
+        'ignore': parsed.ignore,
+        'usernames': username_patterns,
+        'no_degroup': parsed.no_degroup
     }
 
     main(**kwargs)