kmisc 2.1.118__tar.gz → 2.1.120__tar.gz

{kmisc-2.1.118 → kmisc-2.1.120}/PKG-INFO

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

{kmisc-2.1.118 → kmisc-2.1.120}/kmisc/__init__.py

@@ -1973,30 +1973,35 @@ def XML2Dict(root,path=[],sub=0,ignore_value=['\n']):
                 root=tree.getroot()
             except:
                 return False
+    def ignore(a,b):
+        if isinstance(b,tuple) and a not in b:
+            return True
+        elif isinstance(b,set) and a in b:
+            return True
+        elif not isinstance(b,(tuple,set)) and b != '*' and a != b:
+            return True
+        return False
 
     attr_root=root.attrib
     root_tag=attr_root.get('name',root.tag)
-    out={root_tag:attr_root}
+    #Current Path (root)
+    if path and len(path) > sub:
+        if ignore(root_tag,path[sub]): return {}
+    out={root_tag:{}}
+    for i in attr_root:
+        if path and len(path) > sub+1:
+            if ignore(i,path[sub+1]): continue
+        out[root_tag][i]=attr_root[i]
     d_root=out[root_tag]
-    new_sub=sub+1
     for x in root:
         if isinstance(x,str): continue
-        if path:
-            if len(path) <= sub: continue
-            if isinstance(path[sub],tuple): #tuple then choose only same key
-                if root_tag not in path[sub]:
-                    continue
-            elif isinstance(path[sub],set): #set then ignore the choosed names, but others are ok
-                if root_tag in path[sub]:
-                    continue
-            else:
-                if path[sub] != '*' and root_tag != path[sub]:
-                    continue
-
+        #Sub Path
         # for <Subtitle> ... </Subtitle>
         # for <Text> ... </Text>
         attr_x=x.attrib
         x_tag=attr_x.get('name',attr_x.get('id',x.tag)) #<tag name=xxx  id=xxx> ... </tag>
+        if path and len(path) > sub+1:
+            if ignore(x_tag,path[sub+1]): continue
         #Special Tag
         if x_tag == 'Subtitle': #Subtitle
             if x_tag not in d_root:
@@ -2025,18 +2030,26 @@ def XML2Dict(root,path=[],sub=0,ignore_value=['\n']):
                     continue
             d_root[x_tag][-1]['data']=x_data  
         else: #Normal case
-            d_root[x_tag]=attr_x # {...}
+            #Sub's sub
+            #d_root[x_tag]=attr_x # {...}
+            d_root[x_tag]={}
+            for i in attr_x:
+                if path and len(path) > sub+1:
+                    if ignore(i,path[sub+1]): continue
+                d_root[x_tag][i]=attr_x[i]
+
             x_data=x.text # <>text</>  put at "data" key's value
             sub_data=XML2Dict(x,path=path,sub=sub+1,ignore_value=ignore_value)
-            sub_key=next(iter(sub_data))
-            if sub_data[sub_key]:
-                for i in sub_data[sub_key]:
-                    d_root[x_tag][i]=sub_data[sub_key][i]
-            if isinstance(x_data,str):
-                _x_=x_data.strip()
-                if _x_ == '\n' or not _x_ or _x_ in ignore_value:
-                    continue
-            d_root[x_tag]['data']=x_data  
+            if len(sub_data) == 1:
+                sub_key=next(iter(sub_data))
+                if sub_data[sub_key]:
+                    for i in sub_data[sub_key]:
+                        d_root[x_tag][i]=sub_data[sub_key][i]
+                if isinstance(x_data,str):
+                    _x_=x_data.strip()
+                    if _x_ == '\n' or not _x_ or _x_ in ignore_value:
+                        continue
+                d_root[x_tag]['data']=x_data  
     return out
 
 def findXML(xmlfile,find_name=None,find_path=None,default=None,out='xmlobj',get_opt=None,find_all=False):
@@ -3434,6 +3447,51 @@ 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):
+    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
+        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
+        # Initialize the Chrome driver
+        driver = selenium.webdriver.Chrome(options=chrome_options)
+        rc=False,output_file
+        try:
+            # Navigate to the URL
+            driver.get(url)
+            
+            # 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.118 → kmisc-2.1.120}/kmisc.egg-info/PKG-INFO

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