# !/usr/bin/python3
|
# -*- coding: utf-8 -*-
|
|
import json
|
import logging
|
|
import pydash
|
|
# from loguru import logger
|
|
logger = logging.getLogger(__name__)
|
|
|
def _load_json(in_data):
|
if isinstance(in_data, (dict, list)) is False:
|
try:
|
in_data = json.loads(in_data)
|
except Exception as e:
|
logger.error(str(e))
|
raise e
|
return in_data
|
|
|
def set_json(request_obj, in_data={}, include="", json_path="."):
|
"""
|
修改请求体的json, 包含模式
|
"""
|
in_data = _load_json(in_data)
|
request_data = pydash.get(request_obj["json"], json_path)
|
include_keys = include.split("-")
|
for k in include_keys:
|
v = pydash.get(in_data, k)
|
path = k.split(".")[-1]
|
pydash.set_(request_data, path, v)
|
|
|
def set_json_e(request_obj, in_data={}, exclude="", in_path="."):
|
"""
|
修改请求体的json， 排除模式
|
"""
|
in_data = _load_json(in_data)
|
request_data = pydash.get(request_obj["json"], in_path)
|
exclude_keys = exclude.split("-")
|
if isinstance(in_data, dict):
|
for k, v in in_data.items():
|
if k in exclude_keys:
|
continue
|
pydash.set_(request_data, k, v)
|
else:
|
for index, value in enumerate(in_data):
|
if index in exclude_keys:
|
continue
|
pydash.set_(request_data, index, value)
|
