将一个Python字典转换为正确的Python基础模型（BaseModel）pydantic类

huangapple
go评论143阅读模式
英文:

Convert a python dict to correct python BaseModel pydantic class

问题

我要创建一个SendNotificationChannel实例，其中正确设置conditional_config，以使condition_prop成为ConditionalExpressionProps或CycleDurationTrendProps，根据字典的结构而定。例如，如果字典如下：

  1. {
  2.     "id": "1",
  3.     "customer_id": "abc",
  4.     "conditional_config": {
  5.         "cond_type": "EXPRESSION",
  6.         "conditional_expression": ".s_num>10"
  7.     }
  8. }

那么执行类似SendNotificationChannel(**dict)的操作应该将conditional_config.condition_prop设置为ConditionalExpressionProps，对于CycleDurationTrendProps也是类似的。

然而，当我这样做时，它却采用了BaseConditionalProps。

  1. >>> channel = {"conditional_config": {"cond_type": "EXPRESSION", "condition_prop":{"conditional_expression":"ALPHA"}}}
  2. >>> channel_obj = SendNotificationChannel(**channel)
  3. >>> channel_obj
  4. SendNotificationChannel(conditional_config=ConditionalConfig(cond_type=<ConditionType.EXPRESSION: 'EXPRESSION'>, condition_prop=BaseConditionalProps()))
  5. >>> channel_obj.conditional_config.condition_prop
  6. BaseConditionalProps()
  7. >>>

我想知道有人如何解决这个问题，或者我是否在处理这个问题的方式上有误。

英文:

My requirement is to convert python dictionary which can take multiple forms into appropriate pydantic BaseModel class instance. Following are details:

  1. class ConditionType(str, Enum):
  2.     EXPRESSION = &#39;EXPRESSION&#39;
  3.     CYCLE_DUR_TREND = &#39;CYCLE_DUR_TREND&#39;
  5. class ConditionalExpressionProps(BaseConditionalProps):
  6.     conditional_expression: str
  8. class CycleDurationTrendProps(BaseConditionalProps):
  9.     direction_up : bool = True
  10.     n : int = Field(1, ge=1, le=1000)
  12. class ConditionalConfig(BaseModel):
  13.     cond_type: ConditionType = ConditionType.EXPRESSION
  14.     condition_prop: BaseConditionalProps
  16. class SendNotificationChannel(BaseModel):
  17.     id: str
  18.     customer_id: str
  19.     conditional_config: Optional[ConditionalConfig]

I want to create a SendNotificationChannel instance with correct conditional_config set such that condition_prop becomes ConditionalExpressionProps or CycleDurationTrendProps based on the structure of dict.

For example, if dict is:

  1. {
  2.     &quot;id&quot; : &quot;1&quot;, 
  3.     &quot;customer_id&quot; : &quot;abc&quot;, 
  4.     &quot;conditional_config&quot; : {
  5.         &quot;cond_type&quot;: &quot;EXPRESSION&quot;,
  6.         &quot;conditional_expression&quot; : &quot;.s_num&gt;10&quot;
  7.     }
  8. }

then on doing something like SendNotificationChannel(**dict) should make the conditional_config.condition_prop of type ConditionalExpressionProps, and similarly for CycleDurationTrendProps.

