| Server IP : 188.40.95.74 / Your IP : 216.73.216.142 Web Server : Apache System : Linux cp01.striminghost.net 3.10.0-1160.119.1.el7.tuxcare.els13.x86_64 #1 SMP Fri Nov 22 06:29:45 UTC 2024 x86_64 User : vlasotin ( 1054) PHP Version : 5.6.40 Disable Function : NONE MySQL : ON | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /opt/imunify360/venv/lib/python3.11/site-packages/playhouse/ |
Upload File : |
"""
Provide django-style hooks for model events.
"""
from peewee import Model as _Model
class Signal(object):
def __init__(self):
self._flush()
def _flush(self):
self._receivers = set()
self._receiver_list = []
def connect(self, receiver, name=None, sender=None):
name = name or receiver.__name__
key = (name, sender)
if key not in self._receivers:
self._receivers.add(key)
self._receiver_list.append((name, receiver, sender))
else:
raise ValueError('receiver named %s (for sender=%s) already '
'connected' % (name, sender or 'any'))
def disconnect(self, receiver=None, name=None, sender=None):
if receiver:
name = name or receiver.__name__
if not name:
raise ValueError('a receiver or a name must be provided')
key = (name, sender)
if key not in self._receivers:
raise ValueError('receiver named %s for sender=%s not found.' %
(name, sender or 'any'))
self._receivers.remove(key)
self._receiver_list = [(n, r, s) for n, r, s in self._receiver_list
if n != name and s != sender]
def __call__(self, name=None, sender=None):
def decorator(fn):
self.connect(fn, name, sender)
return fn
return decorator
def send(self, instance, *args, **kwargs):
sender = type(instance)
responses = []
for n, r, s in self._receiver_list:
if s is None or isinstance(instance, s):
responses.append((r, r(sender, instance, *args, **kwargs)))
return responses
pre_save = Signal()
post_save = Signal()
pre_delete = Signal()
post_delete = Signal()
pre_init = Signal()
class Model(_Model):
def __init__(self, *args, **kwargs):
super(Model, self).__init__(*args, **kwargs)
pre_init.send(self)
def save(self, *args, **kwargs):
pk_value = self._pk if self._meta.primary_key else True
created = kwargs.get('force_insert', False) or not bool(pk_value)
pre_save.send(self, created=created)
ret = super(Model, self).save(*args, **kwargs)
post_save.send(self, created=created)
return ret
def delete_instance(self, *args, **kwargs):
pre_delete.send(self)
ret = super(Model, self).delete_instance(*args, **kwargs)
post_delete.send(self)
return ret