# coding=utf-8 import os from django.http import Http404, StreamingHttpResponse, HttpResponse from django.shortcuts import render from PixivSearch import pixiv from PixivSearch.module import config from PixivSearch.pixiv import get_nodes logger = pixiv.logging.getLogger('file') def search(request): if request.POST.get('word') != '': word = request.POST.get('word') else: word = 'R-18' if request.POST.get('pageSize') != '': pageSize = request.POST.get('pageSize') else: pageSize = 10 if request.POST.get('order') != '': order = request.POST.get('order') else: order = 10 if request.POST.get('type') != '': type = request.POST.get('type') else: type = None try: logger.info("word:%s,pageSize:%d,order:%d,type:%s", word, pageSize, order, type) array = get_nodes([0, word, pageSize, order, type]) if len(array[0]) > 0: if request.POST.get('type') == 'img': nodes = {'imageNodes': array[0], 'msg': array[1], 'download': word + '.zip', 'size': array[2]} else: nodes = {'imageNodes': array[0], 'msg': array[1]} else: nodes = {'tip': '没有返回结果'} return render(request, 'test.html', nodes) except ValueError: raise Http404() def index(request): return render(request, 'test.html', {'tip': '输入参数进行搜索'}) def download(request): # do something... def file_iterator(file_name, chunk_size=512): with open(file_name) as f: while True: c = f.read(chunk_size) if c: yield c else: break fileName = os.path.join(os.path.dirname(__file__), 'static/download/').replace('\\', '/') + request.GET.get( 'fileName').encode('utf-8') response = StreamingHttpResponse(file_iterator(fileName)) response['Content-Type'] = 'application/octet-stream' response['Content-Disposition'] = 'attachment;filename="{0}"'.format(request.GET.get('fileName').encode('utf-8')) return response def haha(request): p=config.param(param_name='123',param_value='456') p.save() return HttpResponse("%s" % (config.param.param_name))