However when I'm doing this, its taking BaseConditionalProps.

  1. &gt;&gt;&gt; channel = {&quot;conditional_config&quot; : {&quot;cond_type&quot;: &quot;EXPRESSION&quot;, &quot;condition_prop&quot;:{&quot;conditional_expression&quot;:&quot;ALPHA&quot;}}}
  2. &gt;&gt;&gt; channel_obj = SendNotificationChannel(**channel)
  3. &gt;&gt;&gt; channel_obj
  4. SendNotificationChannel(conditional_config=ConditionalConfig(cond_type=&lt;ConditionType.EXPRESSION: &#39;EXPRESSION&#39;&gt;, condition_prop=BaseConditionalProps()))
  5. &gt;&gt;&gt; channel_obj.conditional_config.condition_prop
  6. BaseConditionalProps()
  7. &gt;&gt;&gt;

I'd want to know how would someone solve this problem, or if I'm tackling this the incorrect way.

答案1

得分: 1

如果我正确理解您的问题，您可以使用 .parse_obj 方法来解决问题。此外，您需要更新 condition_prop 字段的类型。以下是示例代码：

  1. from enum import Enum
  2. from typing import Optional, Union
  4. from pydantic import BaseModel, Field
  7. class ConditionType(str, Enum):
  8.     EXPRESSION = 'EXPRESSION'
  9.     CYCLE_DUR_TREND = 'CYCLE_DUR_TREND'
  11. class BaseConditionalProps(BaseModel):
  12.     pass
  14. class ConditionalExpressionProps(BaseConditionalProps):
  15.     conditional_expression: str
  17. class CycleDurationTrendProps(BaseConditionalProps):
  18.     direction_up : bool = True
  19.     n : int = Field(1, ge=1, le=1000)
  21. class ConditionalConfig(BaseModel):
  22.     cond_type: ConditionType = ConditionType.EXPRESSION
  23.     condition_prop: Union[ConditionalExpressionProps, CycleDurationTrendProps]
  25. class SendNotificationChannel(BaseModel):
  26.     id: str
  27.     customer_id: str
  28.     conditional_config: Optional[ConditionalConfig]
  30. channel_1 = {
  31.     "id" : "1", 
  32.     "customer_id" : "abc", 
  33.     "conditional_config" : {
  34.         "cond_type": "EXPRESSION",
  35.         "condition_prop": {
  36.             "conditional_expression" : ".s_num>10"
  37.         }
  38.     }
  39. }
  41. channel_1_obj = SendNotificationChannel.parse_obj(channel_1)
  42. print(channel_1_obj)
  43. # id='1' customer_id='abc' conditional_config=ConditionalConfig(cond_type=<ConditionType.EXPRESSION: 'EXPRESSION'>, condition_prop=ConditionalExpressionProps(conditional_expression='.s_num>10'))
  45. channel_2 = {
  46.     "id" : "2", 
  47.     "customer_id" : "def", 
  48.     "conditional_config" : {
  49.         "cond_type": "CYCLE_DUR_TREND",
  50.         "condition_prop": {
  51.             "n" : 10
  52.         }
  53.     }
  54. }
  56. channel_2_obj = SendNotificationChannel.parse_obj(channel_2)
  57. print(channel_2_obj)
  58. # id='2' customer_id='def' conditional_config=ConditionalConfig(cond_type=<ConditionType.CYCLE_DUR_TREND: 'CYCLE_DUR_TREND'>, condition_prop=CycleDurationTrendProps(direction_up=True, n=10))
英文:

If I understand your question correctly you can to use .parse_obj method for your problem. Also you need to update the condition_prop field type. The following code it is example:

  1. from enum import Enum
  2. from typing import Optional, Union
  4. from pydantic import BaseModel, Field
  7. class ConditionType(str, Enum):
  8.     EXPRESSION = &#39;EXPRESSION&#39;
  9.     CYCLE_DUR_TREND = &#39;CYCLE_DUR_TREND&#39;
  11. class BaseConditionalProps(BaseModel):
  12.     pass
  14. class ConditionalExpressionProps(BaseConditionalProps):
  15.     conditional_expression: str
  17. class CycleDurationTrendProps(BaseConditionalProps):
  18.     direction_up : bool = True
  19.     n : int = Field(1, ge=1, le=1000)
  21. class ConditionalConfig(BaseModel):
  22.     cond_type: ConditionType = ConditionType.EXPRESSION
  23.     condition_prop: Union[ConditionalExpressionProps, CycleDurationTrendProps]
  25. class SendNotificationChannel(BaseModel):
  26.     id: str
  27.     customer_id: str
  28.     conditional_config: Optional[ConditionalConfig]
  31. channel_1 = {
  32.     &quot;id&quot; : &quot;1&quot;, 
  33.     &quot;customer_id&quot; : &quot;abc&quot;, 
  34.     &quot;conditional_config&quot; : {
  35.         &quot;cond_type&quot;: &quot;EXPRESSION&quot;,
  36.         &quot;condition_prop&quot;: {
  37.             &quot;conditional_expression&quot; : &quot;.s_num&gt;10&quot;
  38.         }
  39.     }
  40. }
  42. channel_1_obj = SendNotificationChannel.parse_obj(channel_1)
  43. print(channel_1_obj)
  44. # id=&#39;1&#39; customer_id=&#39;abc&#39; conditional_config=ConditionalConfig(cond_type=&lt;ConditionType.EXPRESSION: &#39;EXPRESSION&#39;&gt;, condition_prop=ConditionalExpressionProps(conditional_expression=&#39;.s_num&gt;10&#39;))
  46. channel_2 = {
  47.     &quot;id&quot; : &quot;2&quot;, 
  48.     &quot;customer_id&quot; : &quot;def&quot;, 
  49.     &quot;conditional_config&quot; : {
  50.         &quot;cond_type&quot;: &quot;CYCLE_DUR_TREND&quot;,
  51.         &quot;condition_prop&quot;: {
  52.             &quot;n&quot; : 10
  53.         }
  54.     }
  55. }
  57. channel_2_obj = SendNotificationChannel.parse_obj(channel_2)
  58. print(channel_2_obj)
  59. # id=&#39;2&#39; customer_id=&#39;def&#39; conditional_config=ConditionalConfig(cond_type=&lt;ConditionType.CYCLE_DUR_TREND: &#39;CYCLE_DUR_TREND&#39;&gt;, condition_prop=CycleDurationTrendProps(direction_up=True, n=10))

huangapple
  • 本文由 发表于 2023年1月6日 14:53:15
  • 转载请务必保留本文链接：https://go.coder-hub.com/75027840.html
  • inheritance
  • pydantic
  • python
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定