pythonprobr · renzon · Apr 18, 2019 · Apr 18, 2019
diff --git a/Pipfile b/Pipfile
@@ -18,6 +18,7 @@ python-decouple = "*"
 raven = "*"
 Pillow = "*"
 ipython = "*"
+django-sitemaps = "*"
 
 [dev-packages]
 faker = "*"
@@ -30,4 +31,4 @@ pytest-freezegun = "*"
 pytest-mock = "*"
 
 [requires]
-python_version = "3.7"
+python_version = "3.7"
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/pythonpro/core/tests/test_sitemaps_and_robots.py b/pythonpro/core/tests/test_sitemaps_and_robots.py
@@ -0,0 +1,21 @@
+import pytest
+from django.test import Client
+from django.urls import reverse
+
+
+@pytest.fixture
+def resp_sitemap(client: Client, db):
+    return client.get(reverse('core:sitemap'), secure=True)
+
+
+def test_sitemap_status_code(resp_sitemap):
+    assert resp_sitemap.status_code == 200
+
+
+@pytest.fixture
+def resp_robots(client: Client, db):
+    return client.get(reverse('core:robots'), secure=True)
+
+
+def test_robots_status_code(resp_robots):
+    assert resp_robots.status_code == 200
diff --git a/pythonpro/core/urls.py b/pythonpro/core/urls.py
@@ -1,10 +1,13 @@
 from django.urls import path
+from django_sitemaps import robots_txt
 
 from . import views
 
 app_name = 'core'
 urlpatterns = [
     path('', views.index, name='index'),
+    path('sitemap.xml', views.sitemap, name='sitemap'),
+    path('robots.txt', robots_txt(timeout=86400), name='robots'),
     path('tech-talks', views.teck_talks, name='tech_talks'),
     path('podcast', views.podcast, name='podcast'),
     path('obrigado', views.thanks, name='thanks'),

diff --git a/pythonpro/core/views.py b/pythonpro/core/views.py
@@ -1,11 +1,14 @@
+from django.conf import settings
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.views import PasswordChangeView
 from django.shortcuts import render
-from django.urls import reverse_lazy
+from django.urls import reverse_lazy, reverse
 from django.views.generic import UpdateView
+from django_sitemaps import Sitemap
 
 from pythonpro.core.forms import UserEmailForm
 from pythonpro.core.models import User
+from pythonpro.promos.facade import find_all_videos
 
 
 def index(request):
@@ -29,6 +32,22 @@ def profile(request):
     return render(request, 'core/profile_detail.html', {})
 
 
+def sitemap(request):
+    map = Sitemap(
+        build_absolute_uri=request.build_absolute_uri,
+    )
+
+    for section in 'core:index core:podcast core:tech_talks modules:index'.split():
+        map.add(reverse(section), changefreq='weekly')
+
+    for video in find_all_videos():
+        map.add(video.get_absolute_url(), changefreq='monthly')
+
+    return map.response(
+        pretty_print=settings.DEBUG,
+    )
+
+
 class _ProfileUpdateName(UpdateView):
     model = User
     fields = ('first_name',)

diff --git a/pythonpro/promos/facade.py b/pythonpro/promos/facade.py
@@ -0,0 +1,5 @@
+from pythonpro.promos.models import Video as _Video
+
+
+def find_all_videos():
+    return _Video.objects.all()
diff --git a/pythonpro/promos/models.py b/pythonpro/promos/models.py
@@ -1,7 +1,11 @@
 from django.db import models
+from django.urls import reverse
 
 
 class Video(models.Model):
     title = models.CharField(max_length=100)
     vimeo_id = models.CharField(max_length=100)
     slug = models.SlugField(max_length=100, unique=True)
+
+    def get_absolute_url(self):
+        return reverse('promos:video', kwargs={'slug': self.slug})