newmediamonitoring1/polls/views/task.py

68 lines
2.3 KiB
Python
Raw Normal View History

2020-09-18 06:03:50 +00:00
from django.shortcuts import render
from django.http import HttpResponse, JsonResponse
from django.views.decorators.csrf import csrf_exempt
import datetime
2020-09-20 11:14:54 +00:00
from channels.layers import get_channel_layer
from asgiref.sync import async_to_sync
2020-09-18 06:03:50 +00:00
from polls.decorators import polls_login_required
2020-09-18 07:52:06 +00:00
from polls.models import Task, TaskAddition
2020-09-20 11:14:54 +00:00
from polls.tasks import process_task
2020-09-18 06:03:50 +00:00
from django.core.exceptions import ObjectDoesNotExist
2020-09-20 11:14:54 +00:00
from polls.utils import model_to_dict, queryset_to_list
2020-09-18 06:03:50 +00:00
@csrf_exempt
@polls_login_required
def tasks(request):
if request.method == 'POST':
return HttpResponse(status=405)
id = request.user.id
tasks = Task.objects.filter(created_by__id=id)
results = []
for o in tasks:
result = dict()
result['id'] = o.id
m = []
for g in o.groups:
n = model_to_dict(g, ['id', 'name'])
m.append(n)
result['groups'] = m
result['added'] = o.added.strftime("%Y-%m-%d %H:%M:%S")
results.append(result)
return JsonResponse(results, safe=False)
2020-09-18 07:52:06 +00:00
@csrf_exempt
@polls_login_required
def create_task(request):
if request.method == 'GET':
return HttpResponse(status=405)
2020-09-18 16:11:19 +00:00
user = request.user
2020-09-18 07:52:06 +00:00
content = request.POST.get('content')
if not content:
return JsonResponse({'status': 'error', 'message': '内容不能为空'})
2020-09-18 16:11:19 +00:00
groups = request.POST.getlist('groups', [])
task = Task.objects.create(created_by=user, content=content)
2020-09-18 07:52:06 +00:00
task.add_groups(groups)
url = request.POST.get('url')
file = request.FILES.get('file')
picture = request.FILES.get('picture')
2020-09-20 11:14:54 +00:00
if url:
2020-09-18 16:11:19 +00:00
urlAddtion = TaskAddition.objects.create(
task=task, category=0, url=url)
2020-09-20 11:14:54 +00:00
if file:
2020-09-18 16:11:19 +00:00
fileAddtion = TaskAddition.objects.create(
task=task, category=1, file=file)
2020-09-20 11:14:54 +00:00
if picture:
2020-09-18 16:11:19 +00:00
pictureAddtion = TaskAddition.objects.create(
task=task, category=2, image=picture)
2020-09-20 11:14:54 +00:00
t = model_to_dict(task, ["id", "content"])
gs = queryset_to_list(task.groups.all(), ["id"])
channel_layer = get_channel_layer()
# async_to_sync(channel_layer.send)('test_channel', {'type': 'hello'})
async_to_sync(channel_layer.group_send)("chat", {"type": "update_price", "content":"helo"})
2020-09-18 16:11:19 +00:00
return JsonResponse({'status': 'success'})