Az/pascal voc loader (#613)
parent
85ae933af5
commit
f17847ff33
@ -0,0 +1,74 @@
|
|||||||
|
# Generated by Django 2.1.9 on 2019-08-05 06:27
|
||||||
|
|
||||||
|
import cvat.apps.engine.models
|
||||||
|
from django.db import migrations, models
|
||||||
|
import django.db.models.deletion
|
||||||
|
|
||||||
|
def split_handlers(apps, schema_editor):
|
||||||
|
db_alias = schema_editor.connection.alias
|
||||||
|
handler_model = apps.get_model('annotation', 'AnnotationHandler')
|
||||||
|
dumper_model = apps.get_model('annotation', "AnnotationDumper")
|
||||||
|
loader_model = apps.get_model('annotation', 'AnnotationLoader')
|
||||||
|
|
||||||
|
|
||||||
|
for db_handler in handler_model.objects.all():
|
||||||
|
if db_handler.type == "dumper":
|
||||||
|
new_handler = dumper_model()
|
||||||
|
else:
|
||||||
|
new_handler = loader_model()
|
||||||
|
|
||||||
|
new_handler.display_name = db_handler.display_name
|
||||||
|
new_handler.format = db_handler.format
|
||||||
|
new_handler.version = db_handler.version
|
||||||
|
new_handler.handler = db_handler.handler
|
||||||
|
new_handler.annotation_format = db_handler.annotation_format
|
||||||
|
|
||||||
|
new_handler.save()
|
||||||
|
db_handler.delete()
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('annotation', '0001_initial'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='AnnotationDumper',
|
||||||
|
fields=[
|
||||||
|
('display_name', cvat.apps.engine.models.SafeCharField(max_length=256, primary_key=True, serialize=False)),
|
||||||
|
('format', models.CharField(max_length=16)),
|
||||||
|
('version', models.CharField(max_length=16)),
|
||||||
|
('handler', models.CharField(max_length=256)),
|
||||||
|
('annotation_format', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='annotation.AnnotationFormat')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'abstract': False,
|
||||||
|
'default_permissions': (),
|
||||||
|
},
|
||||||
|
),
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='AnnotationLoader',
|
||||||
|
fields=[
|
||||||
|
('display_name', cvat.apps.engine.models.SafeCharField(max_length=256, primary_key=True, serialize=False)),
|
||||||
|
('format', models.CharField(max_length=16)),
|
||||||
|
('version', models.CharField(max_length=16)),
|
||||||
|
('handler', models.CharField(max_length=256)),
|
||||||
|
('annotation_format', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='annotation.AnnotationFormat')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'abstract': False,
|
||||||
|
'default_permissions': (),
|
||||||
|
},
|
||||||
|
),
|
||||||
|
migrations.RunPython(
|
||||||
|
code=split_handlers,
|
||||||
|
),
|
||||||
|
migrations.RemoveField(
|
||||||
|
model_name='annotationhandler',
|
||||||
|
name='annotation_format',
|
||||||
|
),
|
||||||
|
migrations.DeleteModel(
|
||||||
|
name='AnnotationHandler',
|
||||||
|
),
|
||||||
|
]
|
||||||
Loading…
Reference in New Issue