mirror of
https://git.linux-kernel.at/oliver/ivatar.git
synced 2025-11-11 18:56:23 +00:00
Carve out the AvatarCreator code into a separate module
This commit is contained in:
Oliver Falk
196
ivatar/ivataraccount/avatar_creator_views.py
Normal file
196
ivatar/ivataraccount/avatar_creator_views.py
Normal file
@@ -0,0 +1,196 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
View classes for the avatar creator
|
||||||
|
"""
|
||||||
|
from django.utils.decorators import method_decorator
|
||||||
|
from django.contrib.auth.decorators import login_required
|
||||||
|
from django.views.generic.base import View, TemplateView
|
||||||
|
from django.http import HttpResponseRedirect, HttpResponse
|
||||||
|
from django.urls import reverse_lazy
|
||||||
|
from jinja2 import Environment, PackageLoader
|
||||||
|
import py_avataaars as pa
|
||||||
|
|
||||||
|
|
||||||
|
@method_decorator(login_required, name="dispatch")
|
||||||
|
class AvatarCreatorView(TemplateView):
|
||||||
|
"""
|
||||||
|
View class responsible for handling avatar creation
|
||||||
|
"""
|
||||||
|
|
||||||
|
template_name = "avatar_creator.html"
|
||||||
|
|
||||||
|
def get(self, request, *args, **kwargs):
|
||||||
|
"""
|
||||||
|
Handle get for create view
|
||||||
|
"""
|
||||||
|
if request.user:
|
||||||
|
if not request.user.is_authenticated:
|
||||||
|
return HttpResponseRedirect(reverse_lazy("profile"))
|
||||||
|
|
||||||
|
return super().get(self, request, args, kwargs)
|
||||||
|
|
||||||
|
def get_context_data(self, **kwargs):
|
||||||
|
"""
|
||||||
|
Provide additional context data
|
||||||
|
"""
|
||||||
|
context = super().get_context_data(**kwargs)
|
||||||
|
context["SkinColor"] = list(pa.SkinColor)
|
||||||
|
context["HairColor"] = list(pa.HairColor)
|
||||||
|
context["FacialHairType"] = list(pa.FacialHairType)
|
||||||
|
context["TopType"] = list(pa.TopType)
|
||||||
|
context["HatColor"] = list(pa.Color)
|
||||||
|
context["MouthType"] = list(pa.MouthType)
|
||||||
|
context["EyesType"] = list(pa.EyesType)
|
||||||
|
context["EyebrowType"] = list(pa.EyebrowType)
|
||||||
|
context["NoseType"] = list(pa.NoseType)
|
||||||
|
context["AccessoriesType"] = list(pa.AccessoriesType)
|
||||||
|
context["ClotheType"] = list(pa.ClotheType)
|
||||||
|
context["ClotheColor"] = list(pa.Color)
|
||||||
|
context["ClotheGraphicType"] = list(pa.ClotheGraphicType)
|
||||||
|
return context
|
||||||
|
|
||||||
|
|
||||||
|
@method_decorator(login_required, name="dispatch")
|
||||||
|
class AvatarView(View):
|
||||||
|
"""
|
||||||
|
View class responsible for handling avatar view
|
||||||
|
"""
|
||||||
|
|
||||||
|
def get(
|
||||||
|
self, request, *args, **kwargs
|
||||||
|
): # pylint: disable=too-many-locals,too-many-branches,too-many-statements,unused-argument
|
||||||
|
"""
|
||||||
|
Handle get for create view
|
||||||
|
"""
|
||||||
|
output_format = "svg+xml"
|
||||||
|
|
||||||
|
avatar_style = list(pa.AvatarStyle)[0]
|
||||||
|
skin_color = list(pa.SkinColor)[0]
|
||||||
|
hair_color = list(pa.HairColor)[0]
|
||||||
|
facial_hair_type = list(pa.FacialHairType)[0]
|
||||||
|
top_type = pa.TopType.SHORT_HAIR_SHORT_FLAT
|
||||||
|
hat_color = list(pa.Color)[0]
|
||||||
|
mouth_type = list(pa.MouthType)[0]
|
||||||
|
eyes_type = list(pa.EyesType)[0]
|
||||||
|
eyebrow_type = list(pa.EyebrowType)[0]
|
||||||
|
nose_type = list(pa.NoseType)[0]
|
||||||
|
accessories_type = list(pa.AccessoriesType)[0]
|
||||||
|
clothe_type = list(pa.ClotheType)[0]
|
||||||
|
clothe_color = list(pa.Color)[0]
|
||||||
|
clothe_graphic_type = list(pa.ClotheGraphicType)[0]
|
||||||
|
|
||||||
|
if "avatar_style" in request.GET:
|
||||||
|
avatar_style = list(pa.AvatarStyle)[int(request.GET["avatar_style"])]
|
||||||
|
if "skin_color" in request.GET:
|
||||||
|
skin_color = list(pa.SkinColor)[int(request.GET["skin_color"])]
|
||||||
|
if "hair_color" in request.GET:
|
||||||
|
hair_color = list(pa.HairColor)[int(request.GET["hair_color"])]
|
||||||
|
if "facial_hair_type" in request.GET:
|
||||||
|
facial_hair_type = list(pa.FacialHairType)[
|
||||||
|
int(request.GET["facial_hair_type"])
|
||||||
|
]
|
||||||
|
if "facial_hair_color" in request.GET:
|
||||||
|
facial_hair_color = list(pa.HairColor)[
|
||||||
|
int(request.GET["facial_hair_color"])
|
||||||
|
]
|
||||||
|
if "top_type" in request.GET:
|
||||||
|
top_type = list(pa.TopType)[int(request.GET["top_type"])]
|
||||||
|
if "hat_color" in request.GET:
|
||||||
|
hat_color = list(pa.Color)[int(request.GET["hat_color"])]
|
||||||
|
if "mouth_type" in request.GET:
|
||||||
|
mouth_type = list(pa.MouthType)[int(request.GET["mouth_type"])]
|
||||||
|
if "eyes_type" in request.GET:
|
||||||
|
eyes_type = list(pa.EyesType)[int(request.GET["eyes_type"])]
|
||||||
|
if "eyebrow_type" in request.GET:
|
||||||
|
eyebrow_type = list(pa.EyebrowType)[int(request.GET["eyebrow_type"])]
|
||||||
|
if "nose_type" in request.GET:
|
||||||
|
nose_type = list(pa.NoseType)[int(request.GET["nose_type"])]
|
||||||
|
if "accessories_type" in request.GET:
|
||||||
|
accessories_type = list(pa.AccessoriesType)[
|
||||||
|
int(request.GET["accessories_type"])
|
||||||
|
]
|
||||||
|
if "clothe_type" in request.GET:
|
||||||
|
clothe_type = list(pa.ClotheType)[int(request.GET["clothe_type"])]
|
||||||
|
if "clothe_color" in request.GET:
|
||||||
|
clothe_color = list(pa.Color)[int(request.GET["clothe_color"])]
|
||||||
|
if "clothe_graphic_type" in request.GET:
|
||||||
|
clothe_graphic_type = list(pa.ClotheGraphicType)[
|
||||||
|
int(request.GET["clothe_graphic_type"])
|
||||||
|
]
|
||||||
|
if "format" in request.GET:
|
||||||
|
if request.GET["format"] == "png":
|
||||||
|
output_format = request.GET["format"]
|
||||||
|
elif request.GET["format"] in ("svg", "svg+xml"):
|
||||||
|
output_format = "svg+xml"
|
||||||
|
else:
|
||||||
|
print(
|
||||||
|
"Format: '%s' isn't supported" % request.GET["format"]
|
||||||
|
) # pylint: disable=consider-using-f-string
|
||||||
|
|
||||||
|
avatar = pa.PyAvataaar(
|
||||||
|
style=avatar_style,
|
||||||
|
skin_color=skin_color,
|
||||||
|
hair_color=hair_color,
|
||||||
|
facial_hair_type=facial_hair_type,
|
||||||
|
facial_hair_color=facial_hair_color,
|
||||||
|
top_type=top_type,
|
||||||
|
hat_color=hat_color,
|
||||||
|
mouth_type=mouth_type,
|
||||||
|
eye_type=eyes_type,
|
||||||
|
eyebrow_type=eyebrow_type,
|
||||||
|
nose_type=nose_type,
|
||||||
|
accessories_type=accessories_type,
|
||||||
|
clothe_type=clothe_type,
|
||||||
|
clothe_color=clothe_color,
|
||||||
|
clothe_graphic_type=clothe_graphic_type,
|
||||||
|
)
|
||||||
|
if output_format == "png":
|
||||||
|
return HttpResponse(avatar.render_png(), content_type="image/png")
|
||||||
|
|
||||||
|
return HttpResponse(avatar.render_svg(), content_type="image/svg+xml")
|
||||||
|
|
||||||
|
|
||||||
|
@method_decorator(login_required, name="dispatch")
|
||||||
|
class AvatarItemView(View):
|
||||||
|
"""
|
||||||
|
View class responsible for providing access to single avatar items
|
||||||
|
"""
|
||||||
|
|
||||||
|
def get(self, request, *args, **kwargs): # pylint: disable=unused-argument
|
||||||
|
"""
|
||||||
|
Handle get for create view
|
||||||
|
"""
|
||||||
|
item = request.GET["item"]
|
||||||
|
|
||||||
|
env = Environment(
|
||||||
|
loader=PackageLoader("py_avataaars", "templates"),
|
||||||
|
trim_blocks=True,
|
||||||
|
lstrip_blocks=True,
|
||||||
|
autoescape=True,
|
||||||
|
keep_trailing_newline=False,
|
||||||
|
extensions=[],
|
||||||
|
)
|
||||||
|
template = env.get_template(item)
|
||||||
|
|
||||||
|
def uni(attr): # pylint: disable=unused-argument
|
||||||
|
return None
|
||||||
|
|
||||||
|
rendered_template = template.render(
|
||||||
|
unique_id=uni,
|
||||||
|
template_path=pa.PyAvataaar._PyAvataaar__template_path, # pylint: disable=protected-access
|
||||||
|
template_name=pa.PyAvataaar._PyAvataaar__template_name, # pylint: disable=protected-access
|
||||||
|
facial_hair_type=pa.FacialHairType.DEFAULT,
|
||||||
|
hat_color=pa.Color.BLACK,
|
||||||
|
clothe_color=pa.Color.HEATHER,
|
||||||
|
hair_color=pa.HairColor.BROWN,
|
||||||
|
facial_hair_color=pa.HairColor.BROWN,
|
||||||
|
clothe_graphic_type=pa.ClotheGraphicType.BAT,
|
||||||
|
accessories_type=pa.AccessoriesType.DEFAULT,
|
||||||
|
).replace("\n", "")
|
||||||
|
rendered_template = (
|
||||||
|
'<?xml version="1.0" encoding="utf-8" ?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg width="280px" height="280px" viewBox="-6 0 274 280" version="1.1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">'
|
||||||
|
+ rendered_template
|
||||||
|
+ "</svg>"
|
||||||
|
) # pylint: disable=line-too-long
|
||||||
|
|
||||||
|
return HttpResponse(rendered_template, content_type="image/svg+xml")
|
||||||
@@ -26,7 +26,9 @@ from .views import ResendConfirmationMailView
|
|||||||
from .views import IvatarLoginView
|
from .views import IvatarLoginView
|
||||||
from .views import DeleteAccountView
|
from .views import DeleteAccountView
|
||||||
from .views import ExportView
|
from .views import ExportView
|
||||||
from .views import AvatarCreatorView, AvatarView
|
from .avatar_creator_views import AvatarCreatorView, AvatarView
|
||||||
|
|
||||||
|
# from .avatar_creator_views import AvatarItemView
|
||||||
|
|
||||||
# Define URL patterns, self documenting
|
# Define URL patterns, self documenting
|
||||||
# To see the fancy, colorful evaluation of these use:
|
# To see the fancy, colorful evaluation of these use:
|
||||||
@@ -83,6 +85,8 @@ urlpatterns = [ # pylint: disable=invalid-name
|
|||||||
path("password_set/", PasswordSetView.as_view(), name="password_set"),
|
path("password_set/", PasswordSetView.as_view(), name="password_set"),
|
||||||
path("avatar_creator/", AvatarCreatorView.as_view(), name="avatar_creator"),
|
path("avatar_creator/", AvatarCreatorView.as_view(), name="avatar_creator"),
|
||||||
path("avatar_view/", AvatarView.as_view(), name="avataaar"),
|
path("avatar_view/", AvatarView.as_view(), name="avataaar"),
|
||||||
|
# This is for testing purpose only and shall not be used in production at all
|
||||||
|
# path("avatar_item_view/", AvatarItemView.as_view(), name="avataaar_item"),
|
||||||
re_path(
|
re_path(
|
||||||
r"remove_unconfirmed_openid/(?P<openid_id>\d+)",
|
r"remove_unconfirmed_openid/(?P<openid_id>\d+)",
|
||||||
RemoveUnconfirmedOpenIDView.as_view(),
|
RemoveUnconfirmedOpenIDView.as_view(),
|
||||||
|
|||||||
@@ -40,8 +40,6 @@ from ipware import get_client_ip
|
|||||||
|
|
||||||
from email_validator import validate_email
|
from email_validator import validate_email
|
||||||
|
|
||||||
import py_avataaars as pa
|
|
||||||
|
|
||||||
from libravatar import libravatar_url
|
from libravatar import libravatar_url
|
||||||
from ivatar.settings import (
|
from ivatar.settings import (
|
||||||
MAX_NUM_PHOTOS,
|
MAX_NUM_PHOTOS,
|
||||||
@@ -1250,138 +1248,3 @@ class ExportView(SuccessMessageMixin, TemplateView):
|
|||||||
)
|
)
|
||||||
response.write(bytesobj.read())
|
response.write(bytesobj.read())
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|
||||||
@method_decorator(login_required, name="dispatch")
|
|
||||||
class AvatarCreatorView(TemplateView):
|
|
||||||
"""
|
|
||||||
View class responsible for handling avatar creation
|
|
||||||
"""
|
|
||||||
|
|
||||||
template_name = "avatar_creator.html"
|
|
||||||
|
|
||||||
def get(self, request, *args, **kwargs):
|
|
||||||
"""
|
|
||||||
Handle get for create view
|
|
||||||
"""
|
|
||||||
if request.user:
|
|
||||||
if not request.user.is_authenticated:
|
|
||||||
return HttpResponseRedirect(reverse_lazy("profile"))
|
|
||||||
|
|
||||||
return super().get(self, request, args, kwargs)
|
|
||||||
|
|
||||||
def get_context_data(self, **kwargs):
|
|
||||||
"""
|
|
||||||
Provide additional context data
|
|
||||||
"""
|
|
||||||
context = super().get_context_data(**kwargs)
|
|
||||||
context["SkinColor"] = list(pa.SkinColor)
|
|
||||||
context["HairColor"] = list(pa.HairColor)
|
|
||||||
context["FacialHairType"] = list(pa.FacialHairType)
|
|
||||||
context["TopType"] = list(pa.TopType)
|
|
||||||
context["HatColor"] = list(pa.Color)
|
|
||||||
context["MouthType"] = list(pa.MouthType)
|
|
||||||
context["EyesType"] = list(pa.EyesType)
|
|
||||||
context["EyebrowType"] = list(pa.EyebrowType)
|
|
||||||
context["NoseType"] = list(pa.NoseType)
|
|
||||||
context["AccessoriesType"] = list(pa.AccessoriesType)
|
|
||||||
context["ClotheType"] = list(pa.ClotheType)
|
|
||||||
context["ClotheColor"] = list(pa.Color)
|
|
||||||
context["ClotheGraphicType"] = list(pa.ClotheGraphicType)
|
|
||||||
return context
|
|
||||||
|
|
||||||
|
|
||||||
@method_decorator(login_required, name="dispatch")
|
|
||||||
class AvatarView(View):
|
|
||||||
"""
|
|
||||||
View class responsible for handling avatar view
|
|
||||||
"""
|
|
||||||
|
|
||||||
def get(self, request, *args, **kwargs):
|
|
||||||
"""
|
|
||||||
Handle get for create view
|
|
||||||
"""
|
|
||||||
format = "svg+xml"
|
|
||||||
|
|
||||||
avatar_style = list(pa.AvatarStyle)[0]
|
|
||||||
skin_color = list(pa.SkinColor)[0]
|
|
||||||
hair_color = list(pa.HairColor)[0]
|
|
||||||
facial_hair_type = list(pa.FacialHairType)[0]
|
|
||||||
top_type = pa.TopType.SHORT_HAIR_SHORT_FLAT
|
|
||||||
hat_color = list(pa.Color)[0]
|
|
||||||
mouth_type = list(pa.MouthType)[0]
|
|
||||||
eyes_type = list(pa.EyesType)[0]
|
|
||||||
eyebrow_type = list(pa.EyebrowType)[0]
|
|
||||||
nose_type = list(pa.NoseType)[0]
|
|
||||||
accessories_type = list(pa.AccessoriesType)[0]
|
|
||||||
clothe_type = list(pa.ClotheType)[0]
|
|
||||||
clothe_color = list(pa.Color)[0]
|
|
||||||
clothe_graphic_type = list(pa.ClotheGraphicType)[0]
|
|
||||||
|
|
||||||
if "avatar_style" in request.GET:
|
|
||||||
avatar_style = list(pa.AvatarStyle)[int(request.GET["avatar_style"])]
|
|
||||||
if "skin_color" in request.GET:
|
|
||||||
skin_color = list(pa.SkinColor)[int(request.GET["skin_color"])]
|
|
||||||
if "hair_color" in request.GET:
|
|
||||||
hair_color = list(pa.HairColor)[int(request.GET["hair_color"])]
|
|
||||||
if "facial_hair_type" in request.GET:
|
|
||||||
facial_hair_type = list(pa.FacialHairType)[
|
|
||||||
int(request.GET["facial_hair_type"])
|
|
||||||
]
|
|
||||||
if "facial_hair_color" in request.GET:
|
|
||||||
facial_hair_color = list(pa.HairColor)[
|
|
||||||
int(request.GET["facial_hair_color"])
|
|
||||||
]
|
|
||||||
if "top_type" in request.GET:
|
|
||||||
top_type = list(pa.TopType)[int(request.GET["top_type"])]
|
|
||||||
if "hat_color" in request.GET:
|
|
||||||
hat_color = list(pa.Color)[int(request.GET["hat_color"])]
|
|
||||||
if "mouth_type" in request.GET:
|
|
||||||
mouth_type = list(pa.MouthType)[int(request.GET["mouth_type"])]
|
|
||||||
if "eyes_type" in request.GET:
|
|
||||||
eyes_type = list(pa.EyesType)[int(request.GET["eyes_type"])]
|
|
||||||
if "eyebrow_type" in request.GET:
|
|
||||||
eyebrow_type = list(pa.EyebrowType)[int(request.GET["eyebrow_type"])]
|
|
||||||
if "nose_type" in request.GET:
|
|
||||||
nose_type = list(pa.NoseType)[int(request.GET["nose_type"])]
|
|
||||||
if "accessories_type" in request.GET:
|
|
||||||
accessories_type = list(pa.AccessoriesType)[
|
|
||||||
int(request.GET["accessories_type"])
|
|
||||||
]
|
|
||||||
if "clothe_type" in request.GET:
|
|
||||||
clothe_type = list(pa.ClotheType)[int(request.GET["clothe_type"])]
|
|
||||||
if "clothe_color" in request.GET:
|
|
||||||
clothe_color = list(pa.Color)[int(request.GET["clothe_color"])]
|
|
||||||
if "clothe_graphic_type" in request.GET:
|
|
||||||
clothe_graphic_type = list(pa.ClotheGraphicType)[
|
|
||||||
int(request.GET["clothe_graphic_type"])
|
|
||||||
]
|
|
||||||
if "format" in request.GET:
|
|
||||||
if request.GET["format"] == "png":
|
|
||||||
format = request.GET["format"]
|
|
||||||
elif request.GET["format"] in ("svg", "svg+xml"):
|
|
||||||
format = "svg+xml"
|
|
||||||
else:
|
|
||||||
print("Format: '%s' isn't supported" % request.GET["format"])
|
|
||||||
|
|
||||||
avatar = pa.PyAvataaar(
|
|
||||||
style=avatar_style,
|
|
||||||
skin_color=skin_color,
|
|
||||||
hair_color=hair_color,
|
|
||||||
facial_hair_type=facial_hair_type,
|
|
||||||
facial_hair_color=facial_hair_color,
|
|
||||||
top_type=top_type,
|
|
||||||
hat_color=hat_color,
|
|
||||||
mouth_type=mouth_type,
|
|
||||||
eye_type=eyes_type,
|
|
||||||
eyebrow_type=eyebrow_type,
|
|
||||||
nose_type=nose_type,
|
|
||||||
accessories_type=accessories_type,
|
|
||||||
clothe_type=clothe_type,
|
|
||||||
clothe_color=clothe_color,
|
|
||||||
clothe_graphic_type=clothe_graphic_type,
|
|
||||||
)
|
|
||||||
if format == "png":
|
|
||||||
return HttpResponse(avatar.render_png(), content_type="image/png")
|
|
||||||
|
|
||||||
return HttpResponse(avatar.render_svg(), content_type="image/svg+xml")
|
|
||||||
|
|||||||
Reference in New Issue
Block a user