diff options
author | Youyuan <83439858+Yuanandcircle@users.noreply.github.com> | 2023-02-28 19:17:44 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-02-28 19:17:44 +0800 |
commit | 112fabb4bbb0aebe91e67d34d9f0678b39c1d1bb (patch) | |
tree | 629c9e00e16dd8027d3928333bb17608b8f2e8f9 /src/services/api/graphql/session/session.py | |
parent | efd51fb7876d21b59f24a785d4b9e63bf8e4eded (diff) | |
parent | 9b88a68001b79583cc2be59b4d6e1af3bed4aecf (diff) | |
download | vyos-1x-112fabb4bbb0aebe91e67d34d9f0678b39c1d1bb.tar.gz vyos-1x-112fabb4bbb0aebe91e67d34d9f0678b39c1d1bb.zip |
Merge branch 'vyos:current' into current
Diffstat (limited to 'src/services/api/graphql/session/session.py')
-rw-r--r-- | src/services/api/graphql/session/session.py | 38 |
1 files changed, 23 insertions, 15 deletions
diff --git a/src/services/api/graphql/session/session.py b/src/services/api/graphql/session/session.py index b2aef9bd9..3c5a062b6 100644 --- a/src/services/api/graphql/session/session.py +++ b/src/services/api/graphql/session/session.py @@ -29,6 +29,28 @@ from api.graphql.libs.op_mode import normalize_output op_mode_include_file = os.path.join(directories['data'], 'op-mode-standardized.json') +def get_config_dict(path=[], effective=False, key_mangling=None, + get_first_key=False, no_multi_convert=False, + no_tag_node_value_mangle=False): + config = Config() + return config.get_config_dict(path=path, effective=effective, + key_mangling=key_mangling, + get_first_key=get_first_key, + no_multi_convert=no_multi_convert, + no_tag_node_value_mangle=no_tag_node_value_mangle) + +def get_user_info(user): + user_info = {} + info = get_config_dict(['system', 'login', 'user', user], + get_first_key=True) + if not info: + raise ValueError("No such user") + + user_info['user'] = user + user_info['full_name'] = info.get('full-name', '') + + return user_info + class Session: """ Wrapper for calling configsession functions based on GraphQL requests. @@ -46,17 +68,6 @@ class Session: except Exception: self._op_mode_list = None - @staticmethod - def _get_config_dict(path=[], effective=False, key_mangling=None, - get_first_key=False, no_multi_convert=False, - no_tag_node_value_mangle=False): - config = Config() - return config.get_config_dict(path=path, effective=effective, - key_mangling=key_mangling, - get_first_key=get_first_key, - no_multi_convert=no_multi_convert, - no_tag_node_value_mangle=no_tag_node_value_mangle) - def show_config(self): session = self._session data = self._data @@ -134,10 +145,7 @@ class Session: user_info = {} user = data['user'] try: - info = self._get_config_dict(['system', 'login', 'user', user, - 'full-name']) - user_info['user'] = user - user_info['full_name'] = info.get('full-name', '') + user_info = get_user_info(user) except Exception as error: raise error |