core-framework 0.3.0__py3-none-any.whl

core_framework/alembic/comment/alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration with an async dbapi.

core_framework/alembic/comment/alembic/env.py

@@ -0,0 +1,72 @@
+import asyncio
+from logging.config import fileConfig
+
+from sqlalchemy import pool, text
+from sqlalchemy.engine import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+
+from alembic import context
+from core_framework.core import settings
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Interpret the config file for Python logging.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+target_metadata = None
+
+config.set_main_option("sqlalchemy.url", settings.write_postgres.alembic_postgres_url)
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode."""
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def do_run_migrations(connection: Connection) -> None:
+    quoted_schema_name = f'"{settings.postgres_schemas.schema_comment.replace('"', '""')}"'
+
+    connection.execute(text(f"create schema if not exists {quoted_schema_name}"))
+    connection.execute(text(f"set search_path to {quoted_schema_name}"))
+    connection.commit()
+
+    context.configure(connection=connection, target_metadata=target_metadata)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations() -> None:
+    """Run migrations in 'online' mode."""
+    connectable = async_engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    async with connectable.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await connectable.dispose()
+
+
+def run_migrations_online() -> None:
+    asyncio.run(run_async_migrations())
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

core_framework/alembic/comment/alembic/script.py.mako

@@ -0,0 +1,28 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, Sequence[str], None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    ${downgrades if downgrades else "pass"}

core_framework/alembic/comment/alembic/versions/30334fd1347b_init.py

@@ -0,0 +1,59 @@
+"""init
+
+Revision ID: 30334fd1347b
+Revises:
+Create Date: 2026-03-10 06:25:01.802249
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "30334fd1347b"
+down_revision: str | Sequence[str] | None = None
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        create type comment_subject_type as enum (
+            'post',
+            'user_wall'
+        );
+    """)
+    op.execute("""
+        create type comment_status as enum (
+            'active',
+            'deleted'
+        );
+    """)
+    op.execute("""
+        create table comments (
+            id varchar(26) primary key,
+            author_id varchar(128) not null,
+            content text not null,
+            subject_type comment_subject_type not null,
+            subject_id varchar(128) not null,
+            parent_comment_id varchar(26) null references comments(id),
+            path text not null,
+            level int not null,
+            status comment_status not null default 'active',
+            edited_at timestamptz null,
+            edited_count int not null default 0,
+            created_at timestamptz not null default now(),
+            updated_at timestamptz not null default now(),
+            constraint comments_edited_count_non_negative check (edited_count >= 0),
+            constraint comments_level_non_negative check (level >= 0),
+            constraint comments_content_not_blank check (length(trim(content)) > 0),
+            constraint comments_subject_type_subject_id_path_unique unique (subject_type, subject_id, path)
+        );
+    """)
+
+
+def downgrade() -> None:
+    op.execute("drop table if exists comments;")
+    op.execute("drop type if exists comment_status;")
+    op.execute("drop type if exists comment_subject_type;")

core_framework/alembic/comment/alembic/versions/a2b3c4d5e6f7_improve_comment_indexes.py

@@ -0,0 +1,54 @@
+"""improve comment indexes
+
+Revision ID: a2b3c4d5e6f7
+Revises: f7e6d5c4b3a2
+Create Date: 2026-03-22 20:00:00.000000
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "a2b3c4d5e6f7"
+down_revision: str | Sequence[str] | None = "f7e6d5c4b3a2"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        create index idx_comments_subject_created
+        on comments (subject_type, subject_id, created_at desc)
+        where status = 'active'
+    """)
+
+    op.execute("""
+        create index idx_comments_author_created
+        on comments (author_id, created_at desc)
+        where status = 'active'
+    """)
+
+    op.execute("drop index if exists idx_comments_parent_comment_id")
+    op.execute("""
+        create index idx_comments_parent_created
+        on comments (parent_comment_id, created_at desc)
+        where status = 'active'
+    """)
+
+    op.execute("""
+        create index idx_comment_likes_liker_id
+        on comment_likes (liker_id)
+    """)
+
+
+def downgrade() -> None:
+    op.execute("drop index if exists idx_comment_likes_liker_id")
+    op.execute("drop index if exists idx_comments_parent_created")
+    op.execute("""
+        create index idx_comments_parent_comment_id
+        on comments (parent_comment_id)
+    """)
+    op.execute("drop index if exists idx_comments_author_created")
+    op.execute("drop index if exists idx_comments_subject_created")

core_framework/alembic/comment/alembic/versions/bcc8e00cfc8b_add_extra_tables.py

@@ -0,0 +1,64 @@
+"""add extra tables
+
+Revision ID: bcc8e00cfc8b
+Revises: 30334fd1347b
+Create Date: 2026-03-12 06:08:09.681774
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "bcc8e00cfc8b"
+down_revision: str | Sequence[str] | None = "30334fd1347b"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        create table comment_likes (
+            comment_id varchar(26) not null references comments(id) on delete cascade,
+            liker_id varchar(128) not null,
+            created_at timestamptz not null default now(),
+            primary key (comment_id, liker_id)
+        );
+    """)
+    op.execute("""
+        create table comment_stats (
+            comment_id varchar(26) primary key references comments(id) on delete cascade,
+            like_count bigint not null default 0,
+            reply_count bigint not null default 0,
+            report_count bigint not null default 0,
+            created_at timestamptz not null default now(),
+            updated_at timestamptz not null default now(),
+            constraint comment_stats_like_count_non_negative check (like_count >= 0),
+            constraint comment_stats_reply_count_non_negative check (reply_count >= 0),
+            constraint comment_stats_report_count_non_negative check (report_count >= 0)
+        );
+    """)
+    op.execute("""
+        create table user_restrictions_lookup (
+            user_id varchar(128) primary key,
+            restriction_type varchar(16) not null,
+            constraint user_restrictions_lookup_restriction_type_valid
+                check (restriction_type in ('muted', 'banned'))
+        );
+    """)
+    op.execute("""
+        create table user_blocks_lookup (
+            blocker_id varchar(128) not null,
+            blocked_id varchar(128) not null,
+            primary key (blocker_id, blocked_id),
+            constraint user_blocks_lookup_no_self_block check (blocker_id != blocked_id)
+        );
+    """)
+
+
+def downgrade() -> None:
+    op.execute("drop table if exists user_blocks_lookup;")
+    op.execute("drop table if exists user_restrictions_lookup;")
+    op.execute("drop table if exists comment_stats;")
+    op.execute("drop table if exists comment_likes;")

core_framework/alembic/comment/alembic/versions/d1e2f3a4b5c6_add_comment_stats_dirty_table.py

@@ -0,0 +1,29 @@
+"""add comment_stats_dirty table
+
+Revision ID: d1e2f3a4b5c6
+Revises: bcc8e00cfc8b
+Create Date: 2026-03-14
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "d1e2f3a4b5c6"
+down_revision: str | Sequence[str] | None = "bcc8e00cfc8b"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        create table comment_stats_dirty (
+            comment_id varchar(26) primary key references comments(id) on delete cascade
+        );
+    """)
+
+
+def downgrade() -> None:
+    op.execute("drop table if exists comment_stats_dirty;")

core_framework/alembic/comment/alembic/versions/e3f4a5b6c7d8_cascade_delete_comment_descendants.py

@@ -0,0 +1,49 @@
+"""cascade delete comment descendants
+
+Revision ID: e3f4a5b6c7d8
+Revises: d1e2f3a4b5c6
+Create Date: 2026-03-17 18:40:00.000000
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "e3f4a5b6c7d8"
+down_revision: str | Sequence[str] | None = "d1e2f3a4b5c6"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        create index if not exists idx_comments_parent_comment_id
+        on comments (parent_comment_id);
+    """)
+    op.execute("""
+        alter table comments
+        drop constraint if exists comments_parent_comment_id_fkey;
+    """)
+    op.execute("""
+        alter table comments
+        add constraint comments_parent_comment_id_fkey
+        foreign key (parent_comment_id)
+        references comments(id)
+        on delete cascade;
+    """)
+
+
+def downgrade() -> None:
+    op.execute("""
+        alter table comments
+        drop constraint if exists comments_parent_comment_id_fkey;
+    """)
+    op.execute("""
+        alter table comments
+        add constraint comments_parent_comment_id_fkey
+        foreign key (parent_comment_id)
+        references comments(id);
+    """)
+    op.execute("drop index if exists idx_comments_parent_comment_id;")

core_framework/alembic/comment/alembic/versions/f7e6d5c4b3a2_comments_path_to_ltree.py

@@ -0,0 +1,60 @@
+"""comments path to ltree
+
+Revision ID: f7e6d5c4b3a2
+Revises: e3f4a5b6c7d8
+Create Date: 2026-03-19 12:00:00.000000
+
+Requires: ltree extension installed in the extension schema (see
+alembic/extension migration c9d0e1f2a3b4). Apply extension migrations
+before this revision.
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "f7e6d5c4b3a2"
+down_revision: str | Sequence[str] | None = "e3f4a5b6c7d8"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("""
+        alter table comments
+        drop constraint if exists comments_subject_type_subject_id_path_unique;
+    """)
+    op.execute("""
+        alter table comments
+        alter column path type extension.ltree
+        using extension.text2ltree(path);
+    """)
+    op.execute("""
+        alter table comments
+        add constraint comments_subject_type_subject_id_path_unique
+        unique (subject_type, subject_id, path);
+    """)
+    op.execute("""
+        create index idx_comments_path_gist
+        on comments using gist (path);
+    """)
+
+
+def downgrade() -> None:
+    op.execute("drop index if exists idx_comments_path_gist;")
+    op.execute("""
+        alter table comments
+        drop constraint if exists comments_subject_type_subject_id_path_unique;
+    """)
+    op.execute("""
+        alter table comments
+        alter column path type text
+        using path::text;
+    """)
+    op.execute("""
+        alter table comments
+        add constraint comments_subject_type_subject_id_path_unique
+        unique (subject_type, subject_id, path);
+    """)

