kmisc 2.1.119__py3-none-any.whl → 2.1.121__py3-none-any.whl

kmisc/__init__.py

@@ -3447,6 +3447,115 @@ def Upper(src,default='org'):
     if isinstance(src,str): return src.upper()
     if default in ['org',{'org'}]: return src
     return default
+
+def web_capture(url,output_file,image_size='1920,1080',wait_time=3,ignore_certificate_error=False,username=None,password=None,auth_fields={'auth':{'type':'name','name':('username','password')},'submit':{'type':'submit','name':None}},next_do={}):
+    #auth_fields.submit.type : name    : login button with name 
+    #                        : id      : login button with id 
+    #                        : submit  : submit button without name or id
+    #auth_fields.submit.name : <string>: name or id's submit name string
+    #auth_fields.auth.type   : name    : username/password with name field
+    #                        : id      : username/password with name id
+    #auth_fields.auth.name   : (usernane,password)    : username/password field string for name or id
+    #next_do                 : put data and click submit
+
+    if isinstance(image_size,str):
+        if 'x' in image_size:
+            image_size=image_size.split('x')
+        elif ',' in image_size:
+            image_size=image_size.split(',')
+    if isinstance(image_size,(list,tuple)) and len(image_size) == 2:
+        image_size=','.join([str(i) for i in image_size])
+    else:
+        #Set it to default image size
+        image_size='1920,1080'
+
+    if Import('import selenium'):
+        return False,'Can not install selenium package'
+    else:
+        # Configure Chrome options for headless mode
+        from selenium.webdriver.chrome.options import Options
+        from selenium.webdriver.common.by import By
+        from selenium.webdriver.support.ui import WebDriverWait
+        from selenium.webdriver.support import expected_conditions as EC
+        chrome_options = Options()
+        chrome_options.add_argument('--headless')  # Run in headless mode
+        chrome_options.add_argument('--no-sandbox')
+        chrome_options.add_argument('--disable-dev-shm-usage')
+        chrome_options.add_argument(f"--window-size={image_size}")  # Set window size
+        if ignore_certificate_error:
+            chrome_options.add_argument('--ignore-certificate-errors') # gnore-certificate-errors
+            chrome_options.add_argument('--allow-insecure-localhost')  # gnore-certificate-errors
+        # Initialize the Chrome driver
+        driver = selenium.webdriver.Chrome(options=chrome_options)
+        rc=False,output_file
+        try:
+            # Navigate to the URL
+            driver.get(url)
+
+            #Login to web page
+            if username and password:
+                wait = WebDriverWait(driver, 10)  # Wait up to 10 seconds
+                #Search field name for username,password
+                if auth_fields.get('auth').get('type') == 'id':
+                    by_type=By.ID
+                else:
+                    by_type=By.NAME
+                username_field = wait.until(EC.presence_of_element_located((by_type, auth_fields.get('auth').get('name')[0])))
+                password_field = driver.find_element(by_type, auth_fields.get('auth').get('name')[1])
+                #login submit 
+                if auth_fields.get('submit',{}).get('type') in ['id','name']:
+                    if auth_fields.get('submit',{}).get('type') == 'id':
+                        by_type=By.ID
+                    else:
+                        by_type=By.NAME
+                    login_button = driver.find_element(by_type, auth_fields.get('submit').get('name'))
+                else: #button submit type
+                    login_button = driver.find_element(By.XPATH, "//button[@type='submit']")
+                #put username/password into field
+                username_field.send_keys(username) 
+                password_field.send_keys(password) 
+                #Login 
+                login_button.click()
+
+            if next_do and isinstance(next_do.get('datas'),(list,tuple)):
+                wait = WebDriverWait(driver, 10)  # Wait up to 10 seconds
+                for do_i in next_do.get('datas'):
+                    if not isinstance(do_i,dict): continue
+                    if do_i.get('type') == 'id':
+                        do_type=By.ID
+                    else:
+                        do_type=By.NAME
+                    user_do_field = wait.until(EC.presence_of_element_located((do_type, do_i.get('name'))))
+                    if do_i.get('click'):
+                        user_do_field.click()
+                    else:
+                        user_do_field.send_keys(do_i.get('data')) 
+                #Submit data
+                if next_do.get('type') in ['id','name']:
+                    if next_do.get('type') == 'id':
+                        by_type=By.ID
+                    else:
+                        by_type=By.NAME
+                    next_do_button = driver.find_element(by_type, next_do.get('name'))
+                else: #button submit type
+                    next_do_button = driver.find_element(By.XPATH, "//button[@type='submit']")
+                next_do_button.click()
+
+            # Wait for the page to load
+            time.sleep(wait_time)
+
+            # Capture screenshot
+            driver.save_screenshot(output_file)
+            #print(f"Screenshot saved to {output_file}")
+            rc=True,output_file
+        except Exception as e:
+            #print(f"Error capturing screenshot: {str(e)}")
+            rc=False,str(e)
+        finally:
+            # Close the browser
+            driver.quit()
+        return rc
+
 ############################################
 #Temporary function map for replacement
 ############################################

{kmisc-2.1.119.dist-info → kmisc-2.1.121.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kmisc
-Version: 2.1.119
+Version: 2.1.121
 Summary: Enginering useful library
 Home-page: https://github.com/kagepark/kmisc
 Author: Kage Park

kmisc-2.1.121.dist-info/RECORD

@@ -0,0 +1,6 @@
+kmisc/__init__.py,sha256=uXh4rvX7twUmd9o7TqICr5UCWlAa2i8kzKmpa4BpbSA,150362
+kmisc-2.1.121.dist-info/LICENSE,sha256=mn9ekhb34HJxsrVhcxrLXJUzy55T62zg-Gh9Ro0mVJI,1066
+kmisc-2.1.121.dist-info/METADATA,sha256=L084X4RVPGQGXITXXVrYSswc1agQNfiKVJRdkDu6S3s,5523
+kmisc-2.1.121.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+kmisc-2.1.121.dist-info/top_level.txt,sha256=wvdHf5aQTqcGYvxk-F9E_BMWLMhlwC8INBmwO-V6_X4,6
+kmisc-2.1.121.dist-info/RECORD,,

kmisc-2.1.119.dist-info/RECORD

@@ -1,6 +0,0 @@
-kmisc/__init__.py,sha256=y5Zhnl1MrphTwpiFDDqb1Wik16xRMSVa-E245qIvgyQ,144993
-kmisc-2.1.119.dist-info/LICENSE,sha256=mn9ekhb34HJxsrVhcxrLXJUzy55T62zg-Gh9Ro0mVJI,1066
-kmisc-2.1.119.dist-info/METADATA,sha256=xSeiDK45vbXDx2QUB1aWT4KhKScIJwPcd683w8ZoiX8,5523
-kmisc-2.1.119.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-kmisc-2.1.119.dist-info/top_level.txt,sha256=wvdHf5aQTqcGYvxk-F9E_BMWLMhlwC8INBmwO-V6_X4,6
-kmisc-2.1.119.dist-info/RECORD,,