Skip to content

Commit

Permalink
OpenConceptLab/ocl_issues#1777 | queue should be saved as user submitted
Browse files Browse the repository at this point in the history
  • Loading branch information
snyaggarwal committed Mar 27, 2024
1 parent 05985f3 commit 73eb895
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions core/tasks/models.py
Expand Up @@ -109,10 +109,11 @@ def before_start(cls, task_id, args, kwargs, name=None):
task.save()
else:
return None
parsed_task = task.parse_id() if '~' in task_id else {}
task.created_by = UserProfile.objects.filter(username=kwargs.pop('username', None)).first() or task.created_by
task.name = name or task.name
task.state = STARTED
task.queue = kwargs.get('queue', None) or task.queue or 'default'
task.queue = parsed_task.get('queue', None) or kwargs.get('queue', None) or task.queue or 'default'
task.args = args
task.kwargs = kwargs
task.started_at = timezone.now()
Expand Down Expand Up @@ -214,9 +215,9 @@ def make_new(cls, queue='default', user=None, username=None, import_queue=None,
from core.users.models import UserProfile
user = UserProfile.objects.filter(username=username).first()
username = user.username if user else username
task = cls(
id=cls.generate_user_task_id(username, import_queue or queue or 'default'),
created_by=user, queue=queue or 'default', **kwargs)
queue_name = import_queue or queue or 'default'
task_id = cls.generate_user_task_id(username, queue_name)
task = cls(id=task_id, created_by=user, queue=queue_name, **kwargs)
task.save()
return task

Expand Down

0 comments on commit 73eb895

Please sign in to comment.