def register_delete_rules(self):
"""Registers rules for how Mongoengine handles the deletion of objects
that are being referenced by other objects.
See the documentation for
:func:`mongoengine.model.register_delete_rule` for more information.
All delete rules for User fields must by DENY, because User objects
should never be deleted. Lists of reference fields should PULL, to
remove deleted objects from the list, and all others should NULLIFY
"""
from eventum.models import (Event, EventSeries, User, Post, BlogPost,
Image)
from mongoengine import NULLIFY, PULL, DENY
Event.register_delete_rule(EventSeries, 'events', PULL)
Image.register_delete_rule(BlogPost, 'images', PULL)
Image.register_delete_rule(User, 'image', NULLIFY)
Image.register_delete_rule(BlogPost, 'featured_image', NULLIFY)
Image.register_delete_rule(Event, 'image', NULLIFY)
EventSeries.register_delete_rule(Event, 'parent_series', NULLIFY)
User.register_delete_rule(Event, 'creator', DENY)
User.register_delete_rule(Image, 'creator', DENY)
User.register_delete_rule(Post, 'author', DENY)
User.register_delete_rule(Post, 'posted_by', DENY)
def register_delete_rules(self):
"""Registers rules for how Mongoengine handles the deletion of objects
that are being referenced by other objects.
See the documentation for
:func:`mongoengine.model.register_delete_rule` for more information.
All delete rules for User fields must by DENY, because User objects
should never be deleted. Lists of reference fields should PULL, to
remove deleted objects from the list, and all others should NULLIFY
"""
from eventum.models import (Event, EventSeries, User, Post, BlogPost,
Image)
from mongoengine import NULLIFY, PULL, DENY
Event.register_delete_rule(EventSeries, 'events', PULL)
Image.register_delete_rule(BlogPost, 'images', PULL)
Image.register_delete_rule(User, 'image', NULLIFY)
Image.register_delete_rule(BlogPost, 'featured_image', NULLIFY)
Image.register_delete_rule(Event, 'image', NULLIFY)
EventSeries.register_delete_rule(Event, 'parent_series', NULLIFY)
User.register_delete_rule(Event, 'creator', DENY)
User.register_delete_rule(Image, 'creator', DENY)
User.register_delete_rule(Post, 'author', DENY)
User.register_delete_rule(Post, 'posted_by', DENY)
def run(self):
"""Run the generation. Uses the configurations passed to
func:`__init__`.
"""
# Setup: db connection, superuser, and printer.
connect(config['MONGODB_SETTINGS']['DB'])
try:
superuser = User.objects().get(gplus_id='super')
except DoesNotExist:
print ('Failed to get superuser. Try running:\n'
'\texport GOOGLE_AUTH_ENABLED=TRUE')
printer = ProgressPrinter(self.quiet)
# Images
if self.should_gen_images:
if self.wipe:
self.warn('Image')
print CLIColor.warning('Wiping Image database.')
Image.drop_collection()
create_images(12, superuser, printer)
# Blog posts
if self.should_gen_posts:
if self.wipe:
self.warn('BlogPost')
print CLIColor.warning('Wiping BlogPost database.')
BlogPost.drop_collection()
create_posts(10, superuser, printer)
# Events and event series
if self.should_gen_events:
if self.wipe:
self.warn('Event and EventSeries')
print CLIColor.warning('Wiping Event database.')
Event.drop_collection()
print CLIColor.warning('Wiping EventSeries database.')
EventSeries.drop_collection()
create_events(superuser, printer)
def _make_series(cls, form, **kwargs):
"""Create a new :class:`EventSeries` object and save it to Mongoengine.
The event is made by creating ``series_data`` and then unpacking it
into the constructor for :class:`EventSeries`.
:param form: The WTForm form to fetch series data from.
:type form: :class:`CreateEventForm` or a subclass.
:param dict kwargs: Any other arguments that should be applied on top
of the form data.
:returns: The newly created series.
:rtype: :class:`EventSeries`
"""
series_data = DataBuilder.series_data_from_form(form)
series_data.update(kwargs)
cls._validate_series_data(series_data)
series_data = cls._remove_none_fields(series_data)
series = EventSeries(**series_data)
series.save()
return series
def _make_series(cls, form, **kwargs):
"""Create a new :class:`EventSeries` object and save it to Mongoengine.
The event is made by creating ``series_data`` and then unpacking it
into the constructor for :class:`EventSeries`.
:param form: The WTForm form to fetch series data from.
:type form: :class:`CreateEventForm` or a subclass.
:param dict kwargs: Any other arguments that should be applied on top
of the form data.
:returns: The newly created series.
:rtype: :class:`EventSeries`
"""
series_data = DataBuilder.series_data_from_form(form)
series_data.update(kwargs)
cls._validate_series_data(series_data)
series_data = cls._remove_none_fields(series_data)
series = EventSeries(**series_data)
series.save()
return series
def __call__(self, form, field):
"""Called internally by :mod:`wtforms` on validation of the field.
:param form: The parent form
:type form: :class:`Form`
:param field: The field to validate
:type field: :class:`Field`
:raises: :class:`wtforms.validators.ValidationError`
"""
from eventum.models import Event, EventSeries
if EventSeries.objects(slug=field.data).count():
raise ValidationError(self.message)
if Event.objects(slug=field.data).count():
raise ValidationError(self.message)
def __call__(self, form, field):
"""Called internally by :mod:`wtforms` on validation of the field.
:param form: The parent form
:type form: :class:`Form`
:param field: The field to validate
:type field: :class:`Field`
:raises: :class:`wtforms.validators.ValidationError`
"""
from eventum.models import Event, EventSeries
if EventSeries.objects(slug=field.data).count():
raise ValidationError(self.message)
if Event.objects(slug=field.data).count():
raise ValidationError(self.message)
def __call__(self, form, field):
"""Called internally by :mod:`wtforms` on validation of the field.
:param form: The parent form
:type form: :class:`Form`
:param field: The field to validate
:type field: :class:`Field`
:raises: :class:`wtforms.validators.ValidationError`
"""
from eventum.models import Event, EventSeries
# If we change the slug, make sure the new slug doesn't exist
if self.original.slug != field.data:
if EventSeries.objects(slug=field.data).count():
raise ValidationError(self.message)
if Event.objects(slug=field.data).count():
raise ValidationError(self.message)
def __call__(self, form, field):
"""Called internally by :mod:`wtforms` on validation of the field.
:param form: The parent form
:type form: :class:`Form`
:param field: The field to validate
:type field: :class:`Field`
:raises: :class:`wtforms.validators.ValidationError`
"""
from eventum.models import Event, EventSeries
# If we change the slug, make sure the new slug doesn't exist
if self.original.slug != field.data:
if EventSeries.objects(slug=field.data).count():
raise ValidationError(self.message)
if Event.objects(slug=field.data).count():
raise ValidationError(self.message)