feat: 添加JoyHub运费模板和Banner管理接口用例
This commit is contained in:
337
dulizhan/library/Dlizhan_interface.py
Normal file
337
dulizhan/library/Dlizhan_interface.py
Normal file
@@ -0,0 +1,337 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import os
|
||||
import sys
|
||||
|
||||
current_file_path = os.path.abspath(__file__)
|
||||
project_root = os.path.abspath(os.path.join(os.path.dirname(current_file_path), '../../'))
|
||||
if project_root not in sys.path:
|
||||
sys.path.insert(0, project_root)
|
||||
|
||||
from base_framework.public_tools import log
|
||||
from base_framework.public_tools.runner import Runner
|
||||
from base_framework.public_tools.eureka_api import EurekaAPI
|
||||
from base_framework.public_tools import utils
|
||||
from base_framework.public_tools.pgsqlhelper import PgSqlHelper
|
||||
import requests
|
||||
import json
|
||||
|
||||
obj_log = log.get_logger()
|
||||
obj_runner = Runner()
|
||||
eureka = EurekaAPI()
|
||||
|
||||
|
||||
class DlzhanInterface:
|
||||
def __init__(self):
|
||||
self.domain_url = eureka.get_url_from_config()
|
||||
self.pg_db = PgSqlHelper()
|
||||
self.joyhub_domain = "https://joyhub-website-manager-api-test.best-envision.com"
|
||||
self.token = None
|
||||
|
||||
def _get_joyhub_headers(self):
|
||||
headers = {
|
||||
'Content-Type': 'application/json',
|
||||
'tenant-id': '126'
|
||||
}
|
||||
if self.token:
|
||||
headers['Authorization'] = 'Bearer ' + self.token
|
||||
return headers
|
||||
|
||||
def set_joyhub_token(self, token):
|
||||
self.token = token
|
||||
|
||||
def _joyhub_request(self, method, path, is_check='', note='', **kwargs):
|
||||
url = "{}{}".format(self.joyhub_domain, path)
|
||||
headers = self._get_joyhub_headers()
|
||||
obj_log.info("=========== {} ===========".format(note or path))
|
||||
|
||||
req_params = {}
|
||||
for key, value in kwargs.items():
|
||||
if value is not None and value != '':
|
||||
req_params[key] = value
|
||||
|
||||
req_map = {
|
||||
'GET': lambda: requests.get(url, headers=headers, params=req_params, verify=False),
|
||||
'POST': lambda: requests.post(url, headers=headers, json=req_params, verify=False),
|
||||
'PUT': lambda: requests.put(url, headers=headers, json=req_params, verify=False),
|
||||
'DELETE': lambda: requests.delete(url, headers=headers, verify=False)
|
||||
}
|
||||
|
||||
resp = req_map.get(method.upper(), lambda: None)()
|
||||
self._check_resp(is_check, resp)
|
||||
return resp.json()
|
||||
|
||||
def _clear_user_fingerprint(self, username):
|
||||
try:
|
||||
sql = "UPDATE system_users SET fingerprint = '' WHERE username = %s"
|
||||
cursor, conn, count = self.pg_db.execute(sql, (username,), choose_db='joyhub')
|
||||
obj_log.info(f"清除用户 {username} 的指纹锁，影响行数: {count}")
|
||||
self.pg_db.close(cursor, conn)
|
||||
return count >= 0
|
||||
except Exception as e:
|
||||
obj_log.error(f"清除指纹锁失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def kw_in_zhyy_purchase_todo_get(self, is_check='', **kwargs):
|
||||
user, kwargs = self._get_user(kwargs)
|
||||
kwargs = self._convert_json(kwargs)
|
||||
url = "%s/erp/purchase-workbench/get-todo" % self.domain_url
|
||||
obj_log.info("your input:{0}".format(kwargs))
|
||||
resp = obj_runner.call_rest_api(API_URL=url, req_type="GET", user=user)
|
||||
self._check_resp(is_check, resp)
|
||||
return resp
|
||||
|
||||
def kw_in_zhyy_purchase_order_page_post(self, is_check='', **kwargs):
|
||||
user, kwargs = self._get_user(kwargs)
|
||||
kwargs = self._convert_json(kwargs)
|
||||
url = "%s/erp/purchase-order/page" % self.domain_url
|
||||
obj_log.info("your input:{0}".format(kwargs))
|
||||
resp = obj_runner.call_rest_api(API_URL=url, req_type="POST", json=kwargs, user=user)
|
||||
self._check_resp(is_check, resp)
|
||||
return resp
|
||||
|
||||
def _get_user(self, kwargs):
|
||||
user = kwargs.get("user", "purchase")
|
||||
if "user" in kwargs:
|
||||
del kwargs["user"]
|
||||
return user, kwargs
|
||||
|
||||
def _convert_json(self, kwargs):
|
||||
return kwargs
|
||||
|
||||
def _check_resp(self, is_check, resp):
|
||||
if is_check == 'true':
|
||||
assert resp is not None, "响应为空"
|
||||
if hasattr(resp, 'json'):
|
||||
resp_json = resp.json()
|
||||
assert resp_json.get('code') == 0, f"请求失败，code={resp_json.get('code')}"
|
||||
return resp
|
||||
|
||||
def kw_in_joyhub_auth_login_post(self, is_check='', **kwargs):
|
||||
obj_log.info("=========== JoyHub 登录接口 ===========")
|
||||
|
||||
username = kwargs.get("username", "")
|
||||
password = kwargs.get("password", "")
|
||||
max_retries = kwargs.get("max_retries", 3)
|
||||
token = kwargs.get("token", "")
|
||||
|
||||
url = "https://joyhub-website-manager-api-test.best-envision.com/admin-api/system/auth/login-dev"
|
||||
|
||||
session = requests.session()
|
||||
session.headers.update({
|
||||
'isencrypt': 'true',
|
||||
'x-api-encrypt': 'true',
|
||||
'tenant-id': '126'
|
||||
})
|
||||
|
||||
resp = obj_runner._Runner__call_api(session, url, req_type="POST", json=kwargs)
|
||||
|
||||
if resp is not None and not isinstance(resp, Exception):
|
||||
try:
|
||||
resp_json = resp.json()
|
||||
except Exception:
|
||||
try:
|
||||
resp_json = json.loads(resp.text)
|
||||
except Exception:
|
||||
resp_json = None
|
||||
|
||||
if resp_json is not None:
|
||||
code = resp_json.get('code')
|
||||
if code == 0 or code == 200:
|
||||
obj_log.info("登录成功")
|
||||
self._check_resp(is_check, resp_json)
|
||||
return resp_json
|
||||
elif code == 401:
|
||||
obj_log.warning("登录失败，尝试清除指纹后重新登录")
|
||||
|
||||
if resp is not None and not isinstance(resp, Exception):
|
||||
try:
|
||||
resp_json = resp.json()
|
||||
except Exception:
|
||||
resp_json = None
|
||||
else:
|
||||
resp_json = None
|
||||
|
||||
self._check_resp(is_check, resp_json)
|
||||
return resp_json
|
||||
|
||||
def kw_in_joyhub_role_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/system/role/create', is_check, '创建角色', **kwargs)
|
||||
|
||||
def kw_in_joyhub_role_delete_post(self, is_check='', **kwargs):
|
||||
role_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/role/delete?id={role_id}', is_check, '删除角色')
|
||||
|
||||
def kw_in_joyhub_role_delete_list_post(self, is_check='', **kwargs):
|
||||
ids = kwargs.get('ids', [])
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/role/delete-list?ids={ids_str}', is_check, '批量删除角色')
|
||||
|
||||
def kw_in_joyhub_role_get_get(self, is_check='', **kwargs):
|
||||
role_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('GET', f'/admin-api/system/role/get?id={role_id}', is_check, '获得角色信息')
|
||||
|
||||
def kw_in_joyhub_role_page_get(self, is_check='', **kwargs):
|
||||
page_no = kwargs.get('pageNo', 1)
|
||||
page_size = kwargs.get('pageSize', 10)
|
||||
return self._joyhub_request('GET', f'/admin-api/system/role/page?pageNo={page_no}&pageSize={page_size}', is_check, '获得角色分页', **kwargs)
|
||||
|
||||
def kw_in_joyhub_role_simple_list_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/role/simple-list', is_check, '获取角色精简信息列表')
|
||||
|
||||
def kw_in_joyhub_role_list_all_simple_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/role/list-all-simple', is_check, '获取角色精简信息列表')
|
||||
|
||||
def kw_in_joyhub_role_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/role/update', is_check, '修改角色', **kwargs)
|
||||
|
||||
def kw_in_joyhub_post_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/system/post/create', is_check, '创建岗位', **kwargs)
|
||||
|
||||
def kw_in_joyhub_post_delete_post(self, is_check='', **kwargs):
|
||||
post_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/post/delete?id={post_id}', is_check, '删除岗位')
|
||||
|
||||
def kw_in_joyhub_post_delete_list_post(self, is_check='', **kwargs):
|
||||
ids = kwargs.get('ids', [])
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/post/delete-list?ids={ids_str}', is_check, '批量删除岗位')
|
||||
|
||||
def kw_in_joyhub_post_get_get(self, is_check='', **kwargs):
|
||||
post_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('GET', f'/admin-api/system/post/get?id={post_id}', is_check, '获得岗位信息')
|
||||
|
||||
def kw_in_joyhub_post_list_all_simple_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/post/list-all-simple', is_check, '获取岗位全列表')
|
||||
|
||||
def kw_in_joyhub_post_page_get(self, is_check='', **kwargs):
|
||||
page_no = kwargs.get('pageNo', 1)
|
||||
page_size = kwargs.get('pageSize', 10)
|
||||
return self._joyhub_request('GET', f'/admin-api/system/post/page?pageNo={page_no}&pageSize={page_size}', is_check, '获得岗位分页列表', **kwargs)
|
||||
|
||||
def kw_in_joyhub_post_simple_list_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/post/simple-list', is_check, '获取岗位精简列表')
|
||||
|
||||
def kw_in_joyhub_post_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/post/update', is_check, '修改岗位', **kwargs)
|
||||
|
||||
# ============ 运费模板信息接口 ============
|
||||
def kw_in_joyhub_shipping_template_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/jh/shipping-template/create', is_check, '创建运费模板信息', **kwargs)
|
||||
|
||||
def kw_in_joyhub_shipping_template_delete_post(self, shipping_template_id, is_check=''):
|
||||
return self._joyhub_request('DELETE', f'/admin-api/jh/shipping-template/delete?id={shipping_template_id}', is_check, '删除运费模板信息')
|
||||
|
||||
def kw_in_joyhub_shipping_template_delete_list_post(self, ids, is_check=''):
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/jh/shipping-template/delete-list?ids={ids_str}', is_check, '批量删除运费模板信息')
|
||||
|
||||
def kw_in_joyhub_shipping_template_get_detail_get(self, shipping_template_id, is_check=''):
|
||||
return self._joyhub_request('GET', f'/admin-api/jh/shipping-template/get-detail?id={shipping_template_id}', is_check, '获得运费模板详情')
|
||||
|
||||
def kw_in_joyhub_shipping_template_page_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/jh/shipping-template/page', is_check, '获得运费模板分页', **kwargs)
|
||||
|
||||
def kw_in_joyhub_shipping_template_save_with_children_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/jh/shipping-template/save-with-children', is_check, '保存运费模板信息（含规则与子表）', **kwargs)
|
||||
|
||||
def kw_in_joyhub_shipping_template_shipping_rule_list_get(self, shipping_template_id, is_check=''):
|
||||
return self._joyhub_request('GET', f'/admin-api/jh/shipping-template/shipping-rule/list-by-shipping-template-id?shippingTemplateId={shipping_template_id}', is_check, '获得运费规则列表')
|
||||
|
||||
def kw_in_joyhub_shipping_template_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/jh/shipping-template/update', is_check, '更新运费模板信息', **kwargs)
|
||||
|
||||
def kw_in_joyhub_banner_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/jh/banner/create', is_check, '创建Banner管理', **kwargs)
|
||||
|
||||
def kw_in_joyhub_banner_delete_delete(self, banner_id, is_check=''):
|
||||
return self._joyhub_request('DELETE', f'/admin-api/jh/banner/delete?id={banner_id}', is_check, '删除Banner管理')
|
||||
|
||||
def kw_in_joyhub_banner_delete_list_delete(self, ids, is_check=''):
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/jh/banner/delete-list?ids={ids_str}', is_check, '批量删除Banner管理')
|
||||
|
||||
def kw_in_joyhub_banner_get_import_template_get(self, is_check=''):
|
||||
return self._joyhub_request('GET', '/admin-api/jh/banner/get-import-template', is_check, '获得导入Banner管理模板')
|
||||
|
||||
def kw_in_joyhub_banner_get_get(self, banner_id, is_check=''):
|
||||
return self._joyhub_request('GET', f'/admin-api/jh/banner/get?id={banner_id}', is_check, '获得Banner管理详情')
|
||||
|
||||
def kw_in_joyhub_banner_page_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/jh/banner/page', is_check, '获得Banner管理分页', **kwargs)
|
||||
|
||||
def kw_in_joyhub_banner_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/jh/banner/update', is_check, '更新Banner管理', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/system/user/create', is_check, '创建用户', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_delete_post(self, is_check='', **kwargs):
|
||||
user_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/user/delete?id={user_id}', is_check, '删除用户')
|
||||
|
||||
def kw_in_joyhub_user_delete_list_post(self, is_check='', **kwargs):
|
||||
ids = kwargs.get('ids', [])
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/user/delete-list?ids={ids_str}', is_check, '批量删除用户')
|
||||
|
||||
def kw_in_joyhub_user_get_get(self, is_check='', **kwargs):
|
||||
user_id = kwargs.get('id', '')
|
||||
return self._joyhub_request('GET', f'/admin-api/system/user/get?id={user_id}', is_check, '获得用户详情')
|
||||
|
||||
def kw_in_joyhub_user_page_get(self, is_check='', **kwargs):
|
||||
page_no = kwargs.get('pageNo', 1)
|
||||
page_size = kwargs.get('pageSize', 10)
|
||||
return self._joyhub_request('GET', f'/admin-api/system/user/page?pageNo={page_no}&pageSize={page_size}', is_check, '获得用户分页列表', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/user/update', is_check, '修改用户', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_update_password_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/user/update-password', is_check, '重置用户密码', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_update_status_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/user/update-status', is_check, '修改用户状态', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_simple_list_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/user/simple-list', is_check, '获取用户精简信息列表')
|
||||
|
||||
def kw_in_joyhub_user_profile_get_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/user/profile/get', is_check, '获得登录用户信息')
|
||||
|
||||
def kw_in_joyhub_user_profile_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/user/profile/update', is_check, '修改用户个人信息', **kwargs)
|
||||
|
||||
def kw_in_joyhub_user_profile_update_password_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/user/profile/update-password', is_check, '修改用户个人密码', **kwargs)
|
||||
|
||||
# ============ 部门管理接口 ============
|
||||
def kw_in_joyhub_dept_create_post(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('POST', '/admin-api/system/dept/create', is_check, '创建部门', **kwargs)
|
||||
|
||||
def kw_in_joyhub_dept_delete_post(self, dept_id, is_check=''):
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/dept/delete?id={dept_id}', is_check, '删除部门')
|
||||
|
||||
def kw_in_joyhub_dept_delete_list_post(self, ids, is_check=''):
|
||||
ids_str = ','.join(map(str, ids))
|
||||
return self._joyhub_request('DELETE', f'/admin-api/system/dept/delete-list?ids={ids_str}', is_check, '批量删除部门')
|
||||
|
||||
def kw_in_joyhub_dept_get_get(self, dept_id, is_check=''):
|
||||
return self._joyhub_request('GET', f'/admin-api/system/dept/get?id={dept_id}', is_check, '获得部门信息')
|
||||
|
||||
def kw_in_joyhub_dept_list_get(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('GET', '/admin-api/system/dept/list', is_check, '获取部门列表', **kwargs)
|
||||
|
||||
def kw_in_joyhub_dept_list_all_simple_get(self, is_check=''):
|
||||
return self._joyhub_request('GET', '/admin-api/system/dept/list-all-simple', is_check, '获取部门精简信息列表')
|
||||
|
||||
def kw_in_joyhub_dept_simple_list_get(self, is_check=''):
|
||||
return self._joyhub_request('GET', '/admin-api/system/dept/simple-list', is_check, '获取部门精简信息列表')
|
||||
|
||||
def kw_in_joyhub_dept_update_put(self, is_check='', **kwargs):
|
||||
return self._joyhub_request('PUT', '/admin-api/system/dept/update', is_check, '更新部门', **kwargs)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
test = DlzhanInterface()
|
||||
a = test.kw_in_zhyy_purchase_todo_get(user="purchase")
|
||||
print(a)
|
||||
Reference in New Issue
Block a user