SQLAlchemy - lunghezza massima della colonna

Question

in SQLAlchemy è possibile applicare la lunghezza massima della stringa di valore assegnata alla colonna mappata? Tutto quello che voglio è di generare un'eccezione se un valore di stringa assegnato è più lungo della lunghezza della colonna della tabella corrispondente di tipo STRING.

Grazie

Answer 1

E 'più facile per rinominare solo la colonna mappata e procura attraverso una proprietà:

class Something(Base): 
    ... 
    _foo = Column('foo', String(123)) 

    @property 
    def foo(self): 
     return self._foo 

    @foo.setter 
    def foo(self, value): 
     if len(value) > _foo.type.length: 
      raise Exception("Value too long") 
     self._foo = value 

Si può facilmente scomporre la creazione di proprietà, e anche utilizzare un quadro di convalida generico come FormEncode .

---

Se avete bisogno di una soluzione specifica più SQLAlchemy e non mente utilizzando interfacce specifiche, quindi SQLAlchemy ha un meccanismo di estensione per catturare gli eventi su attributi. Un validatore usando che sarebbe simile a questa:

from sqlalchemy.orm.interfaces import AttributeExtension, InstrumentationManager 
from sqlalchemy.orm import ColumnProperty 

class InstallValidatorListeners(InstrumentationManager): 
    def post_configure_attribute(self, class_, key, inst): 
     """Add validators for any attributes that can be validated.""" 
     prop = inst.prop 
     # Only interested in simple columns, not relations 
     if isinstance(prop, ColumnProperty) and len(prop.columns) == 1: 
      col = prop.columns[0] 
      # if we have string column with a length, install a length validator 
      if isinstance(col.type, String) and col.type.length: 
       inst.impl.extensions.insert(0, LengthValidator(col.type.length)) 

class ValidationError(Exception): 
    pass 

class LengthValidator(AttributeExtension): 
    def __init__(self, max_length): 
     self.max_length = max_length 

    def set(self, state, value, oldvalue, initiator): 
     if len(value) > self.max_length: 
      raise ValidationError("Length %d exceeds allowed %d" % 
           (len(value), self.max_length)) 
     return value 

Si potrebbe quindi utilizzare questa estensione impostando __sa_instrumentation_manager__ = InstallValidatorListeners su qualsiasi classe che si desidera convalidato. Puoi anche impostarlo sulla classe Base se vuoi che venga applicato a tutte le classi derivate da esso.

Answer 2

Ecco una versione aggiornata che si inserisce il sistema caso di versioni più recenti SQLAlchemy:

class InstallValidatorListeners(InstrumentationManager): 
    def post_configure_attribute(self, class_, key, inst): 
     """Add validators for any attributes that can be validated.""" 
     prop = inst.prop 
     # Only interested in simple columns, not relations 
     if isinstance(prop, ColumnProperty) and len(prop.columns) == 1: 
      col = prop.columns[0] 
      if isinstance(col.type, String) and col.type.length: 
       sqlalchemy.event.listen(
        getattr(class_, key), 'set', LengthValidator(col.type.length), retval=True) 


class ValidationError(Exception): 
    pass 


class LengthValidator(AttributeExtension): 
    def __init__(self, max_length): 
     self.max_length = max_length 

    def __call__(self, state, value, oldvalue, initiator): 
     if len(value) > self.max_length: 
      raise ValidationError(
       "Length %d exceeds allowed %d" % (len(value), self.max_length)) 
     return value