newmediamonitoring1/polls/views/media.py

from django.http import HttpResponse, JsonResponse
from django.views.decorators.csrf import csrf_exempt
from django.core.paginator import Paginator

from dashboard.models import Douyin, NewMedia, Qita, Toutiao, Userprofile, Weibo, Weixin
from polls.decorators import polls_login_required
from polls.models import Notice


@csrf_exempt
@polls_login_required
def medias(request):
    if request.method == 'POST':
        return HttpResponse(status=405)
    organization_id = request.GET.get('organization_id')
    new_medias = NewMedia.media_list(organization_id)
    return JsonResponse({'status': 'success', 'message': new_medias}, safe=False)


@csrf_exempt
@polls_login_required
def my_medias(request):
    if request.method == 'POST':
        return HttpResponse(status=405)
    user = request.user
    profile = user.userprofile_set.first()
    organization_id = profile.organization.id
    new_medias = NewMedia.media_list(organization_id)
    return JsonResponse({'status': 'success', 'message': new_medias}, safe=False)


@csrf_exempt
@polls_login_required
def create_media(request):
    profile = request.user.userprofile_set.first()
    o = profile.organization
    if request.method == 'GET':
        return HttpResponse(status=405)
    f = request.FILES.get('file')
    category = request.POST.get('category')
    if not category:
        return JsonResponse({'status': 'error', 'message': '类型错误'})
    code = request.POST.get('code')
    if not code:
        return JsonResponse({'status': 'error', 'message': '名称不能为空'})
    media_id = request.POST.get('media_id')
    if not media_id:
        return JsonResponse({'status': 'error', 'message': '监测ID不能为空'})
    alias = request.POST.get('alias')
    if not alias:
        return JsonResponse({'status': 'error', 'message': '别名不能为空'})
    attention = request.POST.get('attention', '')
    remark = request.POST.get('remark', '')
    model_name = ''
    status = 2 if o.directly == '省直部门' else 1

    if category == 'weixin':
        model_name = 'Weixin'
        instance = Weixin(code=code, weixinid=media_id, alias=alias,
                          image=f, status=status, organization=o, attention=attention, remark=remark)
    elif category == 'weibo':
        model_name = 'Weibo'
        instance = Weibo(code=code, weiboid=media_id, alias=alias,
                         image=f, status=status, organization=o, attention=attention, remark=remark)
    elif category == 'toutiao':
        model_name = 'Toutiao'
        instance = Toutiao(code=code, toutiaoid=media_id, alias=alias,
                           image=f, status=status, organization=o, attention=attention, remark=remark)
    elif category == 'douyin':
        model_name = 'Douyin'
        instance = Douyin(code=code, douyinid=media_id, alias=alias,
                          image=f, status=status, organization=o, attention=attention, remark=remark)
    else:
        model_name = 'Qita'
        type = request.POST.get('type')
        instance = Qita(code=code, qitaid=media_id, alias=alias,
                        image=f, status=status, organization=o, attention=attention, remark=remark)
    instance.save()
    if o.directly == '省直部门':
        admins = Userprofile.level1_admin()
        for a in admins:
            content = '%s创建了新媒体，请审核' % (profile,)
            Notice.create_reply_notice(
                a.user.id, content, 'dashboard', model_name, 'status', instance.id, '3', '0')
    else:
        admins = Userprofile.level2_admin(instance.organization.cities)
        for a in admins:
            content = '%s创建了新媒体，请审核' % (profile,)
            Notice.create_reply_notice(
                a.user.id, content, 'dashboard', model_name, 'status', instance.id, '2', '0')

    return JsonResponse({'status': 'success'})


@csrf_exempt
@polls_login_required
def update_media(request):
    if request.method == 'GET':
        return HttpResponse(status=405)
    category = request.POST.get('category')
    if not category:
        return JsonResponse({'status': 'error', 'message': '参数错误'})
    media_id = request.POST.get('media_id')
    if not media_id:
        return JsonResponse({'status': 'error', 'message': '参数错误'})
    name = request.POST.get('name')
    if not name:
        return JsonResponse({'status': 'error', 'message': '名称不能为空'})
    value = request.POST.get('value')
    if not value:
        return JsonResponse({'status': 'error', 'message': '值不能为空'})
    result = dict()
    if category == 'weixin':
        instance = Weixin.objects.get(pk=media_id)
        result['media_id'] = instance.weixinid
    elif category == 'weibo':
        instance = Weibo.objects.get(pk=media_id)
        result['media_id'] = instance.weiboid
    elif category == 'toutiao':
        instance = Toutiao.objects.get(pk=media_id)
        result['media_id'] = instance.toutiaoid
    elif category == 'douyin':
        instance = Douyin.objects.get(pk=media_id)
        result['media_id'] = instance.douyinid
    else:
        instance = Qita.objects.get(pk=media_id)
        result['media_id'] = instance.qitaid

    setattr(instance, name, value)
    instance.save(update_fields=[name])
    admins = []
    for admin in instance.organization.userprofile_set.all():
        u = dict()
        u['id'] = admin.user.id
        u['name'] = admin.name
        u['phone'] = admin.user.username
        admins.append(u)

    result['id'] = instance.id
    result['code'] = instance.code
    result['alias'] = instance.alias
    result['status'] = instance.status
    result['attention'] = instance.attention
    result['remark'] = instance.remark
    result['organization'] = instance.organization.name
    result['admin'] = admins
    return JsonResponse({'status': 'success',  'message': result})


