帮助创建一个自定义创建和自定义get方法

Question

我有两个型号，像这样：

class Visit(models.Model): 
    id = models.AutoField(primary_key=True) 
    name = models.CharField(max_length=65535, null=False) 

class Session: 
    id = models.AutoField(primary_key=True) 
    visit = models.ForeignKey(Visit) 
    sequence_no = models.IntegerField(null = False) 

---

我要两个编写自定义的Session模型建立方法，所以当我写这篇文章：

visitor = Vistor.objects.get(id=1) 
new_session = Session.objects.create_new_session(visit=visitor) 

...我得到的Session表中的新纪录，为游客即3。这里的下一个连续的序列号是一些示例数据

VISITOR SEQUENCE_NO 
------- ----------- 
1   1 
1   2 
2   1 
2   2 
1   3 (This would be the row that would be created) 

---

另一个是写一个自定义得到的Session模型的方法，让我写：

visitor = Vistor.objects.get(id=1) 
new_session = Session.objects.get_session(visit=visitor, sequence_no=3) 

...我得到最高的序列号是访问者的纪录。下面是一些示例数据

VISITOR SEQUENCE_NO 
------- ----------- 
1   1 
1   2 (This would be the row that would be fetched) 
2   1 
2   2 
1   3 

---

你能告诉我如何做到这一点吗？这个代码是否应该放在Model或Manager中？

谢谢大家。

Answer 1

这将出现在会话管理器中。它看起来是这样的（未经测试）：

class SessionManager(models.Manager): 
    def new_session(self, visit, **kwargs): 
     vs = self.model.objects.filter(visit=visit).order_by("-sequence_no")[:1] 
     if len(vs): 
      kwargs.update({"sequence_no": vs.sequence_no + 1}) 
     return Super(SessionManager, self).create(visit=visit, **kwargs) 

至于得到一个会话，当你有访问，并sequence_no不应该有需要的任何自定义代码。

Answer 2

这将不得不在经理，因为这是get和create驻留。这些不是模型的方法。

class SessionManager(models.Manager): 
    def get(self, *args, **kwargs): 
     # your stuff here 
     return super(SessionManager, self).get(*args, **kwargs) 

    def create(self, *args, **kwargs): 
     # your stuff here 
     return super(SessionManager, self).create(*args, **kwargs) 

class Session(models.Model): 
    ... 
    objects = SessionManager()