def create_payload()

in cvm-attestation/src/AttestationProvider.py [0:0]

• 25 lines of code
• 7 McCabe index (conditional complexity)

  def create_payload(self, evidence: str, runtimes_data: str):
    # Check if evidence and runtimes_data are strings
    if not isinstance(evidence, str):
      raise ValueError("The 'evidence' argument must be an encoded string.")
    if not isinstance(runtimes_data, str):
      raise ValueError("The 'runtimes_data' argument must be an encoded string.")

    payload = ''
    runtime_data_format = {
      'data': runtimes_data,
      'dataType': 'JSON'
    }

    if self.isolation == IsolationType.TDX:
      payload = {
        'quote': evidence,
        'runtimeData': runtime_data_format
      }
    elif self.isolation == IsolationType.SEV_SNP:
      payload = {
        'report': evidence,
        'runtimeData': runtime_data_format
      }
    else:
      raise ValueError(
        f"Invalid Isolation Type. Valid Types: {IsolationType.TDX}, {IsolationType.SEV_SNP}"
      )
    return payload