finished auth overhaul and started tenants
This commit is contained in:
@@ -5,9 +5,8 @@ from typing import List, Optional
|
||||
|
||||
import pytz
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from sqlalchemy import Boolean, Column, DateTime, String, TypeDecorator
|
||||
from sqlalchemy import Boolean, Column, DateTime, String, TypeDecorator, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
db : SQLAlchemy = SQLAlchemy()
|
||||
|
||||
deck_password = db.Table(
|
||||
@@ -69,19 +68,25 @@ class Deck(db.Model):
|
||||
|
||||
passwords : Mapped[List[Password]] = relationship("Password", secondary=deck_password, back_populates='decks')
|
||||
tokens : Mapped[List[Token]] = relationship("Token", secondary=deck_token, back_populates='decks')
|
||||
course_id: Mapped[int] = mapped_column(ForeignKey("course.id"))
|
||||
course: Mapped['Course'] = relationship("Course", back_populates='decks')
|
||||
|
||||
def __init__(self, name: str, deck: str, description: str = "") -> None:
|
||||
|
||||
def __init__(self, name: str, deck: str, course: Course, description: str = "") -> None:
|
||||
self.name = name
|
||||
self.deck = deck
|
||||
self.description = description
|
||||
self.course = course
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Deck(name={self.name}, deck={self.deck})>"
|
||||
return f"<Deck(name={self.name}, deck={self.deck}, course={self.course})>"
|
||||
|
||||
class Token(db.Model):
|
||||
__tablename__ = "token"
|
||||
id : Mapped[int] = mapped_column(primary_key=True)
|
||||
name: Mapped[str] = mapped_column(String(60), unique=True, nullable=False)
|
||||
active_course_id: Mapped[int] = mapped_column(ForeignKey("course.id"), nullable=True)
|
||||
active_course: Mapped[Course] = relationship("Course", foreign_keys=[active_course_id])
|
||||
decks : Mapped[List[Deck]]= relationship("Deck", secondary=deck_token, back_populates='tokens')
|
||||
|
||||
def __init__(self, name: str) -> None:
|
||||
@@ -93,6 +98,19 @@ class Token(db.Model):
|
||||
def is_unlocked(self, deck: str) -> bool :
|
||||
return deck in [deck.deck for deck in self.decks]
|
||||
|
||||
class Course(db.Model):
|
||||
__tablename__ = "course"
|
||||
id : Mapped[int] = mapped_column(primary_key=True)
|
||||
name: Mapped[str] = mapped_column(String(60), unique=True, nullable=False)
|
||||
decks: Mapped[List['Deck']] = relationship(back_populates='course')
|
||||
|
||||
def __init__(self, name: str) -> None:
|
||||
self.name = name
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Course(id={self.id}, name={self.name})>"
|
||||
|
||||
|
||||
class InfoBanner(db.Model):
|
||||
__tablename__ = "info_banner"
|
||||
id : Mapped[int] = mapped_column(primary_key=True)
|
||||
|
||||
Reference in New Issue
Block a user