mongoengine中如何规定list(embeddeddocumentfield)中以某个域为unique?

新手上路,请多包涵

项目使用mongoengine作ORM层,现在有这样的需求:

有两个集合,User和Group,在User里有一个ListField(EmbeddedDocumentField)用于表达User和Group的关系,一个User可以属于多个Group,且对应一个Role(管理员or用户,等等)

大概如下:

# group.py
class Group(Document):
    # ...
# user.py
class GroupInfo(EmbeddedDocumentField):
    group = StringField(required=True)
    role = String(choices=['owner', 'manager', 'user'], required=True)

class User(Document):
    # ...
    groups = ListField(EmbeddedDocumentField(GroupInfo))
    # ...

我想加一个限制,即在User.groups中,不会有重复的组信息
例如

// 合法
{
    {"group": "g1", "role": "manager"},
    {"group": "g2", "role": "user"}
}
// 不合法,因为g1重复出现
{
    {"group": "g1", "role": "manager"},
    {"group": "g1", "role": "user"}
}

我尝试

class GroupInfo(EmbeddedDocumentField):
    group = StringField(required=True, unique=True)    # 添加unique=True
    role = String(choices=['owner', 'manager', 'user'], required=True)

结果却是,每个user的groups只能有一个GroupInfo,否则就会出NotUnique错误。
所以正确的实现方法是什么?

顺便这种需求有没有更好的写法?

阅读 4.2k
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题