def Delete_ALLSysRole(self, url, pageSize):
'''删除所有角色'''
page_idx = 1
while True:
getallrole = self.GetRoleList(url,
roleName="",
orderBy="createTime_asc",
pageSize=pageSize,
pageNum=page_idx)
# 删除时作判断，如果无子帐号直接删除，如果有子帐号，先查看该角色下的所有子帐号，循环一个一个对子帐号进行冻结，然后再次删除
cnt = 0
for one in getallrole["data"]["list"]:
rolelist = self.DeleteSysRole(url, roleId=one["roleId"])
log1.info(rolelist)
if ("error_msg" in rolelist
) and rolelist["error_msg"] == "该角色下含有子账号，您无法删除！":
pass
else:
cnt += 1
if cnt == 0:
#获取页数，把记录条数的总数对一页的数量取余，能整除就算除法，不能整除就取整
if getallrole["data"]["totalSize"] % pageSize == 0:
pages = getallrole["data"]["totalSize"] / pageSize
else:
pages = getallrole["data"]["totalSize"] // pageSize + 1
# pages=(getallrole["data"]["totalSize"]+9)/pageSize
if page_idx < pages:
page_idx += 1
else:
break
def check_tokentime(self,url,username,password):
'''检查token时效性'''
self.user_login(url,userName=username,password=password)
before=self.token
self.user_login(url,userName=username,password=password)
after =self.token
if before==after:
log1.info("token时效性测试正确")
else:
log1.error("token时效性测试错误")
def loginout(self,url,username):
'''退出'''
URL=f"{url}/api/user/LogoutAdmin"
param={"userName":username}
resp=requests.get(URL,headers={'Content-Type': 'application/json','Authorization':self.token},params=param)
result=resp.json()
print(result)
if result["error_msg"]=="退出成功！":
log1.info(f"退出成功，返回值为{result}")
return result
else:
log1.error("退出失败",exc_info=1)
def DeleteSysRole(self, url, **data):
'''删除某个角色'''
URL = f'{url}/api/SysRoles/DeleteSysRole'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
if ("error_msg" in result) and result["error_msg"] == '删除成功！':
log1.info(f"删除成功，返回信息是{result}")
return result
elif ("error_msg"
in result) and result["error_msg"] == "该角色下含有子账号，您无法删除！":
log1.info("该角色下含有子账号，您无法删除")
return result
elif "traceId" in result:
log1.info(f"参数错误，传入的参数为{data}")
return result
elif "error_msg" in result:
log1.info(f"传入的参数是{data}")
return result
else:
log1.error("删除失败", exc_info=1)
def getPackageList(self, url, **params):
'''获取套餐列表'''
URL = f'{url}/api/vipManage/getPackageList'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if "data" in result and result["data"]["totalSize"] >= 0:
log1.info("成功获取套餐列表")
return result
else:
log1.error(f"获取套餐列表失败,返回结果是{result},传入的参数是{params}", exc_info=1)
def updatePackageIndex(self, url, **params):
'''更新套餐显示顺序'''
URL = f'{url}/api/vipManage/updatePackageIndex'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if "data" in result:
log1.info("更新套餐显示顺序成功")
return result
else:
log1.error(f"更新套餐显示顺序失败，返回值是{result},传入的参数是{params}", exc_info=1)
def GetCurrentSysUser(self, url, **params):
'''获取当前登录用户信息'''
URL = f'{url}/api/SysUsers/GetCurrentSysUser'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if ("data" in result) and ("userId" in result["data"]):
log1.info("成功获取到获取当前用户信息")
return result
else:
log1.error(f'获取当前用户信息失败，传入的参数为{params}')
def GetSysUserList(self, url, **params):
'''获取子帐号列表'''
URL = f'{url}/api/SysUsers/GetSysUserList'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if result["data"]["totalSize"] >= 0:
log1.info("成功获取子帐号列表")
return result
else:
log1.error("获取子帐号列表失败", exc_info=1)
def getOrderList(self, url, **params):
'''订单列表'''
URL = f'{url}/api/order/getOrderList'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if "totalSize" in result["data"]:
log1.info("成功获取订单列表")
return result
else:
log1.error("获取订单列表失败")
def deletedata(self, tabls_name, name):
cursor = self.connection.cursor()
# SQL 删除语句
sql = f"DELETE FROM {tabls_name} WHERE USER_ID='{name}'"
try:
# 执行SQL语句
cursor.execute(sql)
#提交到数据库执行
self.connection.commit()
for one in self.query(tabls_name):
if one[0] == name:
log1.log("没有删除掉")
log1.info(f"delete OK,删除的数据USER_ID={name}")
except:
# 发生错误时回滚
self.connection.rollback()
def getUserList(self, url, **params):
'''获取用户列表接口'''
URL = f'{url}/api/userManage/getUserList'
# params={"type":type,"phone":phone,"registDate":registDate,"orderBy":orderBy,"pageSize":pageSize,"pageNum":pageNum}
resq = requests.get(URL,
params=params,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
})
result = resq.json()
if ("data" in result) and result["data"]["totalSize"] >= 0:
log1.info("成功获取用户列表接口")
return result
elif "title" in result:
log1.info(f"参数格式错误，传入的参数是{params}")
return result
else:
log1.error(f"获取用户列表接口失败，返回信息为{result},传入的参数为{params}")
def UpdateSysUserPassword(self, url, **data):
'''重置密码'''
URL = f'{url}/api/SysUsers/UpdateSysUserPassword'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
if ("error_msg" in result) and result["error_msg"] == "重置密码成功！":
log1.info("密码重置成功")
return result
elif "error_msg" in result:
log1.info(f"参数错误，返回信息是{result},传入的参数是{data}")
return result
else:
log1.error(f"密码重置失败，传入的参数为{data}", exc_info=1)
def AddSysUser(self, url, **data):
'''创建子帐号'''
URL = f'{url}/api/SysUsers/AddSysUser'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
if ("data" in result) and "userId" in result["data"]:
log1.info(f"子帐号创建成功，返回信息为{result}")
return result
elif "error_msg" in result:
log1.info(f"参数数错，返回信息是{result},传入的参数是{data}")
return result
else:
log1.error(f"子帐号创建失败，传入的参数为{data}", exc_info=1)
def GetRoleList(self, url, **params):
'''获取角色列表'''
URL = f'{url}/api/SysRoles/GetRoleList'
# data = {"roleName": roleName, "remark": remark, "roleRights": roleRights}
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
print(result)
if "data" in result and result["data"]["totalSize"] >= 1:
log1.info(f"获取角色列表成功，返回信息是{result}")
return result
elif "title" in result:
log1.info(f"参数错误，传入的参数是{params}")
return result
else:
log1.error("获取角色列表失败", exc_info=1)
def UpdateSysUser(self, url, **data):
'''编辑子帐号'''
URL = f'{url}/api/SysUsers/UpdateSysUser'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
print(result)
if (
"error_msg" in result
) and result["error_msg"] == "更新成功！" and "userId" in result["data"]:
log1.info(f"子帐号编辑成功，返回信息为{result}")
return result
elif "error_msg" in result:
log1.info(f"参数错误，返回的消息是{result},传入的参数是{data}")
return result
else:
log1.error(f"子帐号编辑失败，传入参数为{data}", exc_info=1)
def UpdateSysRole(self, url, **data):
'''编辑角色'''
URL = f'{url}/api/SysRoles/UpdateSysRole'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
print(result)
if (
"error_msg" in result
) and result["error_msg"] == '更新成功！' and "roleId" in result["data"]:
log1.info(f"编辑角色成功，返回信息是{result}")
return result
elif "error_msg" in result:
log1.info(f"返回的信息是{result},传入的参数是{data}")
return result
else:
log1.error("编辑角色失败", exc_info=1)
def query(self, tabls_name):
#查询表里所有的列
# 使用 execute() 方法执行 SQL 查询
try:
cursor = self.connection.cursor()
cursor.execute("select * from %s" % tabls_name)
# 获取所有记录列表
data = cursor.fetchall()
for row in data:
USER_ID = row[0]
USERNAME = row[1]
PASSWORD = row[2]
NAME = row[3]
ROLE_ID = row[5]
STATUS = row[8]
CreateTime = row[15]
UpdateTime = row[16]
log1.info(
f"USER_ID={USER_ID},USERNAME={USERNAME},PASSWORD={PASSWORD},NAME={NAME},ROLE_ID={ROLE_ID},STATUS={STATUS},CreateTime={CreateTime},UpdateTime={UpdateTime}"
)
except:
print("Error: unable to fetch data")
return data
def closeUser(self, url, **params):
'''注销账号'''
URL = f'{url}/api/userManage/closeUser'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if "data" in result:
log1.info("成功注销账号")
return result
elif "title" in result:
log1.info(f"参数格式错误，传入的参数是{params}")
elif "error_msg" in result and result["error_msg"] == "参数错误！":
log1.info(f"参数错误，传入的参数是{params}")
else:
log1.error(f"注销帐号失败，返回信息是{result},传入的参数是{params}")
def putOnPackageInfo(self, url, **params):
'''上架套餐'''
URL = f'{url}/api/vipManage/putOnPackageInfo'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if "error_msg" in result and result["error_msg"] == "上架成功！":
log1.info("上架套餐成功")
return result
elif "error_msg" in result:
log1.info(f"参数错误，传入的参数是{params}")
return result
elif "title" in result:
log1.info(f"参数格式错误，传入的参数是{params}")
return result
else:
log1.error(f"上架套餐失败，返回值是{result},传入的参数是{params}", exc_info=1)
def GetPackage(self, url, **params):
'''获取套餐信息'''
URL = f'{url}/api/vipManage/GetPackage'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if ("data" in result) and ("packageId" in result["data"]):
log1.info("删除套餐成功")
return result
elif "error_msg" in result:
log1.info(f"参数错误，传入的参数是{params}")
return result
elif "title" in result:
log1.info(f"参数格式错误，传入的参数是{params}")
return result
else:
log1.error(f"删除套餐失败，返回值是{result},传入的参数是{params}", exc_info=1)
def GetAllSysMenu(self, url, **params):
'''获取权限菜单列表'''
URL = f'{url}/api/SysMenus/GetAllSysMenu'
# params={"menuId":menuId}
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
if ("data" in result) and result["data"]["totalSize"] >= 1:
log1.info("成功获取权限菜单列表")
return result
elif ("data" in result) and result["data"]["totalSize"] == 0:
log1.info(f"传入的参数错误，传入值为{params}")
return result
elif "error_msg" in result:
log1.info(f"传入的参数是{params}")
return result
else:
log1.error("获取权限菜单列表失败", exc_info=1)
def GetRoleUserList(self, url, **params):
'''查看子账号'''
URL = f'{url}/api/SysRoles/GetRoleUserList'
resp = requests.get(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
params=params)
result = resp.json()
print(result)
if "data" in result and result["data"]["totalSize"] >= 0:
log1.info(f"查看子帐号成功，返回信息是{result}")
return result
elif "data" in result and result["data"]["totalSize"] == 0:
log1.info(f"返回的消息是{result},传入的参数是{params}")
return result
elif "error_msg" in result:
log1.info(f"返回的消息是{result},传入的参数是{params}")
return result
else:
log1.error("无法查看子帐号", exc_info=1)
def updatePackageInfo(self, url, **data):
'''编辑套餐'''
URL = f'{url}/api/vipManage/updatePackageInfo'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
if "error_msg" in result and result["error_msg"] == "更新成功！":
log1.info("套餐编辑成功")
return result
elif "error_msg" in result:
log1.info(f"参数错误，传入的参数是{data}")
return result
elif "title" in result:
log1.info(f"参数格式错误，传入的参数是{data}")
return result
else:
log1.error(f"套餐编辑失败，返回值是{result},传入的参数是{data}", exc_info=1)
def UpdateSysUserStatus(self, url, **data):
'''冻结/解冻子账号'''
URL = f'{url}/api/SysUsers/UpdateSysUserStatus'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
print(result)
if ("error_msg" in result) and result["error_msg"] == "冻结用户成功！":
log1.info(f"子帐号冻结成功")
return result
elif ("error_msg" in result) and result["error_msg"] == "解冻用户成功！":
log1.info(f"子帐号解冻成功")
return result
elif "error_msg" in result:
log1.info(f"参数错误，返回的信息是{result},传入的参数是{data}")
return result
else:
log1.error(f'冻结/解冻失败，返回结果是{result}', exc_info=1)
log1.error(f"子帐号冻结/解冻失败，传入的参数为{data}", exc_info=1)
def AddSysRole(self, url, **data):
'''创建角色'''
URL = f'{url}/api/SysRoles/AddSysRole'
jsoninfo = json.dumps(data)
resp = requests.post(URL,
headers={
'Content-Type': 'application/json',
'Authorization': self.token
},
data=jsoninfo)
result = resp.json()
print(result)
if (
"error_msg" in result
) and result["error_msg"] == '新增成功！' and "roleId" in result["data"]:
log1.info(f"新增成功，返回信息是{result}")
return result
elif ("error_msg" in result) and "已占用" in result["error_msg"]:
log1.info(f"角色名已占用,传入的参数是{data}")
return result
elif ("error_msg" in result) and result["error_msg"] == "角色名不能为空！":
log1.info(f"角色名不能为空,传入的参数是{data}")
return result
elif ("error_msg"
in result) and result["error_msg"] == "角色名最多不能超过8个字符！":
log1.info(f"角色名不能超过8个字符,传入的参数是{data}")
return result
elif ("error_msg"
in result) and result["error_msg"] == "角色说明最多不能超过21个字符！":
log1.info(f"角色说明不能超过21个字符,传入的参数是{data}")
return result
elif "error_msg" in result:
log1.info(f"返回的响应结果是{result},传入的参数是{data}")
return result
else:
log1.error("新增失败", exc_info=1)
def user_login(self,url,**data):
URL=f"{url}/api/user/LoginAdmin"
# data={"userName":username,"password":password}
log1.info(data)
jsoninfo=json.dumps(data)
resp=requests.post(URL,headers={'Content-Type': 'application/json'},data=jsoninfo)
result=resp.json()
log1.info(result)
if ("error_msg" in result) and result["error_msg"]=="登陆成功" and "token" in result:
log1.info(f"登录成功，返回值为{result}")
token="Bearer "+result["token"]
self.token=token
log1.info(f'token值为{token}')
return result
elif ("error_msg" in result) and result["error_msg"]=="用户名或密码错误！":
log1.info(f"登录的用户名或者密码错误,传入的参数是{data}")
return result
elif ("errors" in result) and ("Password" in result["errors"]):
log1.info(f"密码错误，传入的参数是{data}")
return result
elif ("errors" in result) and ("UserName" in result["errors"]):
log1.info(f"用户名错误，传入的参数是{data}")
return result
else:
log1.error("登录失败",exc_info=1)
