def ready(self):
# WARNING: AS THIS IS NOT A FUNCTIONAL PROGRAMMING LANGUAGE,
# OPERATIONS MAY HAVE SIDE EFFECTS.
# DO NOT REMOVE THINKING THE IMPORT IS UNUSED.
# noinspection PyUnresolvedReferences
from . import signals, jinja2 # noqa: F401, imported for side effects
from django.contrib.flatpages.models import FlatPage
from django.contrib.flatpages.admin import FlatPageAdmin
from django.contrib import admin
from reversion.admin import VersionAdmin
class FlatPageVersionAdmin(VersionAdmin, FlatPageAdmin):
pass
admin.site.unregister(FlatPage)
admin.site.register(FlatPage, FlatPageVersionAdmin)
from judge.models import Language, Profile
from django.contrib.auth.models import User
try:
lang = Language.get_default_language()
for user in User.objects.filter(profile=None):
# These poor profileless users
profile = Profile(user=user, language=lang)
profile.save()
except DatabaseError:
pass
def register(self, form):
user = super(RegistrationView, self).register(form)
profile, _ = Profile.objects.get_or_create(user=user, defaults={
'language': Language.get_default_language(),
})
cleaned_data = form.cleaned_data
profile.timezone = cleaned_data['timezone']
profile.language = cleaned_data['language']
profile.organizations.add(*cleaned_data['organizations'])
profile.save()
if newsletter_id is not None and cleaned_data['newsletter']:
Subscription(user=user, newsletter_id=newsletter_id, subscribed=True).save()
return user
def make_profile(backend, user, response, is_new=False, *args, **kwargs):
if is_new:
if not hasattr(user, 'profile'):
profile = Profile(user=user)
profile.language = Language.get_default_language()
logger.info('Info from %s: %s', backend.name, response)
profile.save()
form = ProfileForm(instance=profile, user=user)
else:
data = backend.strategy.request_data()
logger.info(data)
form = ProfileForm(data, instance=user.profile, user=user)
if form.is_valid():
with transaction.atomic(), revisions.create_revision():
form.save()
revisions.set_user(user)
revisions.set_comment('Updated on registration')
return
return render(backend.strategy.request, 'registration/profile_creation.html', {
'title': 'Create your profile', 'form': form,
})
