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

関連記事

: 「大阪Pythonユーザの集まり」に行ってきました

「大阪Pythonユーザの集まり」に行ってきました。あんまりメモ取れてません …

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

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

: 個人のGoogleカレンダーの予定をPythonで取得する

Google Calendar Twilio ReminderのGoogleカレ …

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

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

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

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

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

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

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

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

: Redmine REST APIを操作するpython_redmineを試してみた

Redmine のGUIの登録が面倒なのでコマンドで登録、更新、時間の記録が出来 …

: PythonでHTMLエスケープされた全角ハイフンをアンエスケープする

PythonでHTMLエスケープとアンエスケープについて調べていると、 xml. …

: Python SQLAlchemy MySQLでテーブルを指定せずにFunctionを使う

難しく考えて悩んでいましたが、queryに書いてやればいいだけでした。 sess …

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