hypern 0.2.0__cp310-none-win32.whl

hypern/datastructures.py

@@ -0,0 +1,40 @@
+from typing import Optional
+from enum import Enum
+from pydantic import BaseModel, AnyUrl, EmailStr
+
+
+class BaseModelWithConfig(BaseModel):
+    model_config = {"extra": "allow"}
+
+
+class Contact(BaseModelWithConfig):
+    name: Optional[str] = None
+    url: Optional[AnyUrl] = None
+    email: Optional[EmailStr] = None
+
+
+class License(BaseModelWithConfig):
+    name: str
+    identifier: Optional[str] = None
+    url: Optional[AnyUrl] = None
+
+
+class Info(BaseModelWithConfig):
+    title: str
+    summary: Optional[str] = None
+    description: Optional[str] = None
+    contact: Optional[Contact] = None
+    license: Optional[License] = None
+    version: str
+
+
+class HTTPMethod(Enum):
+    GET = "GET"
+    POST = "POST"
+    PUT = "PUT"
+    DELETE = "DELETE"
+    PATCH = "PATCH"
+    OPTIONS = "OPTIONS"
+    HEAD = "HEAD"
+    TRACE = "TRACE"
+    CONNECT = "CONNECT"

hypern/db/nosql/__init__.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+from typing import List, TypedDict
+
+from uuid import uuid4
+from mongoengine import connect
+
+
+class TypedDictModel(TypedDict):
+    host: str
+    alias: str
+
+
+class NoSqlConfig:
+    def __init__(self, dbs_config: List[TypedDictModel]):
+        self.dbs_config = dbs_config
+
+    def _connect_db(self, db_config: TypedDictModel):
+        _alias = db_config.get("alias", str(uuid4()))
+        connect(host=db_config["host"], alias=_alias)
+
+    def init_app(self, app):
+        self.app = app  # noqa
+        # connect
+        for db_config in self.dbs_config:
+            self._connect_db(db_config)

hypern/db/nosql/addons/__init__.py

@@ -0,0 +1,4 @@
+from .encrypted import EncryptedField
+from .password import PasswordField
+
+__all__ = ["EncryptedField", "PasswordField"]

hypern/db/nosql/addons/color.py

@@ -0,0 +1,16 @@
+from mongoengine import BaseField
+import re
+
+
+class ColorField(BaseField):
+    def validate(self, value):
+        color_regex = r"^#([A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})$"
+        if not re.match(color_regex, value):
+            self.error("Invalid color format. Use hexadecimal color codes (e.g., #FF0000)")
+        return True
+
+    def to_mongo(self, value):
+        return value
+
+    def to_python(self, value):
+        return value

hypern/db/nosql/addons/daterange.py

@@ -0,0 +1,30 @@
+from mongoengine import BaseField
+from datetime import datetime
+
+
+class DateRangeField(BaseField):
+    def __init__(self, **kwargs):
+        super(DateRangeField, self).__init__(**kwargs)
+
+    def validate(self, value):
+        if not isinstance(value, dict) or "start" not in value or "end" not in value:
+            self.error('DateRangeField must be a dictionary with "start" and "end" keys')
+        # Use get to safely access keys
+        start = value.get("start")
+        end = value.get("end")
+        # Check if both "start" and "end" are present
+        if start is None or end is None:
+            self.error('DateRangeField must contain both "start" and "end" keys')
+
+        # Check if "start" and "end" are datetime objects
+        if not isinstance(value["start"], datetime) or not isinstance(value["end"], datetime):
+            self.error('DateRangeField "start" and "end" must be datetime objects')
+        if value["start"] > value["end"]:
+            self.error('DateRangeField "start" must be earlier than "end"')
+        return True
+
+    def to_mongo(self, value):
+        return value
+
+    def to_python(self, value):
+        return value

hypern/db/nosql/addons/encrypted.py

