Issues with CompositeKey

I have two models. Model A has a CompositeKey as primary_key, defined as indicated in [0]. Model B references model A with a ForeignKeyField(A) as indicated in [1]. Its seems that something is wrongly defined and raises an exception when tables are created in sqlite engine.

Below is the full code reproducing the problem.

Any help would be appreciated, Thanks in advance!

[0] http://peewee.readthedocs.org/en/latest/peewee/api.html#CompositeKey [1] http://peewee.readthedocs.org/en/latest/peewee/api.html#ForeignKeyField

context_db = SqliteDatabase(":memory:")

class ContextModel(Model):
    class Meta:
    database = context_db

class A(ContextModel):
    id = IntegerField()
    another_thing = CharField()
    description = TextField()

    class Meta:   
        primary_key = CompositeKey('id','description')         

class B(ContextModel):
    fk = ForeignKeyField(A)

A.create_table()
B.create_table()

And the second invocation raises:

/.../python2.7/site-packages/peewee-2.1.6-py2.7.egg/peewee.pyc in get_db_field(self)
    709         to_pk = self.rel_model._meta.primary_key
    710         if not isinstance(to_pk, PrimaryKeyField):
--> 711             return to_pk.get_db_field()
    712         return super(ForeignKeyField, self).get_db_field()
    713 

AttributeError: 'CompositeKey' object has no attribute 'get_db_field'

[imakiro]

I have this issue too, it’s still a problem. Is there any news?

[coleifer]

I still dislike having some kind of a ForeignKeyField that points to a CompositePrimaryKey. You can always add a foreign key constraint and a handy property in-place of some kind of “CompositeForeignKeyField”.