flask model层之通用的base_model

from basesite.exc import db

SUCCESS = 1
FAILURE = 0


class BaseModel:

    def save(self):
        try:
            db.session.add(self)
            db.session.commit()
        except Exception as e:
            # print(e)
            db.session.rollback()
            return FAILURE
        else:
            return SUCCESS

    def update(self):
        try:
            db.session.merge(self)
            db.session.commit()
        except Exception as e:
            # print(e)
            db.session.rollback()
            return FAILURE
        else:
            return SUCCESS

    def delete(self):
        try:
            db.session.delete(self)
            db.session.commit()
        except Exception as e:
            # print(e)
            db.session.rollback()
            return FAILURE
        else:
            return SUCCESS

    @staticmethod
    def save_all(model_list):
        try:
            db.session.add_all(model_list)
            db.session.commit()
        except Exception as e:
            # print(e)
            db.session.rollback()
            return FAILURE
        else:
            return SUCCESS

    @staticmethod
    def delete_all(model_list):
        try:
            for model in model_list:
                db.session.delete(model)
            db.session.commit()
        except Exception as e:
            # print(e)
            db.session.rollback()
            return FAILURE
        else:
            return SUCCESS