vchrome 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

vchrome/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.0.7'
+__version__ = '0.0.8'
 __author__ = 'v'
 # ----------------------------------------------------------------------------------------------------
 _allowed = {'Chrome'}
@@ -16,6 +16,7 @@ import types
 import queue
 import time
 import inspect
+import textwrap
 import threading
 from time import perf_counter
 from math import factorial, sin, pi
@@ -25,7 +26,6 @@ import socket
 import base64
 import traceback
 from json import JSONDecodeError
-from functools import lru_cache
 from threading import Thread, Event, RLock
 from urllib import request
 import builtins
@@ -148,9 +148,6 @@ def cdp_client(hostname, port, debug=False):
         def close_all(self):
             for i in range(self.num):
                 self._pool.put(self.KillThreadParams)
-    @lru_cache(maxsize=1024)
-    def get_cached_signature(func):
-        return inspect.signature(func)
     def to_human_read(size_in_bytes, decimal_places= 2):
         if size_in_bytes < 0:
             raise ValueError("byte size cannot be negative!")
@@ -466,30 +463,36 @@ def cdp_client(hostname, port, debug=False):
         class FetchFakeResponse:
             def __init__(self, rinfo, encoding='utf8'):
                 self.rinfo = rinfo
-                self.encoding = encoding
+                self.encoding = rinfo.get('encoding') or encoding
                 self._url = rinfo.get('url')
-                self._responseCode = 200
-                self._responseHeaders = {"fake-header": "fake-header"}
-                self._body = b''
+                self._responseCode = rinfo.get('status_code') or 200
+                self._responseHeaders = rinfo.get('headers') or {"fake-header": "fake-header"}
+                if rinfo.get('text'):
+                    self._body = rinfo.get('text').encode(self.encoding)
+                elif rinfo.get('content'):
+                    assert isinstance(rinfo.get('content'),  bytes), '.content must be type: bytes'
+                    self._body = rinfo.get('content')
+                else:
+                    self._body = b''
                 self.request = rinfo.get('request')
             @property
             def url(self): return self._url
             @property
             def status_code(self): return self._responseCode
             @status_code.setter
-            def status_code(self, value): assert isinstance(value, int), 'must be type: int'; self._responseCode = value
+            def status_code(self, value): assert isinstance(value, int), '.status_code must be type: int'; self._responseCode = value
             @property
             def headers(self): return self._responseHeaders
             @headers.setter
-            def headers(self, value): assert isinstance(value, dict), 'must be type: dict'; self._responseHeaders = value
+            def headers(self, value): assert isinstance(value, dict), '.headers must be type: dict'; self._responseHeaders = value
             @property
             def content(self): return self._body
             @content.setter
-            def content(self, value): assert isinstance(value, bytes), 'must be type: bytes'; self._body = value
+            def content(self, value): assert isinstance(value, bytes), '.content must be type: bytes'; self._body = value
             @property
             def text(self): return self._body.decode(self.encoding)
             @text.setter
-            def text(self, value): assert isinstance(value, str), 'must be type: str'; self._body = value.encode(self.encoding)
+            def text(self, value): assert isinstance(value, str), '.text must be type: str'; self._body = value.encode(self.encoding)
             def json(self):
                 c = self.text
                 return json.loads(c[c.find('{'):c.rfind('}')+1])
@@ -509,24 +512,36 @@ def cdp_client(hostname, port, debug=False):
                 self._postData = rinfo.get('postData')
                 self._postData_copy = self._postData
                 self._fake = None
+                self._alert = textwrap.dedent('''
+                The fake_response() function is already running! 
+                req object cannot be modified. maybe you're using the feature in the wrong way.
+                below is an example of a modification:
+
+                def on_request(req):
+                    fk_resp = req.fake_response(); 
+                    fk_resp.content = b'xxxxxx'
+                    # If you need to modify the return content, 
+                    # you need to use the return value object of the fake_response() function.
+                ''').strip()
+            def _A(self): assert self._fake == None, self._alert
             @property
             def url(self): return self._url
             @property
             def method(self): return self._method
             @method.setter
-            def method(self, value): assert isinstance(value, str), 'must be type: str'; self._method = value
+            def method(self, value): self._A(); assert isinstance(value, str), 'must be type: str'; self._method = value
             @property
             def headers(self): return self._headers
             @headers.setter
-            def headers(self, value): assert isinstance(value, dict), 'must be type: dict'; self._headers = value
+            def headers(self, value): self._A(); assert isinstance(value, dict), 'must be type: dict'; self._headers = value
             @property
             def content(self): return self._postData
             @content.setter
-            def content(self, value): assert isinstance(value, bytes), 'must be type: bytes'; self._postData = value
+            def content(self, value): self._A(); assert isinstance(value, bytes), 'must be type: bytes'; self._postData = value
             @property
             def text(self): return self._postData.decode(self.encoding)
             @text.setter
-            def text(self, value): assert isinstance(value, str), 'must be type: str'; self._postData = value.encode(self.encoding)
+            def text(self, value): self._A(); assert isinstance(value, str), 'must be type: str'; self._postData = value.encode(self.encoding)
             def json(self):
                 c = self.text
                 return json.loads(c[c.find('{'):c.rfind('}')+1])
@@ -541,8 +556,11 @@ def cdp_client(hostname, port, debug=False):
                 )
             def get(self, k):
                 return getattr(self, '_'+k, None)
-            def fake_response(self):
-                self._fake = SniffFetch.FetchFakeResponse({"url": self._url, "request": self})
+            def fake_response(self, data=None):
+                data = data or dict()
+                data['request'] = self
+                data['url'] = self._url
+                self._fake = SniffFetch.FetchFakeResponse(data)
                 return self._fake
         class FetchResponse:
             def __init__(self, rinfo, encoding='utf8'):

{vchrome-0.0.7.dist-info → vchrome-0.0.8.dist-info}/METADATA

@@ -1,3 +1,3 @@
 Metadata-Version: 2.4
 Name: vchrome
-Version: 0.0.7
+Version: 0.0.8

vchrome-0.0.8.dist-info/RECORD

@@ -0,0 +1,5 @@
+vchrome/__init__.py,sha256=54rvOagRIZrrCOF7LrudzF85S8pzgAEZIsVq-bbpQMM,210094
+vchrome-0.0.8.dist-info/METADATA,sha256=tKmwbpzUM3M5bQZJ0biHSgcoChSI49lVRZ2Z9z0PLjk,54
+vchrome-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vchrome-0.0.8.dist-info/top_level.txt,sha256=oB919Fa09PCy48Ptj8iy-1QUcvhEeY97MOR42281Fk0,8
+vchrome-0.0.8.dist-info/RECORD,,

vchrome-0.0.7.dist-info/RECORD

@@ -1,5 +0,0 @@
-vchrome/__init__.py,sha256=rZTIhIcODRjNL8PUMvk6XO3mHzwMyqsjdfuKuJPsjwk,208966
-vchrome-0.0.7.dist-info/METADATA,sha256=mdNZkOVidXUB1H8G3Rhgp_3SvSKTqXApqZ4Ei3QAXR0,54
-vchrome-0.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vchrome-0.0.7.dist-info/top_level.txt,sha256=oB919Fa09PCy48Ptj8iy-1QUcvhEeY97MOR42281Fk0,8
-vchrome-0.0.7.dist-info/RECORD,,