Browse Source

Update Flask-Whooshee

See #261.
sh4nks 8 years ago
parent
commit
27f6a78160
2 changed files with 22 additions and 21 deletions
  1. 21 20
      flaskbb/utils/search.py
  2. 1 1
      requirements.txt

+ 21 - 20
flaskbb/utils/search.py

@@ -12,6 +12,7 @@
 import whoosh
 from flask_whooshee import AbstractWhoosheer
 
+from flaskbb._compat import text_type
 from flaskbb.forum.models import Forum, Topic, Post
 from flaskbb.user.models import User
 
@@ -30,18 +31,18 @@ class PostWhoosheer(AbstractWhoosheer):
     def update_post(cls, writer, post):
         writer.update_document(
             post_id=post.id,
-            username=post.username,
-            modified_by=post.modified_by,
-            content=post.content
+            username=text_type(post.username),
+            modified_by=text_type(post.modified_by),
+            content=text_type(post.content)
         )
 
     @classmethod
     def insert_post(cls, writer, post):
         writer.add_document(
             post_id=post.id,
-            username=post.username,
-            modified_by=post.modified_by,
-            content=post.content
+            username=text_type(post.username),
+            modified_by=text_type(post.modified_by),
+            content=text_type(post.content)
         )
 
     @classmethod
@@ -63,18 +64,18 @@ class TopicWhoosheer(AbstractWhoosheer):
     def update_topic(cls, writer, topic):
         writer.update_document(
             topic_id=topic.id,
-            title=topic.title,
-            username=topic.username,
-            content=getattr(topic.first_post,'content',None)
+            title=text_type(topic.title),
+            username=text_type(topic.username),
+            content=text_type(getattr(topic.first_post, 'content', None))
         )
 
     @classmethod
     def insert_topic(cls, writer, topic):
         writer.add_document(
             topic_id=topic.id,
-            title=topic.title,
-            username=topic.username,
-            content=getattr(topic.first_post,'content',None)
+            title=text_type(topic.title),
+            username=text_type(topic.username),
+            content=text_type(getattr(topic.first_post, 'content', None))
         )
 
     @classmethod
@@ -95,16 +96,16 @@ class ForumWhoosheer(AbstractWhoosheer):
     def update_forum(cls, writer, forum):
         writer.update_document(
             forum_id=forum.id,
-            title=forum.title,
-            description=forum.description
+            title=text_type(forum.title),
+            description=text_type(forum.description)
         )
 
     @classmethod
     def insert_forum(cls, writer, forum):
         writer.add_document(
             forum_id=forum.id,
-            title=forum.title,
-            description=forum.description
+            title=text_type(forum.title),
+            description=text_type(forum.description)
         )
 
     @classmethod
@@ -125,16 +126,16 @@ class UserWhoosheer(AbstractWhoosheer):
     def update_user(cls, writer, user):
         writer.update_document(
             user_id=user.id,
-            username=user.username,
-            email=user.email
+            username=text_type(user.username),
+            email=text_type(user.email)
         )
 
     @classmethod
     def insert_user(cls, writer, user):
         writer.add_document(
             user_id=user.id,
-            username=user.username,
-            email=user.email
+            username=text_type(user.username),
+            email=text_type(user.email)
         )
 
     @classmethod

+ 1 - 1
requirements.txt

@@ -19,7 +19,7 @@ Flask-Plugins==1.6.1
 Flask-Redis==0.3.0
 Flask-SQLAlchemy==2.1
 Flask-Themes2==0.1.4
-flask-whooshee==0.2.3
+flask-whooshee==0.4.0
 Flask-WTF==0.13.1
 itsdangerous==0.24
 Jinja2==2.8