CODEWITHSUNDEEP
pythonsqlalchemy
Charleess
Charleess

Reputation: 73

Adding data to a inherited class in SQLAlchemy

I'm trying to create a database to store people and addresses: home and work. I've used inherited classes to create both addresses.

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import ForeignKey, Column, Integer, String, Float
from sqlalchemy.orm import relationship

Base = declarative_base()

class Colleague(Base):
""" Generic Colleague """
    __tablename__ = 'colleagues'
    id = Column(Integer, primary_key=True)
    first_name = Column(String(50))
    last_name = Column(String(50))
    home_address = relationship('Home_Address', back_populates='colleague')
    work_address = relationship('Work_Address', back_populates='colleague')

class Address(Base):
    """ Generic Address """
    __tablename__ = 'addresses'
    id = Column(Integer, primary_key=True)
    adr_type = Column('type', String(50))
    country = Column(String(50))
    state = Column(String(50))
    county = Column(String(50))
    city = Column(String(50))
    district = Column(String(50))
    street = Column(String(50))
    house_number = Column(Integer)
    postal_code = Column(String(50))

    __mapper_args__ = {
        'polymorphic_on' : adr_type
    }

    class Home_Address(Address):
    __tablename__ = 'home_addresses'
    id = Column(Integer, ForeignKey('addresses.id'), primary_key=True)
    colleague_id = Column(Integer, ForeignKey('colleagues.id'))
    colleague = relationship('Colleague', back_populates='home_address')
    __mapper_args__ = {
        'polymorphic_identity' : 'home_addresses'
    }

class Work_Address(Address):
    __tablename__ = 'work_addresses'
    id = Column(Integer, ForeignKey('addresses.id'), primary_key=True)
    colleague_id = Column(Integer, ForeignKey('colleagues.id'))
    colleague = relationship('Colleague', back_populates='work_address')
    __mapper_args__ = {
        'polymorphic_identity' : 'work_addresses'
    }

I've then tried to add an address to a colleague,

adr1 = Home_Address(country='France', city='Paris')
adr2 = Work_Address(country='France', city='Lyon')
col = Colleague(first_name='Joe', last_name='Dalton')
col.home_address = adr1
col.work_address = adr2

but this happens:

TypeError: Incompatible collection type: Home_Address is not list-like

col.home_address gives an empty list, instead of giving me the object Home_Address, how can I set it up to work? Or maybe there is a simpler way of doing it ?

Upvotes: 2

Views: 188

Answers (1)

dizzyf
dizzyf

Reputation: 3593

Colleague -> Address(es) is a one-to-many relationship, right? That would expect a list, so maybe try something like this:

adr1 = Home_Address(country='France', city='Paris')
adr2 = Work_Address(country='France', city='Lyon')
col = Colleague(first_name='Joe', last_name='Dalton')
col.home_address = [adr1]
col.work_address = [adr2]

Upvotes: 1

Related Questions