growing hard days.

SQLAlchemy Session.executeでinsertしてAUTO INCREMENTでセットされたプライマリキーを取得したい

2017/09/30

PythohのORMライブラリSQLAlchemyでMySQLにインサートしたレコードのプライマリキーを、再検索なしに取得したいと思って調べたらちゃんとレスポンスの中にありました。

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql.schema import Column
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.dialects.mysql.base import (
    VARCHAR,
    INTEGER,
    BIGINT
)


class Employee(Base):
    __tablename__ = 'employee'

    employee_id = Column(BIGINT(unsigned=True), primary_key=True, nullable=True)
    name = Column(VARCHAR(length=45), nullable=False)


url = 'mysql+pymysql://username:password@db_host/db_name?charset=utf8'
engine = create_engine(url,echo=True)
Session = sessionmaker(bind=engine)
session = Session()

employee_tbl = Employee

insert_list = [
    {
        'name': 'chie jyarinko'
    },
    {
        'name': 'hiroshi miyamoto'
    }
]

response = self.session.execute(employee_tbl.__table__.insert(), insert_list)
primary_keys = response.inserted_primary_key

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy.sql.schema import Column

from sqlalchemy import create_engine

from sqlalchemy.orm import sessionmaker

from sqlalchemy.dialects.mysql.base import (

VARCHAR,

INTEGER,

BIGINT

)

class Employee(Base):

__tablename__ = 'employee'

employee_id = Column(BIGINT(unsigned=True), primary_key=True, nullable=True)

name = Column(VARCHAR(length=45), nullable=False)

url = 'mysql+pymysql://username:password@db_host/db_name?charset=utf8'

engine = create_engine(url,echo=True)

Session = sessionmaker(bind=engine)

session = Session()

employee_tbl = Employee

insert_list = [

{

'name': 'chie jyarinko'

},

{

'name': 'hiroshi miyamoto'

}

]

response = self.session.execute(employee_tbl.__table__.insert(), insert_list)

primary_keys = response.inserted_primary_key

上記の例では社員名をEmployeeテーブルに登録して、Auto Incrementで自動採番されたemployee_idを配列で返しています。

開発ベンダー5年、ユーザ企業システム部門通算9年、ITトレーナー1年目のSoftware Engineerです。
質問はコメントかSNSなどからお気軽にどうぞ。
出来る限りなるべく答えます。

このブログの内容/発言の一切は個人の見解であり、所属する組織とは関係ありません。

また、勉強会やイベントのレポートは自分が気になったことをメモしたり、聞いて思ったことを書いていますので、登壇者の意見や発表内容ではありません。

- Python python, sqlalchemy

Tweet

関連記事

: PyCharmでテキストを折り返す

超小ネタです。コーディングしているときは1行でそんなに長いコードを書かないので …

: Feedlyのフィードを自動でSlackへ投稿する(AWS Lambda , Amazon DynamoDB)

やりたいこと Feedlyで共有したいフィードに特定のタグを付けます。特定のタ …

: 「MonotaRo Tech #3 テスト自動化」に行ってきました

モノタロウさんの「MonotaRo Tech #3 テスト自動化」に行ってきまし …

: SQLAlchemyでjoinする

PythonのSQLAlchemyでMySQLのテーブルをjoinするときの覚書 …

: CentOS 6にPython実行環境を構築

CentOS6(32bit)にPython2.7実行環境を構築した際のメモです。 …

: feedlyでタグ付けした記事を自動でメール配信する

feedlyを使っていて、良記事や知っておいて欲しい記事とかを部門とかチームとか …

: 「神戸Pythonの会 #2 meetup」に行ってきました

「神戸Pythonの会 #2 meetup」に行ってきました。所感など Ato …

: 「【第２回】データ分析勉強会（尼崎Pythonの会#1）」に行ってきました

「【第２回】データ分析勉強会（尼崎Pythonの会#1）」に行ってきました。略 …

: SendGrid(sendgrid_python)でメール送信してイベントの情報からメール本文をたどれるようにしておく

動的に生成するメールで実際にどんな本文が送信されたかを記録しておきたいときもある …

: SQLAlchemyでシンプルにjoinする

過去のSQLAlchemyでjoinするという記事でPythonのSQLAlch …

PREV: Amazon RDS MySQLでCSVをload data するときに「Access denied」発生
NEXT: Python SQLAlchemy MySQLでcase文を使う