self/sprint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set settings

Browse Source
This commit is contained in:
Egor Matveev
2022-01-21 12:27:53 +03:00
parent f836f37f55
commit 375426dd38
3 changed files with 99 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Main/models/set.py
View File

@@ -17,6 +17,18 @@ class Set(models.Model):
end_time = models.DateTimeField(default=None, null=True)
editors = ArrayField(models.TextField(), default=list)
@property
def start_time_format(self):
if self.start_time is None:
return None
return self.start_time.astimezone(timezone.get_current_timezone()).strftime("%Y-%m-%dT%H:%M")
@property
def end_time_format(self):
if self.end_time is None:
return None
return self.end_time.astimezone(timezone.get_current_timezone()).strftime("%Y-%m-%dT%H:%M")
@property
def available(self):
return (

77
Main/views/SetSettingsView.py
View File

@@ -1,35 +1,86 @@
from django.db.models import Q
import datetime
from typing import Optional
from Main.models import SetTask, Task
from SprintLib.BaseView import BaseView
import pytz
from django.utils import timezone
from Main.models import SetTask, Set
from SprintLib.BaseView import BaseView, AccessError
class SetSettingsView(BaseView):
required_login = True
view_file = "set_settings.html"
endpoint = "admin/set"
current_set: Optional[Set] = None
def pre_handle(self):
self.current_set = self.entities.set
if (
self.request.user != self.current_set.creator
and self.request.user.username not in self.current_set.editors
):
raise AccessError()
def get(self):
self.context['settasks'] = SetTask.objects.filter(set=self.entities.set).order_by('name')
self.context["settasks"] = SetTask.objects.filter(
set=self.current_set
).order_by("name")
self.context["start_time"] = (
self.current_set.start_time_format
if self.current_set.start_time
else timezone.now().strftime("%Y-%m-%dT%H:%M")
)
self.context["end_time"] = (
self.current_set.end_time_format
if self.current_set.end_time
else timezone.now().strftime("%Y-%m-%dT%H:%M")
)
def post_save(self):
for key, value in self.request.POST.items():
if key.startswith('settask_'):
st = SetTask.objects.get(id=key.split('_')[1])
if key.startswith("settask_"):
st = SetTask.objects.get(id=key.split("_")[1])
st.name = value
st.save()
self.entities.set.name = self.request.POST['name']
self.entities.set.save()
return '/admin/set?set_id=' + str(self.entities.set.id)
self.current_set.name = self.request.POST["name"]
self.current_set.save()
return "/admin/set?set_id=" + str(self.current_set.id)
def post_edit(self):
current_tasks = self.entities.set.tasks
task_ids = [task.id for task in current_tasks]
for key, value in self.request.POST.items():
if key.startswith('task_'):
i = int(key.split('_')[1])
if key.startswith("task_"):
i = int(key.split("_")[1])
if i not in task_ids:
SetTask.objects.create(set=self.entities.set, task_id=i)
to_delete = [i for i in task_ids if 'task_' + str(i) not in self.request.POST]
to_delete = [i for i in task_ids if "task_" + str(i) not in self.request.POST]
SetTask.objects.filter(task_id__in=to_delete).delete()
return '/admin/set?set_id=' + str(self.entities.set.id)
return "/admin/set?set_id=" + str(self.entities.set.id)
def post_time(self):
try:
tz = pytz.timezone("Europe/Moscow")
if "start_time_check" in self.request.POST:
self.current_set.start_time = None
else:
self.current_set.start_time = tz.localize(
datetime.datetime.strptime(
self.request.POST["start_time"], "%Y-%m-%dT%H:%M"
)
)
if "end_time_check" in self.request.POST:
self.current_set.end_time = None
else:
self.current_set.end_time = tz.localize(
datetime.datetime.strptime(
self.request.POST["end_time"], "%Y-%m-%dT%H:%M"
)
)
self.current_set.opened = 'opened' in self.request.POST.keys()
self.current_set.public = 'public' in self.request.POST.keys()
except ValueError:
return "/admin/set?set_id=" + str(self.current_set.id)
self.current_set.save()
return "/admin/set?set_id=" + str(self.current_set.id)