Adds Twitter image to web app

lib/twitter.py

@@ -16,6 +16,7 @@ def __post_init__(self):
         # shortcuts
         self.screen_name = self.target.screen_name
         self.total_followers = self.target.followers_count
+        self.image = self.target.profile_image_url_https
 
     @property
     def followers(self):

web/core/migrations/0006_add_image_field_to_job_model.py

@@ -0,0 +1,16 @@
+# Generated by Django 3.0 on 2019-12-29 18:22
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [("core", "0005_add_related_name_to_account_relationship")]
+
+    operations = [
+        migrations.AddField(
+            model_name="job",
+            name="image",
+            field=models.URLField(blank=True, default=""),
+        )
+    ]

web/core/migrations/0007_fill_image_field_on_job_model.py

@@ -0,0 +1,16 @@
+# Generated by Django 3.0 on 2019-12-29 18:24
+
+from django.db import migrations
+
+from lib.twitter import Twitter
+
+
+def collect_image_urls(apps, _schema_editor):
+    for job in apps.get_model("core", "Job").objects.filter(image=""):
+        job.image = Twitter(job.screen_name).image
+        job.save()
+
+
+class Migration(migrations.Migration):
+    dependencies = [("core", "0006_add_image_field_to_job_model")]
+    operations = [migrations.RunPython(collect_image_urls)]

web/core/models.py

@@ -11,6 +11,7 @@
 
 class Job(models.Model, JobSerializer):
     screen_name = LowerCaseCharField(max_length=15, unique=True, db_index=True)
+    image = models.URLField(default="", blank=True)
     total_followers = models.IntegerField(null=True)
     celery_task_id = models.CharField(
         max_length=155, unique=True, db_index=True, default=None, null=True

web/core/serializers.py

@@ -5,6 +5,7 @@ class JobSerializer:
     def json(self):
         serialized = {
             "twitter_account": self.screen_name,
+            "image": self.image,
             "followers": self.total_followers,
             "analyzed": self.followers.analyzed().count(),
             "analyzed_percent": None,

web/core/tasks.py

@@ -27,6 +27,7 @@ def start(pk):
         return restart(job)
 
     job.total_followers = twitter.total_followers
+    job.image = twitter.image
     job.save()
 
     try:

web/core/tests/test_api_view.py

@@ -7,7 +7,9 @@
 
 class TestApiView(TestCase):
     def setUp(self):
-        self.job = mixer.blend(Job, screen_name="Borsalino")
+        self.job = mixer.blend(
+            Job, screen_name="Borsalino", image="https://borsalino.png"
+        )
         accounts = (
             mixer.blend(Account, botometer=0.62),
             mixer.blend(Account, botometer=0.99),
@@ -27,6 +29,7 @@ def test_content(self):
         user, *_ = contents["data"]
         expected = {
             "twitter_account": str,
+            "image": str,
             "followers": int,
             "analyzed": int,
             "analyzed_percent": float,