Annikka/evennia
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed the tag migrations. Didn't need to fool with transactions after all.

Browse source
This commit is contained in:
Kelketek 2013-07-21 17:22:38 -05:00
parent 08ee399ea2
commit 55c0e40309
1 changed files with 14 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/typeclasses/migrations/0004_copy_nicks_to_liteattrs_aliases_to_tags.py
View file

@ -4,10 +4,12 @@ from south.db import db
from south.v2 import DataMigration from south.v2 import DataMigration
from django.db import models, IntegrityError from django.db import models, IntegrityError
class Migration(DataMigration): class Migration(DataMigration):
depends_on = (("objects", "0022_add_db_liteattributes_db_tags"), depends_on = (("objects", "0022_add_db_liteattributes_db_tags"),
("players", "0025_auto__add_db_liteattributes_db_tags")) ("players", "0025_auto__add_db_liteattributes_db_tags"))
def forwards(self, orm): def forwards(self, orm):
"Write your forwards methods here." "Write your forwards methods here."
# Note: Don't use "from appname.models import ModelName". # Note: Don't use "from appname.models import ModelName".
@ -16,47 +18,27 @@ class Migration(DataMigration):
# Each alias and nick is its own case. By default, this function starts # Each alias and nick is its own case. By default, this function starts
# in a transaction, so we'll close that and make our own transactions. # in a transaction, so we'll close that and make our own transactions.
db.commit_transaction()
for alias in orm['objects.Alias'].objects.all(): for alias in orm['objects.Alias'].objects.all():
# convert all Aliases to tags # convert all Aliases to tags
try: try:
db.start_transaction() tag = orm.Tag.objects.get(db_key=alias.db_key, db_category="object_alias")
except orm.Tag.DoesNotExist:
tag = orm.Tag(db_key=alias.db_key, db_category="object_alias", db_data=None) tag = orm.Tag(db_key=alias.db_key, db_category="object_alias", db_data=None)
tag.save() tag.save()
obj = alias.db_obj obj = alias.db_obj
obj.db_tags.add(tag) obj.db_tags.add(tag)
db.commit_transaction()
except IntegrityError:
db.commit_transaction()
print "Tag already exists: %s for %s" % (alias.db_key, alias.db_obj.db_key)
# convert all nicks to LiteAttrs # convert all nicks to LiteAttrs
for nick in orm['objects.ObjectNick'].objects.all(): for nick in orm['objects.ObjectNick'].objects.all():
try:
db.start_transaction()
lattr = orm.LiteAttribute(db_key=nick.db_nick, db_category="object_nick_%s" % nick.db_type, db_data=nick.db_real) lattr = orm.LiteAttribute(db_key=nick.db_nick, db_category="object_nick_%s" % nick.db_type, db_data=nick.db_real)
lattr.save() lattr.save()
obj = nick.db_obj obj = nick.db_obj
obj.db_liteattributes.add(lattr) obj.db_liteattributes.add(lattr)
db.commit_transaction()
except IntegrityError:
db.commit_transaction()
print "Nick already exists: %s for %s" % (nick.db_nick, nick.db_obj.db_key)
for nick in orm['players.PlayerNick'].objects.all(): for nick in orm['players.PlayerNick'].objects.all():
try:
db.start_transaction()
lattr = orm.LiteAttribute(db_key=nick.db_nick, db_category="player_nick_%s" % nick.db_type, db_data=nick.db_real) lattr = orm.LiteAttribute(db_key=nick.db_nick, db_category="player_nick_%s" % nick.db_type, db_data=nick.db_real)
lattr.save() lattr.save()
obj = nick.db_obj obj = nick.db_obj
obj.db_liteattributes.add(lattr) obj.db_liteattributes.add(lattr)
db.commit_transaction()
except IntegrityError:
db.commit_transaction()
print "Nick already exists: %s for %s" % (nick.db_nick, nick.db_obj.db_key)
# South expects the migration to be transaction managed. So start up a
# new transaction for it to close immediately when it exits this function.
db.start_transaction()
def backwards(self, orm): def backwards(self, orm):
"Write your backwards methods here." "Write your backwards methods here."