Обратный атрибут не найден в PonyORM - PullRequest
0 голосов
/ 10 июня 2019

Функция generate_mapping выдает ошибку. У меня есть модели, описанные отдельно в двух модулях.

У меня есть модель:

# ./message/models.py
from pony import orm
from datetime import datetime
from db import db

model_names = ['Message', ]


class Message(db.Entity):
    id = orm.PrimaryKey(int, auto=True)
    time = orm.Required(datetime)
    text = orm.Required(str)
    user_from = orm.Required('User', cascade_delete=False)
    user_to = orm.Required('User', cascade_delete=True)

И еще:

# ./user/models.py
from pony import orm
from datetime import datetime
from db import db

model_names = ['User', ]


class User(db.Entity):
    id = orm.PrimaryKey(int, auto=True)
    login = orm.Required(str, unique=True)
    pswdhash = orm.Required(str)
    last_login = orm.Required(datetime)
    is_active = orm.Required(bool, default=True)

Запуск программы происходит через общий модуль db.py:

# ./db.py
from functools import reduce

from pony import orm

from settings import DATABASE, INSTALLED_APPS

db = orm.Database()
if DATABASE['engine'] == 'sqlite':
    db.bind(provider='sqlite', filename=DATABASE['path'], create_db=True)



pkgs = reduce(  # INSTALLED_APPS list
    lambda value, item: value + [__import__(f'{item}.models')],
    INSTALLED_APPS,
    [],
)
modules = reduce(  # pkg.models list
    lambda value, item: value + [getattr(item, 'models', [])],
    pkgs,
    [],
)
mod_names = reduce(  # models name in all pkg.models
    lambda value, item: value + getattr(item, 'model_names', []),
    modules,
    [],
)

classes = dict()  # dict with all models

for cn in mod_names:
    for m in modules:
        if cn in m.__dict__:
            classes[cn] = m.__dict__[cn]

print(classes)
db.generate_mapping(create_tables=True)

Когда я запускаю программу в терминале, я вижу это:

Traceback (most recent call last):
  File "D:\Programming\Python\cpython\v372\Lib\runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "D:\Programming\Python\cpython\v372\Lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File ".\__main__.py", line 8, in <module>
    from db import db
  File ".\db.py", line 37, in <module>
    db.generate_mapping(create_tables=True)
  File "D:\Envs\hwpy2\lib\site-packages\pony\orm\core.py", line 967, in generate_mapping
    entity._link_reverse_attrs_()
  File "D:\Envs\hwpy2\lib\site-packages\pony\orm\core.py", line 3893, in _link_reverse_attrs_
    else: throw(ERDiagramError, 'Reverse attribute for %s not found' % attr)
  File "D:\Envs\hwpy2\lib\site-packages\pony\utils\utils.py", line 106, in throw
    raise exc
pony.orm.core.ERDiagramError: Reverse attribute for Message.user_from not found

Импорт моделей нормален

In [24]: classes
Out[24]: {'Message': messager.models.Message, 'User': users.models.User}

Что я делаю не так?

1 Ответ

1 голос
/ 20 июня 2019

В PonyORM вы должны установить обратный атрибут с обеих сторон отношения.В вашем случае ваша User модель должна быть

class Message(db.Entity):
    id = orm.PrimaryKey(int, auto=True)
    time = orm.Required(datetime)
    text = orm.Required(str)
    user_from = orm.Required('User', cascade_delete=False, reverse='messages_sent')
    user_to = orm.Required('User', cascade_delete=True, reverse='messages_recieved')

Больше деталей здесь

class User(db.Entity):
    id = orm.PrimaryKey(int, auto=True)
    login = orm.Required(str, unique=True)
    pswdhash = orm.Required(str)
    last_login = orm.Required(datetime)
    is_active = orm.Required(bool, default=True)
    messages_sent = Set('Message', reverse='user_from')
    messages_recieved = Set('Message', reverse='user_to')

Кроме того, поскольку у вас есть 2 ссылки, вы должны точно указать, какой атрибуткарта.

...