def save(self, *args, **kwargs):
created = self.pk is None
with transaction.atomic():
# set group if not already set
state = {
'saved': False,
}
if self.group_id is None:
name = self.pid
def pre_save(group, obj, _state=state):
obj.group_id=group.pk
_state['saved'] = True
self.group, _ = Group.objects.get_or_create_unique(
parent=self, name=self.pid,
defaults={'description': self.name},
pre_parent_save=pre_save,
)
if not state['saved']:
# save the object
super(Project, self).save(*args, **kwargs)
if created:
log.add(self, 'Created')
for field in self._tracker.changed():
log.change(self, 'Changed %s to %s'
% (field, getattr(self, field)))
# has pid changed?
self._tracker.has_changed("pid")
if self._tracker.has_changed("pid"):
old_pid = self._tracker.previous('pid')
if old_pid is not None:
from karaage.datastores import machine_category_set_project_pid
machine_category_set_project_pid(self, old_pid, self.pid)
log.change(self, 'Renamed %s to %s' % (old_pid, self.pid))
# update the datastore
from karaage.datastores import machine_category_save_project
machine_category_save_project(self)
# has group changed?
if self._tracker.has_changed("group_id"):
old_group_pk = self._tracker.previous("group_id")
new_group = self.group
if old_group_pk is not None:
old_group = Group.objects.get(pk=old_group_pk)
from karaage.datastores import remove_accounts_from_project
query = Account.objects.filter(person__groups=old_group)
remove_accounts_from_project(query, self)
if new_group is not None:
from karaage.datastores import add_accounts_to_project
query = Account.objects.filter(person__groups=new_group)
add_accounts_to_project(query, self)
def _remove_person_from_group(person, group):
""" Call datastores after removing a person from a group. """
from karaage.datastores import remove_accounts_from_group
from karaage.datastores import remove_accounts_from_project
from karaage.datastores import remove_accounts_from_institute
a_list = person.account_set
remove_accounts_from_group(a_list, group)
for project in group.project_set.all():
remove_accounts_from_project(a_list, project)
for institute in group.institute_set.all():
remove_accounts_from_institute(a_list, institute)
def _remove_person_from_group(person, group):
""" Call datastores after removing a person from a group. """
from karaage.datastores import remove_accounts_from_group
from karaage.datastores import remove_accounts_from_project
from karaage.datastores import remove_accounts_from_institute
a_list = person.account_set
remove_accounts_from_group(a_list, group)
for project in group.project_set.all():
remove_accounts_from_project(a_list, project)
for institute in group.institute_set.all():
remove_accounts_from_institute(a_list, institute)
def delete(self, *args, **kwargs):
log.delete(
self.project,
'Quota %s: Deleted' % self.machine_category)
from karaage.datastores import remove_accounts_from_project
query = Account.objects.filter(person__groups=self.project.group)
remove_accounts_from_project(query, self.project)
super(ProjectQuota, self).delete(*args, **kwargs)
from karaage.datastores import machine_category_delete_project
machine_category_delete_project(self.project, self.machine_category)
def save(self, *args, **kwargs):
created = self.pk is None
# set group if not already set
if self.group_id is None:
name = self.pid
self.group, _ = Group.objects.get_or_create(name=name)
# save the object
super(Project, self).save(*args, **kwargs)
if created:
log.add(self, 'Created')
for field in self._tracker.changed():
log.change(self, 'Changed %s to %s'
% (field, getattr(self, field)))
# has pid changed?
self._tracker.has_changed("pid")
if self._tracker.has_changed("pid"):
old_pid = self._tracker.previous('pid')
if old_pid is not None:
from karaage.datastores import set_project_pid
set_project_pid(self, old_pid, self.pid)
log.change(self, 'Renamed %s to %s' % (old_pid, self.pid))
# update the datastore
from karaage.datastores import save_project
save_project(self)
# has group changed?
if self._tracker.has_changed("group_id"):
old_group_pk = self._tracker.previous("group_id")
new_group = self.group
if old_group_pk is not None:
old_group = Group.objects.get(pk=old_group_pk)
from karaage.datastores import remove_accounts_from_project
query = Account.objects.filter(person__groups=old_group)
remove_accounts_from_project(query, self)
if new_group is not None:
from karaage.datastores import add_accounts_to_project
query = Account.objects.filter(person__groups=new_group)
add_accounts_to_project(query, self)
def save(self, *args, **kwargs):
created = self.pk is None
# set group if not already set
if self.group_id is None:
name = self.pid
self.group, _ = Group.objects.get_or_create(name=name)
# save the object
super(Project, self).save(*args, **kwargs)
if created:
log.add(self, 'Created')
for field in self._tracker.changed():
log.change(self,
'Changed %s to %s' % (field, getattr(self, field)))
# has pid changed?
self._tracker.has_changed("pid")
if self._tracker.has_changed("pid"):
old_pid = self._tracker.previous('pid')
if old_pid is not None:
from karaage.datastores import set_project_pid
set_project_pid(self, old_pid, self.pid)
log.change(self, 'Renamed %s to %s' % (old_pid, self.pid))
# update the datastore
from karaage.datastores import save_project
save_project(self)
# has group changed?
if self._tracker.has_changed("group_id"):
old_group_pk = self._tracker.previous("group_id")
new_group = self.group
if old_group_pk is not None:
old_group = Group.objects.get(pk=old_group_pk)
from karaage.datastores import remove_accounts_from_project
query = Account.objects.filter(person__groups=old_group)
remove_accounts_from_project(query, self)
if new_group is not None:
from karaage.datastores import add_accounts_to_project
query = Account.objects.filter(person__groups=new_group)
add_accounts_to_project(query, self)
def delete(self, *args, **kwargs):
# ensure nothing connects to this project
query = Account.objects.filter(default_project=self)
query.update(default_project=None)
# delete the object
log.delete(self, 'Deleted')
super(Project, self).delete(*args, **kwargs)
# update datastore associations
old_group_pk = self._tracker.previous("group_id")
if old_group_pk is not None:
old_group = Group.objects.get(pk=old_group_pk)
from karaage.datastores import remove_accounts_from_project
query = Account.objects.filter(person__groups=old_group)
remove_accounts_from_project(query, self)
# update the datastore
from karaage.datastores import machine_category_delete_project
machine_category_delete_project(self)
