tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-09 12:37:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(db): update db orm classes

Browse Source
This commit is contained in:
arkohut 2024-05-31 18:06:09 +08:00
parent 3d7cbe03b4
commit 7dc205ef4f
1 changed files with 32 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
memos/models.py
View File

@ -4,107 +4,92 @@ from sqlalchemy import (
String, String,
Text, Text,
DateTime, DateTime,
Enum Enum,
func
) )
from datetime import datetime from datetime import datetime
from sqlalchemy.orm import relationship, DeclarativeBase, Mapped, mapped_column from sqlalchemy.orm import relationship, DeclarativeBase, Mapped, mapped_column
from enum import Enum as PyEnum
from typing import List from typing import List
from .config import get_database_path from .config import get_database_path
from .schemas import MetadataSource, MetadataType
class MetadataSource(PyEnum):
USER_GENERATED = "user_generated"
SYSTEM_GENERATED = "system_generated"
PLUGIN_GENERATED = "plugin_generated"
class MetadataType(PyEnum):
EXTRACONTENT = "extra_content"
ATTRIBUTE = "attribute"
class Base(DeclarativeBase): class Base(DeclarativeBase):
pass id: Mapped[int] = mapped_column(Integer, primary_key=True)
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), nullable=False)
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now(), nullable=False)
class Library(Base): class LibraryModel(Base):
__tablename__ = "libraries" __tablename__ = "libraries"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
name: Mapped[str] = mapped_column(String, nullable=False) name: Mapped[str] = mapped_column(String, nullable=False)
folders: Mapped[List["Folder"]] = relationship("Folder", back_populates="library") folders: Mapped[List["FolderModel"]] = relationship("FolderModel", back_populates="library")
plugins: Mapped[List["Plugin"]] = relationship("LibraryPlugin", back_populates="library") plugins: Mapped[List["PluginModel"]] = relationship("LibraryPluginModel", back_populates="library")
class Folder(Base): class FolderModel(Base):
__tablename__ = "folders" __tablename__ = "folders"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
path: Mapped[str] = mapped_column(String, nullable=False) path: Mapped[str] = mapped_column(String, nullable=False)
library_id: Mapped[int] = mapped_column(Integer, nullable=False) library_id: Mapped[int] = mapped_column(Integer, nullable=False)
library: Mapped["Library"] = relationship("Library", back_populates="folders") library: Mapped["LibraryModel"] = relationship("LibraryModel", back_populates="folders")
entities: Mapped[List["Entity"]] = relationship("Entity", back_populates="folder") entities: Mapped[List["EntityModel"]] = relationship("EntityModel", back_populates="folder")
class Entity(Base): class EntityModel(Base):
__tablename__ = "entities" __tablename__ = "entities"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True) filepath: Mapped[str] = mapped_column(String, nullable=False)
path: Mapped[str] = mapped_column(String, nullable=False)
filename: Mapped[str] = mapped_column(String, nullable=False) filename: Mapped[str] = mapped_column(String, nullable=False)
size: Mapped[int] = mapped_column(Integer, nullable=False) size: Mapped[int] = mapped_column(Integer, nullable=False)
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False) file_created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
last_modified_at: Mapped[datetime] = mapped_column(DateTime, nullable=False) file_last_modified_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
filetype: Mapped[str] = mapped_column(String, nullable=False) file_type: Mapped[str] = mapped_column(String, nullable=False)
last_scan_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True) last_scan_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
folder_id: Mapped[int] = mapped_column(Integer, nullable=False) folder_id: Mapped[int] = mapped_column(Integer, nullable=False)
folder: Mapped["Folder"] = relationship("Folder", back_populates="entities") folder: Mapped["FolderModel"] = relationship("FolderModel", back_populates="entities")
metadata_entries: Mapped[List["EntityMetadata"]] = relationship("EntityMetadata", back_populates="entity") metadata_entries: Mapped[List["EntityMetadataModel"]] = relationship("EntityMetadataModel", back_populates="entity")
tags: Mapped[List["TagModel"]] = relationship("EntityTagModel", back_populates="entity")
class Tag(Base): class TagModel(Base):
__tablename__ = "tags" __tablename__ = "tags"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
name: Mapped[str] = mapped_column(String, nullable=False) name: Mapped[str] = mapped_column(String, nullable=False)
description: Mapped[str | None] = mapped_column(Text, nullable=True) description: Mapped[str | None] = mapped_column(Text, nullable=True)
color: Mapped[str | None] = mapped_column(String, nullable=True) color: Mapped[str | None] = mapped_column(String, nullable=True)
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False) source: Mapped[str | None] = mapped_column(String, nullable=True)
class EntityTag(Base): class EntityTagModel(Base):
__tablename__ = "entity_tags" __tablename__ = "entity_tags"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
entity_id: Mapped[int] = mapped_column(Integer, nullable=False) entity_id: Mapped[int] = mapped_column(Integer, nullable=False)
tag_id: Mapped[int] = mapped_column(Integer, nullable=False) tag_id: Mapped[int] = mapped_column(Integer, nullable=False)
source: Mapped[MetadataSource] = mapped_column(Enum(MetadataSource), nullable=False) source: Mapped[MetadataSource] = mapped_column(Enum(MetadataSource), nullable=False)
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
class EntityMetadata(Base): class EntityMetadataModel(Base):
__tablename__ = "metadata_entries" __tablename__ = "metadata_entries"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
entity_id: Mapped[int] = mapped_column(Integer, nullable=False) entity_id: Mapped[int] = mapped_column(Integer, nullable=False)
key: Mapped[str] = mapped_column(String, nullable=False) key: Mapped[str] = mapped_column(String, nullable=False)
value: Mapped[str] = mapped_column(Text, nullable=False) value: Mapped[str] = mapped_column(Text, nullable=False)
source: Mapped[MetadataSource] = mapped_column(Enum(MetadataSource), nullable=False) source_type: Mapped[MetadataSource] = mapped_column(Enum(MetadataSource), nullable=False)
source: Mapped[str | None] = mapped_column(String, nullable=True)
datetype: Mapped[MetadataType] = mapped_column(Enum(MetadataType), nullable=False) datetype: Mapped[MetadataType] = mapped_column(Enum(MetadataType), nullable=False)
entity = relationship("Entity", back_populates="metadata_entries") entity = relationship("EntityModel", back_populates="metadata_entries")
class Plugin(Base): class PluginModel(Base):
__tablename__ = "plugins" __tablename__ = "plugins"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
name: Mapped[str] = mapped_column(String, nullable=False) name: Mapped[str] = mapped_column(String, nullable=False)
description: Mapped[str | None] = mapped_column(Text, nullable=True) description: Mapped[str | None] = mapped_column(Text, nullable=True)
webhook_url: Mapped[str] = mapped_column(String, nullable=False) webhook_url: Mapped[str] = mapped_column(String, nullable=False)
libraries = relationship("LibraryPlugin", back_populates="plugin") libraries = relationship("LibraryPluginModel", back_populates="plugin")
class LibraryPlugin(Base): class LibraryPluginModel(Base):
__tablename__ = "library_plugins" __tablename__ = "library_plugins"
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
library_id: Mapped[int] = mapped_column(Integer, nullable=False) library_id: Mapped[int] = mapped_column(Integer, nullable=False)
plugin_id: Mapped[int] = mapped_column(Integer, nullable=False) plugin_id: Mapped[int] = mapped_column(Integer, nullable=False)
library = relationship("Library", back_populates="plugins") library: Mapped["LibraryModel"] = relationship("LibraryModel", back_populates="plugins")
plugin = relationship("Plugin", back_populates="libraries") plugin: Mapped["PluginModel"] = relationship("PluginModel", back_populates="libraries")
# Create the database engine with the path from config # Create the database engine with the path from config