retracesoftware-proxy 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

retracesoftware/proxy/stubfactory.py

@@ -0,0 +1,141 @@
+import sys
+import os
+
+import retracesoftware.functional as functional
+import retracesoftware.utils as utils
+
+class Stub:
+    __slots__ = []
+
+class ExtendedRef:
+    def __init__(self, module, name):
+        self.type = type
+        self.module = module
+
+class StubRef:
+    def __init__(self, module, name, methods, members):
+        self.name = name
+        self.module = module
+        self.methods = methods
+        self.members = members
+
+    def __str__(self):
+        return f'StubRef(module = {self.module}, name = {self.name}, methods = {self.methods}, members = {self.members})'
+
+def resolve(module, name):
+    try:
+        return getattr(sys.modules[module], name)
+    except:
+        return None
+
+class StubMethodDescriptor(functional.repeatedly):
+    def __init__(self, name, next_result):
+        super().__init__(next_result)
+        self.__name__ = name
+
+    def __str__(self):
+        return f"stub - {__name__}"
+    
+class StubMemberDescriptor:
+    def __init__(self, name, next_result):
+        self.next_result = next_result
+        self.__name__ = name
+
+    def __get__(self, instance, owner):
+        if instance is None:
+            return self
+        
+        return self.next_result()
+
+    def __set__(self, instance, value):
+        return self.next_result()
+
+    def __delete__(self, instance):
+        return self.next_result()
+
+    def __str__(self):
+        return f"stub member - {__name__}"
+
+class StubFunction(functional.repeatedly):
+    def __init__(self, name, next_result):
+        super().__init__(next_result)
+        self.__name__ = name
+
+    def __str__(self):
+        return f"stub function - {__name__}"
+
+class StubFactory:
+
+    __slots__ = ['next_result', 'thread_state', 'cache']
+
+    def __init__(self, thread_state, next_result):
+        self.next_result = next_result
+        self.thread_state = thread_state
+        self.cache = {}
+
+    def create_member(self, name):
+        def disabled(*args, **kwargs):
+            if self.thread_state.value == 'disabled' and name == '__repr__':
+                return f"stub member - {name}"
+            else:
+                print(f'Error trying to call member descriptor: {name} {args} {kwargs}, retrace mode: {self.thread_state.value}')
+                utils.sigtrap(None)
+                os._exit(1)
+            
+        next_result = self.thread_state.dispatch(disabled, external = self.next_result)
+
+        return StubMemberDescriptor(name = name, next_result = next_result)
+
+    def create_method(self, name):
+
+        def disabled(*args, **kwargs):
+            if self.thread_state.value == 'disabled' and name == '__repr__':
+                return f"stub - {name}"
+            else:
+                print(f'Error trying to call descriptor: {name} {args} {kwargs}, retrace mode: {self.thread_state.value}')
+                utils.sigtrap(None)
+                os._exit(1)
+            
+        next_result = self.thread_state.dispatch(disabled, external = self.next_result)
+
+        func = functional.repeatedly(next_result)
+        func.__name__ = name
+
+        return func
+
+    def create_stubtype(self, spec):
+
+        print(f'FOOO: {spec.module}')
+        slots = {
+            '__module__': spec.module,
+            '__qualname__': spec.name,
+        }
+
+        for method in spec.methods:
+            slots[method] = self.create_method(method)
+            assert utils.is_method_descriptor(slots[method])
+
+        for member in spec.members:
+            slots[member] = self.create_member(member)
+
+        resolved = resolve(spec.module, spec.name)
+
+        if isinstance(resolved, type):
+            slots['__class__'] = property(functional.repeatedly(resolved))
+
+        stubtype = type(spec.name, (Stub, ), slots)
+
+        for method in spec.methods:
+            slots[method].__objclass__ = stubtype
+
+        return stubtype
+
+    def __call__(self, spec):
+        print(f'In stubFactory.__call__ {spec}')
+        if spec not in self.cache:
+            self.cache[spec] = self.create_stubtype(spec)
+        
+        stubtype = self.cache[spec]
+        stub = stubtype.__new__(stubtype)
+        print(f'In stubFactory created new')
+        return stub

retracesoftware/proxy/thread.py

@@ -1,11 +1,46 @@
 import retracesoftware.functional as functional
 import retracesoftware_utils as utils
 
+# def thread_aware_writer(writer):
+#     on_thread_switch = functional.sequence(utils.thread_id(), writer.handle('THREAD_SWITCH'))
+#     return utils.threadawareproxy(on_thread_switch = on_thread_switch, target = writer)
+
 class ThreadSwitch:
