У меня есть метод get_or_create в классе помощника для SQLAlchemy, но сеанс никогда не возвращает никаких данных из запроса, поэтому всегда пытаются создать новую строку, даже если в базе данных есть те же данные (которые затем выдает ошибку UNIQUE CONSTRANT).
Добавление данных через метод create само по себе работает успешно, и данные сохраняются в БД. В методе get_or_create() экземпляр всегда равен None, и поэтому данные пытаются снова добавить в базу данных.
Код ниже является фрагментом полного приложения и должен показывать только то, что нужно, чтобы повторить проблему.
models.py:
from sqlalchemy import Column, Integer, String, ForeignKey, Float, Enum
from sqlalchemy.ext.declarative import declarative_base
import enum
Base = declarative_base()
class DepositStatus(enum.Enum):
not_yet_paid = 1
paid = 2
to_be_refunded = 3
deposit_kept = 4
refunded = 5
class RiderStatus(enum.Enum):
not_yet_started = 1
active = 2
finished = 3
scratched = 4
class RiderCategories(enum.Enum):
solo = 1
pair = 2
class RiderGenders(enum.Enum):
male = 1
female = 2
class TrackerLocations(Base):
__tablename__ = 'tracker_locations'
id = Column('id', Integer, primary_key=True)
types = Column(String)
__mapper_args__ = {
'polymorphic_on': types
}
class Riders(TrackerLocations):
__tablename__ = 'riders'
id = Column(Integer, ForeignKey('tracker_locations.id'), primary_key=True,
autoincrement=True)
first_name = Column('first_name', String)
last_name = Column('last_name', String)
cap_number = Column('cap_number', String)
category = Column('category', Enum(RiderCategories))
status = Column('status', Enum(RiderStatus))
gender = Column('gender', Enum(RiderGenders))
deposit_status = Column('deposit_status', Enum(DepositStatus))
deposit_amount = Column('deposit_amount', Float)
email = Column('email', String)
__mapper_args__ = {
'polymorphic_identity': 'riders'
}
db.py:
from sqlalchemy.orm import sessionmaker, scoped_session
from sqlalchemy import create_engine
from models import Base
class DB(object):
"""
To be used to interact with SQLAlchemy's ORM.
"""
def __init__(self, db_uri):
self.engine = create_engine(db_uri)
Base.metadata.create_all(self.engine)
self._session = scoped_session(sessionmaker(bind=self.engine))
def close(self):
self.session.close()
def rollback(self):
self.session.rollback()
@property
def session(self):
return self._session()
def get_or_create(self, model, commit=True, **kwargs):
instance = self.session.query(model).filter_by(**kwargs).first()
if instance:
return instance, False
else:
instance = self.create(model, commit=commit, **kwargs)
return instance, True
def create(self, model, commit=True, **kwargs):
instance = model(**kwargs)
self.session.add(instance)
if commit:
self.session.commit()
return instance
tests.py:
import os
import sqlite3
from tempfile import mkdtemp
from unittest import TestCase
from models import Riders
from db import DB
class TestDBInteractions(TestCase):
@classmethod
def setUpClass(cls):
cls.temp_ = mkdtemp()
cls.temp_db = os.path.join(cls.temp_, 'test_db.db')
def setUp(self):
self.db_uri = 'sqlite:///' + self.temp_db
self.db = DB(self.db_uri)
self.conn = sqlite3.connect(self.temp_db)
self.cur = self.conn.cursor()
def tearDown(self):
self.db.close()
def test_get_or_create(self):
test_data = {
'id': 1,
'first_name': 'Bobby',
'last_name': 'Hill',
'cap_number': '123',
'category': 'solo',
'status': 'finished',
'gender': 'male',
'deposit_status': 'paid',
'deposit_amount': 123.1,
'email': 'hello@bob.com'
}
params = tuple(test_data.values())
query = "INSERT INTO riders VALUES (?,?,?,?,?,?,?,?,?,?)"
self.cur.execute(query, params)
# # sqlite doesn't allow multiple write connections to the same DB, so close this connection
self.conn.commit()
self.conn.close()
self.db1 = DB(db_uri=self.db_uri)
response = self.db1.get_or_create(Riders, **{'id': 1})
print(response)