自动化创建ETX用户帐号

在芯片设计行业,ETX是常见的远程访问环境。用户在通过ETX访问远程环境前必须首先加入ETX系统,然后通过profile分配相关的环境的访问权限。

通常这些操作在ETX WEB页面手工操作,如果我们期望实现用户帐号注册全自动化,就需要将以上操作通过脚本来实现。

经过调查发现ETX提供了REST访问接口,因此帐号注册全自动化就可以实现。

主要流程如下:

在调用ETX REST接口前还需要获取ETX API的URL以及访问密钥。登录ETX后,点击右上部用户头像旁边的齿轮图标,在用户设置中选择创建 API keys即可生成ETX API访问密钥。

ETX登录界面左上角选择 REST API即可显示所有ETX API,点击还可查看API的详细说明和示例。

通过 Python 自动创建ETX帐号的示例代码如下:

python 复制代码
#!/opt/miniconda3/bin/python
################################################################################
#
# Author: Shuguangbo
#
################################################################################

import traceback
import logging
import logging.config
import urllib3
import requests
import json

urllib3.disable_warnings()

class userHandler():
    def __init__(self):
        self._account = ''
        self._userID = ''
        self._etxUserID = ''
        self.getETXGroup()
        ...
        
    def getETXUserID(self):
        session = None
        userData = dict()
        userID = ''
        account = self._account
        try:
            apiurl = self._config['ETX']['ETX_URL'] + f'/etx/api/v2/users?where_login={account}'
            session = requests.Session()
            header = {'Content-Type': 'application/json', 'Authorization': f"ETX {self._config['ETX']['API_KEY']}"}
            result = session.get(apiurl, headers=header, verify=False)
            
            if result.status_code == 200:
                logging.info('Get ETX user ID succeeded.')
                userList = json.loads(result.text)
                userID = userList[0]['id']
                self._etxUserID = userID
            else:
                logging.error('Get ETX user ID failed.')
        except Exception as e:
            logging.error(f"Get ETX user ID failed. error: {str(e)}, stack: {traceback.format_exc()}")
        finally:
            if session:
                session.close()
        return userID

    def getETXGroup(self):
        groupData = dict()
        session = None
        try:
            apiurl = self._config['ETX']['ETX_URL'] + '/etx/api/v2/usergroups'
            session = requests.Session()
            header = {'Content-Type': 'application/json', 'Authorization': f"ETX {self._config['ETX']['API_KEY']}"}
            result = session.get(apiurl, headers=header, verify=False)
            if result.status_code == 200:
                logging.info('Get ETX group info succeeded.')
                groupList = json.loads(result.text)
                for group in groupList:
                    groupData[group['name']] = group
            else:
                logging.error(f'Get ETX group info failed. Error: {result.text}')
                exit(1)

        except Exception as e:
            logging.error(f"GET ETX user group info failed. error: {str(e)}, stack: {traceback.format_exc()}")
            exit(1)
        finally:
            if session:
                session.close()
        self._etxGroups = groupData

    def addETXUserIntoGroup(self, userID, groupID):
        result = True
        session = None
        try:
            apiurl = self._config['ETX']['ETX_URL'] + f'/etx/api/v2/usergroups/{groupID}/members/addbyid/{userID}'
            session = requests.Session()
            header = {'Content-Type': 'application/json', 'Authorization': f"ETX {self._config['ETX']['API_KEY']}"}
            result = session.post(apiurl, headers=header, verify=False)
            if result and result.status_code == 200:
                logging.info(f'Add user {userID} into ETX group {groupID} succeeded.')
            else:
                logging.error(f'Add user {userID} into ETX group {groupID} failed.')
                result = False
        except Exception as e:
            logging.error(f"Add ETX user into user group failed. error: {str(e)}, stack: {traceback.format_exc()}")
            result = False
        finally:
            if session:
                session.close()
        return result


    def registerETXUser(self):
        EMSG = ''
        session = None
        try:
            account = self._account
            name = self._name
            email = self._email
            apiurl = self._config['ETX']['ETX_URL'] + '/etx/api/v2/users'
            session = requests.Session()
            header = {'Content-Type': 'application/json', 'Authorization': f"ETX {self._config['ETX']['API_KEY']}"}
            data = {'login': account, 'name': name, 'email': email, 'role':{'name':'User'}}
            result = session.post(apiurl, headers=header, data=json.dumps(data), verify=False)
            if result and result.status_code == 200:
                userData = json.loads(result.text)
                self._etxUserID = userData['id']
                logging.info(f'Register ETX user {account} succeeded. User id: {userData["id"]}')
            else:
                logging.error(f'Register ETX user {account} failed. Error: {result.text}')
                EMSG = f"Register ETX user {account} failed."
        except Exception as e:
            logging.error(f"Register ETX user {account} failed. Error: {str(e)}")
            EMSG = f"Register ETX user {account} failed."
        finally:
            if session:
                session.close()
        return EMSG
        
    def addETXUser(self):
        EMSG = ''
        try:
            user = self._account

            ret = self._registerETXUser()
            if len(ret):
                EMSG += ret
                return EMSG

            userID = self._etxUserID
            if len(userID) == 0:
                EMSG = "ETX user id is invalid.\n"
                return EMSG

            etxgroup = self._config['ETX']['DEFAULT_GROUP']                     
            groupID = self._etxGroups[etxgroup]['id']
            ret = self.addETXUserIntoGroup(userID, groupID)
            if not ret:
                EMSG += f'Failed to add ETX profile for {user} .\n'             

        except Exception as e:
            logging.error(f"Add ETX User profile failed. error: {str(e)}, stack: {traceback.format_exc()}"
            EMSG += "Add ETX User profile failed.\n"
        return EMSG
相关推荐
费弗里30 分钟前
Python全栈应用开发利器Dash 3.x新版本介绍(3)
python·dash
dme.1 小时前
Javascript之DOM操作
开发语言·javascript·爬虫·python·ecmascript
加油吧zkf1 小时前
AI大模型如何重塑软件开发流程?——结合目标检测的深度实践与代码示例
开发语言·图像处理·人工智能·python·yolo
t_hj1 小时前
python规划
python
czhc11400756631 小时前
Linux 76 rsync
linux·运维·python
悠悠小茉莉2 小时前
Win11 安装 Visual Studio(保姆教程 - 更新至2025.07)
c++·ide·vscode·python·visualstudio·visual studio
m0_625686552 小时前
day53
python
Real_man3 小时前
告别 requirements.txt,拥抱 pyproject.toml和uv的现代Python工作流
python
站大爷IP3 小时前
Python文件操作的"保险箱":with语句深度实战指南
python
运器1233 小时前
【一起来学AI大模型】算法核心:数组/哈希表/树/排序/动态规划(LeetCode精练)
开发语言·人工智能·python·算法·ai·散列表·ai编程