Flask-SQLAlchemy 多对多关系：如何插入数据

Question

我在 Flask-Sqlalchemy 模型中创建了多对多关系。但是，我似乎不太了解如何填充我的数据库。你能告诉我我应该如何编写代码来正确填充我的数据库吗？

这是我的模型：

company_contact = db.Table('company_contact', db.Model.metadata,
    db.Column('company_id', Integer, ForeignKey('company.id')),
    db.Column('contact_id', Integer, ForeignKey('contact.id'))
)

class Company(db.Model):
    __tablename__ = 'company'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    contacts = db.relationship('Contact', secondary=company_contact, back_populates="companies")
    company_name = db.Column(db.String(100))

    def __repr__(self):
        return '<Company {}{}>'.format(self.contacts, self.company_name)


class Contact(db.Model):
    __tablename__ = 'contact'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    last_name = db.Column(db.String(255))
    first_name = db.Column(db.String(255))
    job_position = db.Column(db.String(255))
    companies = db.relationship('Company', secondary=company_contact, back_populates="contacts")
    contact_status = db.Column(db.Enum('Client', 'Prospect', 'Churner', 'Prestataire', 'Reperage',
                                       name='contact_status'))
    start_date = db.Column(db.DateTime, server_default=db.func.now())
    stop_date = db.Column(db.DateTime, default=None)

    def __repr__(self):
        return '<Contact {}>'.format(self.last_name, self.first_name, self.job_position, self.companys,
                                     self.contact_status, self.start_date, self.stop_date)

我想要做的是添加两个具有相同 company_name 的联系人。添加第一个不是问题，一切正常。但是，对于第二个联系人，当我添加此代码时会出现问题。

com = Company(company_name='bla')
con = Contact(last_name='Blabla', first_name='Jeff', job_position='Resp Marketing',contact_status='Client')
con.companies.append(com)
db.session.add(con)
db.session.commit()

sqlalchemy.exc.IntegrityError: (psycopg2.errors.UniqueViolation) 重复键值违反唯一约束“company_company_name_key”详细信息：键 (company_name)=(bla) 已经存在。

我不知道如何写它来允许，两个与同一家公司的联系人以及允许两个具有相同联系人的公司。

非常感谢你的帮助。

Answer 1

所以：你的数据库是空的。你初始化数据库，用 sqlalchemy 创建表...（不记得命令了）

然后你这样做：

com = Company(company_name='bla')
con = Contact(last_name='Blabla', first_name='Jeff', job_position='Resp Marketing',contact_status='Client')
con.companies.append(com)
db.session.add(con)
db.session.commit()

然后你这样做：

#can't remember how to get just one row, is it first instaid of all ? sorry
com_already_created = Company().query.all()[0] 

con = Contact(last_name='Blabla2', first_name='Jeff2', job_position='Resp Marketing',contact_status='Client')

con.companies.append(com_already_created)
#or
com_already_created.contacts.append(con)

db.session.add(con)
db.session.commit()

您的问题是您正在创建一个元素以添加到已经存在的表 Company 中（我认为您从“Company”模型实现中的“company_name”中删除了 PK 或 unique）

另外：检查您的 __repr__，使用 python3.7 f"string" 示例：

    def __repr__(self):
        return '<Contact {}>'.format(self.last_name, self.first_name, 
    self.job_position, self.companys, self.contact_status, self.start_date, 
    self.stop_date)

它应该是：

def __repr__(self):
    return f'<Contact PK:{self.id}, lst_nm:{self.last_name}, 
    frst_nm:{self.first_name}, jb_pos:{self.job_position>}, 
    cnt_comp:{len(self.companys)}'

#(add more if you want but its getting messy)