Как Django Dumpdata вложенных моделей в YAML
Так что у меня есть это удивительное приложение Django, которое приносит мне удовлетворение и тому подобное. Но теперь проблема, я хочу использовать dumpdata (или что-то, что делает то же самое) для экспорта модели с другой вложенной моделью в формате yaml.
Допустим, у меня есть две модели, Project а также Questions, каждый Project может иметь свой собственный набор Questions,
Код выглядит примерно так:
Модель проекта:
class Projects(SortableMixin):
"""
Some docstring
"""
slug = models.SlugField(
_('slug'),
help_text=_('Short name to address this projects from templates.'))
# Basic fields
object_id = models.PositiveIntegerField(blank=True, null=True)
name = models.TextField(_('name'), help_text=_('The question.'))
# Some other fields...
question = models.ForeignKey(Question, null=True, blank=True)
Модель вопросов:
class Question(SortableMixin):
"""
Some docstring
"""
slug = models.SlugField(
_('slug'),
help_text=_('Short name to address this question from templates.'))
# Basic fields
object_id = models.PositiveIntegerField(blank=True, null=True)
name = models.TextField(_('name'), help_text=_('The question.'))
input = models.TextField()
Project Модель имеет свое собственное приложение и Questions, Структура выглядит так:
- Django
- Apps
- Project
- Questions
Всякий раз, когда я хочу экспортировать свою базу данных, я буду делать следующее:
./manage.py dumpdata --indent 4 --format yaml > dbdump.yaml
Хотя это работает, и я могу позже импортировать его с LoadData, это не то, что я хочу, вывод файла yaml выглядит дерьмово. Я хочу иметь хороший вложенный файл модели yaml для обзора, ниже дерьмового файла:
Раздел проектов:
- model: project.projects
pk: 1
fields: {slug: "slugproject1", object_id: 10, name: "some project 1", question: ["slugquestion1"]}
- model: project.projects
pk: 2
fields: {slug: "slugproject2", object_id: 11, name: "some project 2", question: ["slugquestion2"]}
- model: project.projects
pk: 3
fields: {slug: "slugproject3", object_id: 12, name: "some project 3", question: ["slugquestion3"]}
Раздел вопросов:
- model: question.question
pk: 1
fields: {slug: "slugquestion1", object_id: 100, name: "some question 1", input: "q1"}
- model: question.question
pk: 1
fields: {slug: "slugquestion2", object_id: 200, name: "some question 2", input: "q2"}
- model: question.question
pk: 1
fields: {slug: "slugquestion3", object_id: 300, name: "some question 3", input: "q3"}
Что я действительно хочу, так это экспортировать файл yaml так:
- model: project.projects
pk: 1
fields: {
slug: "slugproject1",
object_id: 10,
name: "some project 1",
questions: {
model: question.question
pk: 1
fields: {
slug: "slugquestion1"
object_id: 100
name: "some question 1"
input: "q1"
}
}
}
- model: project.projects
pk: 2
fields: {
slug: "slugproject2",
object_id: 11,
name: "some project 2",
questions: {
model: question.question
pk: 2
fields: {
slug: "slugquestion2"
object_id: 200
name: "some question 2"
input: "q2"
}
}
}
- model: project.projects
pk: 3
fields: {
slug: "slugproject3",
object_id: 13,
name: "some project 3",
questions: {
model: question.question
pk: 3
fields: {
slug: "slugquestion3"
object_id: 300
name: "some question 3"
input: "q3"
}
}
}
Для этого я реализовал собственный сериализатор внутри проектов:
- Django
- Apps
- Project
- Management
- Commands
- test.py
- Questions
Код выглядит так:
from django.core.management.base import BaseCommand, CommandError
from apps.project import Projects
from apps.questions import Question
from rest_framework import serializers
import yaml
class QuestionSerialier(serializers.ModelSerializer):
class Meta:
model = Question
fields = ('pk', 'slug', 'object_id', 'name', 'input')
class ProjectsSerializer(serializers.ModelSerializer):
questions = QuestionSerialier(many=True, read_only=True)
class Meta:
model = Projects
fields = ('pk', 'slug','object_id', 'name', 'questions')
class Command(BaseCommand):
help = ''
def add_arguments(self, parser):
pass
def handle(self, *args, **options):
with open('result.yaml', 'w') as yaml_file:
for i in Projects.objects.filter():
yaml.dump(ProjectsSerializer(i).data,
yaml_file,
default_flow_style=False,
allow_unicode=False,
encoding=None)
Я могу запустить код, запустив:
./manage.py test
Только это экспортирует мои модели так:
- project: 1
pk: 1
slug: "slugproject1"
object_id: 10
name: "some project 1"
questions:
- !!python/object/apply:collections.OrderedDict
- - - pk
- - slug
- - object_id
- - name
- - input
- project: 2
pk: 2
slug: "slugproject2"
object_id: 11
name: "some project 2"
questions:
- !!python/object/apply:collections.OrderedDict
- - - pk
- - slug
- - object_id
- - name
- - input
- project: 3
pk: 3
slug: "slugproject3"
object_id: 12
name: "some project 3"
questions:
- !!python/object/apply:collections.OrderedDict
- - - pk
- - slug
- - object_id
- - name
- - input
Как вы можете видеть, вышеприведенное не пригодно для импорта или даже удобочитаемого экспорта...
Не могли бы вы, ребята, указать мне правильное направление в том, как добиться экспорта вложенных моделей dumpdata yaml в django?
Спасибо!
1 ответ
Поэтому я разобрался, как справиться с экспортом / импортом.
Для экспорта в правильный формат файла yaml я сделал следующее:
...
class Command(BaseCommand):
def handle(self, *args, **options):
with open('result.yaml', 'w') as yaml_file:
model = serializer.Meta.model
json_object = []
json_data = json.dumps(serializer(model_data).data)
json_object.append(json.loads(json_data))
yaml.dump(
json_object,
yaml_file,
default_flow_style=False,
allow_unicode=False,
encoding=None
)
Это экспорт в правильный файл yaml.
Затем, чтобы импортировать файл yaml, я сделал следующее:
...
class Command(BaseCommand):
def handle(self, *args, **options):
with open('result.yaml', 'r') as yaml_file:
yaml_list = yaml.load(yaml_file.read())
for data in yaml_list:
...process file
И это все!