Advertisement
Shell_Casing

models.py

Jun 29th, 2018
595
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.14 KB | None | 0 0
  1. from blog_app import db, login_manager
  2. from datetime import datetime
  3. from flask_login import UserMixin
  4.  
  5. @login_manager.user_loader
  6. def load_user(user_id):
  7.     return User.query.get(int(user_id))
  8.  
  9.  
  10. class User(db.Model, UserMixin):
  11.     id = db.Column(db.Integer, primary_key=True)
  12.     username = db.Column(db.String(20), unique=True, nullable=False)
  13.     email = db.Column(db.String(120), unique=True, nullable=False)
  14.     image = db.Column(db.String(20), nullable=False, default='default.jpg')
  15.     password = db.Column(db.String(60), nullable=False)
  16.     posts = db.relationship('Post', backref='author', lazy=True)
  17.  
  18.     def __repr__(self):
  19.         return f"User('{self.username}', '{self.email}', '{self.image}'')"
  20.  
  21.  
  22. class Post(db.Model):
  23.     id = db.Column(db.Integer, primary_key=True)
  24.     title = db.Column(db.String(100), nullable=False)
  25.     date_posted = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
  26.     content = db.Column(db.Text(100), nullable=False)
  27.     user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
  28.  
  29.     def __repr__(self):
  30.         return f"User('{self.title}', '{self.date_posted}')"
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement