Merge pull request doccano#422 from CatalystCode/enhancement/mysql

Enhancement/Add support for MySQL

Author: Hironsan

.travis.yml

@@ -7,6 +7,7 @@ env:
   - DATABASE=sqlite
   - DATABASE=postgres
   - DATABASE=mssql
+  - DATABASE=mysql
 
 services:
   - docker
@@ -22,6 +23,10 @@ before_install:
       docker run --rm --name=mssql --network=doccano -d -e ACCEPT_EULA=y -e SA_PASSWORD=sUp3rS3cr3t mcr.microsoft.com/mssql/server:2017-latest
       docker exec -it mssql sh -c "while ! /opt/mssql-tools/bin/sqlcmd -U SA -P sUp3rS3cr3t -Q 'CREATE DATABASE db;'; do sleep 3; done"
       export DATABASE_URL="mssql://SA:sUp3rS3cr3t@mssql:1433/db?sslmode=disable"
+
+    elif [[ "${DATABASE}" = "mysql" ]]; then
+      docker run --rm --name=mysql --network=doccano -d -e MYSQL_DATABASE=db -e MYSQL_ROOT_PASSWORD=mysqlr00t mysql:5.7
+      export DATABASE_URL="mysql://root:mysqlr00t@mysql:3306/db"
     fi
 
 install:

app/api/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 2.1.7 on 2019-06-21 06:05
+# Generated by Django 2.1.7 on 2019-10-25 16:26
 
 from django.conf import settings
 from django.db import migrations, models
@@ -10,8 +10,8 @@ class Migration(migrations.Migration):
     initial = True
 
     dependencies = [
-        ('contenttypes', '0002_remove_content_type_name'),
         migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('contenttypes', '0002_remove_content_type_name'),
     ]
 
     operations = [
@@ -23,6 +23,7 @@ class Migration(migrations.Migration):
                 ('meta', models.TextField(default='{}')),
                 ('created_at', models.DateTimeField(auto_now_add=True)),
                 ('updated_at', models.DateTimeField(auto_now=True)),
+                ('annotations_approved_by', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL)),
             ],
         ),
         migrations.CreateModel(
@@ -60,12 +61,31 @@ class Migration(migrations.Migration):
                 ('updated_at', models.DateTimeField(auto_now=True)),
                 ('project_type', models.CharField(choices=[('DocumentClassification', 'document classification'), ('SequenceLabeling', 'sequence labeling'), ('Seq2seq', 'sequence to sequence')], max_length=30)),
                 ('randomize_document_order', models.BooleanField(default=False)),
+                ('collaborative_annotation', models.BooleanField(default=False)),
             ],
             options={
                 'abstract': False,
                 'base_manager_name': 'objects',
             },
         ),
+        migrations.CreateModel(
+            name='Role',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=100, unique=True)),
+                ('description', models.TextField(default='')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='RoleMapping',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+            ],
+        ),
         migrations.CreateModel(
             name='Seq2seqAnnotation',
             fields=[
@@ -74,7 +94,7 @@ class Migration(migrations.Migration):
                 ('manual', models.BooleanField(default=False)),
                 ('created_at', models.DateTimeField(auto_now_add=True)),
                 ('updated_at', models.DateTimeField(auto_now=True)),
-                ('text', models.TextField()),
+                ('text', models.CharField(max_length=500)),
                 ('document', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='seq2seq_annotations', to='api.Document')),
                 ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
             ],
@@ -127,6 +147,21 @@ class Migration(migrations.Migration):
             },
             bases=('api.project',),
         ),
+        migrations.AddField(
+            model_name='rolemapping',
+            name='project',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='role_mappings', to='api.Project'),
+        ),
+        migrations.AddField(
+            model_name='rolemapping',
+            name='role',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.Role'),
+        ),
+        migrations.AddField(
+            model_name='rolemapping',
+            name='user',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='role_mappings', to=settings.AUTH_USER_MODEL),
+        ),
         migrations.AddField(
             model_name='project',
             name='polymorphic_ctype',
@@ -161,9 +196,17 @@ class Migration(migrations.Migration):
             name='sequenceannotation',
             unique_together={('document', 'user', 'label', 'start_offset', 'end_offset')},
         ),
+        migrations.AlterUniqueTogether(
+            name='seq2seqannotation',
+            unique_together={('document', 'user', 'text')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='rolemapping',
+            unique_together={('user', 'project', 'role')},
+        ),
         migrations.AlterUniqueTogether(
             name='label',
-            unique_together={('project', 'text'), ('project', 'prefix_key', 'suffix_key')},
+            unique_together={('project', 'text')},
         ),
         migrations.AlterUniqueTogether(
             name='documentannotation',

app/api/migrations/0002_approve_document_labels.py

@@ -263,9 +263,16 @@
 if DATABASES['default'].get('ENGINE') == 'django.db.backends.sqlite3':
     DATABASES['default'].get('OPTIONS', {}).pop('sslmode', None)
 
+# work-around for dj-database-url: patch ssl for mysql
+if DATABASES['default'].get('ENGINE') == 'django.db.backends.mysql':
+    DATABASES['default'].get('OPTIONS', {}).pop('sslmode', None)
+    if env('MYSQL_SSL_CA', None):
+        DATABASES['default'].setdefault('OPTIONS', {})\
+            .setdefault('ssl', {}).setdefault('ca', env('MYSQL_SSL_CA', None))
+
 # default to a sensible modern driver for Azure SQL
 if DATABASES['default'].get('ENGINE') == 'sql_server.pyodbc':
-    db_options = DATABASES['default'].setdefault('OPTIONS', {})\
+    DATABASES['default'].setdefault('OPTIONS', {})\
         .setdefault('driver', 'ODBC Driver 17 for SQL Server')
 
 # Honor the 'X-Forwarded-Proto' header for request.is_secure()

app/api/migrations/0003_support_sql_server.py

@@ -24,6 +24,7 @@ furl==2.0.0
 gunicorn==19.9.0
 lockfile==0.12.2
 model-mommy==1.6.0
+mysqlclient==1.4.2.post1
 psycopg2-binary==2.7.7
 pyexcel==0.5.14
 pyexcel-xlsx==0.5.7