refactor: orm: directly define __class_getitem__ by assigning with pa…

…rameterized_class_getitem (#66)

src/simple_sqlite3_orm/_orm/_async.py

@@ -91,6 +91,7 @@ async def _gen() -> AsyncGenerator[TableSpecType]:
 
         return _gen()
 
+    _wrapped.__doc__ = func.__doc__
     return _wrapped
 
 
@@ -139,8 +140,7 @@ def __init__(
 
         self._loop = asyncio.get_running_loop()
 
-    def __class_getitem__(cls, params):
-        return parameterized_class_getitem(cls, params)
+    __class_getitem__ = classmethod(parameterized_class_getitem)
 
     @cached_property
     def orm_table_name(self) -> str:

src/simple_sqlite3_orm/_orm/_base.py

@@ -106,8 +106,7 @@ def __init__(
             self._con = con()
         row_factory_setter(self._con, self.orm_table_spec, row_factory)
 
-    def __class_getitem__(cls, params):
-        return parameterized_class_getitem(cls, params)
+    __class_getitem__ = classmethod(parameterized_class_getitem)
 
     @property
     def orm_con(self) -> sqlite3.Connection:

src/simple_sqlite3_orm/_orm/_multi_thread.py

@@ -129,8 +129,7 @@ def __init__(
             thread_name_prefix=thread_name_prefix,
         )
 
-    def __class_getitem__(cls, params):
-        return parameterized_class_getitem(cls, params)
+    __class_getitem__ = classmethod(parameterized_class_getitem)
 
     def _thread_initializer(self, con_factory, row_factory) -> None:
         """Prepare thread_scope ORMBase instance for this worker thread."""

tests/test__orm.py

@@ -16,11 +16,9 @@
 
 
 class ORMTest(ORMBase[SampleTable]):
-    pass
+    _orm_table_name = "test_orm_table"
 
 
-TABLE_NAME = "test_orm_table"
-
 _cur_timestamp = time.time()
 mstr = Mystr(_generate_random_str())
 entry_for_test = SampleTable(
@@ -49,14 +47,14 @@ class TestORMBase:
     @pytest.fixture(scope="class")
     def setup_connection(self):
         with sqlite3.connect(":memory:") as conn:
-            orm_inst = ORMTest(conn, table_name=TABLE_NAME)
+            orm_inst = ORMTest(conn)
             yield orm_inst
 
     def test_create_without_rowid_table(self):
         """NOTE: to test select_all_with_pagination, we cannot specify without_rowid, so we
         create this test case dedicated for creating without_rowid table test."""
         with sqlite3.connect(":memory:") as conn:
-            orm_inst = ORMTest(conn, table_name=TABLE_NAME)
+            orm_inst = ORMTest(conn)
             orm_inst.orm_create_table(without_rowid=True)
 
     def test_create_table(self, setup_connection: ORMTest):
@@ -137,7 +135,7 @@ def test_select_all_entries(self, setup_connection: ORMTest):
 
     def test_function_call(self, setup_connection: ORMTest):
         with setup_connection.orm_con as con:
-            cur = con.execute(f"SELECT count(*) FROM {TABLE_NAME};")
+            cur = con.execute(f"SELECT count(*) FROM {ORMTest._orm_table_name};")
             res = cur.fetchone()
             assert res[0] == 1