Question

如果我有一个如下所示的SQLAlchemy声明模型：

class Test(Model):
    __tablename__ = 'tests'

    id = Column(Integer, Sequence('test_id_seq'), primary_key=True)
    ...
    Atest_id = Column(Integer, ForeignKey('Atests.id'), nullable=True)
    Btest_id = Column(Integer, ForeignKey('Btests.id'), nullable=True)
    Ctest_id = Column(Integer, ForeignKey('Ctests.id'), nullable=True)
    Dtest_id = Column(Integer, ForeignKey('Dtests.id'), nullable=True)
    Etest_id = Column(Integer, ForeignKey('Etests.id'), nullable=True)
    ...
    date = Column(DateTime)
    status = Column(String(20))  # pass, fail, needs_review

我想确保给定行中只有一个*test_id个外键，我如何在SQLAlchemy中完成？

我发现有SQLAlchemy CheckConstraint个对象(see docs)，但MySQL不支持检查约束。

数据模型具有SQLAlchemy之外的交互，因此最好是数据库级别的检查（MySQL）

Answer 1

好吧，考虑到你的必需品＆＃34;数据模型在SQLAlchemy之外有交互，所以最好是数据库级别的检查（MySQL）＆＃34; 和＆＃39;确保只有一个[..]不为空＆＃39; 。我认为最好的方法是编写一个这样的触发器：

DELIMITER $$
CREATE TRIGGER check_null_insert BEFORE INSERT 
ON my_table
FOR EACH ROW BEGIN
IF CHAR_LENGTH(CONCAT_WS('', NEW.a-NEW.a, NEW.b-NEW.b, NEW.c-NEW.c)) = 1 THEN 
    UPDATE `Error: Only one value of *test_id must be not null` SET z=0;
END IF;
END$$
DELIMITER ;

一些技巧和注意事项：

IF STATEMENT ：为了避免繁琐的检查写入，每列都不为空，而其他列为空，我做了这个诀窍：将每列减少到一个字符并检查多少字符存在。请注意，如果NEW.a-NEW.a为NEW.a，则Integer始终返回1个字符，NULL返回0个字符，并且操作NULL-NULL在MySQL上返回NULL。
错误触发：我想你想引发一个错误，那么如何在MySQL上做这个呢？你没有提到MySQL版本。仅on MySQL 5.5 you can use the SIGNAL syntax to throw an exception。因此，更便携的方式是发出一个无效的语句，如：UPDATE xx SET z=0。如果您使用的是MySQL 5.5，则可以使用：signal sqlstate '45000' set message_text = 'Error: Only one value of *test_id must be not null';代替UPDATE `Error: Only one value of *test_id must be not null` SET z=0;

另外，我想你也想在更新时检查一下，所以使用：

DELIMITER $$
CREATE TRIGGER check_null_update BEFORE UPDATE 
ON my_table
FOR EACH ROW BEGIN
IF CHAR_LENGTH(CONCAT_WS('', NEW.a-NEW.a, NEW.b-NEW.b, NEW.c-NEW.c)) = 1 THEN 
    UPDATE `Error: Only one value of *test_id must be not null` SET z=0;
END IF;
END$$
DELIMITER ;

或者创建一个存储过程并调用它。

更新

对于支持检查约束的数据库，代码更简单，请参阅此SQL Server示例：

CREATE TABLE MyTable (col1 INT NULL, col2 INT NULL, col3 INT NULL);
GO

ALTER TABLE MyTable
ADD CONSTRAINT CheckOnlyOneColumnIsNull
CHECK (
  LEN(CONCAT(col1-col1, col2-col2, col3-col3)) = 1
)
GO

检查SQLAlchemy

1 个答案:

更新