def format(world, raw_data):
"""Deserialize a player character saved in ShinyFormat and adds it to
the world.
raw_data - the data to be deserialized into a player object.
world - The World instance
"""
pc = json.loads(_match_shiny_tag('Player', raw_data))
items = json.loads(_match_shiny_tag('Inv Items', raw_data))
# Build the area from the assembled dictionary data
try:
new_pc = Player(('foo', 'bar'))
new_pc.playerize(pc)
new_pc.save()
# Inventory time!
containers = {} # old_container_dbid : [new_containee1, ...]
old_new = {} # old dbid's mapped to their new ones
for item in items:
my_container = item[0].get('container')
old_dbid = item[0]['dbid']
del item[0]['dbid']
if item[0].get('owner'):
item[0]['owner'] = new_pc.dbid
else:
del item[0]['container']
i = GameItem(item[0])
i.save()
load_item_types(i, item[1])
old_new[old_dbid] = i.dbid
if my_container:
if containers.get(my_container):
containers[my_container].append(i)
else:
containers[my_container] = [i]
for old_container_dbid, containees_list in containers.items():
for containee in containees_list:
containee.container = old_new.get(old_container_dbid)
containee.save()
except Exception as e:
# if anything went wrong, make sure we destroy any leftover character
# data. This way, we won't run into problems if they try to import it
# again, and we won't leave orphaned or erroneous data in the db.
world.log.error(traceback.format_exc())
try:
new_pc.destruct()
except:
# if something goes wrong destroying the pc, it probably means we
# didn't get far enough to have anything to destroy. Just ignore any
# errors.
pass
raise SportError('There was a horrible error on import! '
'Aborting! Check logfile for details.')
return 'Character "%s" has been successfully imported.' % new_pc.fancy_name(
)
def test_shiny_player_format(self):
from shinymud.lib.sport_plugins.formatters.player_write_shiny_format import format as writeshiny
from shinymud.lib.sport_plugins.formatters.player_read_shiny_format import format as readshiny
from shinymud.models.player import Player
#create a playa
sven = Player(('foo', 'bar'))
sven.playerize({'name': 'sven', 'password': '******'})
sven.permissions = 17
sven.description = "I'm pretty adorable."
sven.title = 'Super Sven'
sven.save()
area = self._create_area()
sven.item_add(area.get_item('1').load())
txt = writeshiny(sven)
self.world.log.debug(txt)
sven.destruct()
# Sven should have been taken out of the database...
row = self.world.db.select('* from player where name=?', ['sven'])
self.assertFalse(row)
row = self.world.db.select('* from game_item where owner=?',
[sven.dbid])
self.assertFalse(row)
result = readshiny(self.world, txt)
self.world.log.debug(result)
self.assertEqual(result,
'Character "Sven" has been successfully imported.')
# Sven should now be in the database, but not online
row = self.world.db.select('* from player where name=?', ['sven'])[0]
self.assertTrue(row)
self.assertFalse(self.world.get_player('sven'))
isven = Player(('foo', 'bar'))
isven.playerize(row)
row = self.world.db.select('* from game_item where owner=?',
[isven.dbid])
self.assertTrue(row)
# Make sure that all attributes we set got imported correctly
self.assertEqual(sven.password, isven.password)
self.assertEqual(sven.description, isven.description)
self.assertEqual(sven.name, isven.name)
self.assertEqual(sven.title, isven.title)
self.assertEqual(sven.permissions, isven.permissions)
# Make sure that the inventory was correctly loaded
self.assertEqual(len(sven.inventory), len(isven.inventory))
item = isven.inventory[0]
self.world.log.debug(item.create_save_dict())
self.world.log.debug(item.item_types)
self.assertFalse(sven.inventory[0] is isven.inventory[0])
self.assertEqual(item.name, 'chair')
self.assertTrue(item.has_type('furniture'))
self.assertEqual(item.item_types['furniture'].capacity, 5)
def format(world, raw_data):
"""Deserialize a player character saved in ShinyFormat and adds it to
the world.
raw_data - the data to be deserialized into a player object.
world - The World instance
"""
pc = json.loads(_match_shiny_tag('Player', raw_data))
items = json.loads(_match_shiny_tag('Inv Items', raw_data))
# Build the area from the assembled dictionary data
try:
new_pc = Player(('foo', 'bar'))
new_pc.playerize(pc)
new_pc.save()
# Inventory time!
containers = {} # old_container_dbid : [new_containee1, ...]
old_new = {} # old dbid's mapped to their new ones
for item in items:
my_container = item[0].get('container')
old_dbid = item[0]['dbid']
del item[0]['dbid']
if item[0].get('owner'):
item[0]['owner'] = new_pc.dbid
else:
del item[0]['container']
i = GameItem(item[0])
i.save()
load_item_types(i, item[1])
old_new[old_dbid] = i.dbid
if my_container:
if containers.get(my_container):
containers[my_container].append(i)
else:
containers[my_container] = [i]
for old_container_dbid, containees_list in containers.items():
for containee in containees_list:
containee.container = old_new.get(old_container_dbid)
containee.save()
except Exception as e:
# if anything went wrong, make sure we destroy any leftover character
# data. This way, we won't run into problems if they try to import it
# again, and we won't leave orphaned or erroneous data in the db.
world.log.error(traceback.format_exc())
try:
new_pc.destruct()
except:
# if something goes wrong destroying the pc, it probably means we
# didn't get far enough to have anything to destroy. Just ignore any
# errors.
pass
raise SportError('There was a horrible error on import! '
'Aborting! Check logfile for details.')
return 'Character "%s" has been successfully imported.' % new_pc.fancy_name()
def test_shiny_player_format_containers(self):
"""Make sure shiny player format still works when player has containers
in their inventory.
"""
from shinymud.lib.sport_plugins.formatters.player_write_shiny_format import format as writeshiny
from shinymud.lib.sport_plugins.formatters.player_read_shiny_format import format as readshiny
from shinymud.lib.connection_handlers.shiny_connections import ShinyConnection
from shinymud.models.player import Player
sven = Player(('foo', 'bar'))
sven.playerize({'name': 'sven', 'password': '******'})
sven.save()
area = self._create_area()
item1 = area.new_item()
item1.build_set_keywords('item1')
item1.build_add_type('container')
game1 = item1.load()
item2 = area.new_item()
item2.build_set_keywords('item2')
item2.build_add_type('container')
game2 = item2.load()
self.assertTrue(game1.item_types['container'].item_add(game2))
item3 = area.new_item()
item3.build_set_keywords('item3')
game3 = item3.load()
self.assertTrue(game2.item_types['container'].item_add(game3))
sven.item_add(game1)
txt = writeshiny(sven)
self.world.log.debug(txt)
sven.destruct()
row = self.world.db.select('* from player where name=?', ['sven'])
self.assertFalse(row)
row = self.world.db.select('* from game_item where owner=?',
[sven.dbid])
self.assertFalse(row)
result = readshiny(self.world, txt)
self.world.log.debug(result)
self.assertEqual(result,
'Character "Sven" has been successfully imported.')
row = self.world.db.select('* from player where name=?', ['sven'])[0]
isven = Player(('foo', 'bar'))
isven.playerize(row)
self.assertEqual(len(isven.inventory), 1)
s1 = isven.inventory[0]
self.assertEqual(s1.keywords, ['item1'])
s2 = s1.item_types['container'].get_item_by_kw('item2')
self.assertEqual(s2.keywords, ['item2'])
s3 = s2.item_types['container'].get_item_by_kw('item3')
self.assertEqual(s3.keywords, ['item3'])
def test_shiny_player_format(self):
from shinymud.lib.sport_plugins.formatters.player_write_shiny_format import format as writeshiny
from shinymud.lib.sport_plugins.formatters.player_read_shiny_format import format as readshiny
from shinymud.models.player import Player
#create a playa
sven = Player(('foo', 'bar'))
sven.playerize({'name': 'sven', 'password': '******'})
sven.permissions = 17
sven.description = "I'm pretty adorable."
sven.title = 'Super Sven'
sven.save()
area = self._create_area()
sven.item_add(area.get_item('1').load())
txt = writeshiny(sven)
self.world.log.debug(txt)
sven.destruct()
# Sven should have been taken out of the database...
row = self.world.db.select('* from player where name=?', ['sven'])
self.assertFalse(row)
row = self.world.db.select('* from game_item where owner=?', [sven.dbid])
self.assertFalse(row)
result = readshiny(self.world, txt)
self.world.log.debug(result)
self.assertEqual(result, 'Character "Sven" has been successfully imported.')
# Sven should now be in the database, but not online
row = self.world.db.select('* from player where name=?', ['sven'])[0]
self.assertTrue(row)
self.assertFalse(self.world.get_player('sven'))
isven = Player(('foo', 'bar'))
isven.playerize(row)
row = self.world.db.select('* from game_item where owner=?', [isven.dbid])
self.assertTrue(row)
# Make sure that all attributes we set got imported correctly
self.assertEqual(sven.password, isven.password)
self.assertEqual(sven.description, isven.description)
self.assertEqual(sven.name, isven.name)
self.assertEqual(sven.title, isven.title)
self.assertEqual(sven.permissions, isven.permissions)
# Make sure that the inventory was correctly loaded
self.assertEqual(len(sven.inventory), len(isven.inventory))
item = isven.inventory[0]
self.world.log.debug(item.create_save_dict())
self.world.log.debug(item.item_types)
self.assertFalse(sven.inventory[0] is isven.inventory[0])
self.assertEqual(item.name, 'chair')
self.assertTrue(item.has_type('furniture'))
self.assertEqual(item.item_types['furniture'].capacity, 5)
def test_shiny_player_format_containers(self):
"""Make sure shiny player format still works when player has containers
in their inventory.
"""
from shinymud.lib.sport_plugins.formatters.player_write_shiny_format import format as writeshiny
from shinymud.lib.sport_plugins.formatters.player_read_shiny_format import format as readshiny
from shinymud.lib.connection_handlers.shiny_connections import ShinyConnection
from shinymud.models.player import Player
sven = Player(('foo', 'bar'))
sven.playerize({'name': 'sven', 'password': '******'})
sven.save()
area = self._create_area()
item1 = area.new_item()
item1.build_set_keywords('item1')
item1.build_add_type('container')
game1 = item1.load()
item2 = area.new_item()
item2.build_set_keywords('item2')
item2.build_add_type('container')
game2 = item2.load()
self.assertTrue(game1.item_types['container'].item_add(game2))
item3 = area.new_item()
item3.build_set_keywords('item3')
game3 = item3.load()
self.assertTrue(game2.item_types['container'].item_add(game3))
sven.item_add(game1)
txt = writeshiny(sven)
self.world.log.debug(txt)
sven.destruct()
row = self.world.db.select('* from player where name=?', ['sven'])
self.assertFalse(row)
row = self.world.db.select('* from game_item where owner=?', [sven.dbid])
self.assertFalse(row)
result = readshiny(self.world, txt)
self.world.log.debug(result)
self.assertEqual(result, 'Character "Sven" has been successfully imported.')
row = self.world.db.select('* from player where name=?', ['sven'])[0]
isven = Player(('foo', 'bar'))
isven.playerize(row)
self.assertEqual(len(isven.inventory), 1)
s1 = isven.inventory[0]
self.assertEqual(s1.keywords, ['item1'])
s2 = s1.item_types['container'].get_item_by_kw('item2')
self.assertEqual(s2.keywords, ['item2'])
s3 = s2.item_types['container'].get_item_by_kw('item3')
self.assertEqual(s3.keywords, ['item3'])
