OpenSlides/openslides/utils/models.py

113 lines
4.1 KiB
Python
Raw Normal View History

from django.db import models
2017-08-23 20:51:06 +02:00
from .access_permissions import BaseAccessPermissions # noqa
from .utils import convert_camel_case_to_pseudo_snake_case
class MinMaxIntegerField(models.IntegerField):
2013-09-25 12:53:44 +02:00
"""
IntegerField with options to set a min- and a max-value.
"""
def __init__(self, min_value=None, max_value=None, *args, **kwargs):
self.min_value, self.max_value = min_value, max_value
super(MinMaxIntegerField, self).__init__(*args, **kwargs)
def formfield(self, **kwargs):
defaults = {'min_value': self.min_value, 'max_value': self.max_value}
defaults.update(kwargs)
return super(MinMaxIntegerField, self).formfield(**defaults)
2015-06-29 12:08:15 +02:00
class RESTModelMixin:
"""
Mixin for Django models which are used in our REST API.
2015-06-29 12:08:15 +02:00
"""
2017-08-23 20:51:06 +02:00
access_permissions = None # type: BaseAccessPermissions
2015-06-29 12:08:15 +02:00
def get_root_rest_element(self):
"""
Returns the root rest instance.
Uses self as default.
"""
return self
@classmethod
def get_access_permissions(cls):
"""
Returns a container to handle access permissions for this model and
its corresponding viewset.
"""
return cls.access_permissions
@classmethod
def get_collection_string(cls):
2015-06-29 12:08:15 +02:00
"""
Returns the string representing the name of the collection. Returns
None if this is not a so called root rest instance.
2015-06-29 12:08:15 +02:00
"""
# TODO Check if this is a root rest element class and return None if not.
return '/'.join(
(convert_camel_case_to_pseudo_snake_case(cls._meta.app_label),
convert_camel_case_to_pseudo_snake_case(cls._meta.object_name)))
2016-01-03 15:33:51 +01:00
def get_rest_pk(self):
2016-01-03 15:33:51 +01:00
"""
Returns the primary key used in the REST API. By default this is
the database pk.
2016-01-03 15:33:51 +01:00
"""
return self.pk
def save(self, skip_autoupdate=False, information=None, *args, **kwargs):
"""
2016-09-30 20:42:58 +02:00
Calls Django's save() method and afterwards hits the autoupdate system.
If skip_autoupdate is set to True, then the autoupdate system is not
informed about the model changed. This also means, that the model cache
is not updated. You have to do this manually, by creating a collection
element from the instance:
CollectionElement.from_instance(instance)
2016-09-30 20:42:58 +02:00
The optional argument information can be a dictionary that is given to
the autoupdate system.
"""
# We don't know how to fix this circular import
from .autoupdate import inform_changed_data
return_value = super().save(*args, **kwargs)
2016-09-30 20:42:58 +02:00
if not skip_autoupdate:
inform_changed_data(self.get_root_rest_element(), information=information)
return return_value
def delete(self, skip_autoupdate=False, information=None, *args, **kwargs):
"""
2016-09-30 20:42:58 +02:00
Calls Django's delete() method and afterwards hits the autoupdate system.
If skip_autoupdate is set to True, then the autoupdate system is not
informed about the model changed. This also means, that the model cache
is not updated. You have to do this manually, by creating a collection
element from the instance:
CollectionElement.from_instance(instance, deleted=True)
or
CollectionElement.from_values(collection_string, id, deleted=True)
The optional argument information can be a dictionary that is given to
the autoupdate system.
"""
# We don't know how to fix this circular import
from .autoupdate import inform_changed_data, inform_deleted_data
instance_pk = self.pk
return_value = super().delete(*args, **kwargs)
2016-09-30 20:42:58 +02:00
if not skip_autoupdate:
if self != self.get_root_rest_element():
# The deletion of a included element is a change of the root element.
inform_changed_data(self.get_root_rest_element(), information=information)
else:
inform_deleted_data(self.get_collection_string(), instance_pk, information=information)
return return_value