Я могу нормально запустить SQLAlchemy, когда все в одном файле.Теперь я хочу поместить свою модель в другой файл.
Однако это не работает, потому что я не могу найти способ поделиться базой.Я пытался использовать Singleton, но в файле model.py значение Null, и схема никогда не создается в базе данных.
Как это исправить?
Мои файлы (упрощенная версия):
- /main/__init__.py
- /main/main.py
- /utils/__init__.py
- /utils/utils.py
- /model/__init__.py
- /model/model.py
main / main.py:
from model import User
from utils.utils import readConf,createSession,getBase
class Worker(threading.Thread):
def __init__(self, queue, session):
self.__queue = queue
threading.Thread.__init__(self)
self._stopevent = threading.Event( )
def run(self):
session.merge(User(queue.get()))
session.commit()
class Collector(threading.Thread):
def __init__(self, nom = ''):
threading.Thread.__init__(self)
self.nom = nom
self._stopevent = threading.Event( )
def run(self):
while not self._stopevent.isSet():
queue.put("Name")
if __name__ == '__main__':
conf = readConf("file.")
session = createSession(conf)
queue = Queue.Queue(0)
Worker(queue, session).start()
Collector("Start").start()
utils / utils.py
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.schema import MetaData
def createSession(conf):
schema = conf['bdd']['type'] + '://' + conf['bdd']['user'] + ':' + conf['bdd']['password'] + '@' + conf['bdd']['host'] + '/' + conf['bdd']['db']
engine = create_engine(schema, echo=True)
b = getBase("Utils")
b.set_base(declarative_base())
Base = b.get_base()
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()
return session
class getBase(object):
__single = None # the one, true Singleton
__base = None
__text = None
__base = None
def __new__(cls, *args, **kwargs):
# Check to see if a __single exists already for this class
# Compare class types instead of just looking for None so
# that subclasses will create their own __single objects
if cls != type(cls.__single):
cls.__single = object.__new__(cls, *args, **kwargs)
__base = declarative_base()
return cls.__single
def __init__(self,name=None):
self.name = name
def get_base(self):
return self.__base
def set_base(self, value):
self.__base = value
model / model.py
from utils.utils import getBase
b = getBase("model")
b.set_base(declarative_base())
Base = b.get_base()
class User(Base):
__tablename__ = 'users'
def __init__(self, name):
self.screen_name = name
name_id = Column(Integer, primary_key=True, autoincrement=True)
screen_name = Column(BigInteger(), primary_key=True, nullable=False)
РЕДАКТИРОВАТЬ @ Lafaya
Я изменил model/__init__.py
так:
#!/usr/bin/python
Base = declarative_base()
Затем я изменил model/model.py
так:
from model import Base
class User(Base):
etc...
Теперь я не могу импортировать Base из model.py (потому что он позволяет импортировать __init.py__
).Но мне нужна ссылка на базу!
Traceback (most recent call last):
File "../main/main.py", line 12, in <module>
from model.model import User
File "../model/model.py", line 3, in <module>
from model import Base
ImportError: cannot import name Base