@@ -0,0 +1,53 @@
+import os
+from typing import Any, Optional
+from mongoengine.base import BaseField
+
+from cryptography.hazmat.primitives import padding
+
+from hypern.security import EDEngine, AESEngine
+
+
+class EncryptedField(BaseField):
+    """
+    A custom MongoEngine field that encrypts data using AES-256-CBC.
+
+    The field automatically handles encryption when saving to MongoDB and
+    decryption when retrieving data.
+
+    Attributes:
+        engine: Encryption engine to use. If not provided, will use AES-256-CBC
+    """
+
+    def __init__(self, engine: Optional[EDEngine] = None, **kwargs):
+        if not engine:
+            key = os.urandom(32)
+            iv = os.urandom(16)
+            padding_class = padding.PKCS7
+            self.engine = AESEngine(secret_key=key, iv=iv, padding_class=padding_class)
+        else:
+            self.engine = engine  # type: ignore
+        super(EncryptedField, self).__init__(**kwargs)
+
+    def to_mongo(self, value: Any) -> Optional[str]:
+        """Convert a Python object to a MongoDB-compatible format."""
+        if value is None:
+            return None
+        return self.engine.encrypt(value)
+
+    def to_python(self, value: Optional[str]) -> Optional[str]:
+        """Convert a MongoDB-compatible format to a Python object."""
+        if value is None:
+            return None
+        if isinstance(value, bytes):
+            return self.engine.decrypt(value)
+        return value
+
+    def prepare_query_value(self, op, value: Any) -> Optional[str]:
+        """Prepare a value used in a query."""
+        if value is None:
+            return None
+
+        if op in ("set", "upsert"):
+            return self.to_mongo(value)
+
+        return value

hypern/db/nosql/addons/password.py

@@ -0,0 +1,134 @@
+from mongoengine.base import BaseField
+import weakref
+from passlib.context import CryptContext
+import re
+from typing import Optional, Any
+
+
+class PasswordField(BaseField):
+    """
+    A custom password field using passlib for hashing and weakref for reference management.
+    Supports multiple hashing schemes and automatic upgrade of hash algorithms.
+    """
+
+    # Class-level password context - shared across all instances
+    pwd_context = CryptContext(
+        # List of hashing schemes in order of preference
+        schemes=["argon2", "pbkdf2_sha256", "bcrypt_sha256"],
+        # Mark argon2 as default
+        default="argon2",
+        # Argon2 parameters
+        argon2__rounds=4,
+        argon2__memory_cost=65536,
+        argon2__parallelism=2,
+        # PBKDF2 parameters
+        pbkdf2_sha256__rounds=29000,
+    )
+
+    def __init__(
+        self,
+        min_length: int = 8,
+        require_number: bool = False,
+        require_special: bool = False,
+        require_uppercase: bool = False,
+        require_lowercase: bool = False,
+        **kwargs,
+    ):
+        """
+        Initialize the password field with validation rules.
+
+        Args:
+            min_length: Minimum password length
+            require_number: Require at least one number
+            require_special: Require at least one special character
+            require_uppercase: Require at least one uppercase letter
+            require_lowercase: Require at least one lowercase letter
+        """
+        self.min_length = min_length
+        self.require_number = require_number
+        self.require_special = require_special
+        self.require_uppercase = require_uppercase
+        self.require_lowercase = require_lowercase
+
+        # Use weakref to store references to parent documents
+        self.instances = weakref.WeakKeyDictionary()
+
+        kwargs["required"] = True
+        super(PasswordField, self).__init__(**kwargs)
+
+    def validate_password(self, password: str) -> tuple[bool, str]:
+        """Validate password strength."""
+
+        if len(password) < self.min_length:
+            return False, f"Password must be at least {self.min_length} characters long"
+
+        if self.require_number and not re.search(r"\d", password):
+            return False, "Password must contain at least one number"
+
+        if self.require_special and not re.search(r"[!@#$%^&*(),.?\":{}|<>]", password):
+            return False, "Password must contain at least one special character"
+
+        if self.require_uppercase and not re.search(r"[A-Z]", password):
+            return False, "Password must contain at least one uppercase letter"
+
+        if self.require_lowercase and not re.search(r"[a-z]", password):
+            return False, "Password must contain at least one lowercase letter"
+
+        return True, ""
+
+    def hash_password(self, password: str) -> str:
+        """Hash password using the configured passlib context."""
+        return self.pwd_context.hash(password)
+
+    def verify_password(self, password: str, hash: str) -> tuple[bool, Optional[str]]:
+        """
+        Verify password and return tuple of (is_valid, new_hash).
+        new_hash is provided if the hash needs to be upgraded.
+        """
+        try:
+            is_valid = self.pwd_context.verify(password, hash)
+            # Check if the hash needs to be upgraded
+            if is_valid and self.pwd_context.needs_update(hash):
+                return True, self.hash_password(password)
+            return is_valid, None
+        except Exception:
+            return False, None
+
+    def __get__(self, instance, owner):
+        """Custom getter using weakref."""
+        if instance is None:
+            return self
+        return self.instances.get(instance)
+
+    def __set__(self, instance, value):
+        """Custom setter using weakref."""
+        if value and isinstance(value, str):
+            # Validate and hash new password
+            is_valid, error = self.validate_password(value)
+            if not is_valid:
+                raise ValueError(error)
+            hashed = self.hash_password(value)
+            self.instances[instance] = hashed
+            instance._data[self.name] = hashed
+        else:
+            # If it's already hashed or None
+            self.instances[instance] = value
+            instance._data[self.name] = value
+
+    def to_mongo(self, value: str) -> Optional[str]:
+        """Convert to MongoDB-compatible value."""
+        if value is None:
+            return None
+        return self.hash_password(value)
+
+    def to_python(self, value: str) -> str:
+        """Convert from MongoDB to Python."""
+        return value
+
+    def prepare_query_value(self, op, value: Any) -> Optional[str]:
+        """Prepare value for database operations."""
+        if value is None:
+            return None
+        if op == "exact":
+            return self.hash_password(value)
+        return value

