Move CompressedField to peewee_ext.py

1 files changed, 1 insertions, 9 deletions

diff --git a/archivist/model.py b/archivist/model.py
index d660218..2d9476a 100644
--- a/archivist/model.py
+++ b/archivist/model.py
@@ -1,5 +1,4 @@
 from peewee import *
-from playhouse.fields import CompressedField as _CompressedField
 from playhouse.hybrid import *
 from playhouse.sqlite_ext import SqliteExtDatabase, ClosureTable
 
@@ -9,7 +8,7 @@ import datetime
 from enum import Enum, unique
 from pkg_resources import resource_filename
 
-from .peewee_ext import EnumField, SQLIndex
+from .peewee_ext import EnumField, CompressedField, SQLIndex
 from .peewee_fixes import * # dummy to force evaluation of those fixes
 
 db = SqliteExtDatabase('test.db', pragmas=[('foreign_keys', 'ON')])
@@ -56,13 +55,6 @@ class BaseModel(Model):
     class Meta:
         database = db
 
-class CompressedField(_CompressedField):
-    def db_value(self, value):
-        return value if value is None else self.compress(value)
-
-    def python_value(self, value):
-        return value if value is None else self.decompress(value)
-
 @table
 class DocumentContent(BaseModel):
     blob = CompressedField()