# Create your views here.
import json
import os
import subprocess

import git
from django.http import JsonResponse
from django.shortcuts import render

from pt_site.views import pt_spider, logger
from ptools.base import CommonResponse, StatusCodeEnum


def test_import(request):
    if request.method == 'GET':
        return render(request, 'pt_test/test_import.html')
    else:
        data_list = json.loads(request.body).get('user')
        res = pt_spider.parse_ptpp_cookies(data_list)
        if res.code == StatusCodeEnum.OK.code:
            cookies = res.data
            print(cookies)
        else:
            return JsonResponse(res.to_dict(), safe=False)
        message_list = []
        for data in cookies:
            try:
                print(data)
                res = pt_spider.get_uid_and_passkey(data)
                msg = res.msg
                print(msg)
                if res.code == StatusCodeEnum.OK.code:
                    message_list.append({
                        'msg': msg,
                        'tag': 'success'
                    })
                else:
                    # error_messages.append(msg)
                    message_list.append({
                        'msg': msg,
                        'tag': 'error'
                    })
            except Exception as e:
                message = '{} 站点导入失败！{}  \n'.format(data.get('domain'), str(e))
                message_list.append({
                    'msg': message,
                    'tag': 'warning'
                })
        return JsonResponse(CommonResponse.success(data={
            'messages': message_list
        }).to_dict(), safe=False)


def handle_json(request):
    if request.method == 'GET':
        return render(request, 'pt_test/test_json.html')


def test_post(request):
    if request.method == 'GET':
        return render(request, 'pt_test/test_post.html')
    else:
        r = request.POST
        for i, j in r.items():
            print(i, type(i))
            print(j, type(j))
        print(r)
        return JsonResponse(CommonResponse.success(data=r).to_dict(), safe=False)


def test_subprocess(request):
    """
    pip install -r requirements.txt
    python manage.py makemigrations
    python manage.py migrate
    :param request:
    :return:
    """
    update_command = [
        # 'cp db/db.sqlite3 db/db.sqlite3-$(date "+%Y%m%d%H%M%S")',
        'git pull',
        # 'pip install -r requirements.txt',
        # 'python manage.py makemigrations',
        'python manage.py migrate',
    ]
    result = []
    for command in update_command:
        p = subprocess.run(command, shell=True)
        print(p.stderr)
        print(p.args)
        print(p.returncode)
        result.append({
            'command': p.args,
            'res': p.returncode
        })
    # print(result)
    return JsonResponse(CommonResponse.success(data=result).to_dict(), safe=False)