Merge pull request #6422 from readthedocs/humitos/update-repos-task

Use trigger_build for update_repos command

Author: humitos

readthedocs/core/management/commands/update_repos.py

@@ -1,5 +1,3 @@
-# -*- coding: utf-8 -*-
-
 """
 Custom management command to rebuild documentation for all projects.
 
@@ -11,12 +9,10 @@
 from django.core.management.base import BaseCommand
 
 from readthedocs.builds.constants import EXTERNAL, INTERNAL
-from readthedocs.builds.models import Build, Version
+from readthedocs.builds.models import Version
 from readthedocs.core.utils import trigger_build
-from readthedocs.projects import tasks
 from readthedocs.projects.models import Project
 
-
 log = logging.getLogger(__name__)
 
 
@@ -64,61 +60,23 @@ def handle(self, *args, **options):
                             active=True,
                             uploaded=False,
                     ):
-
-                        build = Build.objects.create(
-                            project=version.project,
-                            version=version,
-                            type='html',
-                            state='triggered',
-                        )
-
-                        # pylint: disable=no-value-for-parameter
-                        tasks.update_docs_task(
-                            version.pk,
-                            build_pk=build.pk,
-                        )
+                        trigger_build(project=version.project, version=version)
                 elif version == INTERNAL:
                     log.info('Updating all internal versions for %s', slug)
                     for version in Version.internal.filter(
                             project__slug=slug,
                             active=True,
                             uploaded=False,
                     ):
-
-                        build = Build.objects.create(
-                            project=version.project,
-                            version=version,
-                            type='html',
-                            state='triggered',
-                        )
-
-                        # pylint: disable=no-value-for-parameter
-                        tasks.update_docs_task(
-                            version.project_id,
-                            build_pk=build.pk,
-                            version_pk=version.pk,
-                        )
+                        trigger_build(project=version.project, version=version)
                 elif version == EXTERNAL:
                     log.info('Updating all external versions for %s', slug)
                     for version in Version.external.filter(
                             project__slug=slug,
                             active=True,
                             uploaded=False,
                     ):
-
-                        build = Build.objects.create(
-                            project=version.project,
-                            version=version,
-                            type='html',
-                            state='triggered',
-                        )
-
-                        # pylint: disable=no-value-for-parameter
-                        tasks.update_docs_task(
-                            version.project_id,
-                            build_pk=build.pk,
-                            version_pk=version.pk,
-                        )
+                        trigger_build(project=version.project, version=version)
                 else:
                     p = Project.all_objects.get(slug=slug)
                     log.info('Building %s', p)
@@ -130,18 +88,12 @@ def handle(self, *args, **options):
                         active=True,
                         uploaded=False,
                 ):
-                    # pylint: disable=no-value-for-parameter
-                    tasks.update_docs_task(
-                        version.pk,
-                        force=force,
-                    )
+                    trigger_build(project=version.project, version=version)
+
             else:
                 log.info('Updating all docs')
                 for project in Project.objects.all():
                     # pylint: disable=no-value-for-parameter
                     default_version = project.get_default_version()
                     version = project.versions.get(slug=default_version)
-                    tasks.update_docs_task(
-                        version.pk,
-                        force=force,
-                    )
+                    trigger_build(project=version.project, version=version)