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

関連記事

: Python openpyxlで結合セルを含むExcelファイルを開くと罫線が消える

PythonのExcelを読み書きするためのライブラリ、openpyxlで結合セ …

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

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

: Apple Silicon M1 MacBook ProにJupyter Notebookをインストールしました

Python – Jupyter NotebookをMac M1 搭 …

: SQLAlchemyでシンプルにjoinする

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

: pythonのsqlalchemyで嵌った話

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

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

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

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

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

: SQLAlchemyのModelクラスをMySQLから自動生成する

SQLAlchemyのMySQLモデルを書くのが面倒で、きっと何かツールがあるの …

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

Google Calendar Twilio ReminderのGoogleカレ …

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

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

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