hypern/db/nosql/addons/unicode.py

@@ -0,0 +1,10 @@
+from mongoengine import StringField
+
+
+class UnicodeField(StringField):
+    def validate(self, value):
+        try:
+            value.encode("utf-8")
+        except UnicodeEncodeError:
+            self.error("Value must be valid Unicode")
+        return True

hypern/db/sql/__init__.py

@@ -0,0 +1,179 @@
+# -*- coding: utf-8 -*-
+import asyncio
+import threading
+import traceback
+from contextlib import asynccontextmanager
+from contextvars import ContextVar, Token
+from datetime import datetime
+from typing import Dict, Optional, Union
+from uuid import uuid4
+
+from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_scoped_session
+from sqlalchemy.orm import Session, sessionmaker
+from sqlalchemy.sql.expression import Delete, Insert, Update
+
+from hypern.hypern import Request, Response
+
+from .repository import Model, PostgresRepository
+
+
+class ContextStore:
+    def __init__(self, cleanup_interval: int = 300, max_age: int = 3600):
+        """
+        Initialize ContextStore with automatic session cleanup.
+
+        :param cleanup_interval: Interval between cleanup checks (in seconds)
+        :param max_age: Maximum age of a session before it's considered expired (in seconds)
+        """
+        self._session_times: Dict[str, datetime] = {}
+        self.session_var = ContextVar("session_id", default=None)
+
+        self._max_age = max_age
+        self._cleanup_interval = cleanup_interval
+        self._cleanup_thread: Optional[threading.Thread] = None
+        self._stop_event = threading.Event()
+
+        # Start the cleanup thread
+        self._start_cleanup_thread()
+
+    def _start_cleanup_thread(self):
+        """Start a background thread for periodic session cleanup."""
+
+        def cleanup_worker():
+            while not self._stop_event.is_set():
+                self._perform_cleanup()
+                self._stop_event.wait(self._cleanup_interval)
+
+        self._cleanup_thread = threading.Thread(
+            target=cleanup_worker,
+            daemon=True,  # Allows the thread to be automatically terminated when the main program exits
+        )
+        self._cleanup_thread.start()
+
+    def _perform_cleanup(self):
+        """Perform cleanup of expired sessions."""
+        current_time = datetime.now()
+        expired_sessions = [
+            session_id for session_id, timestamp in list(self._session_times.items()) if (current_time - timestamp).total_seconds() > self._max_age
+        ]
+
+        for session_id in expired_sessions:
+            self.remove_session(session_id)
+
+    def remove_session(self, session_id: str):
+        """Remove a specific session."""
+        self._session_times.pop(session_id, None)
+
+    def set_context(self, session_id: str):
+        """
+        Context manager for setting and resetting session context.
+
+        :param session_id: Unique identifier for the session
+        :return: Context manager for session
+        """
+        self.session_var.set(session_id)
+        self._session_times[session_id] = datetime.now()
+
+    def get_context(self) -> str:
+        """
+        Get the current session context.
+
+        :return: Current session ID
+        :raises RuntimeError: If no session context is available
+        """
+        return self.session_var.get()
+
+    def reset_context(self):
+        """Reset the session context."""
+        token = self.get_context()
+        if token is not None:
+            self.session_var.reset(token)
+
+    def stop_cleanup(self):
+        """
+        Stop the cleanup thread.
+        Useful for graceful shutdown of the application.
+        """
+        self._stop_event.set()
+        if self._cleanup_thread:
+            self._cleanup_thread.join()
+
+    def __del__(self):
+        """
+        Ensure cleanup thread is stopped when the object is deleted.
+        """
+        self.stop_cleanup()
+
+
+class SqlConfig:
+    def __init__(self, default_engine: AsyncEngine | None = None, reader_engine: AsyncEngine | None = None, writer_engine: AsyncEngine | None = None):
+        """
+        Initialize the SQL configuration.
+        You can provide a default engine, a reader engine, and a writer engine.
+        If only one engine is provided (default_engine), it will be used for both reading and writing.
+        If both reader and writer engines are provided, they will be used for reading and writing respectively.
+        Note: The reader and writer engines must be different.
+        """
+
+        assert default_engine or reader_engine or writer_engine, "At least one engine must be provided."
+        assert not (reader_engine and writer_engine and id(reader_engine) == id(writer_engine)), "Reader and writer engines must be different."
+
+        engines = {
+            "writer": writer_engine or default_engine,
+            "reader": reader_engine or default_engine,
+        }
+        self.session_store = ContextStore()
+
+        class RoutingSession(Session):
+            def get_bind(this, mapper=None, clause=None, **kwargs):
+                if this._flushing or isinstance(clause, (Update, Delete, Insert)):
+                    return engines["writer"].sync_engine
+                return engines["reader"].sync_engine
+
+        async_session_factory = sessionmaker(
+            class_=AsyncSession,
+            sync_session_class=RoutingSession,
+            expire_on_commit=False,
+        )
+
+        session_scope: Union[AsyncSession, async_scoped_session] = async_scoped_session(
+            session_factory=async_session_factory,
+            scopefunc=asyncio.current_task,
+        )
+
+        @asynccontextmanager
+        async def get_session():
+            """
+            Get the database session.
+            This can be used for dependency injection.
+
+            :return: The database session.
+            """
+            try:
+                yield session_scope
+            except Exception:
+                traceback.print_exc()
+                await session_scope.rollback()
+            finally:
+                await session_scope.remove()
+                await session_scope.close()
+
+        self.get_session = get_session
+        self._context_token: Optional[Token] = None
+
+    def before_request(self, request: Request):
+        token = str(uuid4())
+        self.session_store.set_context(token)
+        return request
+
+    def after_request(self, response: Response):
+        self.session_store.reset_context()
+        return response
+
+    def init_app(self, app):
+        app.inject("get_session", self.get_session)
+        app.before_request()(self.before_request)
+        app.after_request()(self.after_request)
+
+
+__all__ = ["Model", "PostgresRepository", "SqlConfig"]

