Fix #2 Define JSONField

Since peewee>3.0 JSONField was moved to extensions modules for both SQLite and Postgres. All we need is a generic one so we wrap TextField. Signed-off-by: David Brown <[email protected]>
python-social-auth · Jan 3, 2019 · b0654a3 · b0654a3
1 parent 8a5c849
commit b0654a3
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/social_peewee/storage.py b/social_peewee/storage.py
@@ -1,12 +1,20 @@
 import six
 import base64
+import json
 
-from peewee import CharField, IntegerField, Model, Proxy, IntegrityError
-from playhouse.kv import JSONField
-
+from peewee import CharField, IntegerField, Model, Proxy, IntegrityError, \
+ TextField
 from social_core.storage import UserMixin, AssociationMixin, NonceMixin, \
  CodeMixin, PartialMixin, BaseStorage
 
+class JSONField(TextField):
+ def db_value(self, value):
+ return json.dumps(value)
+
+ def python_value(self, value):
+ if value is not None:
+ return json.loads(value)
+
 
 def get_query_by_dict_param(cls, params):
  query = True