363 lines
11 KiB
Python
363 lines
11 KiB
Python
import os
|
|
import base64
|
|
import qrcode
|
|
import logging
|
|
import weasyprint
|
|
import qrcode.image.svg
|
|
|
|
from io import BytesIO
|
|
from pathlib import Path
|
|
from pyhanko.sign import fields, signers
|
|
from pyhanko import stamp
|
|
from pyhanko.pdf_utils import text
|
|
from pyhanko.pdf_utils.incremental_writer import IncrementalPdfFileWriter
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.views.generic.edit import (
|
|
UpdateView,
|
|
CreateView,
|
|
DeleteView,
|
|
FormView
|
|
)
|
|
from django.views.generic.base import TemplateView
|
|
from django.shortcuts import get_object_or_404, redirect
|
|
from django.urls import reverse_lazy
|
|
from django.http import HttpResponse
|
|
from django.contrib import messages
|
|
from idhub.user.forms import (
|
|
ProfileForm,
|
|
RequestCredentialForm,
|
|
DemandAuthorizationForm
|
|
)
|
|
from idhub.mixins import UserView
|
|
from idhub.models import DID, VerificableCredential, Event
|
|
|
|
|
|
class MyProfile(UserView):
|
|
title = _("My profile")
|
|
section = "MyProfile"
|
|
|
|
|
|
class MyWallet(UserView):
|
|
title = _("My wallet")
|
|
section = "MyWallet"
|
|
|
|
|
|
class DashboardView(UserView, TemplateView):
|
|
template_name = "idhub/user/dashboard.html"
|
|
title = _('Dashboard')
|
|
subtitle = _('Events')
|
|
icon = 'bi bi-bell'
|
|
section = "Home"
|
|
|
|
|
|
class ProfileView(MyProfile, UpdateView):
|
|
template_name = "idhub/user/profile.html"
|
|
subtitle = _('My personal data')
|
|
icon = 'bi bi-person-gear'
|
|
from_class = ProfileForm
|
|
fields = ('first_name', 'last_name', 'email')
|
|
success_url = reverse_lazy('idhub:user_profile')
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
def get_form(self):
|
|
form = super().get_form()
|
|
form.fields['first_name'].disabled = True
|
|
form.fields['last_name'].disabled = True
|
|
form.fields['email'].disabled = True
|
|
return form
|
|
|
|
def form_valid(self, form):
|
|
return super().form_valid(form)
|
|
|
|
|
|
class RolesView(MyProfile, TemplateView):
|
|
template_name = "idhub/user/roles.html"
|
|
subtitle = _('My roles')
|
|
icon = 'fa-brands fa-critical-role'
|
|
|
|
|
|
class GDPRView(MyProfile, TemplateView):
|
|
template_name = "idhub/user/gdpr.html"
|
|
subtitle = _('GDPR info')
|
|
icon = 'bi bi-file-earmark-medical'
|
|
|
|
|
|
class CredentialsView(MyWallet, TemplateView):
|
|
template_name = "idhub/user/credentials.html"
|
|
subtitle = _('Credential management')
|
|
icon = 'bi bi-patch-check-fill'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
creds = VerificableCredential.objects.filter(
|
|
user=self.request.user
|
|
)
|
|
context.update({
|
|
'credentials': creds,
|
|
})
|
|
return context
|
|
|
|
|
|
class CredentialView(MyWallet, TemplateView):
|
|
template_name = "idhub/user/credential.html"
|
|
subtitle = _('Credential')
|
|
icon = 'bi bi-patch-check-fill'
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.pk = kwargs['pk']
|
|
self.object = get_object_or_404(
|
|
VerificableCredential,
|
|
pk=self.pk,
|
|
user=self.request.user
|
|
)
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context.update({
|
|
'object': self.object,
|
|
})
|
|
return context
|
|
|
|
|
|
class CredentialJsonView(MyWallet, TemplateView):
|
|
template_name = "certificates/4_Model_Certificat.html"
|
|
subtitle = _('Credential management')
|
|
icon = 'bi bi-patch-check-fill'
|
|
file_name = "certificate.pdf"
|
|
_pss = '123456'
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
# pk = kwargs['pk']
|
|
# self.object = get_object_or_404(
|
|
# VerificableCredential,
|
|
# pk=pk,
|
|
# user=self.request.user
|
|
# )
|
|
# return self.render_to_response(context=self.get_context_data())
|
|
|
|
data = self.build_certificate()
|
|
doc = self.insert_signature(data)
|
|
import pdb; pdb.set_trace()
|
|
response = HttpResponse(doc, content_type="application/pdf")
|
|
response['Content-Disposition'] = 'attachment; filename={}'.format(self.file_name)
|
|
return response
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
this_folder = str(Path.cwd())
|
|
path_img_sig = "idhub/static/images/4_Model_Certificat_html_58d7f7eeb828cf29.jpg"
|
|
img_signature = next(Path.cwd().glob(path_img_sig))
|
|
with open(img_signature, 'rb') as _f:
|
|
img_sig = base64.b64encode(_f.read()).decode('utf-8')
|
|
|
|
path_img_head = "idhub/static/images/4_Model_Certificat_html_7a0214c6fc8f2309.jpg"
|
|
img_header= next(Path.cwd().glob(path_img_head))
|
|
with open(img_header, 'rb') as _f:
|
|
img_head = base64.b64encode(_f.read()).decode('utf-8')
|
|
|
|
qr = self.generate_qr_code("http://localhost")
|
|
context.update({
|
|
# 'object': self.object,
|
|
"image_signature": img_sig,
|
|
"image_header": img_head,
|
|
"qr": qr
|
|
})
|
|
return context
|
|
|
|
def build_certificate(self):
|
|
doc = self.render_to_response(context=self.get_context_data())
|
|
doc.render()
|
|
pdf = weasyprint.HTML(string=doc.content)
|
|
return pdf.write_pdf()
|
|
|
|
def generate_qr_code(self, data):
|
|
qr = qrcode.QRCode(
|
|
version=1,
|
|
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
|
box_size=10,
|
|
border=4,
|
|
)
|
|
qr.add_data(data)
|
|
qr.make(fit=True)
|
|
img_buffer = BytesIO()
|
|
img = qr.make_image(fill_color="black", back_color="white")
|
|
img.save(img_buffer, format="PNG")
|
|
|
|
return base64.b64encode(img_buffer.getvalue()).decode('utf-8')
|
|
|
|
def signer_init(self):
|
|
fname = "examples/signerDNIe004.pfx"
|
|
# pfx_buffer = BytesIO()
|
|
pfx_file= next(Path.cwd().glob(fname))
|
|
s = signers.SimpleSigner.load_pkcs12(
|
|
pfx_file=pfx_file, passphrase=self._pss.encode('utf-8')
|
|
)
|
|
return s
|
|
|
|
def insert_signature(self, doc):
|
|
sig = self.signer_init()
|
|
_buffer = BytesIO()
|
|
_buffer.write(doc)
|
|
w = IncrementalPdfFileWriter(_buffer)
|
|
fields.append_signature_field(
|
|
w, sig_field_spec=fields.SigFieldSpec(
|
|
'Signature', box=(150, 100, 450, 150)
|
|
)
|
|
)
|
|
|
|
meta = signers.PdfSignatureMetadata(field_name='Signature')
|
|
pdf_signer = signers.PdfSigner(
|
|
meta, signer=sig, stamp_style=stamp.QRStampStyle(
|
|
stamp_text='Signed by: %(signer)s\nTime: %(ts)s\nURL: %(url)s',
|
|
text_box_style=text.TextBoxStyle()
|
|
)
|
|
)
|
|
_bf_out = BytesIO()
|
|
url = "https://localhost:8000/"
|
|
pdf_signer.sign_pdf(w, output=_bf_out, appearance_text_params={'url': url})
|
|
return _bf_out.read()
|
|
|
|
|
|
class CredentialJsonView2(MyWallet, TemplateView):
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
pk = kwargs['pk']
|
|
self.object = get_object_or_404(
|
|
VerificableCredential,
|
|
pk=pk,
|
|
user=self.request.user
|
|
)
|
|
response = HttpResponse(self.object.data, content_type="application/json")
|
|
response['Content-Disposition'] = 'attachment; filename={}'.format("credential.json")
|
|
return response
|
|
|
|
|
|
class CredentialsRequestView(MyWallet, FormView):
|
|
template_name = "idhub/user/credentials_request.html"
|
|
subtitle = _('Credential request')
|
|
icon = 'bi bi-patch-check-fill'
|
|
form_class = RequestCredentialForm
|
|
success_url = reverse_lazy('idhub:user_credentials')
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super().get_form_kwargs()
|
|
kwargs['user'] = self.request.user
|
|
return kwargs
|
|
|
|
def form_valid(self, form):
|
|
cred = form.save()
|
|
if cred:
|
|
messages.success(self.request, _("The credential was issued successfully!"))
|
|
Event.set_EV_CREDENTIAL_ISSUED_FOR_USER(cred)
|
|
Event.set_EV_CREDENTIAL_ISSUED(cred)
|
|
url = self.request.session.pop('next_url', None)
|
|
if url:
|
|
return redirect(url)
|
|
else:
|
|
messages.error(self.request, _("The credential does not exist!"))
|
|
return super().form_valid(form)
|
|
|
|
|
|
class DemandAuthorizationView(MyWallet, FormView):
|
|
template_name = "idhub/user/credentials_presentation.html"
|
|
subtitle = _('Credential presentation')
|
|
icon = 'bi bi-patch-check-fill'
|
|
form_class = DemandAuthorizationForm
|
|
success_url = reverse_lazy('idhub:user_demand_authorization')
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super().get_form_kwargs()
|
|
kwargs['user'] = self.request.user
|
|
return kwargs
|
|
|
|
def form_valid(self, form):
|
|
try:
|
|
authorization = form.save()
|
|
except Exception:
|
|
txt = _("Problems connecting with {url}").format(
|
|
url=form.org.response_uri
|
|
)
|
|
messages.error(self.request, txt)
|
|
return super().form_valid(form)
|
|
|
|
if authorization:
|
|
return redirect(authorization)
|
|
else:
|
|
messages.error(self.request, _("Error sending credential!"))
|
|
return super().form_valid(form)
|
|
|
|
|
|
class DidsView(MyWallet, TemplateView):
|
|
template_name = "idhub/user/dids.html"
|
|
subtitle = _('Identities (DIDs)')
|
|
icon = 'bi bi-patch-check-fill'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context.update({
|
|
'dids': self.request.user.dids,
|
|
})
|
|
return context
|
|
|
|
|
|
class DidRegisterView(MyWallet, CreateView):
|
|
template_name = "idhub/user/did_register.html"
|
|
subtitle = _('Add a new Identity (DID)')
|
|
icon = 'bi bi-patch-check-fill'
|
|
wallet = True
|
|
model = DID
|
|
fields = ('label',)
|
|
success_url = reverse_lazy('idhub:user_dids')
|
|
object = None
|
|
|
|
def form_valid(self, form):
|
|
form.instance.user = self.request.user
|
|
form.instance.set_did()
|
|
form.save()
|
|
messages.success(self.request, _('DID created successfully'))
|
|
|
|
Event.set_EV_DID_CREATED(form.instance)
|
|
Event.set_EV_DID_CREATED_BY_USER(form.instance)
|
|
return super().form_valid(form)
|
|
|
|
|
|
class DidEditView(MyWallet, UpdateView):
|
|
template_name = "idhub/user/did_register.html"
|
|
subtitle = _('Identities (DIDs)')
|
|
icon = 'bi bi-patch-check-fill'
|
|
wallet = True
|
|
model = DID
|
|
fields = ('label',)
|
|
success_url = reverse_lazy('idhub:user_dids')
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.pk = kwargs['pk']
|
|
self.object = get_object_or_404(self.model, pk=self.pk)
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
def form_valid(self, form):
|
|
user = form.save()
|
|
messages.success(self.request, _('DID updated successfully'))
|
|
return super().form_valid(form)
|
|
|
|
|
|
class DidDeleteView(MyWallet, DeleteView):
|
|
subtitle = _('Identities (DIDs)')
|
|
icon = 'bi bi-patch-check-fill'
|
|
wallet = True
|
|
model = DID
|
|
success_url = reverse_lazy('idhub:user_dids')
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.pk = kwargs['pk']
|
|
self.object = get_object_or_404(self.model, pk=self.pk)
|
|
Event.set_EV_DID_DELETED(self.object)
|
|
self.object.delete()
|
|
messages.success(self.request, _('DID delete successfully'))
|
|
|
|
return redirect(self.success_url)
|
|
|