fixtureqa 0.1.0__py3-none-any.whl

fixture/core/user_store.py

@@ -0,0 +1,186 @@
+"""
+SQLite-backed user store.
+
+Uses Python's built-in sqlite3 — no extra runtime dependency.
+Thread-safe: every call opens and closes its own connection (sqlite3 handles
+concurrent readers/single-writer serialisation via WAL mode).
+"""
+import sqlite3
+import time
+import uuid
+from dataclasses import dataclass
+from typing import List, Optional
+
+
+@dataclass
+class User:
+    uid: str
+    username: str
+    password_hash: str
+    role: str          # platform_admin | user_admin | user
+    created_at: float
+    is_active: bool
+
+    @property
+    def is_admin(self) -> bool:
+        return self.role == "platform_admin"
+
+
+_DDL_USERS = """
+CREATE TABLE IF NOT EXISTS users (
+    uid           TEXT PRIMARY KEY,
+    username      TEXT UNIQUE NOT NULL,
+    password_hash TEXT NOT NULL,
+    role          TEXT NOT NULL DEFAULT 'user',
+    created_at    REAL NOT NULL,
+    is_active     INTEGER NOT NULL DEFAULT 1
+)
+"""
+
+_DDL_SETTINGS = """
+CREATE TABLE IF NOT EXISTS settings (
+    key   TEXT PRIMARY KEY,
+    value TEXT NOT NULL
+)
+"""
+
+_DEFAULT_SETTINGS = {
+    "registration_enabled": "0",
+    "max_users": "50",
+    "housekeeping_enabled": "1",
+    "msg_retention_days": "90",
+    "log_retention_days": "30",
+    "branding_prefix":     "",   # e.g. "Google" → "Google FIXture"
+    "branding_accent":     "",   # --accent; "" = use CSS default
+    "branding_bg":         "",   # --bg (page background)
+    "branding_bg2":        "",   # --bg2 (header/panel background)
+    "branding_bg3":        "",   # --bg3 (card/input background)
+    "branding_border":     "",   # --border
+    "branding_text":       "",   # --text (primary text)
+    "branding_text2":      "",   # --text2 (secondary text)
+    "branding_logo_dark":  "",   # base64 data URI; "" = use bundled SVG
+    "branding_logo_light": "",   # base64 data URI; "" = use bundled SVG
+    "branding_favicon":    "",   # base64 data URI; "" = use built-in favicon
+}
+
+
+class UserStore:
+    def __init__(self, db_path: str):
+        self._path = db_path
+        self._init_db()
+
+    # ------------------------------------------------------------------
+    # Users
+    # ------------------------------------------------------------------
+
+    def has_any_users(self) -> bool:
+        with self._connect() as conn:
+            row = conn.execute("SELECT 1 FROM users LIMIT 1").fetchone()
+            return row is not None
+
+    def count_users(self) -> int:
+        with self._connect() as conn:
+            row = conn.execute("SELECT COUNT(*) FROM users").fetchone()
+            return row[0]
+
+    def create_user(self, username: str, password_hash: str, role: str = "user") -> User:
+        uid = str(uuid.uuid4())
+        now = time.time()
+        with self._connect() as conn:
+            conn.execute(
+                "INSERT INTO users (uid, username, password_hash, role, created_at, is_active) "
+                "VALUES (?, ?, ?, ?, ?, 1)",
+                (uid, username, password_hash, role, now),
+            )
+        return User(uid=uid, username=username, password_hash=password_hash,
+                    role=role, created_at=now, is_active=True)
+
+    def list_users(self) -> List[User]:
+        with self._connect() as conn:
+            rows = conn.execute(
+                "SELECT uid, username, password_hash, role, created_at, is_active "
+                "FROM users ORDER BY created_at ASC"
+            ).fetchall()
+        return [_row_to_user(r) for r in rows]
+
+    def get_by_username(self, username: str) -> Optional[User]:
+        with self._connect() as conn:
+            row = conn.execute(
+                "SELECT uid, username, password_hash, role, created_at, is_active "
+                "FROM users WHERE username = ?", (username,)
+            ).fetchone()
+        return _row_to_user(row) if row else None
+
+    def get_by_uid(self, uid: str) -> Optional[User]:
+        with self._connect() as conn:
+            row = conn.execute(
+                "SELECT uid, username, password_hash, role, created_at, is_active "
+                "FROM users WHERE uid = ?", (uid,)
+            ).fetchone()
+        return _row_to_user(row) if row else None
+
+    def update_user(self, uid: str, **fields) -> Optional[User]:
+        """Update allowed fields: is_active, role, password_hash, username."""
+        allowed = {"is_active", "role", "password_hash", "username"}
+        updates = {k: v for k, v in fields.items() if k in allowed}
+        if not updates:
+            return self.get_by_uid(uid)
+        cols = ", ".join(f"{k} = ?" for k in updates)
+        vals = list(updates.values()) + [uid]
+        with self._connect() as conn:
+            conn.execute(f"UPDATE users SET {cols} WHERE uid = ?", vals)
+        return self.get_by_uid(uid)
+
+    def delete_user(self, uid: str) -> None:
+        with self._connect() as conn:
+            conn.execute("DELETE FROM users WHERE uid = ?", (uid,))
+
+    # ------------------------------------------------------------------
+    # Settings
+    # ------------------------------------------------------------------
+
+    def get_setting(self, key: str, default: str = "") -> str:
+        with self._connect() as conn:
+            row = conn.execute(
+                "SELECT value FROM settings WHERE key = ?", (key,)
+            ).fetchone()
+        return row["value"] if row else default
+
+    def set_setting(self, key: str, value: str) -> None:
+        with self._connect() as conn:
+            conn.execute(
+                "INSERT INTO settings (key, value) VALUES (?, ?) "
+                "ON CONFLICT(key) DO UPDATE SET value = excluded.value",
+                (key, value),
+            )
+
+    # ------------------------------------------------------------------
+    # Internal
+    # ------------------------------------------------------------------
+
+    def _init_db(self) -> None:
+        with self._connect() as conn:
+            conn.execute("PRAGMA journal_mode=WAL")
+            conn.execute(_DDL_USERS)
+            conn.execute(_DDL_SETTINGS)
+            # Seed defaults without overwriting existing values
+            for k, v in _DEFAULT_SETTINGS.items():
+                conn.execute(
+                    "INSERT OR IGNORE INTO settings (key, value) VALUES (?, ?)", (k, v)
+                )
+
+    def _connect(self) -> sqlite3.Connection:
+        conn = sqlite3.connect(self._path)
+        conn.row_factory = sqlite3.Row
+        return conn
+
+
+def _row_to_user(row: sqlite3.Row) -> User:
+    return User(
+        uid=row["uid"],
+        username=row["username"],
+        password_hash=row["password_hash"],
+        role=row["role"],
+        created_at=row["created_at"],
+        is_active=bool(row["is_active"]),
+    )

