diff --git a/src/adega/management/__init__.py b/src/adega/management/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/src/adega/management/commands/__init__.py b/src/adega/management/commands/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/src/adega/management/commands/analyze.py b/src/adega/management/commands/analyze.py new file mode 100644 index 0000000000000000000000000000000000000000..de038c449320fc57db29d6b5577b0a96891db3fb --- /dev/null +++ b/src/adega/management/commands/analyze.py @@ -0,0 +1,18 @@ +from django.core.management.base import BaseCommand +from uploads.models import Submission + +from script.main import analyze + + +class Command(BaseCommand): + help = 'Makes one specific analysis' + + def add_arguments(self, parser): + parser.add_argument('submission_id', type=int) + + def handle(self, *args, **options): + id = options['submission_id'] + + submission = Submission.objects.get(pk=id) + + analyze(submission) diff --git a/src/script/main.py b/src/script/main.py index b39411fd317929df172afebb7d7c60c5593cf9fe..fb45fa4376ba1c55fd62aa39c2787af16ba594c1 100644 --- a/src/script/main.py +++ b/src/script/main.py @@ -15,6 +15,8 @@ def analyze(submission): build_cache(dataframe) + submission.processed = True + submission.process_time = round(time.clock() - start_time) submission.save() @@ -36,5 +38,6 @@ def main(): print("--- Tempo de CPU: {} ---".format(cpu_time)) print("--- Tempo total: {} ---".format(run_time)) + if __name__ == "__main__": main() diff --git a/src/uploads/admin.py b/src/uploads/admin.py index f6770b002b386aa4a54922d3d8c86872aada55f9..99875aefe6a3f8cef6aa2978c97da412fabf3851 100644 --- a/src/uploads/admin.py +++ b/src/uploads/admin.py @@ -13,11 +13,6 @@ def make_analysis(modeladmin, request, queryset): print('analisando: '+str(submission)) analyze(submission) - submission.processed = True - - print('salvando') - submission.save() - print('OK') except: print('Análise falhou') @@ -33,4 +28,4 @@ class SubmissionAdmin(admin.ModelAdmin): actions = [make_analysis] -admin.site.register(Submission, SubmissionAdmin) \ No newline at end of file +admin.site.register(Submission, SubmissionAdmin)