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

関連記事

: SendGrid(Python)で添付ファイル付きのメールを送信する

SendGridを使って添付ファイル付きのメールを送信するサンプルをPython …

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

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

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

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

: CentOS 6でPythonからOracleに接続する

OracleのデータをPythonでさわりたくて、その実行環境がCentOS6の …

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

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

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

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

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

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

: pythonのsqlalchemyで嵌った話

エラーメッセージ InvalidRequestError: SQL expres …

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

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

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

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

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