160 lines
4.6 KiB
Python
160 lines
4.6 KiB
Python
from datetime import datetime
|
|
from typing import List, Optional
|
|
|
|
from sqlmodel import Field, Relationship, SQLModel # type: ignore
|
|
|
|
|
|
class User(SQLModel, table=True): # type: ignore
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
username: str = Field(unique=True, index=True)
|
|
email: str = Field(unique=True, index=True)
|
|
hashed_password: str
|
|
salt: str
|
|
wrapped_master_key: str
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
|
|
# Add relationships to existing models
|
|
notes: List["Note"] = Relationship(back_populates="user")
|
|
folders: List["Folder"] = Relationship(back_populates="user")
|
|
sessions: List["Session"] = Relationship(back_populates="user")
|
|
tags: List["Tag"] = Relationship(back_populates="user")
|
|
|
|
|
|
class Session(SQLModel, table=True): # type: ignore
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
session_id: str = Field(unique=True, index=True)
|
|
user_id: int = Field(foreign_key="user.id")
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
expires_at: datetime
|
|
ip_address: Optional[str] = None
|
|
user_agent: Optional[str] = None
|
|
|
|
user: User = Relationship(back_populates="sessions")
|
|
|
|
|
|
class Folder(SQLModel, table=True): # type: ignore
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str = Field(max_length=255)
|
|
parent_id: Optional[int] = Field(default=None, foreign_key="folder.id")
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
user_id: int = Field(foreign_key="user.id")
|
|
|
|
# Relationships
|
|
parent: Optional["Folder"] = Relationship(
|
|
back_populates="children", sa_relationship_kwargs={"remote_side": "Folder.id"}
|
|
)
|
|
children: List["Folder"] = Relationship(back_populates="parent")
|
|
notes: List["Note"] = Relationship(back_populates="folder")
|
|
user: User = Relationship(back_populates="folders")
|
|
|
|
|
|
class NoteTag(SQLModel, table=True): #type: ignore
|
|
note_id: int = Field(foreign_key="note.id", primary_key=True)
|
|
tag_id: int = Field(foreign_key="tag.id", primary_key=True)
|
|
|
|
|
|
class Tag(SQLModel, table=True): # type: ignore
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str = Field(max_length=255)
|
|
parent_id: Optional[int] = Field(default=None, foreign_key="tag.id")
|
|
user_id: int = Field(foreign_key="user.id")
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
|
|
# Relationships
|
|
user: User = Relationship(back_populates="tags")
|
|
parent: Optional["Tag"] = Relationship(
|
|
back_populates="children",
|
|
sa_relationship_kwargs={"remote_side": "Tag.id"}
|
|
)
|
|
children: List["Tag"] = Relationship(back_populates="parent")
|
|
notes: List["Note"] = Relationship(back_populates="tags", link_model=NoteTag)
|
|
|
|
|
|
|
|
class Note(SQLModel, table=True): # type: ignore
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
title: str = Field(max_length=255)
|
|
content: str
|
|
folder_id: Optional[int] = Field(default=None, foreign_key="folder.id")
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
updated_at: datetime = Field(default_factory=datetime.now)
|
|
user_id: int = Field(foreign_key="user.id")
|
|
|
|
#Relationships
|
|
folder: Optional[Folder] = Relationship(back_populates="notes")
|
|
user: User = Relationship(back_populates="notes")
|
|
tags: List[Tag] = Relationship(back_populates="notes", link_model=NoteTag)
|
|
|
|
|
|
|
|
# API Response models
|
|
class TagRead(SQLModel):
|
|
id: int
|
|
name: str
|
|
parent_id: Optional[int] = None
|
|
created_at: datetime
|
|
|
|
|
|
class TagCreate(SQLModel):
|
|
name: str
|
|
parent_id: Optional[int] = None
|
|
|
|
|
|
class TagUpdate(SQLModel):
|
|
name: Optional[str] = None
|
|
parent_id: Optional[int] = None
|
|
|
|
|
|
class TagTreeNode(SQLModel):
|
|
id: int
|
|
name: str
|
|
children: List["TagTreeNode"] = []
|
|
|
|
|
|
class TagTreeResponse(SQLModel):
|
|
tags: List[TagTreeNode]
|
|
|
|
|
|
class NoteRead(SQLModel):
|
|
id: int
|
|
title: str
|
|
content: str
|
|
folder_id: Optional[int] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
tags: List[TagRead] = []
|
|
|
|
|
|
class FolderTreeNode(SQLModel):
|
|
id: int
|
|
name: str
|
|
notes: List[NoteRead] = []
|
|
children: List["FolderTreeNode"] = []
|
|
|
|
|
|
class FolderTreeResponse(SQLModel):
|
|
folders: List[FolderTreeNode]
|
|
orphaned_notes: List[NoteRead]
|
|
|
|
|
|
# Create/Update models
|
|
class NoteCreate(SQLModel):
|
|
title: str
|
|
content: str
|
|
folder_id: Optional[int] = None
|
|
|
|
|
|
class NoteUpdate(SQLModel):
|
|
title: Optional[str] = None
|
|
content: Optional[str] = None
|
|
folder_id: Optional[int] = None
|
|
|
|
|
|
class FolderCreate(SQLModel):
|
|
name: str
|
|
parent_id: Optional[int] = None
|
|
|
|
|
|
class FolderUpdate(SQLModel):
|
|
name: Optional[str] = None
|
|
parent_id: Optional[int] = None
|