-    def __init__(self, thread_id):
-        self.thread_id = thread_id
+    __slots__ = ['id']
+
+    def __init__(self, id):
+        self.id = id
+
+def set_thread_id(writer, id):
+    utils.set_thread_id(writer.handle(ThreadSwitch(id)))
+
+def write_thread_switch(writer):
+    on_thread_switch = functional.repeatedly(functional.sequence(utils.thread_id, writer))
+
+    return lambda f: utils.thread_aware_proxy(target = f, on_thread_switch = on_thread_switch, sticky = False)
+
+def prefix_with_thread_id(f, current):
+    def next():
+        nonlocal current, f
+        obj = f()
+
+        while issubclass(type(obj), ThreadSwitch):
+            current = obj.id
+            obj = f()
+
+        return (current, obj)
+
+    return next
+
+def per_thread_messages(messages):
+    thread_id = utils.thread_id
+    # thread_id = lambda: 'FOOOOO!!!'
 
+    demux = utils.demux(source = prefix_with_thread_id(messages, thread_id()), key_function = lambda obj: obj[0])
 
-def thread_aware_writer(writer):
-    on_thread_switch = functional.sequence(utils.thread_id(), writer.handle('THREAD_SWITCH'))
-    return utils.threadawareproxy(on_thread_switch = on_thread_switch, target = writer)
+    # def next():
+    #     thread,message = demux(thread_id())
+    #     return message
+    
+    # return next
+    return functional.repeatedly(lambda: demux(thread_id())[1])

{retracesoftware_proxy-0.1.0.dist-info → retracesoftware_proxy-0.1.2.dist-info}/METADATA

@@ -1,12 +1,7 @@
 Metadata-Version: 2.4
 Name: retracesoftware_proxy
-Version: 0.1.0
-Summary: Proxy layer for retracesoftware runtime
+Version: 0.1.2
 License: Apache-2.0
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
 Requires-Dist: retracesoftware_utils
 Requires-Dist: retracesoftware_functional
 Requires-Dist: retracesoftware_stream
-
-# proxy

retracesoftware_proxy-0.1.2.dist-info/RECORD

@@ -0,0 +1,27 @@
+retracesoftware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+retracesoftware/config.json,sha256=zhi3DZKG0pIT3Mh2EUXlzQai-U6YrknueOjMJLwNKxY,8281
+retracesoftware/install/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+retracesoftware/install/config.py,sha256=EzE5ifQF2lo--hu2njI4T0FJ-zlnWDJV6i7x0DMkVTw,1364
+retracesoftware/install/edgecases.py,sha256=NR3lyvad9sRsyeDv_Ya8V4xMgPsMPOi9rMcnFOJGOEA,6330
+retracesoftware/install/globals.py,sha256=F8XvIoZQQ10gSRalk30dvdKllxlwxkaggYY6FogLDxY,510
+retracesoftware/install/install.py,sha256=nHnXjLnZDcY2untKR7BsPJEAcwoHfvXqgCmdHLiwRPE,5104
+retracesoftware/install/patcher.py,sha256=Bqr1uwo7Kh4a1VASUfiAd0N7CI40GGyxuH6Zv9c5ir8,17259
+retracesoftware/install/predicate.py,sha256=tX7NQc0rGkyyHYO3mduYHcJHbw1wczT53m_Dpkzo6do,2679
+retracesoftware/install/record.py,sha256=tseF_jV4k4HLPTgBPJdjcahl4EQqagoiisMAdGNC52Q,3257
+retracesoftware/install/references.py,sha256=0nUY_9BkeFz2imcw97rcJNq2jqISCB8dysBbCrh1eCo,1623
+retracesoftware/install/replay.py,sha256=SV7-k05xZFiNfBaxrlLrqbB8-i01IPvnJQjRj9zuinI,1828
+retracesoftware/install/tracer.py,sha256=NFilwwtopDrEXjh4ZLkwyIgvPvzApQYC6BA78B11WTQ,4675
+retracesoftware/install/typeutils.py,sha256=_a1PuwdCsYjG1Nkd77V-flqYtwbD4RkJVKn6Z-xABL4,1813
+retracesoftware/proxy/__init__.py,sha256=ZlDZIuUmKFsE9Tvfd2EKGabTepqv8nrbr5pQhCM3IKc,193
+retracesoftware/proxy/gateway.py,sha256=xESohWXkiNm4ZutU0RgWUwxjxcBWRQ4rQyxIGQXv_F4,1590
+retracesoftware/proxy/proxyfactory.py,sha256=qhOqDfMJnLDNkQs26JqDB431MwjjRhGQi8xupJ45asg,12272
+retracesoftware/proxy/proxysystem.py,sha256=JZmi5EWI3B_l5zbq2KsFUTcD9bEsF3EjBhck_IiBmmE,6966
+retracesoftware/proxy/proxytype.py,sha256=XiCFDn6DeMoZ9tOq-VDZnsskkCfhJPSxrguIBxLWmws,13622
+retracesoftware/proxy/record.py,sha256=BGcYEzTXikHPIJXxhfITKA1uz0FcWIv92xfBheK-KCo,7480
+retracesoftware/proxy/replay.py,sha256=GHoXfCr-k23aKCOSjv0TH-nV-bJJ9l279dyTn5-982w,7949
+retracesoftware/proxy/stubfactory.py,sha256=RH4Xw9BTh_mPz81j6lmoRXfXm7RukxCxzk_H3se_shk,4152
+retracesoftware/proxy/thread.py,sha256=-SvnyVbANkmX2lLRpOvFtkpdpAoF6DhnnYdOOs7Q8vo,1379
+retracesoftware_proxy-0.1.2.dist-info/METADATA,sha256=C2ZwqWsiW4Hs0HPbadhuecKiDB8etu9dfBGYrt7Q4q0,202
+retracesoftware_proxy-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+retracesoftware_proxy-0.1.2.dist-info/top_level.txt,sha256=hYHsR6txLidmqvjBMITpIHvmJJbmoCAgr76-IpZPRz8,16
+retracesoftware_proxy-0.1.2.dist-info/RECORD,,

