vchrome 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

vchrome/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.1.9'
+__version__ = '0.1.11'
 __author__ = 'v'
 # ----------------------------------------------------------------------------------------------------
 _allowed = {'Chrome'}
@@ -15,6 +15,7 @@ import copy
 import types
 import queue
 import time
+import ast
 import inspect
 import textwrap
 import threading
@@ -252,6 +253,18 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
             return data
         if is_err:
             raise Exception(is_err)
+    def has_print(func):
+        try:
+            if not func:
+                return False
+            code = textwrap.dedent(inspect.getsource(func))
+            tree = ast.parse(code)
+            for node in ast.walk(tree):
+                if isinstance(node, ast.Call):
+                    if isinstance(node.func, ast.Name) and node.func.id == "print":
+                        return True
+        except:
+            return False
     def is_function(obj):
         return type(obj) == types.FunctionType or type(obj) == types.MethodType
     def create_connection_saf(*a, **kw):
@@ -431,6 +444,8 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
             self.call_listen[pk] = [waiter, on_request, on_response]
             self.call_listen_keys = self.call_listen.keys()
             self.call_listen_vals[pk] = [pattern, is_regex]
+            # To make it easier to synchronize function printing (if there are print calls within the function).
+            if has_print(on_request) or has_print(on_response): monkey_print()
             return waiter
         def _handle_network(self):
             self.f.set_method_callback('Network.requestWillBeSent', self.Network_requestWillBeSent)
@@ -453,8 +468,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                         waiter, on_request, on_response = self.call_listen[k]
                         if on_request:
                             r = SniffNetwork.NetworkRequest(self.info_listen[requestId])
-                            if on_request(r):
-                                waiter.add(r)
+                            if on_request(r): waiter.add(r)
         def Network_requestWillBeSentExtraInfo(self, rdata):
             request_extra = rdata['params']
             requestId = request_extra['requestId']
@@ -467,8 +481,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                         waiter, on_request, on_response = self.call_listen[k]
                         if on_request:
                             r = SniffNetwork.NetworkRequest(self.info_listen[requestId])
-                            if on_request(r):
-                                waiter.add(r)
+                            if on_request(r): waiter.add(r)
             else:
                 self.info_listen[requestId] = {}
                 self.info_listen[requestId]['request_extra'] = request_extra
@@ -504,8 +517,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                         rbody = self.f.cdp('Network.getResponseBody', {"requestId": requestId}, sessionId=rdata.get('sessionId'), limit_time=3)
                         self.info_listen[requestId]['response_body'] = rbody
                         r = SniffNetwork.NetworkResponse(self.info_listen[requestId])
-                        if on_response(r):
-                            waiter.add(r)
+                        if on_response(r): waiter.add(r)
                         self.info_listen.pop(requestId, None)
                         self.qlist.get('V')
         def Network_loadingFailed(self, rdata):
@@ -521,8 +533,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                     if on_response:
                         self.qlist.put('V')
                         r = SniffNetwork.NetworkResponse(self.info_listen[requestId])
-                        if on_response(r):
-                            waiter.add(r)
+                        if on_response(r): waiter.add(r)
                         self.qlist.get('V')
     class SniffFetch:
         class FetchFakeResponse:
@@ -712,6 +723,8 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
             self.call_change[pk] = [waiter, on_request, on_response]
             self.call_change_keys = self.call_change.keys()
             self.call_change_vals[pk] = [pattern, is_regex]
+            # To make it easier to synchronize function printing (if there are print calls within the function).
+            if has_print(on_request) or has_print(on_response): monkey_print()
             return waiter
         def _handle_fetch(self):
             self.f.set_method_callback('Fetch.requestPaused', self.Fetch_requestPaused)
@@ -738,8 +751,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                         })
                         if on_request:
                             try:
-                                if on_request(x):
-                                    waiter.add(x)
+                                if on_request(x): waiter.add(x)
                             except:
                                 self.f.logger.log('[ERROR] in request on_request', traceback.format_exc())
                                 continue
@@ -771,6 +783,15 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                                 self.f.logger.log('[ERROR] in request on_request continue', traceback.format_exc())
                 self.f.cdp('Fetch.continueRequest', {'requestId': requestId}, sessionId=rdata.get('sessionId'))
             else:
+                # request: Redirect.
+                if 'responseHeaders' in rdata['params'] and type(rdata['params']['responseHeaders']) == list:
+                    for kv in rdata['params']['responseHeaders']:
+                        if kv['name'] == 'Location':
+                            return self.f.cdp('Fetch.continueRequest', {'requestId': requestId, 'url':kv['value']}, sessionId=rdata.get('sessionId'))
+                # request: fail Request.
+                if 'responseErrorReason' in rdata['params']:
+                    return self.f.cdp('Fetch.failRequest', {'requestId': requestId, 'errorReason':rdata['params']['responseErrorReason']}, sessionId=rdata.get('sessionId'))
+                # response
                 for k in self.call_change_keys:
                     if self._match_url(self.call_change_vals[k], url):
                         waiter, on_request, on_response = self.call_change[k]