hypern/db/sql/addons/__init__.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+from .ts_vector import TSVector
+from .datetime import DatetimeType
+from .password import PasswordType
+from .encrypted import StringEncryptType, LargeBinaryEncryptType, AESEngine
+
+__all__ = [
+    "TSVector",
+    "DatetimeType",
+    "PasswordType",
+    "StringEncryptType",
+    "LargeBinaryEncryptType",
+    "AESEngine",
+]

hypern/db/sql/addons/color.py

@@ -0,0 +1,16 @@
+import re
+
+from sqlalchemy.types import String, TypeDecorator
+
+
+class ColorField(TypeDecorator):
+    impl = String
+
+    def process_bind_param(self, value, dialect):
+        color_regex = r"^#([A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})$"
+        if not re.match(color_regex, value):
+            raise ValueError("Invalid color format. Use hexadecimal color codes (e.g., #FF0000)")
+        return value
+
+    def process_result_value(self, value, dialect):
+        return value

hypern/db/sql/addons/daterange.py

@@ -0,0 +1,23 @@
+from datetime import datetime
+
+from sqlalchemy.dialects.postgresql import DATERANGE
+from sqlalchemy.types import TypeDecorator
+
+
+class DateRangeField(TypeDecorator):
+    impl = DATERANGE
+
+    def process_bind_param(self, value, dialect):
+        if value is None:
+            return None
+        elif "start" in value and "end" in value:
+            return f"['{value['start']}', '{value['end']}']"
+        else:
+            raise ValueError('DateRangeField must be a dictionary with "start" and "end" keys')
+
+    def process_result_value(self, value, dialect):
+        if value is None:
+            return None
+        else:
+            start, end = value[1:-1].split(",")
+            return {"start": datetime.strptime(start.strip("'"), "%Y-%m-%d %H:%M:%S.%f"), "end": datetime.strptime(end.strip("'"), "%Y-%m-%d %H:%M:%S.%f")}

