postgresql-charms-single-kernel 0.0.1__py3-none-any.whl → 16.0.1__py3-none-any.whl

{postgresql_charms_single_kernel-0.0.1.dist-info → postgresql_charms_single_kernel-16.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: postgresql-charms-single-kernel
-Version: 0.0.1
+Version: 16.0.1
 Summary: Shared and reusable code for PostgreSQL-related charms
 Author-email: Canonical Data Platform <data-platform@lists.launchpad.net>
 License-Expression: Apache-2.0

postgresql_charms_single_kernel-16.0.1.dist-info/RECORD

@@ -0,0 +1,11 @@
+single_kernel_postgresql/__init__.py,sha256=F57JUcML43xgR0kpj9csryZkAoDSBPLLRjzSTHPaTd4,116
+single_kernel_postgresql/abstract_charm.py,sha256=grlJCmpz8rq7_NKVkyuoK7_8I2r6DRkh5DhbtyPWna8,792
+single_kernel_postgresql/config/__init__.py,sha256=k9Ud5ZZNd3l0nn8xi8AIlT45oZk8hJ542BjAFGDJzuc,140
+single_kernel_postgresql/config/literals.py,sha256=4EpAnoxCEMfOCPpFD5dyAU_0h2OwAHgEX9OjTWNcaQg,527
+single_kernel_postgresql/utils/__init__.py,sha256=VwAEW3wYjs99q38bid47aS6Os4s3catK4H5HfdPkp94,121
+single_kernel_postgresql/utils/filesystem.py,sha256=CJ2iXqFPKM0NfqqZSTDlENOrM0RW7rmTmcuzwV0bR9A,552
+single_kernel_postgresql/utils/postgresql.py,sha256=bM57IUWPX4x0u7yL68zYdlPcBs7SIDRKj28zzeMXXeI,75950
+postgresql_charms_single_kernel-16.0.1.dist-info/METADATA,sha256=CcN0WZUEcOdlf2ho3l_K60JTq8qMhzanZ5NW73EcMc4,484
+postgresql_charms_single_kernel-16.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+postgresql_charms_single_kernel-16.0.1.dist-info/top_level.txt,sha256=fH85HKyfDV3--1JuYe-rWJmN5XTlXVXGOBO5iNA36Rk,25
+postgresql_charms_single_kernel-16.0.1.dist-info/RECORD,,

single_kernel_postgresql/config/literals.py

@@ -5,6 +5,9 @@
 This module should contain the literals used in the charms (paths, enums, etc).
 """
 
+# Permissions.
+POSTGRESQL_STORAGE_PERMISSIONS = 0o700
+
 # Relations.
 PEER = "database-peers"
 
@@ -13,5 +16,6 @@ BACKUP_USER = "backup"
 MONITORING_USER = "monitoring"
 REPLICATION_USER = "replication"
 REWIND_USER = "rewind"
+SNAP_USER = "_daemon_"
 USER = "operator"
 SYSTEM_USERS = [MONITORING_USER, REPLICATION_USER, REWIND_USER, USER]

single_kernel_postgresql/utils/filesystem.py

@@ -0,0 +1,20 @@
+# Copyright 2025 Canonical Ltd.
+# See LICENSE file for licensing details.
+"""Filesystem utilities."""
+
+import os
+import pwd
+
+from ..config.literals import SNAP_USER
+
+
+def change_owner(path: str) -> None:
+    """Change the ownership of a file or a directory to the snap user.
+
+    Args:
+        path: path to a file or directory.
+    """
+    # Get the uid/gid for the snap user.
+    user_database = pwd.getpwnam(SNAP_USER)
+    # Set the correct ownership for the file or directory.
+    os.chown(path, uid=user_database.pw_uid, gid=user_database.pw_gid)

single_kernel_postgresql/utils/postgresql.py

@@ -20,14 +20,18 @@ Any charm using this library should import the `psycopg2` or `psycopg2-binary` d
 """
 
 import logging
+import os
+import pwd
 from collections import OrderedDict
+from datetime import datetime, timezone
 from typing import Dict, List, Optional, Set, Tuple
 
 import psycopg2
 from ops import ConfigData
 from psycopg2.sql import SQL, Identifier, Literal
 
-from ..config.literals import BACKUP_USER, SYSTEM_USERS
+from ..config.literals import BACKUP_USER, POSTGRESQL_STORAGE_PERMISSIONS, SNAP_USER, SYSTEM_USERS
+from .filesystem import change_owner
 
 # Groups to distinguish HBA access
 ACCESS_GROUP_IDENTITY = "identity_access"
@@ -1071,6 +1075,32 @@ class PostgreSQL:
             cursor = connection.cursor()
 
             if temp_location is not None:
+                # Fix permissions on the temporary tablespace location when a reboot happens and tmpfs is being used.
+                temp_location_stats = os.stat(temp_location)
+                if (
+                    pwd.getpwuid(temp_location_stats.st_uid).pw_name != SNAP_USER
+                    or temp_location_stats.st_mode != POSTGRESQL_STORAGE_PERMISSIONS
+                ):
+                    change_owner(temp_location)
+                    os.chmod(temp_location, POSTGRESQL_STORAGE_PERMISSIONS)
+                    # Rename existing temp tablespace if it exists, instead of dropping it.
+                    cursor.execute("SELECT TRUE FROM pg_tablespace WHERE spcname='temp';")
+                    if cursor.fetchone() is not None:
+                        new_name = f"temp_{datetime.now(timezone.utc).strftime('%Y%m%d%H%M%S')}"
+                        cursor.execute(f"ALTER TABLESPACE temp RENAME TO {new_name};")
+
+                        # List temp tablespaces with suffix for operator follow-up cleanup and log them.
+                        cursor.execute(
+                            "SELECT spcname FROM pg_tablespace WHERE spcname LIKE 'temp_%';"
+                        )
+                        temp_tbls = sorted([row[0] for row in cursor.fetchall()])
+                        logger.info(
+                            "There are %d temp tablespaces that should be checked and removed: %s",
+                            len(temp_tbls),
+                            ", ".join(temp_tbls),
+                        )
+
+                # Ensure a fresh temp tablespace exists at the expected location.
                 cursor.execute("SELECT TRUE FROM pg_tablespace WHERE spcname='temp';")
                 if cursor.fetchone() is None:
                     cursor.execute(f"CREATE TABLESPACE temp LOCATION '{temp_location}';")

postgresql_charms_single_kernel-0.0.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-single_kernel_postgresql/__init__.py,sha256=F57JUcML43xgR0kpj9csryZkAoDSBPLLRjzSTHPaTd4,116
-single_kernel_postgresql/abstract_charm.py,sha256=grlJCmpz8rq7_NKVkyuoK7_8I2r6DRkh5DhbtyPWna8,792
-single_kernel_postgresql/config/__init__.py,sha256=k9Ud5ZZNd3l0nn8xi8AIlT45oZk8hJ542BjAFGDJzuc,140
-single_kernel_postgresql/config/literals.py,sha256=2McbLq3gCSFEXXgBYu52nYLb-RS088sNMYO4JZFWMSY,449
-single_kernel_postgresql/utils/__init__.py,sha256=VwAEW3wYjs99q38bid47aS6Os4s3catK4H5HfdPkp94,121
-single_kernel_postgresql/utils/postgresql.py,sha256=_N9y_fcr2l8tNscfgZ5dYnWFMp8NvV9bydFmEUmntsY,74191
-postgresql_charms_single_kernel-0.0.1.dist-info/METADATA,sha256=NMis-jmTpyfdIF2MJwaglMCVKJpXxamP47YKw4FLQeU,483
-postgresql_charms_single_kernel-0.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-postgresql_charms_single_kernel-0.0.1.dist-info/top_level.txt,sha256=fH85HKyfDV3--1JuYe-rWJmN5XTlXVXGOBO5iNA36Rk,25
-postgresql_charms_single_kernel-0.0.1.dist-info/RECORD,,