byzh-core 0.0.8.0__py3-none-any.whl → 0.0.9.1__py3-none-any.whl

byzh/core/Brecorder/record1d.py

@@ -1,4 +1,5 @@
 import pandas as pd
+from typing import Literal
 
 from .. import B_os
 
@@ -10,7 +11,7 @@ class B_Record1d:
     recorder.write(name="Bob", age=30, city="Beijing")
     recorder.write(name="Charlie", age=22, city="Shenzhen")
     '''
-    def __init__(self, csv_path, mode="a"):
+    def __init__(self, csv_path, mode:Literal["w", "a"]="w"):
         self.csv_path = csv_path
 
         B_os.makedirs(csv_path)

byzh/core/Brecorder/record2d.py

@@ -1,10 +1,10 @@
 import pandas as pd
-from typing import Any
+from typing import Literal
 
 from .. import B_os
 
 class B_Record2d:
-    def __init__(self, csv_path, mode="a"):
+    def __init__(self, csv_path, mode:Literal["w", "a"]="w"):
         self.csv_path = csv_path
 
         B_os.makedirs(csv_path)
@@ -15,39 +15,39 @@ class B_Record2d:
         elif mode == "a":
             self.__read()
 
-    def write(self, row, col, value):
-        row, col, value = str(row), str(col), str(value)
+    # 支持 recorder[row, col] 访问
+    def __getitem__(self, key):
+        row, col = key
+        row, col = str(row), str(col)
+        return self.data.loc[row, col]
 
-        # 如果行不存在，pandas 会自动创建
-        # 如果列不存在，pandas 也会自动扩展
+    # 支持 recorder[row, col] = value
+    def __setitem__(self, key, value):
+        row, col = key
+        row, col, value = str(row), str(col), str(value)
         self.data.loc[row, col] = value
-
-        # 保存
         self.__save()
 
-    def get(self, row, col) -> Any:
-        row, col = str(row), str(col)
-        result = self.data.loc[row, col]
-        return result
+    def write(self, row, col, value):
+        self[row, col] = value
+
+    def get(self, row, col):
+        return self[row, col]
+
     def get_str(self, row, col) -> str:
-        row, col = str(row), str(col)
-        result = self.data.loc[row, col]
-        return str(result)
+        return str(self[row, col])
 
     def get_int(self, row, col) -> int:
-        row, col = str(row), str(col)
-        result = self.data.loc[row, col]
-        return int(result)
+        return int(self[row, col])
+
     def get_float(self, row, col) -> float:
-        row, col = str(row), str(col)
-        result = self.data.loc[row, col]
-        return float(result)
+        return float(self[row, col])
+
     def get_bool(self, row, col) -> bool:
-        row, col = str(row), str(col)
-        result = self.data.loc[row, col]
-        if result == "True" or result == "true" or result == "1":
+        result = self[row, col]
+        if result in ("True", "true", "1"):
             return True
-        elif result == "False" or result == "false" or result == "0":
+        elif result in ("False", "false", "0"):
             return False
         else:
             raise ValueError(f"无法转换为布尔值 -> {result}")
@@ -55,9 +55,7 @@ class B_Record2d:
     def __read(self):
         try:
             self.data = pd.read_csv(self.csv_path, index_col=0)
-        except FileNotFoundError:
-            self.data = pd.DataFrame()
-        except pd.errors.EmptyDataError:
+        except (FileNotFoundError, pd.errors.EmptyDataError):
             self.data = pd.DataFrame()
 
     def __save(self, csv_path=None):
@@ -68,19 +66,20 @@ class B_Record2d:
     def __str__(self):
         return str(self.data)
 
+
 if __name__ == '__main__':
     csv_file = "test_data.csv"
 
     recorder = B_Record2d(csv_file, mode="w")
 
-    # 写入一些单元格
-    recorder.write("awa", "OvO", 10)
-    recorder.write("awa", "TwT", 20)
-    recorder.write("qwq", "OvO", 30)
-    recorder.write("qwq", "TwT", 40)
+    # 用索引方式赋值
+    recorder["awa", "OvO"] = 10
+    recorder["awa", "TwT"] = 20
+    recorder["qwq", "OvO"] = 30
+    recorder["qwq", "TwT"] = 40
 
     print("当前内容：")
     print(recorder)
 
-    recorder = B_Record2d(csv_file, mode="a")
-    print(recorder)
+    # 用索引方式读取
+    print("awa, OvO =", recorder["awa", "OvO"])

{byzh_core-0.0.8.0.dist-info → byzh_core-0.0.9.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: byzh-core
-Version: 0.0.8.0
-Summary: byzh_core是byzh系列的核心库，包含了一些常用的工具函数和类。
+Version: 0.0.9.1
+Summary: byzh-core是byzh系列的核心库，包含了一些常用的工具函数和类。
 Author: byzh_rc
 License: MIT
 Description-Content-Type: text/markdown

{byzh_core-0.0.8.0.dist-info → byzh_core-0.0.9.1.dist-info}/RECORD

@@ -3,8 +3,8 @@ byzh/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 byzh/core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
 byzh/core/Barchive/archive.py,sha256=S1hy3qYFEZr5RRmx_Mnq1s2IMUEw973ny1rKSL7J_a0,6721
 byzh/core/Brecorder/__init__.py,sha256=rYlgYIhr_AcPuGkZRCitbi2_hZYyoERk4XIZ14CU1dk,108
-byzh/core/Brecorder/record1d.py,sha256=0iPVzX5-x-iX02HMZSq0goekNp-_g2W3uS0ZaHlAEkY,2037
-byzh/core/Brecorder/record2d.py,sha256=SIFDZVGEjizYBbp0QJl7ILCAPHdG3wcVmCvmJ3Kr2w8,2585
+byzh/core/Brecorder/record1d.py,sha256=P7y4Gbc0zYcRsZBT4CeYgv35k0tU4h3ED-3H8RQ6cuU,2083
+byzh/core/Brecorder/record2d.py,sha256=pR2qyoG_Wo40RN9ygB4z-wZfBmCW25EYjGHz3de6lLM,2342
 byzh/core/Btable/__init__.py,sha256=NlQBjp_mvObEUm4y6wXbGipkNM2N3uNIUidaJkTwFsw,62
 byzh/core/Btable/auto_table.py,sha256=avMeRWydw9L8s3fU0-BSl8a2dCyreIUzW--csm5eR0g,14734
 byzh/core/Bterminal/__init__.py,sha256=azRLD-kY8Tv2c3llHRBrEJIdVgYw6hAoLgzJeA4PvE0,142
@@ -18,9 +18,9 @@ byzh/core/Butils/text_style.py,sha256=34u2-Rsur63iNQ0m7tHxpFWqnQZWybSxBx7vnhOhqD
 byzh/core/Butils/timer.py,sha256=OLxbWWRauYjp1KwavACjoyrqrCciUiIG2ae7qgtZrVA,511
 byzh/core/Bwriter/__init__.py,sha256=QcHzKs0305Pr78ZvZgcC_4miSYZh8jJQDt5TD0GIeEg,82
 byzh/core/Bwriter/writer.py,sha256=ss9EWo-l7czBUlY7Fu0buvQgO5fRspYKtC2X6eaRa0c,3213
-byzh_core-0.0.8.0.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
-byzh_core-0.0.8.0.dist-info/METADATA,sha256=R_g0yzV3USWUfVBtvA_G0eQO2uHS8jHbj1GholsRNqQ,371
-byzh_core-0.0.8.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-byzh_core-0.0.8.0.dist-info/entry_points.txt,sha256=qoN3Uvulj0BcOQVDqsFqP2dBVQzG1QrUtC5wFsghSLo,51
-byzh_core-0.0.8.0.dist-info/top_level.txt,sha256=tmaFVY8uApe6apOETSOgwz-4v5Mj4uxgRT8yNnDNZNA,5
-byzh_core-0.0.8.0.dist-info/RECORD,,
+byzh_core-0.0.9.1.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
+byzh_core-0.0.9.1.dist-info/METADATA,sha256=g6HCDJ6JSj8wezcuKeSA5LGHPWswr0WKyTEXh5yHgDQ,371
+byzh_core-0.0.9.1.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+byzh_core-0.0.9.1.dist-info/entry_points.txt,sha256=qoN3Uvulj0BcOQVDqsFqP2dBVQzG1QrUtC5wFsghSLo,51
+byzh_core-0.0.9.1.dist-info/top_level.txt,sha256=tmaFVY8uApe6apOETSOgwz-4v5Mj4uxgRT8yNnDNZNA,5
+byzh_core-0.0.9.1.dist-info/RECORD,,