Add api endpoints to trigger async github/bitbucket repo syncing

Author: gregmuellegger

readthedocs/oauth/tasks.py

@@ -0,0 +1,32 @@
+from django.contrib.auth.models import User
+from djcelery import celery as celery_app
+
+from rtd.utils.tasks import PublicTask
+from rtd.utils.tasks import permission_check
+from rtd.utils.tasks import user_id_matches
+from .utils import import_bitbucket
+from .utils import import_github
+
+
+@permission_check(user_id_matches)
+class SyncGitHubRepositories(PublicTask):
+    public_name = 'sync_github_repositories'
+
+    def run_public(self, user_id):
+        user = User.objects.get(pk=user_id)
+        import_github(user, sync=True)
+
+
+sync_github_repositories = celery_app.tasks[SyncGitHubRepositories.name]
+
+
+@permission_check(user_id_matches)
+class SyncBitBucketRepositories(PublicTask):
+    public_name = 'sync_bitbucket_repositories'
+
+    def run_public(self, user_id):
+        user = User.objects.get(pk=user_id)
+        import_bitbucket(user, sync=True)
+
+
+sync_bitbucket_repositories = celery_app.tasks[SyncBitBucketRepositories.name]

readthedocs/restapi/urls.py

@@ -25,4 +25,6 @@
     url(r'search/project/$', 'restapi.views.search_views.project_search', name='api_project_search'),
     url(r'search/section/$', 'restapi.views.search_views.section_search', name='api_section_search'),
     url(r'jobs/status/(?P<task_id>[^/]+)/', 'restapi.views.task_views.job_status', name='api_job_status'),
+    url(r'jobs/sync-github-repositories/', 'restapi.views.task_views.sync_github_repositories', name='api_sync_github_repositories'),
+    url(r'jobs/sync-bitbucket-repositories/', 'restapi.views.task_views.sync_bitbucket_repositories', name='api_sync_bitbucket_repositories'),
 )

readthedocs/restapi/views/task_views.py

@@ -1,11 +1,13 @@
 import logging
 
+from django.core.urlresolvers import reverse
 from rest_framework import decorators, permissions
 from rest_framework.renderers import JSONPRenderer, JSONRenderer, BrowsableAPIRenderer
 from rest_framework.response import Response
 
 from rtd.utils.tasks import TaskNoPermission
 from rtd.utils.tasks import get_public_task_data
+import oauth.tasks
 
 
 log = logging.getLogger(__name__)
@@ -39,3 +41,31 @@ def job_status(request, task_id):
             get_status_data('unknown', 'PENDING', {}))
     return Response(
         get_status_data(task_name, state, public_data))
+
+
+@decorators.api_view(['POST'])
+@decorators.permission_classes((permissions.IsAuthenticated,))
+@decorators.renderer_classes(
+    (JSONRenderer, JSONPRenderer, BrowsableAPIRenderer))
+def sync_github_repositories(request):
+    result = oauth.tasks.sync_github_repositories.delay(
+        user_id=request.user.id)
+    task_id = result.task_id
+    return Response({
+        'task_id': task_id,
+        'url': reverse('api_job_status', kwargs={'task_id': task_id}),
+    })
+
+
+@decorators.api_view(['POST'])
+@decorators.permission_classes((permissions.IsAuthenticated,))
+@decorators.renderer_classes(
+    (JSONRenderer, JSONPRenderer, BrowsableAPIRenderer))
+def sync_bitbucket_repositories(request):
+    result = oauth.tasks.sync_bitbucket_repositories.delay(
+        user_id=request.user.id)
+    task_id = result.task_id
+    return Response({
+        'task_id': task_id,
+        'url': reverse('api_job_status', kwargs={'task_id': task_id}),
+    })