MapManager.visible(): also include maps where current user is owner o…

…r editor
umap-project · Jan 11, 2016 · e1ef0e4 · e1ef0e4
1 parent 47dd0e5
commit e1ef0e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/leaflet_storage/managers.py b/leaflet_storage/managers.py
@@ -5,4 +5,8 @@ class MapManager(models.GeoManager):
     def visible(self, request):
         filter = Q(share_status=self.model.PUBLIC)
 
+        if request.user.is_authenticated():
+            filter = filter | Q(owner=request.user)
+            filter = filter | Q(editors=request.user)
+
         return self.filter(filter)