Test suite runs and passes!

[zzz-floof.git] / floof / model / search.py

diff --git a/floof/model/search.py b/floof/model/search.py
index 37f2d5f..9dabd58 100644 (file)
--- a/floof/model/search.py
+++ b/floof/model/search.py
@@ -3,8 +3,8 @@ from elixir import *
 
 
 class SavedSearch(Entity):
-    string = Field(Unicode) # I tried calling this query, but it broke elixir
-    author = ManyToOne('User')
+    string = Field(Unicode, required=True)
+    author = ManyToOne('User', required=True)
     fork = ManyToOne("SavedSearch")
 
     def __unicode__(self):
@@ -13,8 +13,20 @@ class SavedSearch(Entity):
 
 
 class GalleryWidget(Entity):
-    page = ManyToOne('UserPage')
-    search = ManyToOne(SavedSearch)
+    page = ManyToOne('UserPage', required=True)
+    search = ManyToOne(SavedSearch, required=True)
+    
+    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
@@ -24,7 +36,7 @@ class GalleryWidget(Entity):
 
     @property
     def string(self):
-        return self.search
+        return self.search.string
 
     @string.setter
     def string(self, value):
@@ -33,4 +45,4 @@ class GalleryWidget(Entity):
         #     # TODO: may have to refactor this into an init if the key ordering is inconvenienc
         #     raise "Oh no!  This gallery needs a displayer to set on the saved search."
 
-        self.search = SavedSearch(author=getattr(self,"author",None), string=value)
+        self.search = SavedSearch(author=getattr(self, "author", None), string=value)