Skip to content

Commit

Permalink
data_quality: Create rewrite_quality_data command
Browse files Browse the repository at this point in the history
Update settings and add app.py to make this into a django app
  • Loading branch information
michaelwood committed May 26, 2021
1 parent fc9cf57 commit 292793c
Show file tree
Hide file tree
Showing 3 changed files with 40 additions and 0 deletions.
5 changes: 5 additions & 0 deletions datastore/data_quality/apps.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
from django.apps import AppConfig


class AdditionalDataConfig(AppConfig):
name = "data_quality"
34 changes: 34 additions & 0 deletions datastore/data_quality/management/commands/rewrite_quality_data.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
from django.core.management.base import BaseCommand

from data_quality import quality_data
import db.models as db


class Command(BaseCommand):
help = (
"Reloads the additional data on grant data specified by datagetter id or latest"
)

def add_arguments(self, parser):
parser.add_argument(
type=str,
action="store",
dest="getter_run",
help="The datagetter run id or latest",
)

def handle(self, *args, **options):

if "latest" in options["getter_run"]:
source_files = db.Latest.objects.get(
series=db.Latest.CURRENT
).sourcefile_set.all()
else:
source_files = db.SourceFile.object.filter(getter_run=options["getter_run"])

for source_file in source_files:
grants_list = {
"grants": list(source_file.grant_set.values_list("data", flat=True))
}
source_file.quality = quality_data.create(grants_list)
source_file.save()
1 change: 1 addition & 0 deletions datastore/settings/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@
"api",
"prometheus",
"additional_data",
"data_quality",
# Useful Django apps:
"django.contrib.admin",
"django.contrib.auth",
Expand Down

0 comments on commit 292793c

Please sign in to comment.