from pytestlab.compliance import Signature
from pytestlab.instruments import PowerSupply

# Create a simulated power supply
psu = PowerSupply(backend='sim')

# Configure the power supply
psu.voltage = 5.0
psu.current_limit = 1.0

# Create a signature of the current state
signature = Signature.create(psu)

print("Instrument Signature:")
print(signature.to_json(indent=2))

from pytestlab.compliance import audit

# Enable auditing for the PowerSupply class
audit.enable(PowerSupply)

# Create a new power supply with auditing enabled
psu_audited = PowerSupply(backend='sim')

# Make some changes
psu_audited.voltage = 3.3
psu_audited.current_limit = 0.5
psu_audited.output = True

# Get the audit trail
trail = audit.get_trail(psu_audited)

print("Audit Trail:")
for record in trail:
    print(record)

# We'll use the signature we created in the first step.
print("Original Signature:")
print(signature.to_json(indent=2))

# Let's use the audited PSU, which has a different state.
print(f"\nVerification against current state of psu_audited: {psu_audited.voltage}V, {psu_audited.current_limit}A")

# Verify the current state against the original signature
is_verified, diff = signature.verify(psu_audited)

if is_verified:
    print("\nInstrument state VERIFIED.")
else:
    print("\nInstrument state MISMATCH.")
    print("Differences:")
    print(diff)

# Now, let's restore the state of the original psu and verify
print(f"\nVerification against original psu state: {psu.voltage}V, {psu.current_limit}A")
is_verified_orig, diff_orig = signature.verify(psu)
if is_verified_orig:
    print("\nInstrument state VERIFIED.")
else:
    print("\nInstrument state MISMATCH.")
    print("Differences:")
    print(diff_orig)