SQLAlchemy, Declarative, PostgreSQL: не удается создать таблицы

Question

Я создал несколько классов, которые я хочу сохранить в базе данных. Я хотел бы иметь возможность запускать приложение с определенным флагом, чтобы взять все эти классы и создать соответствующие таблицы в моем db. С этой целью я пытаюсь импортировать классы, а затем вызывать метод Base.metadata.create_all(engine). Однако это не работает. В то же время, когда я вызываю это из файла с каждым из этих фактических классов, таблицы создаются.

Как мне решить задачу создания этих таблиц? Должен ли я попытаться создать их из одного файла/скрипта или добавить эту строку в каждый из этих классов?

Вот пример моего кода:

1) Item.py

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

Base = declarative_base() 

class Item(Base): 
    __tablename__ = "items" 

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

    def __repr__(self): 
     return "<~ {} ~> name: {}".format(self.id, self.name) 

2) main.py

from sqlalchemy import create_engine 
from sqlalchemy.orm import sessionmaker 

def main(): 
    from sqlalchemy.ext.declarative import declarative_base 
    Base = declarative_base() 
    engine = create_engine('postgresql+psycopg2://me:my_passl@localhost/my_first_database', echo=True) 
    from Item import Item 
    print(Item) 
    print(Item.__table__) 
    Base.metadata.create_all(engine) 

main() 

Извините за странный порядок импорта, но я где-то читал, что порядок импорта ваших классов и базы имеет значение, поэтому я попытался поиграть с ним.

Answer 1

Вы уже создали Base в Item.py, просто импортировать его в main.py:

Если main.py и Item.py находятся на одной и той же папке, а затем в main.py:

from Item import Base, Item

И удалить весь импорт в main функция, поэтому main.py будет выглядеть так:

from sqlalchemy import create_engine 
from sqlalchemy.orm import sessionmaker 
from Item import Base, Item 

def main(): 
    engine = create_engine('postgresql+psycopg2://me:my_passl@localhost/my_first_database', echo=True) 
    print(Item) 
    print(Item.__table__) 
    Base.metadata.create_all(engine) 

main()