Python, SQLAlchemy, Flask-SQLAlchemy에서 발생하는 'Could not assemble any primary key columns for mapped table' 오류 해결하기

2024-07-27

Python, SQLAlchemy, Flask-SQLAlchemy에서 발생하는 "Could not assemble any primary key columns for mapped table" 오류 해결 방법

"Could not assemble any primary key columns for mapped table" 오류는 SQLAlchemy에서 테이블을 매핑하려고 할 때 발생하는 오류입니다. 테이블에 프라이머리 키 열이 정의되지 않은 경우 이 오류가 발생합니다. 프라이머리 키는 테이블의 각 행을 고유하게 식별하는 데 사용되는 열입니다.

해결 방법

이 오류를 해결하려면 테이블에 프라이머리 키 열을 정의해야 합니다. 다음은 두 가지 방법입니다.

선언적 매핑 사용

선언적 매핑을 사용하면 Python 클래스를 사용하여 테이블을 정의할 수 있습니다. 프라이머리 키 열을 정의하려면 primary_key=True 키워드를 사용하십시오.

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(255))
    email = Column(String(255))

반사적 매핑을 사용하면 기존 데이터베이스 테이블을 기반으로 Python 클래스를 생성할 수 있습니다. 프라이머리 키 열을 식별하려면 PrimaryKeyConstraint를 사용하십시오.

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///database.db')
Base = declarative_base(engine)
Session = sessionmaker(bind=engine)

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, PrimaryKeyConstraint())
    name = Column(String(255))
    email = Column(String(255))

Flask-SQLAlchemy에서 오류 해결

Flask-SQLAlchemy를 사용하는 경우 다음과 같이 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False를 설정하여 오류를 해결할 수 있습니다. 이 설정은 SQLAlchemy가 테이블 변경 내용을 추적하지 않도록 하여 오류를 줄이는 데 도움이 될 수 있습니다.

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

예제 코드

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///database.db')
Base = declarative_base(engine)
Session = sessionmaker(bind=engine)

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(255))
    email = Column(String(255))

# 테이블 생성
Base.metadata.create_all(engine)

# 세션 생성
session = Session()

# 사용자 추가
user1 = User(name='Alice', email='[email protected]')
user2 = User(name='Bob', email='[email protected]')
session.add_all([user1, user2])

# 변경 사항 커밋
session.commit()

# 사용자 조회
users = session.query(User).all()
for user in users:
    print(user.name, user.email)

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///database.db')
Base = declarative_base(engine)
Session = sessionmaker(bind=engine)

# 기존 테이블을 기반으로 클래스 생성
Base.metadata.reflect()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(255))
    email = Column(String(255))

# 세션 생성
session = Session()

# 사용자 추가
user1 = User(name='Alice', email='[email protected]')
user2 = User(name='Bob', email='[email protected]')
session.add_all([user1, user2])

# 변경 사항 커밋
session.commit()

# 사용자 조회
users = session.query(User).all()
for user in users:
    print(user.name, user.email)

Flask-SQLAlchemy 사용

from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

class User(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255))
    email = db.Column(db.String(255))

@app.route('/')
def index():
    # 사용자 조회
    users = User.query.all()
    return render_template('index.html', users=users)

if __name__ == '__main__':
    app.run()

"Could not assemble any primary key columns for mapped table" 오류의 대체 해결 방법

테이블에 여러 열로 구성된 복합 프라이머리 키를 정의할 수 있습니다.

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    user_id = Column(Integer, primary_key=True)
    name = Column(String(255))
    email = Column(String(255))

암묵적 프라이머리 키 사용

일부 데이터베이스 관리 시스템에서는 특정 열을 기본적으로 프라이머리 키로 간주합니다. 예를 들어, PostgreSQL에서는 id라는 이름의 열을 기본 프라이머리 키로 간주합니다.

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)  # PostgreSQL에서 기본 프라이머리 키로 간주됨
    name = Column(String(255))
    email = Column(String(255))

프라이머리 키 제약 조건 사용

반사적 매핑을 사용하는 경우 PrimaryKeyConstraint를 사용하여 프라이머리 키 열을 식별할 수 있습니다.

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///database.db')
Base = declarative_base(engine)
Session = sessionmaker(bind=engine)

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, PrimaryKeyConstraint())
    name = Column(String(255))
    email = Column(String(255))

테이블 검증 사용

SQLAlchemy에서 제공하는 inspect.get_table_names() 및 inspect.get_columns() 함수를 사용하여 테이블 및 열이 올바르게 정의되었는지 확인할 수 있습니다.

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import inspect

engine = create_engine('sqlite:///database.db')
Base = declarative_base(engine)

# 테이블 이름 목록 가져오기
table_names = inspect.get_table_names(engine)
print(table_names)

# 특정 테이블의 열 가져오기
table = inspect.get_columns('users')
for column in table:
    print(column.name)

오류 메시지 분석

"Could not assemble any primary key columns for mapped table" 오류 메시지는 테이블에 프라이머리 키 열이 정의되지 않았음을 나타냅니다. 오류 메시지의 나머지 부분을 carefully 분석하면 문제를 해결하는 데 도움이 될 수 있는 추가 정보를 제공할 수 있습니다.

주의 사항