clair/CVM-Sentry
1
0
Fork 0
Code Activity

fix that log.info or whatever

Browse Source
This commit is contained in:
Clair Delafuente
2025-10-12 18:34:12 -04:00
parent f0bd973461
commit 8b64179f41
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
cvmsentry.py
View File

@@ -105,10 +105,13 @@ async def periodic_snapshot_task():
method=6, method=6,
) )
# Get file size in kilobytes
file_size_kb = os.path.getsize(snapshot_path) / 1024
# Update the hash for this VM # Update the hash for this VM
vm_data["last_frame_hash"] = current_hash vm_data["last_frame_hash"] = current_hash
log.info(f"Saved snapshot for VM '{vm_name}' to {snapshot_path}") log.info(f"Saved snapshot for VM '{vm_name}' to {snapshot_path} ({file_size_kb:.2f} KB)")
else: else:
log.warning(f"No framebuffer available for VM '{vm_name}'") log.warning(f"No framebuffer available for VM '{vm_name}'")
@@ -326,7 +329,7 @@ async def connect(vm_name: str):
# Create an image from the decoded data # Create an image from the decoded data
image = Image.open(BytesIO(initial_frame_data)) image = Image.open(BytesIO(initial_frame_data))
# Store the image in memory as a BytesIO object # the image in memory as a BytesIO object
framebuffer = BytesIO() framebuffer = BytesIO()
image.save(framebuffer, format="JPEG") image.save(framebuffer, format="JPEG")
framebuffer.seek(0) framebuffer.seek(0)
@@ -334,9 +337,6 @@ async def connect(vm_name: str):
# Assign the in-memory framebuffer to the VM's dictionary # Assign the in-memory framebuffer to the VM's dictionary
vms[vm_name]["framebuffer"] = framebuffer vms[vm_name]["framebuffer"] = framebuffer
framebuffer_size = framebuffer.getbuffer().nbytes framebuffer_size = framebuffer.getbuffer().nbytes
log.info(
f"({STATE.name} - {vm_name}) !!! WHOLE FRAME UPDATE !!! ({framebuffer_size} bytes)"
)
case ["png", "0", "0", x, y, rect_b64]: case ["png", "0", "0", x, y, rect_b64]:
# Decode the base64 image data for the rectangle # Decode the base64 image data for the rectangle
rect_data = base64.b64decode(rect_b64) rect_data = base64.b64decode(rect_b64)
@@ -363,9 +363,6 @@ async def connect(vm_name: str):
# Log the updated framebuffer size # Log the updated framebuffer size
framebuffer_size = framebuffer.getbuffer().nbytes framebuffer_size = framebuffer.getbuffer().nbytes
log.debug(
f"({STATE.name} - {vm_name}) Updated framebuffer size: {framebuffer_size} bytes"
)
else: else:
continue continue
case ["turn", turn_time, count, current_turn, *queue]: case ["turn", turn_time, count, current_turn, *queue]: