MySQL 如何在SQLAlchemy中实现MySQL数据库的外键引用同一表格的功能

在本文中，我们将介绍如何在SQLAlchemy中实现MySQL数据库的外键引用同一表格的功能。

在MySQL中，我们可以使用外键来引用另一个表格的某一列。而有时，我们需要在同一张表格中的两个不同列之间建立外键关系。这时候，我们就需要使用外键引用同一表格的功能。

阅读更多：MySQL 教程

创建示例表格

为了说明如何在SQLAlchemy中使用外键引用同一表格的功能，我们首先需要创建一个示例表格。我们可以使用以下代码在MySQL数据库中创建一个简单的表格：

CREATE TABLE `person` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(50) NOT NULL,
  `spouse_id` int(11) DEFAULT NULL,
  PRIMARY KEY (`id`),
  CONSTRAINT `spouse_fk` FOREIGN KEY (`spouse_id`) REFERENCES `person` (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;

在这个示例表格中，我们定义了三个列：

id：主键列，用来保存人员的唯一标识符。
name：用来保存人员的姓名。
spouse_id：用来保存人员的配偶的唯一标识符。

同时，我们还定义了一个名为 spouse_fk 的外键，用来将 spouse_id 列和 id 列建立起外键关系。

在SQLAlchemy中引用同一表格的外键

在SQLAlchemy中，我们可以使用 ForeignKey 类来定义外键。而要引用同一表格的外键，我们只需要在 ForeignKey 的构造函数中传入对同一张表格的引用即可。

下面是一个示例代码：

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

Base = declarative_base()

class Person(Base):
    __tablename__ = 'person'

    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)
    spouse_id = Column(Integer, ForeignKey('person.id'))

    spouse = relationship('Person', remote_side=[id])

在这个示例代码中，我们首先定义了一个 SQLAlchemy 的基础类 Base。接着，我们定义了一个名为 Person 的类，用来映射 MySQL 中的 person 表格。在 Person 类中，我们定义了三个列：

id：用来映射 MySQL 中的 id 列。
name：用来映射 MySQL 中的 name 列。
spouse_id：用来映射 MySQL 中的 spouse_id 列，并在构造函数中使用 ForeignKey 类引用同一张表格的 id 列。

此外，我们还使用 relationship 函数定义了一个名为 spouse 的反向关系，用来获取此人员的配偶。在 relationship 函数中，我们使用 remote_side 参数指定了对应的远程列，使得 SQLAlchemy 能够正确地获取配偶的信息。

使用示例代码查询数据

有了示例表格和映射类之后，我们就可以使用 SQLAlchemy 进行数据的查询了。下面是一个查询示例代码：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

engine = create_engine('mysql+pymysql://USERNAME:PASSWORD@HOSTNAME/DATABASE_NAME')
Session = sessionmaker(bind=engine)
session = Session()

person1 = Person(name='John')
person2 = Person(name='Mary', spouse=person1)
person1.spouse = person2

session.add_all([person1, person2])
session.commit()

people = session.query(Person).all()

for person in people:
    print(person.name, person.spouse.name if person.spouse else None)

在这个示例代码中，我们首先创建了一个 MySQL 数据库引擎和一个会话对象。然后，我们创建了两个人员对象 person1 和 person2，并将 person2 设置为 person1 的配偶。最后，我们将这两个人员对象添加到会话中，并进行提交。接着，我们使用 query 方法查询了 Person 表格中的所有数据，并将查到的结果进行输出。

在输出结果中，我们可以看到每个人员的姓名以及他们的配偶的姓名。需要注意的是，在 spouse 列中，我们使用了条件表达式 if person.spouse else None 来判断人员是否有配偶。这是因为，在模型定义中，我们将 spouse_id 列设置为可选的，使得每个人员都可以没有配偶。