From 7a3df8a64bb8015c0c57c7429ef0629a63eee167 Mon Sep 17 00:00:00 2001
From: FoxRefire <155989196+FoxRefire@users.noreply.github.com>
Date: Sun, 28 Apr 2024 06:15:48 +0900
Subject: [PATCH] Refactor: Code cleanup

---
 python/pre.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/python/pre.py b/python/pre.py
index 8dcc050..ad76a85 100644
--- a/python/pre.py
+++ b/python/pre.py
@@ -10,10 +10,8 @@ pssh = PSSH(pssh)
 
 # load device
 try:
-    with open('device.wvd', 'wb') as f:
-        wvdExt=await (await pyfetch("device.wvd")).bytes()
-        f.write(wvdExt)
-    device = Device.load("device.wvd")
+    wvd = await (await pyfetch("device.wvd")).bytes()
+    device = Device.loads(wvd)
 except OSError:
     try:
         print("device.wvd not found! looking for device_client_id_blob and device_private_key...")