diff --git a/ivatar/views.py b/ivatar/views.py index 9d8de33..6629e93 100644 --- a/ivatar/views.py +++ b/ivatar/views.py @@ -133,8 +133,11 @@ class AvatarImageView(TemplateView): content_type='image/png') if str(default) == 'robohash': + roboset = 'any' + if request.GET.get('robohash'): + roboset = request.GET.get('robohash') robohash = Robohash(kwargs['digest']) - robohash.assemble(roboset='any', sizex=size, sizey=size) + robohash.assemble(roboset=roboset, sizex=size, sizey=size) data = BytesIO() robohash.img.save(data, format='png') data.seek(0) diff --git a/requirements.txt b/requirements.txt index aec5e22..f23c634 100644 --- a/requirements.txt +++ b/requirements.txt @@ -34,4 +34,4 @@ psycopg2 notsetuptools git+https://github.com/ofalk/monsterid.git git+https://github.com/azaghal/pydenticon.git -robohash +git+https://github.com/e1ven/Robohash.git