From 28685cba3e4de12a6ba08234efe04a4213f27b1b Mon Sep 17 00:00:00 2001 From: Untone Date: Tue, 23 Jan 2024 10:56:31 +0300 Subject: [PATCH] base-init --- orm/notification.py | 1 + resolvers/schema.py | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/orm/notification.py b/orm/notification.py index 98351ad..d0355a9 100644 --- a/orm/notification.py +++ b/orm/notification.py @@ -3,6 +3,7 @@ from enum import Enum as Enumeration from sqlalchemy import JSON as JSONType, func, cast from sqlalchemy import Column, Enum, ForeignKey, Integer from sqlalchemy.orm import relationship +from sqlalchemy.orm.session import engine from orm.author import Author from services.db import Base diff --git a/resolvers/schema.py b/resolvers/schema.py index 7d05927..7e87145 100644 --- a/resolvers/schema.py +++ b/resolvers/schema.py @@ -1,10 +1,16 @@ + import strawberry from strawberry.schema.config import StrawberryConfig from services.auth import LoginRequiredMiddleware from resolvers.load import Query from resolvers.seen import Mutation +from services.db import Base, engine schema = strawberry.Schema( query=Query, mutation=Mutation, config=StrawberryConfig(auto_camel_case=False), extensions=[LoginRequiredMiddleware] ) + + +# Create the specified tables if they do not already exist +Base.metadata.create_all(bind=engine, checkfirst=True)