retracesoftware_proxy-0.1.0.dist-info/RECORD

@@ -1,26 +0,0 @@
-retracesoftware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-retracesoftware/config.json,sha256=aedf9CXxOBkcQk3BJVrh0GuBkIAjBGBRlrTXdoRJ-MY,8108
-retracesoftware/install/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-retracesoftware/install/config.py,sha256=EzE5ifQF2lo--hu2njI4T0FJ-zlnWDJV6i7x0DMkVTw,1364
-retracesoftware/install/edgecases.py,sha256=NR3lyvad9sRsyeDv_Ya8V4xMgPsMPOi9rMcnFOJGOEA,6330
-retracesoftware/install/globals.py,sha256=F8XvIoZQQ10gSRalk30dvdKllxlwxkaggYY6FogLDxY,510
-retracesoftware/install/install.py,sha256=VNtivSf6A9wtLzIkZKJQ7thPkYTdjtTmxaIkPUOwLhg,5942
-retracesoftware/install/patcher.py,sha256=5DGzVaNM53Z_KWz-wWQxlr1WqeWuuxKoVZyN5tNGuTw,15866
-retracesoftware/install/predicate.py,sha256=tX7NQc0rGkyyHYO3mduYHcJHbw1wczT53m_Dpkzo6do,2679
-retracesoftware/install/record.py,sha256=mJyz8ZFnrveDCvrUHksTXaFZbBX10yqH6UBV1QH5EwA,4370
-retracesoftware/install/references.py,sha256=0nUY_9BkeFz2imcw97rcJNq2jqISCB8dysBbCrh1eCo,1623
-retracesoftware/install/replay.py,sha256=ox43e8k_uvn9w6vf6rt8f3KwaXcBVx2GbvWaLZpwqR4,4952
-retracesoftware/install/tracer.py,sha256=dFhXKkmvGLF-_WzQvZXxNO1Ftj6crH1SM6pGBU4cqMY,4586
-retracesoftware/install/typeutils.py,sha256=_a1PuwdCsYjG1Nkd77V-flqYtwbD4RkJVKn6Z-xABL4,1813
-retracesoftware/proxy/__init__.py,sha256=NaGsqoj4X3NpHfTAmRIfsXM16R6AGE99ciQeknj3sXc,119
-retracesoftware/proxy/gateway.py,sha256=JtSwwllXiUykpNdojbK7okyYadRnILGgpRCHxAQijZs,3911
-retracesoftware/proxy/proxyfactory.py,sha256=2PgC0NtDbRmhmU--caTiQDsaF8s1h9tavQxvPIow_bM,11912
-retracesoftware/proxy/proxysystem.py,sha256=kDdtea-vCcsqdwSwmgGBBvVrIRl6Gwb89Wgx6VSIwgQ,759
-retracesoftware/proxy/proxytype.py,sha256=P78aL3_aemG8p3NKmP4YfAcwC8IDr7LIq5-dkV4n0Ec,9038
-retracesoftware/proxy/record.py,sha256=yI8CH0oOyrAyvjy9JsawgkQh9jYTlg9sQ_CcejrFKO4,4772
-retracesoftware/proxy/replay.py,sha256=usbb2OTX6uJG_zialqHk9en0mxD2xEhjPajMZDjZnkQ,3940
-retracesoftware/proxy/thread.py,sha256=9_6gWiRbir77f9kelpvXaTWviRfZo0ACkKPdGNbrq2Q,394
-retracesoftware_proxy-0.1.0.dist-info/METADATA,sha256=c24hdDKP-miIrX6mM1yVyXCQeu0ozLP4N0ebtqiGccw,324
-retracesoftware_proxy-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-retracesoftware_proxy-0.1.0.dist-info/top_level.txt,sha256=hYHsR6txLidmqvjBMITpIHvmJJbmoCAgr76-IpZPRz8,16
-retracesoftware_proxy-0.1.0.dist-info/RECORD,,