diff options
| author | Dakota Hawkins <dakotahawkins@users.noreply.github.com> | 2018-12-19 06:41:31 -0500 |
|---|---|---|
| committer | Carlton Gibson <carlton.gibson@noumenal.es> | 2018-12-19 12:41:31 +0100 |
| commit | 8d3147e130c4e5638fceb1e6125c040362ce12e8 (patch) | |
| tree | 3e17b6ed50be3466ced3ec73a2745f37f994cebb /django/db | |
| parent | b514dc14f4e1c364341f5931b354e83ef15ee12d (diff) | |
Fixed #30031 -- Added --no-header option to makemigrations/squashmigrations.
Diffstat (limited to 'django/db')
| -rw-r--r-- | django/db/migrations/writer.py | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/django/db/migrations/writer.py b/django/db/migrations/writer.py index d056e00646..6a62b4cbf4 100644 --- a/django/db/migrations/writer.py +++ b/django/db/migrations/writer.py @@ -132,8 +132,9 @@ class MigrationWriter: of the migration file from it. """ - def __init__(self, migration): + def __init__(self, migration, include_header=True): self.migration = migration + self.include_header = include_header self.needs_manual_porting = False def as_string(self): @@ -195,10 +196,13 @@ class MigrationWriter: if self.migration.replaces: items['replaces_str'] = "\n replaces = %s\n" % self.serialize(self.migration.replaces)[0] # Hinting that goes into comment - items.update( - version=get_version(), - timestamp=now().strftime("%Y-%m-%d %H:%M"), - ) + if self.include_header: + items['migration_header'] = MIGRATION_HEADER_TEMPLATE % { + 'version': get_version(), + 'timestamp': now().strftime("%Y-%m-%d %H:%M"), + } + else: + items['migration_header'] = "" if self.migration.initial: items['initial_str'] = "\n initial = True\n" @@ -279,10 +283,14 @@ class MigrationWriter: return serializer_factory(value).serialize() -MIGRATION_TEMPLATE = """\ +MIGRATION_HEADER_TEMPLATE = """\ # Generated by Django %(version)s on %(timestamp)s -%(imports)s +""" + + +MIGRATION_TEMPLATE = """\ +%(migration_header)s%(imports)s class Migration(migrations.Migration): %(replaces_str)s%(initial_str)s |