veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00
Code Issues Projects Releases Wiki Activity

Support association proxies

Browse source
This commit is contained in:
Petr Viktorin 2012-02-12 23:19:40 +01:00
parent ef3fb2f536
commit 391fd1c1ac
2 changed files with 69 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
pokedex/db/tables.py
View file

@ -1337,8 +1337,21 @@ class PokemonForm(TableBase):
@property @property
def name(self): def name(self):
"""Name of this form: the form_name, if set; otherwise the species name"""
return self.pokemon_name or self.species.name return self.pokemon_name or self.species.name
@staticmethod
def _add_relationships(add_relationship, add_association_proxy, **kwargs):
add_relationship('pokemon', Pokemon,
primaryjoin=PokemonForm.pokemon_id==Pokemon.id,
innerjoin=True, lazy='joined')
add_association_proxy('species', 'pokemon', 'species')
add_relationship('version_group', VersionGroup,
innerjoin=True)
add_relationship('pokeathlon_stats', PokemonFormPokeathlonStat,
order_by=PokemonFormPokeathlonStat.pokeathlon_stat_id,
backref='pokemon_form')
create_translation_table('pokemon_form_names', PokemonForm, 'names', create_translation_table('pokemon_form_names', PokemonForm, 'names',
relation_lazy='joined', relation_lazy='joined',
form_name = Column(Unicode(32), nullable=True, index=True, form_name = Column(Unicode(32), nullable=True, index=True,
@ -1818,7 +1831,19 @@ def add_relationships():
argument.relationship_info[backref_name] = backref_info argument.relationship_info[backref_name] = backref_info
setattr(cls, name, relationship(argument, secondary=secondary, setattr(cls, name, relationship(argument, secondary=secondary,
**kwargs)) **kwargs))
add_relationships(add_relationship=add_relationship) def add_association_proxy(name, target_collection, attr, **kwargs):
cls.relationship_info.setdefault('_order', [])
cls.relationship_info['_order'].append(name)
cls.relationship_info[name] = info = kwargs.pop('info', {})
info.update(kwargs)
info.update(dict(
type='association_proxy',
target_collection=target_collection,
attr=attr))
setattr(cls, name, association_proxy(name, target_collection,
**kwargs))
add_relationships(add_relationship=add_relationship,
add_association_proxy=add_association_proxy)
add_relationships() add_relationships()
Ability.changelog = relationship(AbilityChangelog, Ability.changelog = relationship(AbilityChangelog,

74
pokedex/doc/tabledoc.py
View file

@ -79,8 +79,10 @@ def column_header(c, class_name=None, transl_name=None, show_type=True,
for fk in c.foreign_keys: for fk in c.foreign_keys:
if fk.column in column_to_cls: if fk.column in column_to_cls:
foreign_cls = column_to_cls[fk.column] foreign_cls = column_to_cls[fk.column]
if relation_name: if relation_name and relation_name + '_id' == c.name:
result.append(u'(%s →' % c.name) result.append(u'(%s →' % c.name)
elif relation_name:
result.append(u'(**%s** →' % c.name)
else: else:
result.append(u'(→') result.append(u'(→')
result.append(u':class:`~pokedex.db.tables.%s`.%s)' % ( result.append(u':class:`~pokedex.db.tables.%s`.%s)' % (
@ -168,8 +170,8 @@ def generate_columns(cls, remaining_attrs):
relation_name = c.name[:-3] relation_name = c.name[:-3]
if c.name.endswith('_id') and relation_name in remaining_attrs: if c.name.endswith('_id') and relation_name in remaining_attrs:
relation = getattr(cls, relation_name) relation = getattr(cls, relation_name)
yield (column_header(c, name, yield column_header(c, name,
relation=relation, relation_name=relation_name) + ':') relation=relation, relation_name=relation_name)
remaining_attrs.remove(relation_name) remaining_attrs.remove(relation_name)
else: else:
yield column_header(c, name) + ':' yield column_header(c, name) + ':'
@ -187,7 +189,7 @@ def generate_strings(cls, remaining_attrs):
if c.name in common_columns: if c.name in common_columns:
continue continue
yield column_header(c, cls.__name__, yield column_header(c, cls.__name__,
translation_class.__table__.name) + ':' translation_class.__table__.name)
yield u'' yield u''
yield u' ' + unicode(c.info['description']) yield u' ' + unicode(c.info['description'])
yield u'' yield u''
@ -204,33 +206,43 @@ def generate_relationships(cls, remaining_attrs):
for rel_name, info in infos: for rel_name, info in infos:
if rel_name in remaining_attrs: if rel_name in remaining_attrs:
info = cls.relationship_info.get(rel_name) info = cls.relationship_info.get(rel_name)
yield u'%s.\ **%s**' % (cls.__name__, rel_name) if info['type'] in ('relationship', 'backref'):
class_name = u':class:`~pokedex.db.tables.%s`' % info['argument'].__name__ yield u'%s.\ **%s**' % (cls.__name__, rel_name)
if info.get('uselist', True): class_name = u':class:`~pokedex.db.tables.%s`' % info['argument'].__name__
class_name = u'[%s]' % class_name if info.get('uselist', True):
yield u'(→ %s)' % class_name class_name = u'[%s]' % class_name
if 'description' in info: yield u'(→ %s)' % class_name
yield u'' if 'description' in info:
yield u' ' + unicode(info['description']) yield u''
''' yield u' ' + unicode(info['description'])
if info.get('secondary') is not None: '''
yield u'' if info.get('secondary') is not None:
yield ' Association table: ``%s``' % info['secondary'] yield u''
if 'primaryjoin' in info: yield ' Association table: ``%s``' % info['secondary']
yield u'') if 'primaryjoin' in info:
yield ' Join condition: ``%s``' % info['primaryjoin'] yield u'')
if 'secondaryjoin' in info: yield ' Join condition: ``%s``' % info['primaryjoin']
yield ' , ``%s``' % info['secondaryjoin'] if 'secondaryjoin' in info:
''' yield ' , ``%s``' % info['secondaryjoin']
if 'order_by' in info: '''
yield u'' if 'order_by' in info:
try: yield u''
order = iter(info['order_by']) try:
except TypeError: order = iter(info['order_by'])
order = [info['order_by']] except TypeError:
yield u' ' order = [info['order_by']]
yield ' Ordered by: ' + u', '.join( yield u' '
u'``%s``' % o for o in order) yield ' Ordered by: ' + u', '.join(
u'``%s``' % o for o in order)
elif info['type'] == 'association_proxy':
yield u'%s.\ **%s**:' % (cls.__name__, rel_name)
yield '``{info[attr]}`` of ``self.{info[target_collection]}``'.format(
info=info)
if 'description' in info:
yield u''
yield u' ' + unicode(info['description'])
else:
continue
yield u'' yield u''
remaining_attrs.remove(rel_name) remaining_attrs.remove(rel_name)