core_framework/alembic/comment/alembic.ini

@@ -0,0 +1,52 @@
+# A generic, single database configuration.
+
+[alembic]
+# path to migration scripts.
+# this is typically a path given in POSIX (e.g. forward slashes)
+# format, relative to the token %(here)s which refers to the location of this
+# ini file
+script_location = %(here)s/alembic
+
+# sys.path path, will be prepended to sys.path if present.
+prepend_sys_path = .
+
+path_separator = os
+
+sqlalchemy.url = driver://user:pass@localhost/dbname
+
+
+[post_write_hooks]
+
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARNING
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARNING
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

core_framework/alembic/extension/alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration with an async dbapi.

core_framework/alembic/extension/alembic/env.py

@@ -0,0 +1,98 @@
+import asyncio
+from logging.config import fileConfig
+
+from sqlalchemy import pool, text
+from sqlalchemy.engine import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+
+from alembic import context
+from core_framework.core import settings
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Interpret the config file for Python logging.
+# This line sets up loggers basically.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+# from myapp import mymodel
+# target_metadata = mymodel.Base.metadata
+target_metadata = None
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+config.set_main_option("sqlalchemy.url", settings.write_postgres.alembic_postgres_url)
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+
+    Calls to context.execute() here emit the given string to the
+    script output.
+
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def do_run_migrations(connection: Connection) -> None:
+    quoted_schema_name = f'"{settings.postgres_schemas.schema_extension.replace('"', '""')}"'
+
+    connection.execute(text(f"create schema if not exists {quoted_schema_name}"))
+    connection.execute(text(f"set search_path to {quoted_schema_name}"))
+    connection.commit()
+
+    context.configure(connection=connection, target_metadata=target_metadata)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations() -> None:
+    """In this scenario we need to create an Engine
+    and associate a connection with the context.
+
+    """
+
+    connectable = async_engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    async with connectable.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await connectable.dispose()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode."""
+
+    asyncio.run(run_async_migrations())
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

core_framework/alembic/extension/alembic/script.py.mako

@@ -0,0 +1,28 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, Sequence[str], None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    ${downgrades if downgrades else "pass"}

core_framework/alembic/extension/alembic/versions/0389226049cb_add_pg_trgm_extension.py

@@ -0,0 +1,25 @@
+"""add pg_trgm extension
+
+Revision ID: 0389226049cb
+Revises: b0ba0d8a284e
+Create Date: 2026-02-08 13:38:47.721189
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "0389226049cb"
+down_revision: str | Sequence[str] | None = "b0ba0d8a284e"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("create extension pg_trgm")
+
+
+def downgrade() -> None:
+    op.execute("drop extension if exists pg_trgm")

core_framework/alembic/extension/alembic/versions/5dc58b016cf5_add_citext_extension.py

@@ -0,0 +1,25 @@
+"""add citext extension
+
+Revision ID: 5dc58b016cf5
+Revises:
+Create Date: 2025-12-26 08:31:16.109190
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "5dc58b016cf5"
+down_revision: str | Sequence[str] | None = None
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("create extension citext")
+
+
+def downgrade() -> None:
+    op.execute("drop extension if exists citext")

core_framework/alembic/extension/alembic/versions/b0ba0d8a284e_add_pg_stat_statements_extension.py

@@ -0,0 +1,25 @@
+"""add pg_stat_statements extension
+
+Revision ID: b0ba0d8a284e
+Revises: 5dc58b016cf5
+Create Date: 2026-01-06 06:34:34.095569
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "b0ba0d8a284e"
+down_revision: str | Sequence[str] | None = "5dc58b016cf5"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.execute("create extension pg_stat_statements")
+
+
+def downgrade() -> None:
+    op.execute("drop extension if exists pg_stat_statements")