def __get__(self, instance, owner):
super(ForeignKey, self)._set_up(instance, owner)
if not instance:
return self.model
if not self.field:
self.field = '%s_id' % self.model.Meta.table
conditions = {self.model.Meta.pk: getattr(instance, self.field)}
return Query(model=self.model, conditions=conditions)[0]
def __get__(self, instance, owner):
super(OneToMany, self)._set_up(instance, owner)
if not instance:
return self.model
if not self.field:
self.field = '%s_id' % instance.Meta.table
conditions = {self.field: getattr(instance, instance.Meta.pk)}
return Query(model=self.model, conditions=conditions)
def query(self, **kwargs):
'Returns Query object'
return Query(model=self.rclass, conditions=kwargs)