fixture/core/venue_responses.py

@@ -0,0 +1,94 @@
+"""
+Build automatic Execution Report responses for venue sessions.
+
+Both builders take:
+  order_fields  — {tag_str: value_str} parsed from the inbound NewOrderSingle
+  order_id      — venue-assigned OrderID (tag 37); shared across ack + fill
+                  so both reports refer to the same order
+"""
+
+import uuid
+from datetime import datetime, timezone
+
+from fixcore.message.message import Message
+
+
+def _utc_ts() -> str:
+    return datetime.now(timezone.utc).strftime("%Y%m%d-%H:%M:%S")
+
+
+def _new_exec_id() -> str:
+    return "E" + uuid.uuid4().hex[:8].upper()
+
+
+def new_order_id() -> str:
+    return "O" + uuid.uuid4().hex[:8].upper()
+
+
+def _set(msg: Message, tag: int, value: str) -> None:
+    msg.set_field(tag, value)
+
+
+def _copy(msg: Message, fields: dict, tag: int) -> None:
+    val = fields.get(str(tag)) or fields.get(tag)
+    if val:
+        _set(msg, tag, val)
+
+
+def build_ack(order_fields: dict, order_id: str) -> Message:
+    """ExecutionReport — OrdStatus=New (acknowledgement)."""
+    msg = Message()
+    msg.header.set(35, "8")  # MsgType = ExecutionReport
+
+    _copy(msg, order_fields, 11)   # ClOrdID
+    _copy(msg, order_fields, 55)   # Symbol
+    _copy(msg, order_fields, 54)   # Side
+    _copy(msg, order_fields, 38)   # OrderQty
+    _copy(msg, order_fields, 40)   # OrdType
+    _copy(msg, order_fields, 44)   # Price
+    _copy(msg, order_fields, 1)    # Account
+
+    qty = order_fields.get("38") or order_fields.get(38, "0")
+
+    _set(msg, 37, order_id)         # OrderID
+    _set(msg, 17, _new_exec_id())   # ExecID
+    _set(msg, 20, "0")              # ExecTransType = New (FIX 4.2)
+    _set(msg, 150, "0")             # ExecType = New
+    _set(msg, 39, "0")              # OrdStatus = New
+    _set(msg, 14, "0")              # CumQty = 0
+    _set(msg, 151, qty)             # LeavesQty = OrderQty
+    _set(msg, 6, "0")               # AvgPx = 0
+    _set(msg, 60, _utc_ts())        # TransactTime
+
+    return msg
+
+
+def build_fill(order_fields: dict, order_id: str) -> Message:
+    """ExecutionReport — OrdStatus=Filled (full fill)."""
+    msg = Message()
+    msg.header.set(35, "8")  # MsgType = ExecutionReport
+
+    _copy(msg, order_fields, 11)   # ClOrdID
+    _copy(msg, order_fields, 55)   # Symbol
+    _copy(msg, order_fields, 54)   # Side
+    _copy(msg, order_fields, 38)   # OrderQty
+    _copy(msg, order_fields, 40)   # OrdType
+    _copy(msg, order_fields, 44)   # Price
+    _copy(msg, order_fields, 1)    # Account
+
+    qty = order_fields.get("38") or order_fields.get(38, "0")
+    px = order_fields.get("44") or order_fields.get(44, "0")
+
+    _set(msg, 37, order_id)         # OrderID
+    _set(msg, 17, _new_exec_id())   # ExecID
+    _set(msg, 20, "0")              # ExecTransType = New (FIX 4.2)
+    _set(msg, 150, "2")             # ExecType = Fill
+    _set(msg, 39, "2")              # OrdStatus = Filled
+    _set(msg, 32, qty)              # LastShares / LastQty
+    _set(msg, 31, px)               # LastPx
+    _set(msg, 14, qty)              # CumQty = OrderQty
+    _set(msg, 151, "0")             # LeavesQty = 0
+    _set(msg, 6, px)                # AvgPx = LastPx
+    _set(msg, 60, _utc_ts())        # TransactTime
+
+    return msg