Flask SQLAlchemy：如何在向会话中添加新对象之前找到主键值？

Question

我正在使用b64encode 从表中每一行的主键创建一个唯一的引用字符串。这是我在 Flask SQLAlchemy 中的模型：

class Groups(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    ref_id = db.Column(db.String(20), index = True)

    def __init__(self):
        self.ref_id = id_encode(self.id)

但是，传递给id_encode() 函数的self.id 具有“无”值，而不是主键的值。据我所知，只有在提交会话后才能获得新的主键值，或者通过using the flush() method。

这是id_encode()的定义：

def id_encode(num):
        bts = base64.b64encode(bytes(str(num), 'utf-8'))
        print(num)
        return bts.decode()

如果有什么方法可以让它工作，或者有什么更好（但简单）的方法来生成唯一参考 ID？我只是不希望普通用户能够通过输入数字 ID 导航到其他组。

Answer 1

您可以做的是用类函数替换 « __init __ »。

所以而不是：

class Groups(db.Model):
id = db.Column(db.Integer, primary_key = True)
ref_id = db.Column(db.String(20), index = True)

def __init__(self):
    self.ref_id = id_encode(self.id)

你应该试试：

class Groups(db.Model):
id = db.Column(db.Integer, primary_key = True)
ref_id = db.Column(db.String(20), index = True)

def generate_id(self):
    self.ref_id = base64.b64encode(bytes(str(self.id), 'utf-8'))

那么当你创建一个新的组对象时，只要记住在提交之前使用该函数即可。

group = Groups(id=xxx, arg2=xxx, arg3=xxx etc) 
# omit "id" if you set it to autoincrement
group.generate_id()
db.session.commit()