-
Notifications
You must be signed in to change notification settings - Fork 634
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #287 from djcoin/master
Populate user settings w/ data migration, Django1.7 compatible fix #276
- Loading branch information
Showing
6 changed files
with
119 additions
and
21 deletions.
There are no files selected for viewing
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,50 @@ | ||
# -*- coding: utf-8 -*- | ||
from __future__ import unicode_literals | ||
|
||
from django.contrib.auth import get_user_model | ||
from django.db import models, migrations | ||
|
||
from helpdesk.settings import DEFAULT_USER_SETTINGS | ||
|
||
|
||
def picke_settings(data): | ||
"""Pickling as defined at migration's creation time""" | ||
import cPickle | ||
from helpdesk.lib import b64encode | ||
return b64encode(cPickle.dumps(data)) | ||
|
||
|
||
# https://docs.djangoproject.com/en/1.7/topics/migrations/#data-migrations | ||
def populate_usersettings(apps, schema_editor): | ||
"""Create a UserSettings entry for each existing user. | ||
This will only happen once (at install time, or at upgrade) | ||
when the UserSettings model doesn't already exist.""" | ||
|
||
_User = get_user_model() | ||
User = apps.get_model(_User._meta.app_label, _User._meta.model_name) | ||
|
||
# Import historical version of models | ||
UserSettings = apps.get_model("helpdesk", "UserSettings") | ||
|
||
settings_pickled = picke_settings(DEFAULT_USER_SETTINGS) | ||
|
||
for u in User.objects.all(): | ||
try: | ||
UserSettings.objects.get(user=u) | ||
except UserSettings.DoesNotExist: | ||
UserSettings.objects.create(user=u, settings_pickled=settings_pickled) | ||
|
||
|
||
noop = lambda *args, **kwargs: None | ||
|
||
class Migration(migrations.Migration): | ||
|
||
dependencies = [ | ||
('helpdesk', '0002_socks_proxy'), | ||
] | ||
|
||
operations = [ | ||
migrations.RunPython(populate_usersettings, reverse_code=noop), | ||
] | ||
|
||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
# -*- coding: utf-8 -*- | ||
from __future__ import unicode_literals | ||
|
||
import os | ||
from sys import path | ||
|
||
from django.db import models, migrations | ||
from django.core import serializers | ||
|
||
fixture_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '../fixtures')) | ||
fixture_filename = 'emailtemplate.json' | ||
|
||
def deserialize_fixture(): | ||
fixture_file = os.path.join(fixture_dir, fixture_filename) | ||
|
||
with open(fixture_file, 'rb') as fixture: | ||
return list(serializers.deserialize('json', fixture, ignorenonexistent=True)) | ||
|
||
|
||
def load_fixture(apps, schema_editor): | ||
objects = deserialize_fixture() | ||
|
||
for obj in objects: | ||
obj.save() | ||
|
||
|
||
def unload_fixture(apps, schema_editor): | ||
"Delete all EmailTemplate objects" | ||
|
||
objects = deserialize_fixture() | ||
|
||
EmailTemplate = apps.get_model("helpdesk", "emailtemplate") | ||
EmailTemplate.objects.filter(pk__in=[ obj.object.pk for obj in objects ]).delete() | ||
|
||
|
||
class Migration(migrations.Migration): | ||
|
||
dependencies = [ | ||
('helpdesk', '0003_populate_usersettings'), | ||
] | ||
|
||
operations = [ | ||
migrations.RunPython(load_fixture, reverse_code=unload_fixture), | ||
] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters