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,
}
)
