Browse Source

add migrate tasks

devel
Dmitry Novikov 7 years ago
parent
commit
32170d5841
  1. 21
      migrate2v2_dump.py

21
migrate2v2_dump.py

@ -5,13 +5,14 @@ from json import dump
from bitfield import BitField
from django import setup
from django.core.serializers.json import DjangoJSONEncoder
from django.db.models import ImageField, ManyToManyField
from django.db.models import ImageField, ManyToManyField, ManyToOneRel
from djing.fields import MACAddressField
class BatchSaveStreamList(list):
def __init__(self, model_class, model_name, except_fields=None, choice_list_map=None, field_name_map=None, *args, **kwargs):
def __init__(self, model_class, model_name, except_fields=None, choice_list_map=None, field_name_map=None, *args,
**kwargs):
super().__init__(*args, **kwargs)
self._model_class = model_class
self._model_name = model_name
@ -63,6 +64,8 @@ class BatchSaveStreamList(list):
# related fields
if field.is_relation:
if isinstance(field, ManyToOneRel):
return getattr(obj, field.field_name)
val = getattr(obj, field.attname)
if isinstance(field, ManyToManyField):
s = val.only('pk').values_list('pk', flat=True)
@ -246,13 +249,13 @@ def dump_tasks():
'F': 2
},
'mode': {
'na', 0, 'ic', 1,
'yt', 2, 'rc', 3,
'ls', 4, 'cf', 5,
'cn', 6, 'pf', 7,
'cr', 8, 'co', 9,
'fc', 10, 'ni', 11,
'ot', 12
'na': 0, 'ic': 1,
'yt': 2, 'rc': 3,
'ls': 4, 'cf': 5,
'cn': 6, 'pf': 7,
'cr': 8, 'co': 9,
'fc': 10, 'ni': 11,
'ot': 12
}
})
batch_save('task_comments.json', ExtraComment, 'tasks.extracomment')

Loading…
Cancel
Save