rk-digi 2.1.0

Python package with useful stuff for projects in Randers Kommune - Digitalisering

RK-digitalisering-package

Python package with useful stuff for projects in Randers Kommune - Digitalisering.

Classes

ManagedOAuth2Session (sync)

A drop-in replacement for requests.Session/OAuth2Session that handles OAuth2 token acquisition and refresh for both client credentials and refresh token flows.

Example

from rkdigi import ManagedOAuth2Session

session = ManagedOAuth2Session(
	token_url="https://example.com/oauth/token",
	client_id="your-client-id",
	client_secret="your-client-secret"
)
res = session.get("https://example.com/api/data")

DatabaseManager (sync + async)

DatabaseManager is for database connections and session management for both Microsoft SQL Server and PostgreSQL databases. It supports both synchronous and asynchronous usage.

• Connection Profiles: You can create multiple DatabaseManager instances, each identified by a unique profile_name. If a new object is created with the same profile_name, the previously create object will be returned.
• Credential Handling: Credentials and connection details can be provided directly to the constructor, or (if omitted) loaded automatically from environment variables based on the profile_name. For example, if profile_name is mydb, the class will look for environment variables like MYDB_HOST, MYDB_PORT, MYDB_USER, and MYDB_PASSWORD.
• Sync and Async Support: By default, the class operates in synchronous mode. To use asynchronous mode, set async_mode=True when creating the instance. The class will then use SQLAlchemy's async engine and session management.
• Session Management:
  • In sync mode, use get_session() as a context manager to obtain a SQLAlchemy session.
  • In async mode, use get_session_async() as an async context manager to obtain an async session.
• Resource Cleanup:
  • In sync mode, call dispose() to close the engine and clean up resources.
  • In async mode, call await dispose_async() to clean up async resources.
• Dependencies:
  • apache-airflow-providers-postgres (for Airflow Postgres connections)
  • apache-airflow-providers-microsoft-mssql (for Airflow MSSQL connections)
  • psycopg2 (for PostgreSQL, sync)
  • asyncpg (for PostgreSQL, async)
  • pymssql (for Microsoft SQL Server, sync)
  • aioodbc (for Microsoft SQL Server, async)
  • pyodbc (required by aioodbc for async SQL Server)

Sync example

Basic sync example supplying credentials in constructor. DatabaseManager default to sync.

from sqlalchemy import text
from rkdigi import DatabaseManager

db_manager = DatabaseManager(
	profile_name='db_mydb',
	db_type='mssql',
	username='username',
	password='password',
	database="mydatabase",
	host='demo.com',
	port=1433,
)
with db_manager.get_session() as session:
	res = session.execute(text("SELECT 1"))
db_manager.dispose()

Async example

Basic async example getting credentials from environment variables. If no credentials are given then DatabaseManager will try to get them based on profile_name. In this example: DB_MYDB_HOST, DB_MYDB_PORT, etc. are used.

import asyncio
from sqlalchemy import text
from rkdigi import DatabaseManager

async def my_db_func():
	db_manager = DatabaseManager(
		profile_name='db_mydb',
		db_type='postgres',
		async_mode=True
	)
	async with db_manager.get_session_async() as session:
		res = await session.execute(text("SELECT 1"))
	await db_manager.dispose_async()
asyncio.run(my_db_func())

Airflow example

Sync example using an airflow connection.

from sqlalchemy import text
from rkdigi import DatabaseManager

db_manager = DatabaseManager(
	profile_name='db_mydb',
	db_type='postgres',
	airflow_connection_id='mydb_id'
)
with db_manager.get_session() as session:
	res = session.execute(text("SELECT 1"))
db_manager.dispose()

Create tables example

If DatabaseManager init is provided with a base model, it will create the tables in the database after testing it can connect.

from rkdigi import DatabaseManager
from mymodel import MyBaseModel

db_manager = DatabaseManager(
	profile_name='db_mydb',
	db_type='postgres',
	base_model=MyBaseModel
)

Singleton explanation

DatabaseManager implements singleton behavior based on profile_name

from rkdigi import DatabaseManager

db_manager = DatabaseManager(
	profile_name='db_mydb',  # same profile_name
	db_type='postgres',
	username='username',
	password='password',
	host='demo123.com'
)