hypern/db/sql/addons/datetime.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+from sqlalchemy import types
+
+
+class DatetimeType(types.TypeDecorator):
+    impl = types.DateTime
+    cache_ok = True
+
+    def load_dialect_impl(self, dialect):
+        if dialect.name == "sqlite":
+            return dialect.type_descriptor(types.TEXT)
+        return dialect.type_descriptor(self.impl)
+
+    def process_bind_param(self, value, dialect):
+        if dialect.name == "sqlite":
+            return value.isoformat()
+        return value
+
+    def process_result_value(self, value, dialect):
+        if dialect.name != "sqlite":
+            return value.timestamp()
+        return value

hypern/db/sql/addons/encrypted.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+import os
+import typing
+
+from cryptography.hazmat.primitives import padding
+from sqlalchemy.types import LargeBinary, String, TypeDecorator
+
+from hypern.security import AESEngine, EDEngine
+
+
+class StringEncryptType(TypeDecorator):
+    impl = String
+    cache_ok = True
+
+    def __init__(self, engine: typing.Optional[EDEngine] = None, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+        if not engine:
+            key = os.urandom(32)
+            iv = os.urandom(16)
+            padding_class = padding.PKCS7
+            self.engine = AESEngine(secret_key=key, iv=iv, padding_class=padding_class)
+        else:
+            self.engine = engine  # type: ignore
+
+    def process_bind_param(self, value, dialect):
+        if value is None:
+            return value
+        if not isinstance(value, str):
+            raise ValueError("Value String Encrypt Type must be a string")
+        return self.engine.encrypt(value).decode(encoding="utf-8")
+
+    def process_result_value(self, value, dialect):
+        if value is None:
+            return value
+        return self.engine.decrypt(value)
+
+
+class LargeBinaryEncryptType(StringEncryptType):
+    impl = LargeBinary
+    cache_ok = True
+
+    def __init__(self, engine: typing.Optional[EDEngine] = None, *args, **kwargs) -> None:
+        super().__init__(engine=engine, *args, **kwargs)  # type: ignore
+
+    def process_bind_param(self, value, dialect):
+        if value is None:
+            return value
+        value = super().process_bind_param(value, dialect)
+        if isinstance(value, str):
+            return value.encode("utf-8")
+        return value
+
+    def process_result_value(self, value, dialect):
+        if isinstance(value, bytes):
+            value = value.decode("utf-8")
+            return super().process_result_value(value, dialect)
+        return value