You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
190 lines
6.8 KiB
190 lines
6.8 KiB
import json |
|
|
|
from django.http import JsonResponse |
|
from django.db.models import Q |
|
from rest_framework import viewsets, permissions, status, filters |
|
from rest_framework.filters import OrderingFilter |
|
|
|
from update.models import Cinema |
|
from update.serializers import * |
|
from rest_framework.views import APIView |
|
from rest_framework.response import Response |
|
from rest_framework_extensions.cache.mixins import CacheResponseMixin |
|
from django.utils.decorators import method_decorator |
|
from django.views.decorators.cache import cache_page |
|
from django.views.decorators.csrf import csrf_exempt |
|
from django_filters.rest_framework import DjangoFilterBackend |
|
from django.core.cache import cache |
|
from update.utils.get_version import GetVersion |
|
from rest_framework.decorators import action |
|
from update.utils.git_util import GitUtil, GitDbUtil |
|
from update.utils.cmd_extcute import UpdateCommandUtil |
|
from update.utils.db_compare import DbCompare |
|
from update.utils.client_util import ClientUtil |
|
|
|
|
|
# CacheResponseMixin 一定要放第一位 |
|
class CinemaViewSet(CacheResponseMixin, viewsets.ModelViewSet): |
|
# 接口文档的中文注释 |
|
""" |
|
create: 添加测试影院 |
|
list: 获取测试影院列表 |
|
retrieve: 获取某个影院的信息 |
|
update: 更新某个影院的信息 |
|
delete: 删除指定影院 |
|
""" |
|
queryset = Cinema.objects.all() |
|
serializer_class = CinemaSerializer |
|
# permission_classes = (permissions.IsAuthenticated,) |
|
filter_backends = (filters.SearchFilter, DjangoFilterBackend, OrderingFilter) |
|
ordering_fields = ('ip',) |
|
search_fields = ('sys_ver',) |
|
# filter_fields = ('ip',) |
|
filterset_fields = ('ip',) |
|
GetVersion().main_process() |
|
|
|
@action(methods=['get'], detail=False) |
|
@method_decorator(cache_page(60 * 1)) |
|
def refresh(self, request, *args, **kwargs): |
|
GetVersion().main_process() |
|
queryset = Cinema.objects.all().order_by('ip') |
|
serializer = self.get_serializer(instance=queryset, many=True) |
|
return Response(serializer.data) |
|
|
|
|
|
class CinemaUserViewSet(CacheResponseMixin, viewsets.ModelViewSet): |
|
queryset = CinemaUser.objects.order_by('pinyin').all() |
|
serializer_class = CinemaUserSerializer |
|
|
|
|
|
# class CinemaSearchAPIView(APIView, CacheResponseMixin): |
|
# @method_decorator(cache_page(60 * 5)) |
|
# def get(self, request, *args, **kwargs): |
|
# query_params = request.query_params.dict() |
|
# print(query_params) |
|
# query_data = Cinema.objects.filter( |
|
# Q(ip__contains=query_params.get('ip')) & Q(sys_ver__icontains=query_params.get('version'))) |
|
# serializer = CinemaSerializer(instance=query_data, many=True) |
|
# return Response(serializer.data, status=status.HTTP_200_OK) |
|
|
|
@csrf_exempt |
|
def update_user(request, *args, **kwargs): |
|
cinema_id = request.GET.dict().get('id') |
|
user = json.loads(request.body).get('user', '暂无') |
|
result = Cinema.objects.filter(id=cinema_id).update(user=user) |
|
return JsonResponse({'result': 'success'} if result == 1 else {'result': 'fail'}) |
|
|
|
|
|
def get_operation_cmd(request): |
|
update_cmd = UpdateCommandUtil() |
|
no_sys_cmd = update_cmd.get_no_sys_cmd() |
|
serializer = UpdateCommandSerializer(instance=no_sys_cmd, many=True) |
|
return JsonResponse(serializer.data, safe=False) |
|
|
|
|
|
def get_git_version(request): |
|
git_db_util = GitDbUtil() |
|
git_db_util.set_release_to_db() |
|
git_ver = git_db_util.get_short_version() |
|
serializer = ReleaseSerializer(instance=git_ver, many=True) |
|
return JsonResponse(serializer.data, safe=False) |
|
|
|
|
|
def update_cine(request): |
|
req = request.GET.dict() |
|
cinema_ip = req.get('ip') |
|
|
|
update_status = { |
|
'git': {'result': 'success', 'msg': ''}, |
|
'setup': {'result': 'success', 'msg': ''}, |
|
'db_compare': {'result': 'success', 'msg': ''}, |
|
'teardown': {'result': 'success', 'msg': ''}, |
|
'config': {'result': 'success', 'msg': ''}, |
|
'client': {'result': 'success', 'msg': ''}, |
|
} |
|
|
|
short_release = req.get('version') |
|
cmd_list = json.loads(req.get('cmd')) |
|
print(cinema_ip, short_release, cmd_list) |
|
|
|
# cinema_ip = '172.16.3.88' |
|
# short_release = '2.0.33.0338_Release' |
|
# cmd_list = [10, 11, 12, 13, 14, 15, 16, 17, 19, 20] |
|
|
|
# 获取并写入cine.sql |
|
print('获取并写入cine.sql') |
|
git_util = GitUtil(short_release) |
|
git_result = git_util.handle_create_cine() |
|
# 执行setup |
|
print('执行setup') |
|
update_cmd = UpdateCommandUtil() |
|
update_cmd.exec_cmd_by_type(cinema_ip, 'setup', cmd_list, short_release) |
|
# 数据库对比 |
|
if 19 in cmd_list: |
|
print('数据库对比') |
|
db_compare = DbCompare(cinema_ip, short_release) |
|
db_compare.exec_diff_sql() |
|
# 执行teardown |
|
if 17 in cmd_list: |
|
print('执行teardown') |
|
update_cmd.exec_cmd_by_type(cinema_ip, 'teardown', cmd_list, short_release) |
|
# 传输客户端 |
|
if 20 in cmd_list: |
|
print('传输客户端') |
|
client_release = ClientUtil() |
|
client_release.client_process(cinema_ip, short_release) |
|
return JsonResponse({'result': 'success', 'ip': cinema_ip}) |
|
|
|
|
|
# # 测试用接口 |
|
def write_git_version_to_db(request): |
|
git_util = GitDbUtil() |
|
git_util.set_release_to_db() |
|
return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def write_cine_sql(request): |
|
# git_util = GitUtil('2.0.33.0338_Release') |
|
# git_util.handle_create_cine() |
|
# return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def compare_db_structor(request): |
|
# db_compare = DbCompare('172.16.3.88', '2.0.33.0338_Release') |
|
# db_compare.exec_diff_sql() |
|
# return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def get_cmd(request): |
|
# update_cmd = UpdateCommandUtil() |
|
# update_cmd.connect('172.16.3.112') |
|
# update_cmd.disconnect() |
|
# # cmd = update_cmd.get_all_cmd() |
|
# # cmd_ser = UpdateCommandSerializer(instance=cmd, many=True) |
|
# # return JsonResponse(data=cmd_ser.data, safe=False) |
|
# cmd_list = [10, 12, 13, 14, 15, 17, 19, 20] |
|
# r = update_cmd.get_checked_cmd(cmd_list) |
|
# print(r) |
|
# return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def setup_cmd(request): |
|
# update_cmd = UpdateCommandUtil() |
|
# cmd_list = [10, 11, 12, 13, 14, 15, 16, 17, 19, 20] |
|
# # exec_cmd_data = update_cmd.get_checked_cmd(cmd_list) |
|
# update_cmd.exec_cmd_by_type('172.16.3.88', 'setup', cmd_list, '2.0.33.0338_Release') |
|
# return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def teardown_cmd(request): |
|
# update_cmd = UpdateCommandUtil() |
|
# cmd_list = [10, 11, 12, 13, 14, 15, 16, 17, 19, 20] |
|
# # exec_cmd_data = update_cmd.get_checked_cmd(cmd_list) |
|
# update_cmd.exec_cmd_by_type('172.16.3.88', 'teardown', cmd_list, '2.0.33.0338_Release') |
|
# return JsonResponse({'result': 'success'}) |
|
# |
|
# |
|
# def get_client(request): |
|
# client_release = ClientUtil() |
|
# client_release.client_process('172.16.3.88', '2.0.33.0338_Release') |
|
# return JsonResponse({'result': 'success'})
|
|
|