def testCanonizePilots(self): pilots = self.pm.db_connector.get_session().query(Pilot) for pilot in pilots: pilot.canon_name = xwingmetadata.canonize(pilot.name) self.pm.db_connector.get_session().commit()
ships = xwingmetadata.ships session = myapp.db_connector.get_session() for ship_type in ships.keys(): pilots = ships[ship_type] for pilot in pilots: db_pilot = session.query(Pilot).filter(Pilot.name == pilot['name']).first() if db_pilot is None: # print "New pilot!" sql = "insert into pilot ( name, canon_name, cost, pilot_skill ) values ( '%s','%s', '%d', '%d' );" % \ (pilot['name'], pilot['canon_name'], pilot['cost'], pilot['pilot_skill']) print sql sql = "insert ship_pilot ( ship_type, pilot_id) select '%s', id from pilot where name='%s' and canon_name='%s';" % \ (xwingmetadata.canonize(ship_type), pilot['name'], pilot['canon_name']) print sql else: #print "I've seen you before!" continue upgrades = xwingmetadata.upgrades for upgrade_type in upgrades.keys(): for upgrade in upgrades[upgrade_type]: db_upgrade = session.query(Upgrade).filter(Upgrade.name == upgrade['name']).first() if db_upgrade is None: if upgrade.has_key('type') and upgrade['type'] is not None: sql = "insert into upgrade ( upgrade_type, name, canon_name, cost ) values ( '%s', '%s', '%s', '%d');" % \ (upgrade['type'], upgrade['name'], upgrade['canon_name'], upgrade['cost'])
def testCanonizeUpgrades(self): upgrades = self.pm.db_connector.get_session().query(Upgrade) for upgrade in upgrades: upgrade.canon_name = xwingmetadata.canonize(upgrade.name) self.pm.db_connector.get_session().commit()