~*
*.swp
-*.egg-into
+*.egg-info
metadata.create_all()
+ # Oh, mysql-chan.
+ # TODO try to insert data in preorder so we don't need this hack and won't
+ # break similarly on other engines
+ if 'mysql' in engine_uri:
+ session.execute('SET FOREIGN_KEY_CHECKS = 0')
+
# This is a secret attribute on a secret singleton of a secret class that
# appears to hopefully contain all registered classes as keys.
# There is no other way to accomplish this, as far as I can tell.
print table_name
reader = csv.reader(open("%s/%s.csv" % (dir, table_name), 'rb'), lineterminator='\n')
- columns = [unicode(column) for column in reader.next()]
+ column_names = [unicode(column) for column in reader.next()]
for csvs in reader:
row = table()
- for column, value in zip(columns, csvs):
- value = value.decode('utf-8')
- setattr(row, column, value)
+ for column_name, value in zip(column_names, csvs):
+ if table.__table__.c[column_name].nullable and value == '':
+ # Empty string in a nullable column really means NULL
+ value = None
+ else:
+ # Otherwise, unflatten from bytes
+ value = value.decode('utf-8')
+
+ setattr(row, column_name, value)
session.add(row)
session.commit()
+ # Shouldn't matter since this is usually the end of the program and thus
+ # the connection too, but let's change this back just in case
+ if 'mysql' in engine_uri:
+ session.execute('SET FOREIGN_KEY_CHECKS = 1')
+
def csvexport(engine_uri, dir='.'):
import csv
493,Arceus,,,4,254,,,,32,3200,Alpha,gray,8,,-1,3,255,0,,0,0,0
494,Pokemon494,,,,0,,,,0,0,,,0,,-1,3,255,0,,0,0,0
495,Pokemon495,,,,0,,,,0,0,,,0,,-1,3,255,0,,0,0,0
-496,Deoxys,attack,386,,202,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
-497,Deoxys,defense,386,,202,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
-498,Deoxys,speed,386,,202,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
-499,Wormadam,ground,413,,213,,,,5,65,Bagworm,gray,2,,8,45,159,70,,0,0,0
-500,Wormadam,steel,413,,213,,,,5,65,Bagworm,gray,2,,8,45,159,70,,0,0,0
+496,Deoxys,attack,386,,,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
+497,Deoxys,defense,386,,,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
+498,Deoxys,speed,386,,,,,,17,608,DNA,red,12,rare,-1,3,215,0,,0,0,0
+499,Wormadam,ground,413,,,,,,5,65,Bagworm,gray,2,,8,45,159,70,,0,0,0
+500,Wormadam,steel,413,,,,,,5,65,Bagworm,gray,2,,8,45,159,70,,0,0,0
if 'charset' not in uri:
uri += '?charset=utf8'
- # Tables should be InnoDB, in the event that we're creating them
+ # Tables should be InnoDB, in the event that we're creating them, and
+ # use UTF-8 goddammit!
for table in metadata.tables.values():
table.kwargs['mysql_engine'] = 'InnoDB'
+ table.kwargs['mysql_charset'] = 'utf8'
### Connect
engine = create_engine(uri)
from sqlalchemy import Column, ForeignKey, MetaData, Table
from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import relation
+from sqlalchemy.orm import backref, relation
from sqlalchemy.types import *
from sqlalchemy.databases.mysql import *
forme_name = Column(Unicode(16))
forme_base_pokemon_id = Column(Integer, ForeignKey('pokemon.id'))
generation_id = Column(Integer, ForeignKey('generations.id'))
- evolution_chain_id = Column(Integer, ForeignKey('evolution_chains.id'), nullable=False)
+ evolution_chain_id = Column(Integer, ForeignKey('evolution_chains.id'))
evolution_parent_pokemon_id = Column(Integer, ForeignKey('pokemon.id'))
evolution_method_id = Column(Integer, ForeignKey('evolution_methods.id'))
evolution_parameter = Column(Unicode(32))
class PokemonAbility(TableBase):
__tablename__ = 'pokemon_abilities'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
ability_id = Column(Integer, ForeignKey('abilities.id'), nullable=False)
- slot = Column(Integer, primary_key=True, nullable=False)
+ slot = Column(Integer, primary_key=True, nullable=False, autoincrement=False)
class PokemonDexNumber(TableBase):
__tablename__ = 'pokemon_dex_numbers'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
- generation_id = Column(Integer, ForeignKey('generations.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
+ generation_id = Column(Integer, ForeignKey('generations.id'), primary_key=True, nullable=False, autoincrement=False)
pokedex_number = Column(Integer, nullable=False)
class PokemonEggGroup(TableBase):
__tablename__ = 'pokemon_egg_groups'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
- egg_group_id = Column(Integer, ForeignKey('egg_groups.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
+ egg_group_id = Column(Integer, ForeignKey('egg_groups.id'), primary_key=True, nullable=False, autoincrement=False)
class PokemonFlavorText(TableBase):
__tablename__ = 'pokemon_flavor_text'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
- version_id = Column(Integer, ForeignKey('versions.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
+ version_id = Column(Integer, ForeignKey('versions.id'), primary_key=True, nullable=False, autoincrement=False)
flavor_text = Column(Unicode(255), nullable=False)
class PokemonName(TableBase):
__tablename__ = 'pokemon_names'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
- language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
+ language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False, autoincrement=False)
name = Column(Unicode(16), nullable=False)
class PokemonShape(TableBase):
class PokemonStat(TableBase):
__tablename__ = 'pokemon_stats'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
- stat_id = Column(Integer, ForeignKey('stats.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
+ stat_id = Column(Integer, ForeignKey('stats.id'), primary_key=True, nullable=False, autoincrement=False)
base_stat = Column(Integer, nullable=False)
effort = Column(Integer, nullable=False)
class PokemonType(TableBase):
__tablename__ = 'pokemon_types'
- pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False)
+ pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)
type_id = Column(Integer, ForeignKey('types.id'), nullable=False)
- slot = Column(Integer, primary_key=True, nullable=False)
+ slot = Column(Integer, primary_key=True, nullable=False, autoincrement=False)
class Stat(TableBase):
__tablename__ = 'stats'
class TypeEfficacy(TableBase):
__tablename__ = 'type_efficacy'
- damage_type_id = Column(Integer, ForeignKey('types.id'), primary_key=True, nullable=False)
- target_type_id = Column(Integer, ForeignKey('types.id'), primary_key=True, nullable=False)
+ damage_type_id = Column(Integer, ForeignKey('types.id'), primary_key=True, nullable=False, autoincrement=False)
+ target_type_id = Column(Integer, ForeignKey('types.id'), primary_key=True, nullable=False, autoincrement=False)
damage_factor = Column(Integer, nullable=False)
class Type(TableBase):
order_by=PokemonEggGroup.egg_group_id,
backref='pokemon')
Pokemon.evolution_chain = relation(EvolutionChain, backref='pokemon')
+Pokemon.evolution_method = relation(EvolutionMethod)
+Pokemon.evolution_children = relation(Pokemon, primaryjoin=Pokemon.id==Pokemon.evolution_parent_pokemon_id,
+ backref=backref('evolution_parent',
+ remote_side=[Pokemon.id]))
Pokemon.flavor_text = relation(PokemonFlavorText, backref='pokemon')
Pokemon.foreign_names = relation(PokemonName, backref='pokemon')
Pokemon.generation = relation(Generation, backref='pokemon')
version = '0.1',
packages = find_packages(),
package_data = { '': 'data' },
+ install_requires=['SQLAlchemy>=0.5.1'],
entry_points = {
'console_scripts': [
projects / zzz-pokedex.git / commitdiff
