python - Django MSSQL:覆盖外键约束名称生成
问题描述
我正在尝试使用 django-mssql-backend 在 Django 3.0 中创建数据库模型,就像 SQL Server 2019 的 db backend 一样。数据库对其中包含的表使用多个模式,其中一些表是非托管的(已经存在) ,以及其他通过迁移从头开始创建的。为了使多个模式正常工作,我使用了我在此处的另一个答案中找到的技巧,该技巧建议将表名格式化如下:
class MyModel(models.Model):
...
class Meta:
db_table = 'schema].[table'
这是为了使 SQL 编译为使在外部自动形成的包裹方括号完成模式/表定义。这样做的问题是 ForeignKey 对象的约束名称使用此表名称生成,这会导致出现无效的约束名称,从而导致在创建表后迁移失败,并且是时候创建约束了。\
它们生成如下:
[schema1].[table1_colname_id_bc165567_fk2_schema2].[table_colname]
有没有办法覆盖这种行为?如果可以通过分叉后端并添加手动编译代码来覆盖它,我将如何去做呢?否则,如何在使用多个模式并充分利用 Django 附带的 ORM/迁移时在我的模型中拥有外键?
解决方案
我最终弄清楚了。在 fork django-mssql-backend 之后,我覆盖了模块内类中的_create_index_name
and_fk_constraint_name
方法,将表名更改为,这省略了字符串的模式 hack 部分和额外的左括号,但不应该干扰那些没有的表是否也应用了该 hack。DatabaseSchemaEditor
sql_server.pyodbc.schema
split('[')[-1]
这是被覆盖的方法
(大部分代码与 django.db.backends.base.schema 中的原始实现相同):
def _create_index_name(self, table_name, column_names, suffix=""):
"""
Generate a unique name for an index/unique constraint.
The name is divided into 3 parts: the table name, the column names,
and a unique digest and suffix.
"""
# CHANGE HERE (table_name to table_name.split('[')[-1]
_, table_name = split_identifier(table_name.split('[')[-1])
hash_suffix_part = '%s%s' % (names_digest(table_name, *column_names, length=8), suffix)
max_length = self.connection.ops.max_name_length() or 200
# If everything fits into max_length, use that name.
index_name = '%s_%s_%s' % (table_name, '_'.join(column_names), hash_suffix_part)
if len(index_name) <= max_length:
return index_name
# Shorten a long suffix.
if len(hash_suffix_part) > max_length / 3:
hash_suffix_part = hash_suffix_part[:max_length // 3]
other_length = (max_length - len(hash_suffix_part)) // 2 - 1
index_name = '%s_%s_%s' % (
table_name[:other_length],
'_'.join(column_names)[:other_length],
hash_suffix_part,
)
# Prepend D if needed to prevent the name from starting with an
# underscore or a number (not permitted on Oracle).
if index_name[0] == "_" or index_name[0].isdigit():
index_name = "D%s" % index_name[:-1]
return index_name
def _fk_constraint_name(self, model, field, suffix):
def create_fk_name(*args, **kwargs):
return self.quote_name(self._create_index_name(*args, **kwargs))
return ForeignKeyName(
model._meta.db_table.split('[')[-1],
[field.column],
# CHANGE HERE (db_table to db_table.split('[')[-1]
split_identifier(field.target_field.model._meta.db_table.split('[')[-1])[1],
[field.target_field.column],
suffix,
create_fk_name,
)
如果 Django 的未来版本继续不支持多个模式,希望这对将来遇到类似问题的其他人有用。
编辑:我已经在 django-mssql-backend 存储库上创建了一个拉取请求,以合并这些更改,因此如果该问题的上下文获得批准,将来对人们来说可能不是问题。
推荐阅读
- javascript - 我正在研究 flightradar 24 示例并得到 json decode 错误
- abp - 实体框架:访问/调试 SQL
- java - 为什么我的模型类中的 int 返回 null
- firebase - Flutter Firestore 检索和设置嵌套对象
- python - 在无头 Chrome 中下载文件,(python)
- d3.js - 如何让热图中每个元素的文本与 y 轴标签相同?
- javascript - 在控制台中出现错误无法绑定到“ngif”,因为它不是角度应用程序中“div”的已知属性
- powershell - 难以将安全字符串传递给 system.data.odbc.odbcconnection 对象
- java - 使用电容器应用程序的 Angular8 无法从 android studio 登录
- swiftui - 无法使我的状态栏颜色与 SiwftUI 中的 UINavigationBar 颜色匹配