Initial commit

chart/views.py

@@ -0,0 +1,150 @@
+from django.http import JsonResponse, StreamingHttpResponse, HttpResponse
+from django.shortcuts import render
+from django.utils.safestring import mark_safe
+
+from chart import charts
+from chart.models import PlottingLibrary
+from chart.tools import *
+
+url_ = 'http://127.0.0.1:8000'
+
+
+# 处理页面返回问题 页面重定向
+def out_page(request, page):
+
+    if page in ["index.html", 'about.html', 'tools.html']:
+        return render(request, f'./page/{page}', {"url": url_})
+
+    # 验证访问的网页是否存在 防止eval漏洞
+    temp = page.replace("_", ' ')
+    print(temp.replace('.html', ''))
+    db = PlottingLibrary.objects.get(name=f"{temp.replace('.html', '')}")
+    print(temp)
+    if db is None:
+        return HttpResponse("no")
+    else:
+        a = ChartParameter(db.parameter)
+
+        data = {
+            'show_img': f"/{db.show_img}.png",
+            'name': db.name.replace(' ', '_'),
+            'title': db.name,
+            'Basics': mark_safe(a.html_out_Basics()),
+            'Senior': mark_safe(a.html_out_Senior()),
+            'example_file': db.name.replace(" ", "_"),
+            'js': mark_safe(a.js_out()),
+            'url': url_,
+            "color":mark_safe(a.color_js())
+
+        }
+
+
+
+        return render(request, f'./chart/Basic.html', data)
+
+
+# 处理上传文件 返回颜色列表
+def up_file(request):
+    file = request.FILES.get("file")
+    col_name = request.POST.get("col_name", "color")
+    if col_name == "NO":
+        file_path, file_id = path_out("chart")
+        keep_file(file, file_path)
+
+        data = {"flag": file_id}
+        return JsonResponse(data)
+
+    file_path, file_id = path_out("chart")
+
+    keep_file(file, file_path)
+
+    col_list = pick_col(file_path, col_name)
+
+    data = {"flag": file_id, 'data': [{"title": f"color_{i}", 'name': i, } for i in col_list][:8]}
+    return JsonResponse(data)
+
+
+def download(request):
+    file_name = request.GET.get("name", None)
+    sort = request.GET.get("sort", "chart")
+    if sort == "tools":
+        file_id = request.GET.get("file_id")  # 工作编号
+        name_path = os.path.join(BASE_DIR, "file_keep", "tools", "file", f"{file_id}")
+        in_file = f"out.txt"
+
+    elif file_name is None:
+        file_downlode_type = request.GET.get("file_downlode_type")  # 文件类型
+        file_id = request.GET.get("file_id")  # 工作编号
+        name_path = os.path.join(BASE_DIR, "file_keep", "chart", "img", f"{file_id}.{file_downlode_type}")
+        in_file = f"out.{file_downlode_type}"
+    else:
+        name_path = os.path.join(
+            BASE_DIR, "file_keep", "eg_file", "img", file_name)
+        in_file = file_name
+        print(name_path)
+
+    response = StreamingHttpResponse(readFile(name_path))
+    response['Content-Type'] = 'application/octet-stream'
+
+    response['Content-Disposition'] = 'attachment;filename="{0}"'.format(
+        in_file)
+    return response
+
+
+def readFile(filename, chunk_size=512):
+    with open(filename, 'rb') as f:
+        while True:
+            c = f.read(chunk_size)
+            if c:
+                yield c
+            else:
+                break
+
+
+def chart_run(request, get_chart_sort):
+    chart_name = get_chart_sort.replace("_run", '')
+    file_id = request.GET.get("file_id")
+    k = request.GET.get("k")
+    data = {}
+
+    temp = chart_name.replace("_", ' ')
+    db = PlottingLibrary.objects.get(name=temp)
+
+    if db is None:
+        return HttpResponse("no")
+
+    for i in request.GET.lists():
+
+        if i[0] not in ['_', 'file_id', 'k']:
+            data[i[0]] = i[1][0]
+
+    file_path = path_out("chart", file_id)  # 返回文件路径
+    chart_out_path = path_out("chart", file_id, show=1)  # 返回图片路径
+
+    out_path = path_out("chart", file_id, out=1)  # 返回图片路径
+
+    # file_path 输入文件的路径 file
+    # chart 展示图片的路径
+    # out_path 输出文件的路径
+
+    eval(f'charts.{chart_name}')(file_path[0], f"{chart_out_path}_{k}.png", out_path, **data)
+
+    return JsonResponse({"img": f"/static/img_eg/show_chart/{file_id}_{k}.png"})
+
+
+def tools(request):
+    file = request.FILES.get("file", None)  # 标题
+    mg = request.POST.get('mg', None)
+    data = ""
+    if file is None and mg is None:
+        return JsonResponse({"http": "内容为空"})
+    if file is None:
+        data = 反向互补序列(mg)
+    else:
+        p = file.read()
+        p.decode(encoding='utf-8')
+        data = 反向互补序列(p.decode(encoding='utf-8'))
+
+        file_path, file_id = path_out("tools")
+        keep_file(data, file_path, ui=1)
+    return JsonResponse({"work_id": file_id})