@csrf_exempt
@polls_login_required
def media_detail(request, type, media_id):
    result = dict()
    if type == 'weixin':
        media = Weixin.objects.get(pk=media_id)
        result['media_id'] = media.weixinid
    elif type == 'weibo':
        media = Weibo.objects.get(pk=media_id)
        result['media_id'] = media.weiboid
    elif type == 'toutiao':
        media = Toutiao.objects.get(pk=media_id)
        result['media_id'] = media.toutiaoid
    elif type == 'douyin':
        media = Douyin.objects.get(pk=media_id)
        result['media_id'] = media.douyinid
    else:
        media = Qita.objects.get(pk=media_id)
        result['media_id'] = media.qitaid

    admins = []
    for admin in media.organization.userprofile_set.all():
        u = dict()
        u['id'] = admin.user.id
        u['name'] = admin.name
        u['phone'] = admin.user.username
        admins.append(u)

    result['id'] = media.id
    result['code'] = media.code
    result['alias'] = media.alias
    result['status'] = media.status
    result['attention'] = media.attention
    result['remark'] = media.remark
    result['organization'] = media.organization.name
    result['admin'] = admins
    return JsonResponse({'status': 'success', 'message': result}, safe=False)


@csrf_exempt
@polls_login_required
def delete_media(request):
    if request.method == 'GET':
        return HttpResponse(status=405)
    profile = request.user.userprofile_set.first()
    o = profile.organization

    category = request.POST.get('category')
    if not category:
        return JsonResponse({'status': 'error', 'message': '参数错误'})
    media_id = request.POST.get('media_id')
    if not media_id:
        return JsonResponse({'status': 'error', 'message': '参数错误'})
    result = dict()
    if category == 'weixin':
        model_name = 'Weixin'
        instance = Weixin.objects.get(pk=media_id)
        result['media_id'] = instance.weixinid
    elif category == 'weibo':
        model_name = 'Weibo'
        instance = Weibo.objects.get(pk=media_id)
        result['media_id'] = instance.weiboid
    elif category == 'toutiao':
        model_name = 'Toutiao'
        instance = Toutiao.objects.get(pk=media_id)
        result['media_id'] = instance.toutiaoid
    elif category == 'douyin':
        model_name = 'Douyin'
        instance = Douyin.objects.get(pk=media_id)
        result['media_id'] = instance.douyinid
    else:
        model_name = 'Qita'
        instance = Qita.objects.get(pk=media_id)
        result['media_id'] = instance.qitaid

    if o.is_direct():
        instance.status = 5
        instance.save()
        managers = Userprofile.level1_admin()
        for a in managers:
            content = '%s申请注销新媒体，请审核' % (profile,)
            Notice.create_reply_notice(
                a.user.id, content, 'dashboard', model_name, 'status', instance.id, '0', '3')
    else:
        instance.status = 4
        instance.save()
        managers = Userprofile.level2_admin(instance.organization.cities)
        for a in managers:
            content = '%s申请注销新媒体，请审核' % (profile,)
            Notice.create_reply_notice(
                a.user.id, content, 'dashboard', model_name, 'status', instance.id, '5', '3')

    admins = []
    for admin in instance.organization.userprofile_set.all():
        u = dict()
        u['id'] = admin.user.id
        u['name'] = admin.name
        u['phone'] = admin.user.username
        admins.append(u)

    result['id'] = instance.id
    result['code'] = instance.code
    result['alias'] = instance.alias
    result['status'] = instance.status
    result['attention'] = instance.attention
    result['remark'] = instance.remark
    result['organization'] = instance.organization.name
    result['admin'] = admins
    return JsonResponse({'status': 'success',  'message': result})