db_manager = DatabaseManager(
	profile_name='db_mydb',  # same profile_name
	db_type='mssql',
	username='user',
	password='pass',
	host='demoABC.com'
)
# db_manager will still have an engine for the postgres database on demo123.com
db_manager.dispose()  # Only after calling dispose will it be re-initialized
db_manager = DatabaseManager(
	profile_name='db_mydb',  # same profile_name
	db_type='mssql',
	username='user',
	password='pass',
	host='demoABC.com'
)
# Now the new credentials are applied and a new engine created

EmailSender (sync + async)

EmailSender is for sending emails from a SMTP server. It takes both just email addresses (string) and address headers with names (tuple), like; ('Name', '[e-mail]). Emails with html body will get a plain text body added as well, fallback for email clients not supporting html. Attachments can be given either as a path to a file (string) or as filename and data in bytes (tuple).

Address formats:

• A single address can be given as a string email: '[e-mail]'
• A single named address can be given as a tuple: ('Name', '[e-mail]')
• Multiple recipients/CC must be given as a sequence (typically a list), e.g. ['[e-mail]', '[e-mail]'], [('A', '[e-mail]'), '[e-mail]'] or ('[e-mail]', ('B', '[e-mail]'))

Note: A 2-tuple is reserved for the named-address form (name, email). Passing a tuple of two emails like ('[e-mail]', '[e-mail]') is not supported; use a list instead.

Sync example

from rkdigi import EmailSender

email_sender = EmailSender(smtp_server='smtp.example.com', smtp_port=25)
email_sender.send_email(
	sender=('No Reply', '[e-mail]'),
	reply_to='[e-mail]',
	recipients='[e-mail]',
	cc=[('CC', '[e-mail]')],
	subject='Test Subject',
	body='<html><body>Test Body</body></html>',
	attachments=[('myfile.txt', b'<somebytes>')]
)

Async example

import asyncio
from rkdigi import EmailSender

async def send_email_func():
    email_sender = EmailSender()
    await email_sender.send_email_async(
        sender='[e-mail]',
        recipients=[('One', '[e-mail]'), '[e-mail]'],
        subject='Test Subject',
        body='Test Body',
		attachments=['testdir/test.txt']
    )

asyncio.run(send_email_func())

EmailReader (sync + async)

EmailReader is for reading emails from an IMAP server. The class provides a method for getting a list of mailboxes/folders: list_mailboxes / list_mailboxes_async.

Reading/getting emails can be done with get_emails / get_emails_async. The search_criteria parameter will filter the emails in the mail box and follow IMAP RFC 3501, they can be found in the SEARCH section. The set_flags and del_flags parameters are the flags will be set or removed for the returned emails. If set to None no flags will be set or removed (this is the default for del_flags, \\Seen is default for set_flags). Multiple flags must be separated by a space e.g. "\\Seen \\Flagged". Documentation for flags which can be set can be found at imap Enum Flag. Two lists are returned, one with the email data as EmailMessage objects (with the uid in attribute uid e.g. email_obj.uid), the other with a list of ids for emails which could not be fetched. The ids are bytes e.g. [b'1', b'2'].

A single email can be fetched by uid using get_email_by_uid / get_email_by_uid_async, it takes the parameters uid and mailbox and returns a EmailMessage object with uid attribute set.

Sync example

from rkdigi import EmailReader
reader = EmailReader(
	email='[e-mail]',
	password='password',
	imap_server='imap.example.com',
	imap_port=143
)
folders = reader.list_mailboxes()
emails, failed_email_ids = reader.get_emails(
	mailbox=folder[0],
	criteria="UNSEEN",
	set_flags=None,
	del_flags="\\Flagged"
)
first_email = emails[0]
if not first_email.is_multipart():
	charset = first_email.get_content_charset()
	body = first_email.get_payload(decode=True)
	normal_string_body = body.decode(charset)

email_by_id = reader.get_email_by_uid(uid=b"1", mailbox="INBOX")

Async example

import asyncio
from rkdigi import EmailReader

async def get_emails():
	reader = EmailReader(
		email='[e-mail]',
		password='password',
		imap_server='imap.example.com',
		imap_port=143
	)
	folders = await reader.list_mailboxes_async()
	emails, failed_email_ids = await reader.get_emails_async(
		mailbox=folders[0],
		criteria="ALL",
		set_flags="\\Seen",
		max=10
	)
asyncio.run(get_emails())

async def get_by_id():
	reader = EmailReader(
		email='[e-mail]',
		password='password',
		imap_server='imap.example.com',
		imap_port=143
	)
	email_by_id = await reader.get_email_by_uid_async(uid=b"1", mailbox="INBOX")
asyncio.run(get_by_id())

EmailManager (sync + async)

EmailManager just initiate