Rework `WPS332`: allow `:=` in comprehensions

This code should be allowed:

```python
users = [user for user_id in user_ids if (user := get_user(user_id)) is not None]
```

We only allow `:=` in comprehensions / generator expressions now.