'在Django中设置数据库级别的字段默认值?'

2024-03-28 16:50:47 发布

您现在位置:Python中文网/ 问答频道 /正文

在django中使用什么方法来设置数据库级别的默认值?在

我知道这是一个django wontfix决定,但是如果是为特定的后端(postgres)编写的,那么肯定有一些想法在流传。在

我希望通过python来实现这一点,所以我不希望使用/sql/.sql文件。在

我想我可以在现有字段上设置属性,例如:

foo = models.CharField(max_length=256, blank=True)
foo.db_default = ''

然后过滤字段并构造一个ALTER COLUMN SET DEFAULT查询。在

我该把它放哪儿呢?文档说不要修改post_syncdb钩子中的数据库。在

有什么好主意吗?我还想动态生成sql/.sql文件。一个管理指挥总比什么都没有强。自动档就好了。嗯,我想知道post_syncdb是否能及时生成sql文件来执行?将进行测试。在


Tags: 文件django方法数据库sql属性foomodels
1条回答
网友
1楼 · 发布于 2024-03-28 16:50:47

我试着做post_syncdb,但是用.sql文件太晚了。与class_prepared的联系及时解决了syncdb但我意识到我过于复杂了。我还不如在模型实例化时运行一些代码。在

我最终得到了一个直截了当的解决方案,在6个月内,当我需要再次处理这个问题时,可以很容易地理解它。在

def db_default(field, db_default):
    """
    Set attribute __db_default for signal to set default values with.
    """
    field.__db_default = db_default
    return field

class OrderShipLog(models.Model):
    date = db_default(models.DateTimeField(), 'now()')

def generate_default_sql(model):
    """
    Generate SQL for postgresql_psycopg2 default values because the shipping
    database post doesn't guarantee posting certain fields but id like them all to be strings.
    """
    db_fields = filter(lambda x: hasattr(x, '__db_default'), model._meta.fields)
    db_table = model._meta.db_table
    modelname = model._meta.object_name

    sql_dir = os.path.join(os.path.dirname(__file__), 'sql/')
    sql_filename = os.path.join(sql_dir, '{modelname}.postgresql_psycopg2.sql'.format(
        modelname=modelname.lower()))

    from django.db import connection
    if not db_table in connection.introspection.table_names():
        print >> sys.stderr, "{0} not in introspected table list; creating db default sql file : {1}".format(db_table, sql_filename)
        try:
            if not os.path.exists(sql_dir):
                os.makedirs(sql_dir)

            with open(sql_filename, 'w+') as f:
                for field in db_fields:
                    attname, db_column = field.get_attname_column()
                    if not field.__db_default:
                        raise Exception("Must enter value for field {modelname}.{field}.__db_default".format(
                            modelname=modelname,
                            field=attname))
                    f.write('ALTER TABLE {db_table} ALTER COLUMN {db_column} SET DEFAULT {default};\n'.format(
                        db_table=db_table,
                        db_column=db_column,
                        default=field.__db_default,
                        ))
        except Exception, e:
            print >> sys.stderr, "Could not generate SQL file. Manually set the defaults! {0}\n".format(e) * 10

generate_default_sql(OrderShipLog)

相关问题 更多 >