knify 1.8.48__tar.gz → 1.8.50__tar.gz

{knify-1.8.48 → knify-1.8.50}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: knify
-Version: 1.8.48
+Version: 1.8.50
 Summary: Development tools for python
 Home-page: https://github.com/qicongsheng/knify
 Author: qicongsheng
@@ -11,3 +11,4 @@ Platform: any
 License-File: LICENSE
 Requires-Dist: loguru
 Requires-Dist: urllib3
+Requires-Dist: openpyxl

knify-1.8.50/knify/excelutil.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+# Author: qicongsheng
+from typing import Callable
+
+from openpyxl.reader.excel import load_workbook
+
+
+class Header:
+    def __init__(self, index: int, name: str | None, transformer: Callable[[object], object] = None):
+        self.index = index
+        self.name = name
+        self.transformer = transformer
+
+
+class HeaderBuilder:
+    def __init__(self):
+        self.default_transformer = None
+        self.data = []
+
+    def set_default_transformer(self, transformer: Callable[[object], object] = None):
+        self.default_transformer = transformer
+        return self
+
+    def build(self, name: str | None, transformer: Callable[[object], object] = None):
+        target_transformer = transformer if transformer is not None else self.default_transformer
+        self.data.append(Header(len(self.data), name, target_transformer))
+        return self
+
+    def to_headers(self):
+        return self.data
+
+
+def read_excel(file_path: str, sheet: str | int | None = 0, headers: list[Header] | None = None, start_row: int = 1):
+    results = []
+    workbook = load_workbook(filename=file_path)
+    sheet_ = workbook[sheet] if isinstance(sheet, str) else workbook[workbook.sheetnames[sheet]]
+    headers_ = [cell.value for cell in sheet_[1]]
+    for row_idx, row in enumerate(sheet_.rows):
+        if row_idx < start_row:
+            continue
+        result = {}
+        for header_idx, header_ in enumerate(headers_):
+            # 没有传入headers,使用默认header
+            if headers is None or len(headers) == 0:
+                result[header_] = row[header_idx].value
+            # 传入了headers
+            else:
+                target_headers = list(filter(lambda h_: h_.index == header_idx, headers))
+                header = target_headers[0] if target_headers is not None and len(target_headers) > 0 else None
+                if header is None:
+                    continue
+                else:
+                    col_name = header.name if header.name is not None else header_
+                    cell_value = row[header_idx].value
+                    cell_value = cell_value if header.transformer is None else header.transformer(row[header_idx].value)
+                    result[col_name] = cell_value
+            results.append(result)
+    return results

{knify-1.8.48 → knify-1.8.50}/knify/help.py

@@ -8,7 +8,7 @@ def get_pip_name():
 
 
 def get_version():
-    return '1.8.48'
+    return '1.8.50'
 
 
 def print_version():

knify-1.8.50/knify/listutil.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+# Author: qicongsheng
+
+
+def partition(list_obj: list, partition_size):
+    return [list_obj[i:i + partition_size] for i in range(0, len(list_obj), partition_size)]
+
+
+def is_empty(list_obj: list):
+    return list_obj is None or len(list_obj) == 0
+
+
+def is_not_empty(list_obj: list):
+    return list_obj is not None and len(list_obj) > 0

{knify-1.8.48 → knify-1.8.50}/knify.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: knify
-Version: 1.8.48
+Version: 1.8.50
 Summary: Development tools for python
 Home-page: https://github.com/qicongsheng/knify
 Author: qicongsheng
@@ -11,3 +11,4 @@ Platform: any
 License-File: LICENSE
 Requires-Dist: loguru
 Requires-Dist: urllib3
+Requires-Dist: openpyxl

{knify-1.8.48 → knify-1.8.50}/knify.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ README.md
 setup.py
 knify/__init__.py
 knify/dateutil.py
+knify/excelutil.py
 knify/fileutil.py
 knify/help.py
 knify/listutil.py

{knify-1.8.48 → knify-1.8.50}/knify.egg-info/requires.txt

@@ -1,2 +1,3 @@
 loguru
 urllib3
+openpyxl

{knify-1.8.48 → knify-1.8.50}/setup.py

@@ -19,6 +19,7 @@ setup(
     platforms='any',
     install_requires=[
         'loguru',
-        'urllib3'
+        'urllib3',
+        'openpyxl'
     ]
 )

knify-1.8.48/knify/listutil.py

@@ -1,7 +0,0 @@
-#!/usr/bin/env python
-# -*- coding:utf-8 -*-
-# Author: qicongsheng
-
-
-def partition(list_obj, partition_size):
-    return [list_obj[i:i + partition_size] for i in range(0, len(list_obj), partition_size)]