SQLite Import moved to mgmt command for Heroku
This commit is contained in:
		
							parent
							
								
									92445132fe
								
							
						
					
					
						commit
						ffbfdce5b2
					
				
					 12 changed files with 80 additions and 132 deletions
				
			
		| 
						 | 
				
			
			@ -7,7 +7,7 @@ from django.db import migrations, models
 | 
			
		|||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ('projects', '0002_project_data'),
 | 
			
		||||
        ('projects', '0001_initial_project'),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
| 
						 | 
				
			
			@ -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),
 | 
			
		||||
    ]
 | 
			
		||||
| 
						 | 
				
			
			@ -7,7 +7,7 @@ from django.db import migrations, models
 | 
			
		|||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ('projects', '0004_grant_data'),
 | 
			
		||||
        ('projects', '0002_initial_grant'),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
| 
						 | 
				
			
			@ -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),
 | 
			
		||||
    ]
 | 
			
		||||
| 
						 | 
				
			
			@ -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 = [
 | 
			
		||||
| 
						 | 
				
			
			@ -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),
 | 
			
		||||
    ]
 | 
			
		||||
| 
						 | 
				
			
			@ -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),
 | 
			
		||||
    ]
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue