sqlalchemy防sql注入

银行对安全性要求高，其中包括基本的mysql防注入，因此，记录下相关使用方法：

注意：sqlalchemy自带sql防注入，但是在 execute执行手写sql时需要考虑此安全问题

对于 where in 的防sql注入：（in 的内容一定要是tuple类型，否则查询结果不对）

in_str = tuple(input_list)
sql = "(SELECT count(id) FROM {0} WHERE {0}.id IN :in_str);".format(cls.__tablename__)
cursor = db.get_engine(current_app, cls.__bind_key__)
return cursor.execute(text(sql), in_str=in_str).fetchone()[0]

对于 where 一般的防sql注入：

sql = """
                    (select {index}.sec_id,
                    from    {index},
                            {main}
                    where   {index}.sec_id= {main}.sec_id
                    and     {main}.user_id=:user_id);
                """.format(index=TableA.__tablename__,
                           main=TableB.__tablename__)
        cursor = db.get_engine(current_app, TableB.__bind_key__)
        return cursor.execute(text(sql), user_id=user_id).fetchall()

防sql注入 只能对 where里面等于号后面的进行防注入，其他部分的字符串仍然需要拼接

其余关键字中的使用方法参考如下官网教程

官网教程：https://docs.sqlalchemy.org/en/latest/core/tutorial.html#using-textual-sql

扫描二维码关注公众号，回复： 6819708 查看本文章

猜你喜欢