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

Added some more error catching to the attr migrations.

Browse source
This commit is contained in:
Griatch 2012-02-22 11:40:52 +01:00
parent a923a9a061
commit 440d403327
3 changed files with 41 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/objects/migrations/0009_converting_attributes.py
View file

@ -151,17 +151,21 @@ class Migration(DataMigration):
def forwards(self, orm): def forwards(self, orm):
"Write your forwards methods here." "Write your forwards methods here."
for attr in orm.ObjAttribute.objects.all(): for attr in orm.ObjAttribute.objects.all():
# repack attr into new format, and reimport # repack attr into new format, and reimport
val = pickle.loads(to_str(attr.db_value)) try:
if hasattr(val, '__iter__'): val = pickle.loads(to_str(attr.db_value))
val = ("iter", val) if hasattr(val, '__iter__'):
elif type(val) == PackedDBobject: val = ("iter", val)
val = ("dbobj", val) elif type(val) == PackedDBobject:
else: val = ("dbobj", val)
val = ("simple", val) else:
attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val))))) val = ("simple", val)
attr.save() attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val)))))
attr.save()
except TypeError, RuntimeError:
pass
def backwards(self, orm): def backwards(self, orm):
"Write your backwards methods here." "Write your backwards methods here."

24
src/players/migrations/0008_converting_attributes.py
View file

@ -154,16 +154,20 @@ class Migration(DataMigration):
def forwards(self, orm): def forwards(self, orm):
"Write your forwards methods here." "Write your forwards methods here."
for attr in orm.PlayerAttribute.objects.all(): for attr in orm.PlayerAttribute.objects.all():
# repack attr into new format, and reimport try:
val = pickle.loads(to_str(attr.db_value)) # repack attr into new format, and reimport
if hasattr(val, '__iter__'): val = pickle.loads(to_str(attr.db_value))
val = ("iter", val) if hasattr(val, '__iter__'):
elif type(val) == PackedDBobject: val = ("iter", val)
val = ("dbobj", val) elif type(val) == PackedDBobject:
else: val = ("dbobj", val)
val = ("simple", val) else:
attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val))))) val = ("simple", val)
attr.save() attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val)))))
attr.save()
except TypeError, RuntimeError:
pass
def backwards(self, orm): def backwards(self, orm):
"Write your backwards methods here." "Write your backwards methods here."

25
src/scripts/migrations/0005_converting_attributes.py
View file

@ -152,17 +152,20 @@ class Migration(DataMigration):
def forwards(self, orm): def forwards(self, orm):
"Write your forwards methods here." "Write your forwards methods here."
for attr in orm.ScriptAttribute.objects.all(): for attr in orm.ScriptAttribute.objects.all():
# repack attr into new format, and reimport try:
val = pickle.loads(to_str(attr.db_value)) # repack attr into new format, and reimport
if hasattr(val, '__iter__'): val = pickle.loads(to_str(attr.db_value))
val = ("iter", val) if hasattr(val, '__iter__'):
elif type(val) == PackedDBobject: val = ("iter", val)
val = ("dbobj", val) elif type(val) == PackedDBobject:
else: val = ("dbobj", val)
val = ("simple", val) else:
attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val))))) val = ("simple", val)
attr.save() attr.db_value = to_unicode(pickle.dumps(to_str(to_attr(from_attr(attr, val)))))
attr.save()
except TypeError, RuntimeError:
pass
def backwards(self, orm): def backwards(self, orm):
"Write your backwards methods here." "Write your backwards methods here."
raise RuntimeError raise RuntimeError