FM / playhouse

/opt/imunify360/venv/lib/python3.11/site-packages/playhouse/

/opt/imunify360/venv/lib/python3.11/site-packages/playhouse

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
apsw_ext.py	5018	0644	edit dl rm
cockroachdb.py	9169	0644	edit dl rm
dataset.py	14488	0644	edit dl rm
db_url.py	4358	0644	edit dl rm
fields.py	1699	0644	edit dl rm
flask_utils.py	8197	0644	edit dl rm
hybrid.py	1528	0644	edit dl rm
kv.py	5608	0644	edit dl rm
migrate.py	33447	0644	edit dl rm
mysql_ext.py	3874	0644	edit dl rm
pool.py	13092	0644	edit dl rm
postgres_ext.py	15336	0644	edit dl rm
psycopg3_ext.py	5473	0644	edit dl rm
reflection.py	31148	0644	edit dl rm
shortcuts.py	11753	0644	edit dl rm
signals.py	2511	0644	edit dl rm
sqlcipher_ext.py	3632	0644	edit dl rm
sqliteq.py	10819	0644	edit dl rm
sqlite_changelog.py	4793	0644	edit dl rm
sqlite_ext.py	48520	0644	edit dl rm
sqlite_udf.py	13665	0644	edit dl rm
test_utils.py	1854	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imunify360/venv/lib/python3.11/site-packages/playhouse/fields.py (1699B)

try:
    import bz2
except ImportError:
    bz2 = None
try:
    import zlib
except ImportError:
    zlib = None
try:
    import cPickle as pickle
except ImportError:
    import pickle

from peewee import BlobField
from peewee import buffer_type

class CompressedField(BlobField):
    ZLIB = 'zlib'
    BZ2 = 'bz2'
    algorithm_to_import = {
        ZLIB: zlib,
        BZ2: bz2,
    }

def __init__(self, compression_level=6, algorithm=ZLIB, *args,
                 **kwargs):
        self.compression_level = compression_level
        if algorithm not in self.algorithm_to_import:
            raise ValueError('Unrecognized algorithm %s' % algorithm)
        compress_module = self.algorithm_to_import[algorithm]
        if compress_module is None:
            raise ValueError('Missing library required for %s.' % algorithm)

self.algorithm = algorithm
        self.compress = compress_module.compress
        self.decompress = compress_module.decompress
        super(CompressedField, self).__init__(*args, **kwargs)

def python_value(self, value):
        if value is not None:
            return self.decompress(value)

def db_value(self, value):
        if value is not None:
            return self._constructor(
                self.compress(value, self.compression_level))

class PickleField(BlobField):
    def python_value(self, value):
        if value is not None:
            if isinstance(value, buffer_type):
                value = bytes(value)
            return pickle.loads(value)

def db_value(self, value):
        if value is not None:
            pickled = pickle.dumps(value, pickle.HIGHEST_PROTOCOL)
            return self._constructor(pickled)

try: import bz2 except ImportError: bz2 = None try: import zlib except ImportError: zlib = None try: import cPickle as pickle except ImportError: import pickle from peewee import BlobField from peewee import buffer_type class CompressedField(BlobField): ZLIB = 'zlib' BZ2 = 'bz2' algorithm_to_import = { ZLIB: zlib, BZ2: bz2, } def __init__(self, compression_level=6, algorithm=ZLIB, *args, **kwargs): self.compression_level = compression_level if algorithm not in self.algorithm_to_import: raise ValueError('Unrecognized algorithm %s' % algorithm) compress_module = self.algorithm_to_import[algorithm] if compress_module is None: raise ValueError('Missing library required for %s.' % algorithm) self.algorithm = algorithm self.compress = compress_module.compress self.decompress = compress_module.decompress super(CompressedField, self).__init__(*args, **kwargs) def python_value(self, value): if value is not None: return self.decompress(value) def db_value(self, value): if value is not None: return self._constructor( self.compress(value, self.compression_level)) class PickleField(BlobField): def python_value(self, value): if value is not None: if isinstance(value, buffer_type): value = bytes(value) return pickle.loads(value) def db_value(self, value): if value is not None: pickled = pickle.dumps(value, pickle.HIGHEST_PROTOCOL) return self._constructor(pickled)