Untone
1b48675b92
Some checks failed
Deploy on push / deploy (push) Failing after 2m22s
### 🔄 Изменения - **SQLAlchemy KeyError** - исправление ошибки `KeyError: Reaction` при инициализации - **Исправлена ошибка SQLAlchemy**: Устранена проблема `InvalidRequestError: When initializing mapper Mapper[Shout(shout)], expression Reaction failed to locate a name (Reaction)` ### 🧪 Тестирование - **Исправление тестов** - адаптация к новой структуре моделей - **RBAC инициализация** - добавление `rbac.initialize_rbac()` в `conftest.py` - **Создан тест для getSession**: Добавлен комплексный тест `test_getSession_cookies.py` с проверкой всех сценариев - **Покрытие edge cases**: Тесты проверяют работу с валидными/невалидными токенами, отсутствующими пользователями - **Мокирование зависимостей**: Использование unittest.mock для изоляции тестируемого кода ### 🔧 Рефакторинг - **Упрощена архитектура**: Убраны сложные конструкции с отложенными импортами, заменены на чистую архитектуру - **Перемещение моделей** - `Author` и связанные модели перенесены в `orm/author.py`: Вынесены базовые модели пользователей (`Author`, `AuthorFollower`, `AuthorBookmark`, `AuthorRating`) из `orm.author` в отдельный модуль - **Устранены циклические импорты**: Разорван цикл между `auth.core` → `orm.community` → `orm.author` через реструктуризацию архитектуры - **Создан модуль `utils/password.py`**: Класс `Password` вынесен в utils для избежания циклических зависимостей - **Оптимизированы импорты моделей**: Убран прямой импорт `Shout` из `orm/community.py`, заменен на строковые ссылки ### 🔧 Авторизация с cookies - **getSession теперь работает с cookies**: Мутация `getSession` теперь может получать токен из httpOnly cookies даже без заголовка Authorization - **Убрано требование авторизации**: `getSession` больше не требует декоратор `@login_required`, работает автономно - **Поддержка dual-авторизации**: Токен может быть получен как из заголовка Authorization, так и из cookie `session_token` - **Автоматическая установка cookies**: Middleware автоматически устанавливает httpOnly cookies при успешном `getSession` - **Обновлена GraphQL схема**: `SessionInfo` теперь содержит поля `success`, `error` и опциональные `token`, `author` - **Единообразная обработка токенов**: Все модули теперь используют централизованные функции для работы с токенами - **Улучшена обработка ошибок**: Добавлена детальная валидация токенов и пользователей в `getSession` - **Логирование операций**: Добавлены подробные логи для отслеживания процесса авторизации ### 📝 Документация - **Обновлена схема GraphQL**: `SessionInfo` тип теперь соответствует новому формату ответа - Обновлена документация RBAC - Обновлена документация авторизации с cookies
85 lines
3.6 KiB
Python
85 lines
3.6 KiB
Python
from asyncio.log import logger
|
|
from enum import Enum
|
|
|
|
from ariadne import (
|
|
MutationType,
|
|
ObjectType,
|
|
QueryType,
|
|
SchemaBindable,
|
|
load_schema_from_path,
|
|
)
|
|
|
|
from orm import collection, community, draft, invite, notification, reaction, shout, topic
|
|
from orm.author import Author, AuthorBookmark, AuthorFollower, AuthorRating
|
|
from storage.db import create_table_if_not_exists, local_session
|
|
|
|
# Создаем основные типы
|
|
query = QueryType()
|
|
mutation = MutationType()
|
|
type_draft = ObjectType("Draft")
|
|
type_community = ObjectType("Community")
|
|
type_collection = ObjectType("Collection")
|
|
type_author = ObjectType("Author")
|
|
|
|
# Загружаем определения типов из файлов схемы
|
|
type_defs = load_schema_from_path("schema/")
|
|
|
|
# Список всех типов для схемы
|
|
resolvers: SchemaBindable | type[Enum] | list[SchemaBindable | type[Enum]] = [
|
|
query,
|
|
mutation,
|
|
type_draft,
|
|
type_community,
|
|
type_collection,
|
|
type_author,
|
|
]
|
|
|
|
|
|
def create_all_tables() -> None:
|
|
"""Create all database tables in the correct order."""
|
|
# Порядок важен - сначала таблицы без внешних ключей, затем зависимые таблицы
|
|
models_in_order = [
|
|
# user.User, # Базовая таблица auth
|
|
Author, # Базовая таблица
|
|
community.Community, # Базовая таблица
|
|
topic.Topic, # Базовая таблица
|
|
# Связи для базовых таблиц
|
|
AuthorFollower, # Зависит от Author
|
|
community.CommunityFollower, # Зависит от Community
|
|
topic.TopicFollower, # Зависит от Topic
|
|
# Черновики (теперь без зависимости от Shout)
|
|
draft.Draft, # Зависит только от Author
|
|
draft.DraftAuthor, # Зависит от Draft и Author
|
|
draft.DraftTopic, # Зависит от Draft и Topic
|
|
# Основные таблицы контента
|
|
shout.Shout, # Зависит от Author и Draft
|
|
shout.ShoutAuthor, # Зависит от Shout и Author
|
|
shout.ShoutTopic, # Зависит от Shout и Topic
|
|
# Реакции
|
|
reaction.Reaction, # Зависит от Author и Shout
|
|
shout.ShoutReactionsFollower, # Зависит от Shout и Reaction
|
|
# Дополнительные таблицы
|
|
AuthorRating, # Зависит от Author
|
|
AuthorBookmark, # Зависит от Author
|
|
notification.Notification, # Зависит от Author
|
|
notification.NotificationSeen, # Зависит от Notification
|
|
collection.Collection, # Зависит от Author
|
|
collection.ShoutCollection, # Зависит от Collection и Shout
|
|
invite.Invite, # Зависит от Author и Shout
|
|
]
|
|
|
|
with local_session() as session:
|
|
for model in models_in_order:
|
|
try:
|
|
# Ensure model is a type[DeclarativeBase]
|
|
if not hasattr(model, "__tablename__"):
|
|
logger.warning(f"Skipping {model} - not a DeclarativeBase model")
|
|
continue
|
|
|
|
create_table_if_not_exists(session.get_bind(), model) # type: ignore[arg-type]
|
|
# logger.info(f"Created or verified table: {model.__tablename__}")
|
|
except Exception as e:
|
|
table_name = getattr(model, "__tablename__", str(model))
|
|
logger.error(f"Error creating table {table_name}: {e}")
|
|
raise
|