external-repos/ivatar
1
0
Fork 0
mirror of https://git.linux-kernel.at/oliver/ivatar.git synced 2025-11-16 13:08:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

PEP8

Browse Source
This commit is contained in:
Oliver Falk
2018-05-23 10:39:18 +02:00
parent 6819aa7264
commit 6bd4080d58
3 changed files with 54 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
ivatar/ivataraccount/forms.py
View File

@@ -17,6 +17,7 @@ from ipware import get_client_ip
MAX_NUM_UNCONFIRMED_EMAILS_DEFAULT = 5
class AddEmailForm(forms.Form):
'''
Form to handle adding email addresses
@@ -41,12 +42,16 @@ class AddEmailForm(forms.Form):
# Enforce the maximum number of unconfirmed emails a user can have
num_unconfirmed = user.unconfirmedemail_set.count()
max_num_unconfirmed_emails = getattr(settings, 'MAX_NUM_UNCONFIRMED_EMAILS', MAX_NUM_UNCONFIRMED_EMAILS_DEFAULT)
max_num_unconfirmed_emails = getattr(
settings,
'MAX_NUM_UNCONFIRMED_EMAILS',
MAX_NUM_UNCONFIRMED_EMAILS_DEFAULT)
if num_unconfirmed >= max_num_unconfirmed_emails:
return False
# Check whether or not a confirmation email has been sent by this user already
# Check whether or not a confirmation email has been
# sent by this user already
if UnconfirmedEmail.objects.filter(
user=user, email=self.cleaned_data['email']).exists():
return False
@@ -61,15 +66,20 @@ class AddEmailForm(forms.Form):
unconfirmed.user = user
unconfirmed.save()
link = settings.SITE_URL + reverse('confirm_email', kwargs={'verification_key':unconfirmed.verification_key})
email_subject = _('Confirm your email address on %s') % settings.SITE_NAME
link = settings.SITE_URL + \
reverse(
'confirm_email',
kwargs={'verification_key': unconfirmed.verification_key})
email_subject = _('Confirm your email address on %s') % \
settings.SITE_NAME
email_body = render_to_string('email_confirmation.txt', {
'verification_link': link,
'site_name': settings.SITE_NAME,
})
#if settings.DEBUG:
# if settings.DEBUG:
# print('DEBUG: %s' % link)
send_mail(email_subject, email_body, settings.SERVER_EMAIL,
send_mail(
email_subject, email_body, settings.SERVER_EMAIL,
[unconfirmed.email])
return True
@@ -86,14 +96,16 @@ class UploadPhotoForm(forms.Form):
required=True,
error_messages={
'required':
_('We only host "G-rated" images and so this field must be checked.')
_('We only host "G-rated" images and so this field must\
be checked.')
})
can_distribute = forms.BooleanField(
label=_('can be freely copied'),
required=True,
error_messages={
'required':
_('This field must be checked since we need to be able to distribute photos to third parties.')
_('This field must be checked since we need to be able to\
distribute photos to third parties.')
})
def save(self, request, data):
@@ -110,6 +122,7 @@ class UploadPhotoForm(forms.Form):
return None
return photo
class AddOpenIDForm(forms.Form):
'''
Form to handle adding OpenID
@@ -129,7 +142,8 @@ class AddOpenIDForm(forms.Form):
'''
# Lowercase hostname port of the URL
url = urlsplit(self.cleaned_data['openid'])
data = urlunsplit((url.scheme.lower(), url.netloc.lower(), url.path,
data = urlunsplit(
(url.scheme.lower(), url.netloc.lower(), url.path,
url.query, url.fragment))
# TODO: Domain restriction as in libravatar?

16
ivatar/ivataraccount/gravatar.py
View File

@@ -4,14 +4,15 @@ import hashlib
URL_TIMEOUT = 5 # in seconds
def get_photo(email):
'''
Fetch photo from Gravatar, given an email address
'''
hash_object = hashlib.new('md5')
hash_object.update(email.lower().encode('utf-8'))
thumbnail_url = 'https://secure.gravatar.com/avatar/' + hash_object.hexdigest(
) + '?s=80&d=404'
thumbnail_url = 'https://secure.gravatar.com/avatar/' + \
hash_object.hexdigest() + '?s=80&d=404'
image_url = 'https://secure.gravatar.com/avatar/' + hash_object.hexdigest(
) + '?s=512&d=404'
@@ -22,14 +23,19 @@ def get_photo(email):
urlopen(image_url, timeout=URL_TIMEOUT)
except HTTPError as e:
if e.code != 404 and e.code != 503:
print('Gravatar fetch failed with an unexpected %s HTTP error' % # pragma: no cover
print( # pragma: no cover
'Gravatar fetch failed with an unexpected %s HTTP error' %
e.code)
return False
except URLError as e: # pragma: no cover
print('Gravatar fetch failed with URL error: %s' % e.reason) # pragma: no cover
print(
'Gravatar fetch failed with URL error: %s' %
e.reason) # pragma: no cover
return False # pragma: no cover
except SSLError as e: # pragma: no cover
print('Gravatar fetch failed with SSL error: %s' % e.reason) # pragma: no cover
print(
'Gravatar fetch failed with SSL error: %s' %
e.reason) # pragma: no cover
return False # pragma: no cover
return {

12
ivatar/ivataraccount/models.py
View File

@@ -20,7 +20,7 @@ from openid.association import Association as OIDAssociation
from openid.store import nonce as oidnonce
from openid.store.interface import OpenIDStore
from ivatar.settings import MAX_LENGTH_EMAIL
from ivatar.settings import MAX_LENGTH_EMAIL, logger
from .gravatar import get_photo as get_gravatar_photo
@@ -38,7 +38,7 @@ def file_format(image_type):
elif image_type == 'GIF':
return 'gif'
print('Unsupported file format: %s' % image_type)
logger.info('Unsupported file format: %s' % image_type)
return None
@@ -188,7 +188,9 @@ class ConfirmedEmail(BaseAccountModel):
'''
Override save from parent, add digest
'''
self.digest = hashlib.md5(self.email.strip().lower().encode('utf-8')).hexdigest()
self.digest = hashlib.md5(
self.email.strip().lower().encode('utf-8')
).hexdigest()
return super().save(*args, **kwargs)
@@ -254,7 +256,9 @@ class ConfirmedOpenId(BaseAccountModel):
netloc = url.username + ':' + password + '@' + url.hostname
else:
netloc = url.hostname
lowercase_url = urlunsplit((url.scheme.lower(), netloc, url.path, url.query, url.fragment))
lowercase_url = urlunsplit(
(url.scheme.lower(), netloc, url.path, url.query, url.fragment)
)
self.digest = hashlib.sha256(lowercase_url.encode('utf-8')).hexdigest()
return super().save(*args, **kwargs)