python/jc
1
0
Fork 0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-06-17 00:07:37 +02:00
Code Issues Releases Activity

Merge pull request #284 from pbrezina/id

Browse Source 
id: support space in names
This commit is contained in:
Kelly Brazil
2022-09-15 15:46:32 +00:00
committed by GitHub
parent 860167b950 4ebde3af5f
commit c4de1c53b9
2 changed files with 41 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
jc/parsers/id.py
View File

@ -100,12 +100,13 @@ Examples:
} }
} }
""" """
import re
import jc.utils import jc.utils
class info(): class info():
"""Provides parser metadata (version, author, etc.)""" """Provides parser metadata (version, author, etc.)"""
version = '1.5' version = '1.6'
description = '`id` command parser' description = '`id` command parser'
author = 'Kelly Brazil' author = 'Kelly Brazil'
author_email = 'kellyjonbrazil@gmail.com' author_email = 'kellyjonbrazil@gmail.com'
@ -170,28 +171,28 @@ def parse(data, raw=False, quiet=False):
raw_output = {} raw_output = {}
# Clear any blank lines # re.split produces first element empty
cleandata = list(filter(None, data.split())) cleandata = re.split(r' ?(uid|gid|groups|context)=', data.strip())[1:]
if jc.utils.has_data(data): if jc.utils.has_data(data):
for section in cleandata: for key, value in zip(cleandata[0::2], cleandata[1::2]):
if section.startswith('uid'): if key == 'uid':
uid_parsed = section.replace('(', '=').replace(')', '=') uid_parsed = value.replace('(', '=').replace(')', '=')
uid_parsed = uid_parsed.split('=') uid_parsed = uid_parsed.split('=')
raw_output['uid'] = {} raw_output['uid'] = {}
raw_output['uid']['id'] = uid_parsed[1] raw_output['uid']['id'] = uid_parsed[0]
raw_output['uid']['name'] = _get_item(uid_parsed, 2) raw_output['uid']['name'] = _get_item(uid_parsed, 1)
if section.startswith('gid'): if key == 'gid':
gid_parsed = section.replace('(', '=').replace(')', '=') gid_parsed = value.replace('(', '=').replace(')', '=')
gid_parsed = gid_parsed.split('=') gid_parsed = gid_parsed.split('=')
raw_output['gid'] = {} raw_output['gid'] = {}
raw_output['gid']['id'] = gid_parsed[1] raw_output['gid']['id'] = gid_parsed[0]
raw_output['gid']['name'] = _get_item(gid_parsed, 2) raw_output['gid']['name'] = _get_item(gid_parsed, 1)
if section.startswith('groups'): if key == 'groups':
groups_parsed = section.replace('(', '=').replace(')', '=') groups_parsed = value.replace('(', '=').replace(')', '=')
groups_parsed = groups_parsed.replace('groups=', '') groups_parsed = groups_parsed.replace('groups=', '')
groups_parsed = groups_parsed.split(',') groups_parsed = groups_parsed.split(',')
raw_output['groups'] = [] raw_output['groups'] = []
@ -203,9 +204,8 @@ def parse(data, raw=False, quiet=False):
group_dict['name'] = _get_item(grp_parsed, 1) group_dict['name'] = _get_item(grp_parsed, 1)
raw_output['groups'].append(group_dict) raw_output['groups'].append(group_dict)
if section.startswith('context'): if key == 'context':
context_parsed = section.replace('context=', '') context_parsed = value.split(':', maxsplit=3)
context_parsed = context_parsed.split(':', maxsplit=3)
raw_output['context'] = {} raw_output['context'] = {}
raw_output['context']['user'] = context_parsed[0] raw_output['context']['user'] = context_parsed[0]
raw_output['context']['role'] = context_parsed[1] raw_output['context']['role'] = context_parsed[1]

24
tests/test_id.py
View File

@ -43,6 +43,30 @@ class MyTests(unittest.TestCase):
{'uid': {'id': 1000, 'name': 'user'}, 'gid': {'id': 1000, 'name': None}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel'}]} {'uid': {'id': 1000, 'name': 'user'}, 'gid': {'id': 1000, 'name': None}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel'}]}
) )
def test_id_space(self):
"""
Test 'id' with with space in name
"""
self.assertEqual(
jc.parsers.id.parse('uid=1000(user 1) gid=1000 groups=1000,10(wheel)', quiet=True),
{'uid': {'id': 1000, 'name': 'user 1'}, 'gid': {'id': 1000, 'name': None}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel'}]}
)
self.assertEqual(
jc.parsers.id.parse('uid=1000(user) gid=1000(group 1) groups=1000,10(wheel)', quiet=True),
{'uid': {'id': 1000, 'name': 'user'}, 'gid': {'id': 1000, 'name': 'group 1'}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel'}]}
)
self.assertEqual(
jc.parsers.id.parse('uid=1000(user) gid=1000 groups=1000,10(wheel 1)', quiet=True),
{'uid': {'id': 1000, 'name': 'user'}, 'gid': {'id': 1000, 'name': None}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel 1'}]}
)
self.assertEqual(
jc.parsers.id.parse('uid=1000(user 1) gid=1000(group 1) groups=1000,10(wheel 1)', quiet=True),
{'uid': {'id': 1000, 'name': 'user 1'}, 'gid': {'id': 1000, 'name': 'group 1'}, 'groups': [{'id': 1000, 'name': None}, {'id': 10, 'name': 'wheel 1'}]}
)
def test_id_centos_7_7(self): def test_id_centos_7_7(self):
""" """
Test 'id' on Centos 7.7 Test 'id' on Centos 7.7