hockey-blast-common-lib 0.1.17__tar.gz → 0.1.19__tar.gz

{hockey_blast_common_lib-0.1.17 → hockey_blast_common_lib-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hockey-blast-common-lib
-Version: 0.1.17
+Version: 0.1.19
 Summary: Common library for shared functionality and DB models
 Author: Pavel Kletskov
 Author-email: kletskov@gmail.com

hockey_blast_common_lib-0.1.19/hockey_blast_common_lib/db_connection.py

@@ -0,0 +1,42 @@
+import os
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+# Database connection parameters per organization
+DB_PARAMS = {
+    "frontend": {
+        "dbname": os.getenv("DB_NAME", "hockey_blast"),
+        "user": os.getenv("DB_USER", "frontend_user"),
+        "password": os.getenv("DB_PASSWORD", "hockey-blast"),
+        "host": os.getenv("DB_HOST", "localhost"),
+        "port": int(os.getenv("DB_PORT", 5432))
+    },
+
+    "frontend-sample-db": {
+        "dbname": os.getenv("DB_NAME_SAMPLE", "hockey_blast_sample"),
+        "user": os.getenv("DB_USER_SAMPLE", "frontend_user"),
+        "password": os.getenv("DB_PASSWORD_SAMPLE", "hockey-blast"),
+        "host": os.getenv("DB_HOST_SAMPLE", "localhost"),
+        "port": int(os.getenv("DB_PORT_SAMPLE", 5432))
+    },
+
+    "boss": {
+        "dbname": os.getenv("DB_NAME_BOSS", "hockey_blast"),
+        "user": os.getenv("DB_USER_BOSS", "boss"),
+        "password": os.getenv("DB_PASSWORD_BOSS", "WrongPassword"),
+        "host": os.getenv("DB_HOST_BOSS", "localhost"),
+        "port": int(os.getenv("DB_PORT_BOSS", 5432))
+    },
+}
+
+def get_db_params(config_name):
+    if config_name not in DB_PARAMS:
+        raise ValueError(f"Invalid organization: {config_name}")
+    return DB_PARAMS[config_name]
+
+def create_session(config_name):
+    db_params = get_db_params(config_name)
+    db_url = f"postgresql://{db_params['user']}:{db_params['password']}@{db_params['host']}:{db_params['port']}/{db_params['dbname']}"
+    engine = create_engine(db_url)
+    Session = sessionmaker(bind=engine)
+    return Session()

hockey_blast_common_lib-0.1.19/hockey_blast_common_lib/restore_sample_db.sh

@@ -0,0 +1,37 @@
+#!/bin/zsh
+
+# Database credentials from environment variables
+DB_USER=${DB_USER:-"frontend_user"}
+DB_PASSWORD=${DB_PASSWORD:-"hockey-blast"}
+DB_NAME=${DB_NAME:-"hockey_blast_sample"}
+DB_HOST=${DB_HOST:-"localhost"}
+DB_PORT=${DB_PORT:-"5432"}
+COMPRESSED_DUMP_FILE="hockey_blast_sample_backup.sql.gz"
+
+# Superuser credentials
+SUPERUSER="your_superuser"
+SUPERUSER_PASSWORD="your_superuser_password"
+
+# Export PGPASSWORD to avoid password prompt
+export PGPASSWORD=$SUPERUSER_PASSWORD
+
+# Drop the existing database if it exists
+psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT -d postgres --command="SELECT pg_terminate_backend(pg_stat_activity.pid) FROM pg_stat_activity WHERE pg_stat_activity.datname = '$DB_NAME' AND pid <> pg_backend_pid();"
+psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT -d postgres --command="DROP DATABASE IF EXISTS $DB_NAME"
+
+# Create a new database
+psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT -d postgres --command="CREATE DATABASE $DB_NAME OWNER $SUPERUSER"
+
+# Export PGPASSWORD for frontend_user user
+export PGPASSWORD=$DB_PASSWORD
+
+# Restore the database from the dump file with --no-owner option
+gunzip -c $COMPRESSED_DUMP_FILE | pg_restore --username=$DB_USER --host=$DB_HOST --port=$DB_PORT --dbname=$DB_NAME --format=custom --no-owner
+
+# Create the frontend_user if it does not exist
+psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --dbname=$DB_NAME --command="DO \$\$ BEGIN IF NOT EXISTS (SELECT FROM pg_catalog.pg_roles WHERE rolname = 'frontend_user') THEN CREATE ROLE frontend_user LOGIN PASSWORD '$DB_PASSWORD'; END IF; END \$\$;"
+
+# Grant necessary permissions to the  user
+psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --dbname=$DB_NAME  --command="GRANT USAGE, SELECT ON ALL SEQUENCES IN SCHEMA public TO frontend_user"
+
+echo "Database restore completed: $DB_NAME"

{hockey_blast_common_lib-0.1.17 → hockey_blast_common_lib-0.1.19}/hockey_blast_common_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hockey-blast-common-lib
-Version: 0.1.17
+Version: 0.1.19
 Summary: Common library for shared functionality and DB models
 Author: Pavel Kletskov
 Author-email: kletskov@gmail.com

{hockey_blast_common_lib-0.1.17 → hockey_blast_common_lib-0.1.19}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='hockey-blast-common-lib',  # The name of your package
-    version='0.1.17',
+    version='0.1.19',
     description='Common library for shared functionality and DB models',
     author='Pavel Kletskov',
     author_email='kletskov@gmail.com',

hockey_blast_common_lib-0.1.17/hockey_blast_common_lib/db_connection.py

@@ -1,43 +0,0 @@
-# import psycopg2
-from sqlalchemy import create_engine
-from sqlalchemy.orm import sessionmaker
-
-# Database connection parameters per organization
-DB_PARAMS = {
-    "frontend": {
-        "dbname": "hockey_blast",
-        "user": "frontend_user",
-        "password": "hockey-blast",
-        "host": "localhost",
-        "port": 5432
-    },
-
-    "frontend-sample-db": {
-        "dbname": "hockey_blast_sample",
-        "user": "frontend_user",
-        "password": "hockey-blast",
-        "host": "localhost",
-        "port": 5432
-    },
-
-    "boss": {
-        "dbname": "hockey_blast",
-        "user": "boss",
-        "password": "WrongPassword",
-        "host": "localhost",
-        "port": 5432
-    },
-
-}
-
-def get_db_params(config_name):
-    if config_name not in DB_PARAMS:
-        raise ValueError(f"Invalid organization: {config_name}")
-    return DB_PARAMS[config_name]
-
-def create_session(config_name):
-    db_params = get_db_params(config_name)
-    db_url = f"postgresql://{db_params['user']}:{db_params['password']}@{db_params['host']}:{db_params['port']}/{db_params['dbname']}"
-    engine = create_engine(db_url)
-    Session = sessionmaker(bind=engine)
-    return Session()

hockey_blast_common_lib-0.1.17/hockey_blast_common_lib/restore_sample_db.sh

@@ -1,34 +0,0 @@
-#!/bin/zsh
-
-# Database credentials from environment variables
-DB_USER=${DB_USER:-"frontend_user"}
-DB_PASSWORD=${DB_PASSWORD:-"hockey-blast"}
-DB_NAME=${DB_NAME:-"hockey_blast_sample"}
-DB_HOST=${DB_HOST:-"localhost"}
-DB_PORT=${DB_PORT:-"5432"}
-COMPRESSED_DUMP_FILE="hockey_blast_sample_backup.sql.gz"
-
-# Superuser credentials
-SUPERUSER="your_superuser"
-SUPERUSER_PASSWORD="your_superuser_password"
-
-# Export PGPASSWORD to avoid password prompt
-export PGPASSWORD=$SUPERUSER_PASSWORD
-
-# Drop the existing database if it exists
-psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --command="SELECT pg_terminate_backend(pg_stat_activity.pid) FROM pg_stat_activity WHERE pg_stat_activity.datname = '$DB_NAME' AND pid <> pg_backend_pid();"
-psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --command="DROP DATABASE IF EXISTS $DB_NAME"
-
-# Create a new database
-psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --command="CREATE DATABASE $DB_NAME OWNER $SUPERUSER"
-
-# Export PGPASSWORD for read-only user
-export PGPASSWORD=$DB_PASSWORD
-
-# Restore the database from the dump file with --no-owner option
-gunzip -c $COMPRESSED_DUMP_FILE | pg_restore --username=$DB_USER --host=$DB_HOST --port=$DB_PORT --dbname=$DB_NAME --format=custom --no-owner
-
-# Grant necessary permissions to the read-only user
-psql --username=$SUPERUSER --host=$DB_HOST --port=$DB_PORT --dbname=$DB_NAME --command="GRANT USAGE, SELECT ON ALL SEQUENCES IN SCHEMA public TO frontend_user"
-
-echo "Database restore completed: $DB_NAME"