Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """Database module, including the SQLAlchemy database object and DB-related utilities."""
- from sqlalchemy.orm import relationship
- from app import db
- # Alias common SQLAlchemy names
- Column = db.Column
- relationship = relationship
- class CRUDMixin(object):
- """Mixin that adds convenience methods for CRUD (create, read, update, delete) operations."""
- @classmethod
- def create(cls, **kwargs):
- """Create a new record and save it the database."""
- instance = cls(**kwargs)
- return instance.save()
- def update(self, commit=True, **kwargs):
- """Update specific fields of a record."""
- for attr, value in kwargs.items():
- setattr(self, attr, value)
- return commit and self.save() or self
- def save(self, commit=True):
- """Save the record."""
- db.session.add(self)
- if commit:
- db.session.commit()
- return self
- def delete(self, commit=True):
- """Remove the record from the database."""
- db.session.delete(self)
- return commit and db.session.commit()
- class Model(CRUDMixin, db.Model):
- """Base model class that includes CRUD convenience methods."""
- __abstract__ = True
- # From Mike Bayer's "Building the app" talk
- # https://speakerdeck.com/zzzeek/building-the-app
- class SurrogatePK(object):
- """A mixin that adds a surrogate integer 'primary key' column named ``id`` to any declarative-mapped class."""
- __table_args__ = {'extend_existing': True}
- id = db.Column(db.Integer, primary_key=True)
- @classmethod
- def get_by_id(cls, record_id):
- """Get record by ID."""
- if any(
- (isinstance(record_id, basestring) and record_id.isdigit(),
- isinstance(record_id, (int, float))),
- ):
- return cls.query.get(int(record_id))
- return None
- def reference_col(tablename, nullable=False, pk_name='id', **kwargs):
- """Column that adds primary key foreign key reference.
- Usage: ::
- category_id = reference_col('category')
- category = relationship('Category', backref='categories')
- """
- return db.Column(
- db.ForeignKey('{0}.{1}'.format(tablename, pk_name)),
- nullable=nullable, **kwargs)
- class User(Model, SurrogatePK):
- __tablename__ = 'user'
- username = Column(db.Unicode(50), nullable=False, unique=True)
- password = Column(db.Unicode(50), nullable=False)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement