#!/usr/bin/env python
# -*- coding: utf-8 -*-
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,String, DateTime,Boolean,desc,asc
from sqlalchemy.orm import sessionmaker
import datetime
# create_engine()返回一個Engine的例項
engine = create_engine("mysql+pymysql://root:
 
[email protected]:3306/news?charset=utf8",
                       encoding='utf8')
# 通過declarative_base()功能建立一個基類
Base = declarative_base()
#  建立會話，繫結例項
Session = sessionmaker(bind=engine)


class News(Base):
    '''構造一個類，對映表news'''
    __tablename__ = 'news'
    id = Column(Integer, primary_key=True
 
 ,autoincrement=True)
    title = Column(String(200), nullable=False)
    content = Column(String(2000), nullable=False)
    created_at = Column(DateTime,nullable=True)
    types = Column(String(10), nullable=False)
    image = Column(String(300), nullable=False)
    author = Column(String(20), nullable=True
 
)
    view_count = Column(Integer,default=0)
    is_valid = Column(Boolean,default=1)


class OrmTest(object):
    '''呼叫類之前，建立會話'''
    def __init__(self):
        self.session = Session()

    # 新增一條資料
    def add_one(self):
        '''新增一條或多條資料'''
        new_obj = News(
            title = '標題',
            content = '內容',
            types = '百家',
            created_at = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        )
        new_obj2 = News(
            title='標題1',
            content='內容1',
            types='百家1',
            created_at=datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        )
        # 新增一條資料
        self.session.add(new_obj)
        # 新增多條資料，使用[]
        self.session.add_all([new_obj,new_obj2])
        #新增後，提交，或者回滾.rollback()
        self.session.commit()
        return new_obj

    # 查詢一條資料,返回一個object
    def get_one(self):
        '''查詢一條或多條資料，返回'''
        return self.session.query(News).get(2)

    # 查詢多條資料,filter,返回一個object list
    def get_more(self):
        return self.session.query(News).filter(
            News.is_valid==True,News.types=='百家',News.author.is_(None)
        ).order_by(desc(News.id)).all()

    # 查詢多條資料,filter_by
    def get_more2(self):
        return self.session.query(News).filter_by(
            is_valid=True,types='百家'
        ).order_by(desc(News.id)).all()

    # 修改一條資料
    def update_date(self, pk):
        '''修改一條資料'''
        obj_list = self.session.query(News).filter(News.id == pk)
        if obj_list:
            for obj in obj_list:
                obj.is_valid = 1
                self.session.add(obj)
            self.session.commit()
            return True
        return False

    # 修改多條資料
    def update_more_date(self,pk):
        '''修改多條資料'''
        data_list = self.session.query(News).filter(News.types==pk)
        if data_list:
            for item in data_list:
                item.is_valid = 1
                self.session.add(item)
            self.session.commit()
            return True
        return False

    # 刪除資料
    def delete_date(self, pk):
        '''刪除資料'''
        data_list = self.session.query(News).filter(News.id < pk)
        n = 0
        if data_list:
            for item in data_list:
                self.session.delete(item)
                n +=1
            self.session.commit()
            return '刪除%s條資料' % n
        return '條件沒有資料要刪除'

if __name__ == '__main__':
    obj = OrmTest()
    rest=obj.get_one()
    print(rest.title)

查詢：

查詢一條資料，可以使用get(id),
查詢多條資料，filter或者filter_by, 不同的是寫法問題，結果是一樣，