dingxin_toolbox_drf/mock/utils/mock_service.py

import requests
from update.models import Cinema
from django.http.response import JsonResponse
from mock.mock_templates import *

ZZ_URL = 'https://zzcs.yinghezhong.com'


def mock(_ip, _api, *args, **kwargs):
    print('mock')
    cinema_obj = Cinema.objects.filter(ip=_ip).first()
    if _api == 'reportTicket':
        mock_data = report_ticket_tmp.get()
        return JsonResponse(mock_data, json_dumps_params={'ensure_ascii': False})
    if _api == 'reportFilmSchedule':
        mock_data = report_film_schedule_tmp.get()
        return JsonResponse(mock_data, json_dumps_params={'ensure_ascii': False})
    if _api == 'uploadScreenSeatInfo':
        mock_data = upload_screen_seat_info_tmp.get()
        return JsonResponse(mock_data, json_dumps_params={'ensure_ascii': False})
    if _api == 'getCinemaInfo':
        mock_data = get_cinema_info_tmp.get(cinema_obj)
        return JsonResponse(mock_data, json_dumps_params={'ensure_ascii': False})


def bypass(_r):
    print('bypass')
    print(_r.method, _r.path)
    print(_r.GET)
    print(_r.POST)
    if _r.method == 'GET':
        req_url = ZZ_URL + _r.path
        req_params = dict(_r.GET)
        print(req_url)
        print(req_params)
        response_dara = requests.get(req_url, params=req_params)
        print(response_dara.json())
        return JsonResponse(response_dara.json(), json_dumps_params={'ensure_ascii': False})
    if _r.method == 'POST':
        req_url = ZZ_URL + _r.path
        req_params = dict(_r.GET)
        req_data = dict(_r.POST)
        print(req_url)
        print(req_params)
        print(req_data)
        response_dara = requests.post(req_url, params=req_params, data=req_data)
        print(response_dara.json())
        return JsonResponse(response_dara.json(), json_dumps_params={'ensure_ascii': False})