SQLite Import moved to mgmt command for Heroku

This commit is contained in:
Matthew Ryan Dillon 2016-01-28 11:01:06 -07:00
parent 92445132fe
commit ffbfdce5b2
12 changed files with 80 additions and 132 deletions

View file

@ -7,7 +7,7 @@ from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('projects', '0002_project_data'),
('projects', '0001_initial_project'),
]
operations = [

View file

@ -1,31 +0,0 @@
from django.db import migrations, models, transaction
from ccdb.utils.data_import import setup_sqlite
@transaction.atomic
def import_projects(apps, schema_editor):
Project = apps.get_model('projects', 'Project')
c = setup_sqlite()
if c:
for r in c.execute('SELECT * FROM tbl_lu_projects;'):
p = Project(id=r[0], name=r[1], code=r[2], iacuc_number=r[3],
description=r[4], sort_order=r[5])
p.save()
def remove_projects(apps, schema_editor):
print("removing projects...")
Project = apps.get_model("projects", "Project")
Project.objects.all().delete()
class Migration(migrations.Migration):
dependencies = [
('projects', '0001_initial_project'),
]
operations = [
migrations.RunPython(import_projects, remove_projects),
]

View file

@ -7,7 +7,7 @@ from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('projects', '0004_grant_data'),
('projects', '0002_initial_grant'),
]
operations = [

View file

@ -1,30 +0,0 @@
from django.db import migrations, models, transaction
from ccdb.utils.data_import import setup_sqlite
@transaction.atomic
def import_grants(apps, schema_editor):
Grant = apps.get_model('projects', 'Grant')
c = setup_sqlite()
if c:
for r in c.execute('SELECT * FROM tbl_lu_grants;'):
g = Grant(id=r[0], title=r[1], code=r[2],
description=r[3], sort_order=r[4])
g.save()
def remove_grants(apps, schema_editor):
Grant = apps.get_model('projects', 'Grant')
Grant.objects.all().delete()
class Migration(migrations.Migration):
dependencies = [
('projects', '0003_initial_grant'),
]
operations = [
migrations.RunPython(import_grants, remove_grants),
]

View file

@ -7,7 +7,7 @@ from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('projects', '0006_project_grant_data'),
('projects', '0003_grant_projects'),
]
operations = [

View file

@ -1,31 +0,0 @@
from django.db import migrations, models, transaction
from ccdb.utils.data_import import setup_sqlite
@transaction.atomic
def import_project_grant(apps, schema_editor):
Project = apps.get_model('projects', 'Project')
Grant = apps.get_model('projects', 'Grant')
c = setup_sqlite()
if c:
for r in c.execute('SELECT * FROM tbl_hash_project_grants;'):
p = Project.objects.get(id=r[0])
g = Grant.objects.get(id=r[1])
p.grants.add(g)
p.save()
def remove_project_grant(apps, schema_editor):
pass
class Migration(migrations.Migration):
dependencies = [
('projects', '0005_grant_projects'),
]
operations = [
migrations.RunPython(import_project_grant, remove_project_grant),
]

View file

@ -1,37 +0,0 @@
from django.db import migrations, models, transaction
from ccdb.utils.data_import import setup_sqlite
@transaction.atomic
def import_grantreport(apps, schema_editor):
GrantReport = apps.get_model('projects', 'GrantReport')
Grant = apps.get_model('projects', 'Grant')
c = setup_sqlite()
if c:
q = '''
SELECT *, report_due_date AS "due_date [dtdt]"
FROM tbl_lu_grant_reports;
'''
for r in c.execute(q):
g = Grant.objects.get(id=r[0])
gr = GrantReport(grant=g, title=r[1], report_type=r[2],
description=r[3], due_date=r[8], submitted_date=r[5],
attachment=r[6], sort_order=r[7])
gr.save()
def remove_grantreport(apps, schema_editor):
GrantReport = apps.get_model('projects', 'GrantReport')
GrantReport.objects.all().delete()
class Migration(migrations.Migration):
dependencies = [
('projects', '0007_initial_grantreport'),
]
operations = [
migrations.RunPython(import_grantreport, remove_grantreport),
]