Chapter 12: Self-followers (12e)

app/models.py

@@ -105,6 +105,14 @@ class User(UserMixin, db.Model):
  lazy='dynamic',
  cascade='all, delete-orphan')
 
+ @staticmethod
+ def add_self_follows():
+ for user in User.query.all():
+ if not user.is_following(user):
+ user.follow(user)
+ db.session.add(user)
+ db.session.commit()
+
  def __init__(self, **kwargs):
  super(User, self).__init__(**kwargs)
  if self.role is None:
@@ -114,6 +122,7 @@ def __init__(self, **kwargs):
  self.role = Role.query.filter_by(default=True).first()
  if self.email is not None and self.avatar_hash is None:
  self.avatar_hash = self.gravatar_hash()
+ self.follow(self)
 
  @property
  def password(self):

app/templates/followers.html

@@ -10,6 +10,7 @@ <h1>{{ title }} {{ user.username }}</h1>
 <table class="table table-hover followers">
  <thead><tr><th>User</th><th>Since</th></tr></thead>
  {% for follow in follows %}
+ {% if follow.user != user %}
  <tr>
  <td>
  <a href="{{ url_for('.user', username = follow.user.username) }}">
@@ -19,6 +20,7 @@ <h1>{{ title }} {{ user.username }}</h1>
  </td>
  <td>{{ moment(follow.timestamp).format('L') }}</td>
  </tr>
+ {% endif %}
  {% endfor %}
 </table>
 <div class="pagination">

app/templates/user.html

@@ -30,8 +30,8 @@ <h1>{{ user.username }}</h1>
  <a href="{{ url_for('.unfollow', username=user.username) }}" class="btn btn-default">Unfollow</a>
  {% endif %}
  {% endif %}
- <a href="{{ url_for('.followers', username=user.username) }}">Followers: <span class="badge">{{ user.followers.count() }}</span></a>
- <a href="{{ url_for('.followed_by', username=user.username) }}">Following: <span class="badge">{{ user.followed.count() }}</span></a>
+ <a href="{{ url_for('.followers', username=user.username) }}">Followers: <span class="badge">{{ user.followers.count() - 1 }}</span></a>
+ <a href="{{ url_for('.followed_by', username=user.username) }}">Following: <span class="badge">{{ user.followed.count() - 1 }}</span></a>
  {% if current_user.is_authenticated and user != current_user and user.is_following(current_user) %}
  | <span class="label label-default">Follows you</span>
  {% endif %}

tests/test_user_model.py

@@ -186,8 +186,8 @@ def test_follows(self):
  self.assertTrue(u1.is_following(u2))
  self.assertFalse(u1.is_followed_by(u2))
  self.assertTrue(u2.is_followed_by(u1))
- self.assertTrue(u1.followed.count() == 1)
- self.assertTrue(u2.followers.count() == 1)
+ self.assertTrue(u1.followed.count() == 2)
+ self.assertTrue(u2.followers.count() == 2)
  f = u1.followed.all()[-1]
  self.assertTrue(f.followed == u2)
  self.assertTrue(timestamp_before <= f.timestamp <= timestamp_after)
@@ -196,13 +196,13 @@ def test_follows(self):
  u1.unfollow(u2)
  db.session.add(u1)
  db.session.commit()
- self.assertTrue(u1.followed.count() == 0)
- self.assertTrue(u2.followers.count() == 0)
- self.assertTrue(Follow.query.count() == 0)
+ self.assertTrue(u1.followed.count() == 1)
+ self.assertTrue(u2.followers.count() == 1)
+ self.assertTrue(Follow.query.count() == 2)
  u2.follow(u1)
  db.session.add(u1)
  db.session.add(u2)
  db.session.commit()
  db.session.delete(u2)
  db.session.commit()
- self.assertTrue(Follow.query.count() == 0)
+ self.assertTrue(Follow.query.count() == 1)