如何在Cerberus中检查参照完整性?

时间:2019-04-21 13:48:33

标签: python validation referential-integrity cerberus

考虑以下Cerberus模式:

{
  'employee': {
    'type': 'list',
    'schema': {
      'type': 'dict',
      'schema': {
        'id': {'required': True, 'type': 'integer'},
        'name': {'required': True, 'type': 'string'}
      }
    }
  },
  'ceo-employee-id': {'required': True, 'type': 'integer'}
}

1)我如何验证ceo-employee-id与员工列表中的id值之一匹配? (参照完整性)

2)如何验证员工列表中的每个ID是否唯一(即没有重复的员工ID)?

我意识到我可以按照下面@rafael的建议在验证并解析配置后在运行时执行此操作。我想知道我是否可以使用Cerberus验证功能来做到这一点。

2 个答案:

答案 0 :(得分:1)

您需要使用实现check_with方法的custom validator,在其中使用document属性,并修改架构以包括以下内容:

from cerberus import Validator


class CustomValidator(Validator):
    def _check_with_ceo_employee(self, field, value):
        if value not in (x["id"] for x in self.document["employee"]):
            self._error(field, "ID is missing in employee list.")

    def _check_with_employee_id_uniqueness(self, field, value):
        all_ids = [x["id"] for x in self.document["employee"]]
        if len(all_ids) != len(set(all_ids)):
            self._error(field, "Employee IDs are not unique.")


validator = CustomValidator({
    'employee': {
        'type': 'list',
        'schema': {
            'type': 'dict',
            'schema': {
                'id': {'required': True, 'type': 'integer'},
                'name': {'required': True, 'type': 'string'}
             },
        },
        'check_with': 'employee_id_uniqueness'
    },
    'ceo-employee-id': {'required': True, 'type': 'integer', 'check_with': 'ceo_employee'}
})

参考文档包含此处使用的所有部分的提示。

(对于可能缩入示例中的任何缩进错误,我深表歉意。)

答案 1 :(得分:0)

假设您已经验证了json的架构,则可以轻松地检查这样的两个条件。 让doc作为您的json文档。

employee_ids = [employee['id'] for employee in doc['employee']]
ceo_employee_id =  doc['ceo-employee-id']

1)我如何验证ceo-employee-id与员工列表中的id值之一匹配? (参照完整性)

ceo_id_exists_in_employees = any([employee_id == ceo_employee_id for employee_id in employee_ids])

2)如何验证员工列表中的每个ID是否唯一(即没有重复的员工ID)?

employee_id_is_unique = len(set(employee_ids)) == len(employee_ids)

3)断言两个值均为True

if ceo_id_exists_in_employees and employee_id_is_unique:
    print('passed')
else:
    print('failed')