Flask ORM

플라스크에서 ORM으로 디비 모델을 짜본 예시이다.  

from app import db
from .category_model import Category

class Category(db.Model):
    __tablename__ = 'category'
    category_id = db.Column(db.Integer, unique=True, primary_key = True, nullable=False, foreign_key = (Poll.poll, 'poll_id'))
    korean_food = db.Column(db.Boolean, nullable=False)
    chinese_food = db.Column(db.Boolean, nullable=False)
    japanese_food = db.Column(db.Boolean, nullable=False)
    asian_food = db.Column(db.Boolean, nullable=False)
    western_food = db.Column(db.Boolean, nullable=False)

    def __repr__(self):
        return "<Category(category_id='{}', korean_food='{}',chinese_food='{}', japanese_food='{}', asian_food='{}',western_food='{}')>"\
            .format(self.category_id, self.korean_food, self.chinese_food, self.japanese_food, self.asian_food, self.western_food)

 

플라스 ORM으로 디비에 접근해서 수정하고 업데이트한 예시이다. 

def update_user_log(email):
     try:
         user = User.query.filter_by(email=email).first()
         user.cnt_login += 1
         user.last_login = datetime.utcnow() + timedelta(hours=9)
         db.session.commit()
     except Exception as e:
         print(e)
         abort(500)
     return user