David Wobrock
484a775fda
* Use the Django migration classes to manipulate these objects this includes the discovery of migrations through the MigrationLoader (instead of manually walking through the Django project and find the migrations through a regex) and representing a Migration through the Django class. * Get the SQL of a migration through a 'call_command' call of 'sqlmigrate' That speeds up greatly the linter since we don't have to setup Django for each migration * Add the database to the cache path
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
# Copyright 2019 3YOURMIND GmbH
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
from importlib import import_module
|
|
|
|
|
|
def split_path(path):
|
|
decomposed_path = []
|
|
while 1:
|
|
head, tail = os.path.split(path)
|
|
if head == path: # sentinel for absolute paths
|
|
decomposed_path.insert(0, head)
|
|
break
|
|
elif tail == path: # sentinel for relative paths
|
|
decomposed_path.insert(0, tail)
|
|
break
|
|
else:
|
|
path = head
|
|
decomposed_path.insert(0, tail)
|
|
|
|
if not decomposed_path[-1]:
|
|
decomposed_path = decomposed_path[:-1]
|
|
return decomposed_path
|
|
|
|
|
|
def split_migration_path(migration_path):
|
|
from django.db.migrations.loader import MIGRATIONS_MODULE_NAME
|
|
|
|
decomposed_path = split_path(migration_path)
|
|
for i, p in enumerate(decomposed_path):
|
|
if p == MIGRATIONS_MODULE_NAME:
|
|
return decomposed_path[i - 1], os.path.splitext(decomposed_path[i + 1])[0]
|
|
|
|
|
|
def clean_bytes_to_str(byte_input):
|
|
return byte_input.decode("utf-8").strip()
|
|
|
|
|
|
def get_migration_abspath(app_label, migration_name):
|
|
from django.db.migrations.loader import MigrationLoader
|
|
|
|
module_name, _ = MigrationLoader.migrations_module(app_label)
|
|
migration_path = "{}.{}".format(module_name, migration_name)
|
|
migration_module = import_module(migration_path)
|
|
|
|
migration_file = migration_module.__file__
|
|
if migration_file.endswith(".pyc"):
|
|
migration_file = migration_file[:-1]
|
|
return migration_file
|