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を配列で返しています。

最後までお読みいただきましてありがとうございました！

「AWS認定資格試験テキスト＆問題集　AWS認定ソリューションアーキテクト - プロフェッショナル改訂第2版」という本を書きました。

「AWS認定資格試験テキスト AWS認定クラウドプラクティショナー改訂第3版」という本を書きました。

「AWS認定資格試験テキスト　AWS認定AIプラクティショナー」という本を書きました。

「ポケットスタディ AWS認定デベロッパーアソシエイト［DVA-C02対応］」という本を書きました。

「要点整理から攻略するAWS認定ソリューションアーキテクト-アソシエイト」という本を書きました。

「AWSではじめるLinux入門ガイド」という本を書きました。

開発ベンダー5年、ユーザ企業システム部門通算9年、ITインストラクター5年目でプロトタイプビルダーもやりだしたSoftware Engineerです。
質問はコメントかSNSなどからお気軽にどうぞ。
出来る限りなるべく答えます。

このブログの内容/発言の一切は個人の見解であり、所属する組織とは関係ありません。
このブログは経験したことなどの共有を目的としており、手順や結果などを保証するものではありません。
ご参考にされる際は、読者様自身のご判断にてご対応をお願いいたします。

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

- Python python, sqlalchemy

Tweet

関連記事

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

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

: Pepper 開発ワークショップ初級~中級~番外編に1日で一気に参加した

Pepperの開発方法が知りたかったので、Pepperアトリエ秋葉原 with …

: pyenv update で「pyenv: no such command `update’ 」

環境 macOS Sierra 10.12.4 pyenvのアップデート pye …

: Linux Mint 17 MATE 64bit にTensorFlowをインストールする

勉強会でTensorFlowを使用しますので、Linux Mint 17 MAT …

: SQLAlchemyでシンプルにjoinする

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

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

Google Calendar Twilio ReminderのGoogleカレ …

: docomo Developer supportの雑談対話APIから自然対話APIへ移行しました

LINEやSlackのbotで重宝していた、雑談対話APIが終了する、とのことで …

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

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

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

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

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

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

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