def dummy_wagtail_doc(request): if not Collection.objects.exists(): # pragma: no cover Collection.add_root() doc = Document(title='hello') doc.file.save('foo.txt', ContentFile('foo', 'foo.txt')) doc.save() doc = Document.objects.get(pk=doc.pk) # Reload to ensure the upload took def nuke(): try: # Try cleaning up so `/var/media` isn't full of foo doc.file.delete() doc.delete() except: # pragma: no cover pass request.addfinalizer(nuke) return doc
def dummy_wagtail_doc(request): if not Collection.objects.exists(): # pragma: no cover Collection.add_root() doc = Document(title='hello') doc.file.save('foo.txt', ContentFile('foo', 'foo.txt')) doc.save() doc = Document.objects.get(pk=doc.pk) # Reload to ensure the upload took def nuke(): try: # Try cleaning up so `/var/media` isn't full of foo doc.file.delete() doc.delete() except: # pragma: no cover pass request.addfinalizer(nuke) return doc
def save(self, *args, **kwargs): ret = super().save(*args, **kwargs) update_fields = [] if self.root_collection is None: obj = Collection(name=self.name) Collection.add_root(instance=obj) self.root_collection = obj update_fields.append('root_collection') if self.admin_group is None: obj = Group.objects.create(name='%s admins' % self.name) self.admin_group = obj update_fields.append('admin_group') if self.contact_person_group is None: obj = Group.objects.create(name='%s contact persons' % self.name) self.contact_person_group = obj update_fields.append('contact_person_group') if update_fields: super().save(update_fields=update_fields) return ret