feat: write_to_file now handles both bytes and string types automatic…

…ally
i2mint · Jan 17, 2025 · 4d5189b · 4d5189b
1 parent 983aa93
commit 4d5189b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dol/util.py b/dol/util.py
@@ -1884,8 +1884,12 @@ def read_from_bytes(
 
 
 def write_to_file(obj: VT, key: KT):
-    with open(key, "wb") as f:
-        f.write(obj)
+    if isinstance(key, bytes):
+        with open(key, "wb") as f:
+            f.write(obj)
+    else:
+        with open(key, "w") as f:
+            f.write(obj)
 
 
 def written_key(