如何正确地使用多个文件使用相同的dectarative_base（）？

Question

这是我的文件布局的外观：

.
├── main.py
└── models
    ├── base.py
    ├── building.py
    └── user.py

在base.py中，我所拥有的只是所有其他类都使用的常见dectarative_base（）。

base.py：

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

在其他模型文件中，我有一些简单的类来建模数据。

building.py：

from sqlalchemy import Column, Integer, String
from base import Base

class Building(Base):
    __tablename__ = "building"

    id = Column('id', Integer, primary_key=True, autoincrement=True)
    address = Column('address', String(50), unique=True)

if __name__ == "__main__":
    print("Imports successful on building.py")

user.py：

from sqlalchemy import Column, Integer, String
from base import Base

class User(Base):
    __tablename__ = "user"

    id = Column('id', Integer, primary_key=True, autoincrement=True)
    username = Column('username', String(50), unique=True)

if __name__ == "__main__":
    print("Imports successful on user.py")

我的Man.py只是试图导入数据模型类。

main.py：

from models.user import User
from models.building import Building

print("Imports successful on main.py")

我可以运行building.py和user.py毫无问题，但是如果我运行main.py.py 我会得到以下追溯：

Traceback (most recent call last):
  File "main.py", line 1, in <module>
    from models.user import User
  File "/[path to root director]/models/user.py", line 2, in <module>
    from base import Base
ModuleNotFoundError: No module named 'base'

我希望能够将数据模型类跨多个文件拆分，以使项目更有条理和更易于维护。但是，似乎他们都必须使用相同的dectarative_base对base.metadata.create_all（bind = Engine）之类的东西进行工作。我还想将这些文件与使用的文件分开声明。

正确的方法是什么？