@@ -785,8 +806,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
                                 "request": SniffFetch.FetchRequest(rdata['params']['request'])
                             })
                             try:
-                                if on_response(x):
-                                    waiter.add(x)
+                                if on_response(x): waiter.add(x)
                             except:
                                 self.f.logger.log('[ERROR] in response on_response', traceback.format_exc())
                                 continue
@@ -1933,16 +1953,22 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
             pass
         def attach(self, f):
             f.clear_cache = self.clear_cache
+            f.reset = self.reset
         def add_cache_session(self, sessionId):
             self.cache_sesslist.append(sessionId)
             if self.is_enable:
                 self._cache_cdp(sessionId)
-        def clear_cache(self):
+        def reset(self):
             self._enable()
             for sessionId in self.cache_sesslist:
                 self.f.cdp('Network.clearBrowserCookies',sessionId=sessionId)
                 self.f.cdp('Storage.clearDataForOrigin', {'origin':'*','storageTypes':'local_storage,session_storage,indexeddb'},sessionId=sessionId);
             return self.f
+        def clear_cache(self):
+            self._enable()
+            for sessionId in self.cache_sesslist:
+                self.f.cdp('Network.clearBrowserCache',sessionId=sessionId)
+            return self.f
     class Browser:
         def __init__(self, f):
             self.f = f
@@ -1959,6 +1985,7 @@ def cdp_client(hostname, port, debug=False, runtimeEnable=False, cfg={}):
             c.listen = self.f.listen
             c.intercept = self.f.intercept
             c.clear_cache = self.f.clear_cache
+            c.reset = self.f.reset
             c.js = self.f.run_js
             c._js2py = self.f._js2py
             c.tabs = self.f.root.tabs
@@ -2835,6 +2862,7 @@ class Chrome:
     def _connect(self):
         v_cfg = {}
         v_cfg['headless'] = self.headless
+        v_cfg['proxy'] = self.proxy
         v_cfg_fpath = self.user_path / 'v_config.json'
         if not is_port_open(self.port):
             self._init()
@@ -2843,8 +2871,9 @@ class Chrome:
             v_cfg = self.loadSaveF(v_cfg_fpath, v_cfg)
         else:
             self.writeSaveF(v_cfg_fpath, v_cfg)
-        if v_cfg['headless'] != self.headless:
+        if v_cfg.get('headless', 'v') != self.headless or v_cfg.get('proxy', 'v') != self.proxy:
             v_cfg['headless'] = self.headless
+            v_cfg['proxy'] = self.proxy
             self.root = cdp_client(self.hostname, port=self.port, debug=self.debug, runtimeEnable=self.runtimeEnable, cfg=v_cfg)
             self.root.active.quit()
             self._init()

{vchrome-0.1.9.dist-info → vchrome-0.1.11.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.1
 Name: vchrome
-Version: 0.1.9
+Version: 0.1.11
 

vchrome-0.1.11.dist-info/RECORD

@@ -0,0 +1,6 @@
+v/__init__.py,sha256=6ZddLaDcwLudHWmFr2utAcYhperXw9AYoItNPm8EZSM,65
+vchrome/__init__.py,sha256=9LYqVSuZahcPySY1fQZseDU8bR2gYVFlpm7GkZiJjrM,229000
+vchrome-0.1.11.dist-info/METADATA,sha256=xrfKhZxjyhM4rF6QQDkl4t_LTKPwvIMEr7I8MPexovQ,57
+vchrome-0.1.11.dist-info/WHEEL,sha256=hPN0AlP2dZM_3ZJZWP4WooepkmU9wzjGgCLCeFjkHLA,92
+vchrome-0.1.11.dist-info/top_level.txt,sha256=IYbkbnFb2FGoroYj_ZXIitmnJDgYBA29_HmK9BdEUkY,10
+vchrome-0.1.11.dist-info/RECORD,,

vchrome-0.1.9.dist-info/RECORD

@@ -1,6 +0,0 @@
-v/__init__.py,sha256=6ZddLaDcwLudHWmFr2utAcYhperXw9AYoItNPm8EZSM,65
-vchrome/__init__.py,sha256=jSrnEZLp3_atYOrh3Sd-QF2n1ynTIwIuCGXnuXIR_DM,227216
-vchrome-0.1.9.dist-info/METADATA,sha256=s8cXPtNAplO0IEcCir-q9X-souSVzmAR4ocahqNlCFw,56
-vchrome-0.1.9.dist-info/WHEEL,sha256=hPN0AlP2dZM_3ZJZWP4WooepkmU9wzjGgCLCeFjkHLA,92
-vchrome-0.1.9.dist-info/top_level.txt,sha256=IYbkbnFb2FGoroYj_ZXIitmnJDgYBA29_HmK9BdEUkY,10
-vchrome-0.1.9.dist-info/RECORD,,