def pdf_index(request, report_id): pdf_bool = False; if report_id == "index": report = Report.objects.all() pdf_bool = True; return render_to_pdf('reports/index.html', { 'pagesize':'A4', 'report':report, 'pdf_bool':pdf_bool, } )
def pdf_report(request, report_id): report = get_object_or_404(Report, pk=report_id) # list of vulnerabilities belonging to report vulv_vul = Vulnerability.objects.filter(report_id=report_id) # image control strImg = [] strVul = map(str, vulv_vul) for id in strVul: get_img = Vimage.objects.filter(vul_id=id) if get_img: strImg.append(get_img) vul_count = vulv_vul.count() start_date = report.pub_date start_date.replace('/','-') end_date = report.end_date end_date.replace('/','-') print >> sys.stderr, 'pdf_report ' + end_date # list of cve belonging to vulnerabilities listed cve_list = [] cve_list_high = [] cve_list_medium = [] cve_list_low = [] for vuls in vulv_vul: cve_list.append(int(vuls.vul_id)) for i in cve_list: details = Cve.objects.get(id=i) if details.vul_cvescore > 7: cve_list_high.append(details.vul_cvetype) break; elif details.vul_cvescore > 4: cve_list_medium.append(details.vul_cvetype) break; else: cve_list_low.append(details.vul_cvetype) #start_date.append(report.pub_date.day) #start_date.append(report.pub_date.month) #start_date.append(report.pub_date.year) #end_date.append(report.end_date.day) #end_date.append(report.end_date.month) #end_date.append(report.end_date.year) pdf_bool = True; user = UserProfile.objects.get(user=report.creator_name) reviewer = UserProfile.objects.get(user=report.reviewer_name) print >> sys.stderr, 'pdf_report ' + str(user.user_id) full_dir = '/home/coffeecat/django-areport/areport/reports/' return render_to_pdf('reports/indie_pdf.html', { 'pagesize':'A4', 'report':report, 'pdf_bool':pdf_bool, 'start_date': start_date, 'end_date': end_date, 'user':user, 'reviewer':reviewer, 'vul_count': vul_count, 'cve_list_high': cve_list_high, 'cve_list_medium': cve_list_medium, 'cve_list_low': cve_list_low, 'full_dir': full_dir, 'images': strImg, } )