"""Data validation module.""" from trafaret import DataError import trafaret_schema import lambda_exceptions def validate_phf_sales_data_record(phf_record, line_number): """Validate phf publishing escrow data. Args: phf_record (dict): Phf sales data. line_number (int): line that is being validated. Returns: dict: validated phf publishing escrow data. """ try: validated_metadata = ( trafaret_schema.PHF_RECORD_SCHEMA.check( phf_record)) return validated_metadata except DataError as e: trafaret_error = e.error invalid_fields = {line_number: [field for field in trafaret_error]} raise lambda_exceptions.DataValidationFailed(invalid_fields)