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)