X-Git-Url: http://git.veekun.com/zzz-floof.git/blobdiff_plain/a6d7a53b0cc7a8f8629e8e1272a56d3c51179317..ea3d7fcf702e372eac978d61b9af9dbad46543bb:/floof/model/search.py

diff --git a/floof/model/search.py b/floof/model/search.py
index f03de51..7e04ae7 100644
--- a/floof/model/search.py
+++ b/floof/model/search.py
@@ -15,6 +15,18 @@ class SavedSearch(Entity):
 class GalleryWidget(Entity):
     page = ManyToOne('UserPage')
     search = ManyToOne(SavedSearch)
+    
+    def __init__(self, string=None, **kwargs):
+        owner = kwargs.get('owner', None)
+        if string:
+            self.search = SavedSearch(author=owner, string=string)
+        
+        page = kwargs.get('page', None)
+        if owner and not page:
+            page = author.primary_page
+        
+        super(GalleryWidget, self).__init__(**kwargs)
+
 
     # NOTE: no longer needed now that we have pages, I guess.
     # displayer = ManyToOne('User') # determines whose page should it should show up on