Fix a bug in un.rpyc and also add support for RevertableDict's in the…

… AST
CensoredUsername · Jul 1, 2019 · 8fdda39 · 8fdda39
1 parent 129c4e1
commit 8fdda39
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/un.rpyc/unrpyc-compile.py b/un.rpyc/unrpyc-compile.py
@@ -48,7 +48,12 @@ class RevertableList(magic.FakeStrict, list):
     def __new__(cls):
         return list.__new__(cls)
 
-class_factory = magic.FakeClassFactory((PyExpr, PyCode, RevertableList), magic.FakeStrict)
+class RevertableDict(magic.FakeStrict, dict):
+    __module__ = "renpy.python"
+    def __new__(cls):
+        return dict.__new__(cls)
+
+factory = magic.FakeClassFactory((PyExpr, PyCode, RevertableList, RevertableDict), magic.FakeStrict)
 
 def read_ast_from_file(raw_contents):
     # .rpyc files are just zlib compressed pickles of a tuple of some data and the actual AST of the file

diff --git a/unrpyc.py b/unrpyc.py
@@ -57,7 +57,12 @@ class RevertableList(magic.FakeStrict, list):
     def __new__(cls):
         return list.__new__(cls)
 
-class_factory = magic.FakeClassFactory((PyExpr, PyCode, RevertableList), magic.FakeStrict)
+class RevertableDict(magic.FakeStrict, dict):
+    __module__ = "renpy.python"
+    def __new__(cls):
+        return dict.__new__(cls)
+
+class_factory = magic.FakeClassFactory((PyExpr, PyCode, RevertableList, RevertableDict), magic.FakeStrict)
 
 printlock = Lock()