Pārlūkot izejas kodu

Fix PPVM reading issues, add stream support when adding assets to PPUBs

Billy Barrow 3 gadi atpakaļ
vecāks
revīzija
7732ae3939
2 mainītis faili ar 6 papildinājumiem un 3 dzēšanām
  1. 3 0
      ppub_builder.py
  2. 3 3
      ppvm.py

+ 3 - 0
ppub_builder.py

@@ -15,6 +15,9 @@ class PpubBuilder:
     def add_asset(self, path, mimetype, data, flags = None, streamable = False):
         self.assets.append(BuilderAsset(path, mimetype, BytesIO(data), flags, streamable))
 
+    def add_asset_stream(self, path, mimetype, stream, flags = None, streamable = False):
+        self.assets.append(BuilderAsset(path, mimetype, stream, flags, streamable))
+
     def write_to_stream(self, stream):
         stream.write(b"ppub\n")
         data_blob = b""

+ 3 - 3
ppvm.py

@@ -9,9 +9,9 @@ class FileEntry:
 
     @staticmethod
     def from_string(string):
-        parts = string.split(":", 2)
+        parts = string.split(":", 1)
         type = parts[0]
-        parts = parts[1].split(",", 3)
+        parts = parts[1].split(",", 2)
         label = parts[0].strip()
         filename = parts[1].strip()
         
@@ -51,7 +51,7 @@ class Ppvm:
             if(line == ""):
                 break
 
-            parts = line.split(":", 2)
+            parts = line.split(":", 1)
             metadata[parts[0]] = parts[1].strip()
 
         entries = []