✨ xx_or_create add fetch_related arg

cherry/models/models.py

@@ -10,6 +10,7 @@
  Tuple,
  Type,
  TYPE_CHECKING,
+ Union,
 )
 from typing_extensions import dataclass_transform, Self
 
@@ -27,7 +28,6 @@
 )
 from cherry.fields.types import get_sqlalchemy_type_from_field
 from cherry.fields.utils import (
- args_and_kwargs_to_clause_list,
  classproperty,
 )
 from cherry.meta.meta import init_meta_config, MetaConfig, mix_meta_config
@@ -553,10 +553,15 @@ async def get_or_create(
  cls,
  *args: Any,
  defaults: Optional[DictStrAny] = None,
+ fetch_related: Union[bool, Tuple[Any, ...]] = False,
  **kwargs: Any,
  ) -> Tuple[Self, bool]:
  """select one model with filter condition, if not exist, create one"""
  queryset = cls.filter(*args, **kwargs)
+ if fetch_related is True or isinstance(fetch_related, tuple):
+ queryset = queryset.prefetch_related(
+ () if fetch_related is True else fetch_related,
+ )
  try:
  return await queryset.get(), True
  except NoMatchDataError:
@@ -574,12 +579,16 @@ async def update_or_create(
  cls,
  *args: Any,
  defaults: Optional[DictStrAny] = None,
+ fetch_related: Union[bool, Tuple[Any, ...]] = False,
  **kwargs: Any,
  ) -> Tuple[Self, bool]:
  """update one model with filter condition,
  if not exist, create one with filter and defaults values"""
- clause_list = args_and_kwargs_to_clause_list(cls, args, kwargs)
  queryset = cls.filter(*args, **kwargs)
+ if fetch_related is True or isinstance(fetch_related, tuple):
+ queryset = queryset.prefetch_related(
+ () if fetch_related is True else fetch_related,
+ )
  try:
  model = await queryset.get()
  return await model.update(**(defaults or {})), True

cherry/queryset/queryset.py

@@ -345,7 +345,7 @@ async def _fetch_one_related(self, conn: AsyncConnection, now_data: Dict[str, An
  rfield.related_model.__meta__.table.select().where(
  getattr(
  rfield.related_model,
- rfield.related_field_name,
+ rfield.related_field.foreign_key_self_name,
  )
  == now_data[target_field.foreign_key],
  ),
@@ -405,7 +405,7 @@ async def _fetch_many_related(
  rfield.related_model.__meta__.table.select().where(
  getattr(
  rfield.related_model,
- rfield.related_field_name,
+ rfield.related_field.foreign_key_self_name,
  ).in_(related_values),
  ),
  )

tests/test_query.py

@@ -73,3 +73,8 @@ async def test_query_with_one_to_many():
  assert len(school1.students) == 0
  await school2.fetch_related(School.students)
  assert len(school2.students) == 1
+
+ schools = await School.select_related().all()
+ assert len(schools) == 2
+ assert len(schools[0].students) == 0
+ assert len(schools[1].students) == 1