Import ducklm runtime
This commit is contained in:
@@ -0,0 +1,2 @@
|
||||
"""Permission and approval handling."""
|
||||
|
||||
@@ -0,0 +1,67 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import sqlite3
|
||||
from pathlib import Path
|
||||
|
||||
from app.core.contracts import PermissionDecision
|
||||
|
||||
|
||||
class SQLiteApprovalStore:
|
||||
"""Stores persistent user approval decisions."""
|
||||
|
||||
def __init__(self, db_path: str | Path) -> None:
|
||||
self._db_path = Path(db_path)
|
||||
self._db_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
self._initialize()
|
||||
|
||||
def save(self, decision: PermissionDecision) -> PermissionDecision:
|
||||
with sqlite3.connect(self._db_path) as conn:
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT OR REPLACE INTO approvals (action_type, pattern, decision, created_at)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""",
|
||||
(
|
||||
decision.action_type,
|
||||
decision.pattern,
|
||||
decision.decision,
|
||||
decision.created_at.isoformat(),
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
return decision
|
||||
|
||||
def load(self, action_type: str, pattern: str) -> PermissionDecision | None:
|
||||
with sqlite3.connect(self._db_path) as conn:
|
||||
row = conn.execute(
|
||||
"""
|
||||
SELECT action_type, pattern, decision, created_at
|
||||
FROM approvals
|
||||
WHERE action_type = ? AND pattern = ?
|
||||
""",
|
||||
(action_type, pattern),
|
||||
).fetchone()
|
||||
if not row:
|
||||
return None
|
||||
return PermissionDecision(
|
||||
action_type=row[0],
|
||||
pattern=row[1],
|
||||
decision=row[2],
|
||||
created_at=row[3],
|
||||
)
|
||||
|
||||
def _initialize(self) -> None:
|
||||
with sqlite3.connect(self._db_path) as conn:
|
||||
conn.execute(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS approvals (
|
||||
action_type TEXT NOT NULL,
|
||||
pattern TEXT NOT NULL,
|
||||
decision TEXT NOT NULL,
|
||||
created_at TEXT NOT NULL,
|
||||
PRIMARY KEY (action_type, pattern)
|
||||
)
|
||||
"""
|
||||
)
|
||||
conn.commit()
|
||||
|
||||
Reference in New Issue
Block a user