Strip whitespace

pyvecorg/build.py

@@ -44,16 +44,23 @@ def parse_members(rows):
 
 def coerce_member(member):
     return {
-        'name': member.get('nickname', member.get('name')),
-        'role': member.get('role', 'member'),
-        'github': member.get('github'),
-        'twitter': member.get('twitter'),
-        'mastodon': member.get('mastodon'),
-        'linkedin': member.get('linkedin'),
-        'avatar_filename': member.get('avatar_filename'),
+        'name': strip_whitespace(member.get('nickname', member.get('name'))),
+        'role': strip_whitespace(member.get('role', 'member')),
+        'github': strip_whitespace(member.get('github')),
+        'twitter': strip_whitespace(member.get('twitter')),
+        'mastodon': strip_whitespace(member.get('mastodon')),
+        'linkedin': strip_whitespace(member.get('linkedin')),
+        'avatar_filename': strip_whitespace(member.get('avatar_filename')),
     }
 
 
+def strip_whitespace(value):
+    try:
+        return value.strip() or None
+    except AttributeError:
+        return value
+
+
 def generate_yaml(data):
     yaml_contents = dedent('''\
         #

tests/test_data.py

@@ -6,6 +6,7 @@
 import pytest
 import requests
 import jsonschema
+from pyvecorg.build import coerce_member, strip_whitespace
 
 from pyvecorg.data import (load_data, select_language,
                            DATA_PATH, SUPPORTED_LANGS)
@@ -180,3 +181,25 @@ def test_select_language_missing():
     }}]}}
     data_out = {'deep': {'nested': [{'key': None}]}}
     assert select_language(data_in, 'en') == data_out
+
+
+def test_coerce_member_name():
+    member = {'name': 'Jan Python', 'nickname': 'Honza P.'}
+
+    assert coerce_member(member)['name'] == 'Honza P.'
+
+
+def test_coerce_member_whitespace():
+    member = {'mastodon': ' @[email protected]  '}
+
+    assert coerce_member(member)['mastodon'] == '@[email protected]'
+
+
+@pytest.mark.parametrize('value, expected', [
+    (None, None),
+    ('', None),
+    ('      ', None),
+    ('     @pyconcz    ', '@pyconcz'),
+])
+def test_strip_whitespace(value: str, expected: str):
+    assert strip_whitespace(value) == expected