wehub-resource-sync
c889a57b6b
Test Suites / Build CI Environment (push) Has been cancelled
Test Suites / Basic Tests (push) Has been cancelled
Test Suites / End-to-End Tests (push) Has been cancelled
Test Suites / CLI Tests (push) Has been cancelled
Test Suites / Slow End-to-End Tests (push) Has been cancelled
Test Suites / Graph Database Tests (push) Has been cancelled
Test Suites / Vector DB Tests (push) Has been cancelled
Test Suites / Temporal Graph Test (push) Has been cancelled
Test Suites / Search Test on Different DBs (push) Has been cancelled
Test Suites / Example Tests (push) Has been cancelled
Test Suites / Notebook Tests (push) Has been cancelled
Test Suites / OS and Python Tests Ubuntu (push) Has been cancelled
Test Suites / OS and Python Tests Extended (push) Has been cancelled
Test Suites / LLM Test Suite (push) Has been cancelled
Test Suites / S3 File Storage Test (push) Has been cancelled
Test Suites / Run Integration Tests (push) Has been cancelled
Test Suites / MCP Tests (push) Has been cancelled
Test Suites / Docker Compose Test (push) Has been cancelled
Test Suites / Docker CI test (push) Has been cancelled
Test Suites / Relational DB Migration Tests (push) Has been cancelled
Test Suites / Distributed Cognee Test (push) Has been cancelled
Test Suites / DB Examples Tests (push) Has been cancelled
Test Suites / Test Completion Status (push) Has been cancelled
Test Suites / Claude Code Review (push) Has been cancelled
Test Suites / basic checks (push) Has been cancelled
build | Build and Push Cognee MCP Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
build | Build and Push Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.11) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.12) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (kuzu, kuzu) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (neo4j, neo4j) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Examples (push) Has been cancelled
Weighted Edges Tests / Code Quality for Weighted Edges (push) Has been cancelled
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
from datetime import datetime, timezone
|
|
from sqlalchemy import (
|
|
DateTime,
|
|
Index,
|
|
# event,
|
|
Text,
|
|
JSON,
|
|
UUID,
|
|
)
|
|
|
|
# from sqlalchemy.schema import DDL
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from cognee.infrastructure.databases.relational import Base
|
|
|
|
|
|
class Node(Base):
|
|
__tablename__ = "nodes"
|
|
|
|
id: Mapped[UUID] = mapped_column(UUID(as_uuid=True), primary_key=True)
|
|
|
|
slug: Mapped[UUID] = mapped_column(UUID(as_uuid=True), nullable=False)
|
|
|
|
user_id: Mapped[UUID] = mapped_column(UUID(as_uuid=True), nullable=False)
|
|
|
|
data_id: Mapped[UUID] = mapped_column(UUID(as_uuid=True), nullable=False)
|
|
|
|
dataset_id: Mapped[UUID] = mapped_column(UUID(as_uuid=True), index=True, nullable=False)
|
|
pipeline_run_id: Mapped[UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), index=True, nullable=True
|
|
)
|
|
|
|
label: Mapped[str | None] = mapped_column(Text)
|
|
type: Mapped[str] = mapped_column(Text, nullable=False)
|
|
indexed_fields: Mapped[list] = mapped_column(JSON, nullable=False)
|
|
|
|
attributes: Mapped[dict | None] = mapped_column(JSON)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), nullable=False
|
|
)
|
|
|
|
__table_args__ = (
|
|
Index("index_node_dataset_slug", "dataset_id", "slug"),
|
|
Index("index_node_dataset_data", "dataset_id", "data_id"),
|
|
# {"postgresql_partition_by": "HASH (user_id)"}, # HASH partitioning on user_id
|
|
)
|
|
|
|
|
|
# Enable row-level security (RLS) for nodes
|
|
# enable_node_rls = DDL("""
|
|
# ALTER TABLE nodes ENABLE ROW LEVEL SECURITY;
|
|
# """)
|
|
# create_user_isolation_policy = DDL("""
|
|
# CREATE POLICY user_isolation_policy
|
|
# ON nodes
|
|
# USING (user_id = current_setting('app.current_user_id')::uuid)
|
|
# WITH CHECK (user_id = current_setting('app.current_user_id')::uuid);
|
|
# """)
|
|
|
|
# event.listen(Node.__table__, "after_create", enable_node_rls)
|
|
# event.listen(Node.__table__, "after_create", create_user_isolation_policy)
|